From 02632790713eaabd5250ba04283233c8bc078067 Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <e.dolstra@tudelft.nl>
Date: Thu, 30 Nov 2006 20:45:20 +0000
Subject: [PATCH] * More operations.

---
 src/libstore/remote-store.cc    | 23 ++++++++++++++++++-----
 src/libstore/worker-protocol.hh |  8 +++++---
 src/nix-worker/main.cc          | 28 +++++++++++++++++++++++++++-
 3 files changed, 50 insertions(+), 9 deletions(-)

diff --git a/src/libstore/remote-store.cc b/src/libstore/remote-store.cc
index f59fe8b98..99f9589be 100644
--- a/src/libstore/remote-store.cc
+++ b/src/libstore/remote-store.cc
@@ -2,6 +2,7 @@
 #include "util.hh"
 #include "remote-store.hh"
 #include "worker-protocol.hh"
+#include "archive.hh"
 
 #include <iostream>
 #include <unistd.h>
@@ -111,33 +112,45 @@ void RemoteStore::queryReferrers(const Path & storePath,
 
 Path RemoteStore::addToStore(const Path & srcPath)
 {
-    throw Error("not implemented");
+    writeInt(wopAddToStore, to);
+    writeString(baseNameOf(srcPath), to);
+    dumpPath(srcPath, to);
+    Path path = readString(from);
+    return path;
 }
 
 
 Path RemoteStore::addToStoreFixed(bool recursive, string hashAlgo,
     const Path & srcPath)
 {
-    throw Error("not implemented");
+    throw Error("not implemented 4");
 }
 
 
 Path RemoteStore::addTextToStore(const string & suffix, const string & s,
     const PathSet & references)
 {
-    throw Error("not implemented");
+    writeInt(wopAddTextToStore, to);
+    writeString(suffix, to);
+    writeString(s, to);
+    writeInt(references.size(), to);
+    for (PathSet::iterator i = references.begin(); i != references.end(); ++i)
+        writeString(*i, to);
+    
+    Path path = readString(from);
+    return path;
 }
 
 
 void RemoteStore::buildDerivations(const PathSet & drvPaths)
 {
-    throw Error("not implemented");
+    throw Error("not implemented 6");
 }
 
 
 void RemoteStore::ensurePath(const Path & storePath)
 {
-    throw Error("not implemented");
+    throw Error("not implemented 7");
 }
 
 
diff --git a/src/libstore/worker-protocol.hh b/src/libstore/worker-protocol.hh
index 0ba0c374e..dd711f330 100644
--- a/src/libstore/worker-protocol.hh
+++ b/src/libstore/worker-protocol.hh
@@ -7,9 +7,11 @@
 
 
 typedef enum {
-    wopQuit = 0,
-    wopIsValidPath = 1,
-    wopQuerySubstitutes = 2,
+    wopQuit,
+    wopIsValidPath,
+    wopQuerySubstitutes,
+    wopAddToStore,
+    wopAddTextToStore,
 } WorkerOp;
 
 
diff --git a/src/nix-worker/main.cc b/src/nix-worker/main.cc
index d6966c2bf..d834e625e 100644
--- a/src/nix-worker/main.cc
+++ b/src/nix-worker/main.cc
@@ -3,6 +3,7 @@
 #include "util.hh"
 #include "serialise.hh"
 #include "worker-protocol.hh"
+#include "archive.hh"
 
 using namespace nix;
 
@@ -40,8 +41,33 @@ void processConnection(Source & from, Sink & to)
             break;
         }
 
+        case wopAddToStore: {
+            /* !!! uberquick hack */
+            string baseName = readString(from);
+            Path tmp = createTempDir();
+            Path tmp2 = tmp + "/" + baseName;
+            restorePath(tmp2, from);
+            writeString(store->addToStore(tmp2), to);
+            deletePath(tmp);
+            break;
+        }
+
+        case wopAddTextToStore: {
+            string suffix = readString(from);
+            string s = readString(from);
+            unsigned int refCount = readInt(from);
+            PathSet refs;
+            while (refCount--) {
+                Path ref = readString(from);
+                assertStorePath(ref);
+                refs.insert(ref);
+            }
+            writeString(store->addTextToStore(suffix, s, refs), to);
+            break;
+        }
+
         default:
-            throw Error("invalid operation");
+            throw Error(format("invalid operation %1%") % op);
         }
         
     } while (!quit);