Add 'guix-register'.

* nix/guix-register/guix-register.cc, tests/guix-register.sh: New
  files.
* Makefile.am (SH_TESTS)[BUILD_DAEMON]: Add tests/guix-register.sh.
* daemon.am (sbin_PROGRAMS, guix_register_SOURCES,
  guix_register_CPPFLAGS, guix_register_LDADD): New variables.
* test-env.in: Export 'storedir', 'prefix', 'datarootdir', 'datadir',
  and 'localstatedir'.

1 files changed, 168 insertions, 0 deletions

diff --git a/nix/guix-register/guix-register.cc b/nix/guix-register/guix-register.cc
new file mode 100644
index 0000000000..0a028f0cfe
--- /dev/null
+++ b/nix/guix-register/guix-register.cc
@@ -0,0 +1,168 @@
+/* GNU Guix --- Functional package management for GNU
+   Copyright (C) 2013 Ludovic Courtès <ludo@gnu.org>
+   Copyright (C) 2007, 2008, 2009, 2010, 2011, 2012,
+     2013 Eelco Dolstra <eelco.dolstra@logicblox.com>
+
+   This file is part of GNU Guix.
+
+   GNU Guix is free software; you can redistribute it and/or modify it
+   under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or (at
+   your option) any later version.
+
+   GNU Guix is distributed in the hope that it will be useful, but
+   WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with GNU Guix.  If not, see <http://www.gnu.org/licenses/>.  */
+
+/* This file derives from the implementation of 'nix-store
+   --register-validity', by Eelco Dolstra, as found in the Nix package
+   manager's src/nix-store/nix-store.cc.  */
+
+#include <config.h>
+
+#include <globals.hh>
+#include <local-store.hh>
+
+#include <iostream>
+#include <fstream>
+#include <cstdlib>
+#include <cstdio>
+
+#include <argp.h>
+
+using namespace nix;
+
+/* Input stream where we read closure descriptions.  */
+static std::istream *input = &std::cin;
+
+
+
+/* Command-line options.  */
+
+const char *argp_program_version =
+  "guix-register (" PACKAGE_NAME ") " PACKAGE_VERSION;
+const char *argp_program_bug_address = PACKAGE_BUGREPORT;
+
+static char doc[] =
+"guix-register -- register a closure as valid in a store\
+\v\
+This program is used internally when populating a store with data \
+from an existing store.  It updates the new store's database with \
+information about which store files are valid, and what their \
+references are.";
+
+static const struct argp_option options[] =
+  {
+    { "prefix", 'p', "DIRECTORY", 0,
+      "Open the store that lies under DIRECTORY" },
+    { 0, 0, 0, 0, 0 }
+  };
+
+/* Parse a single option. */
+static error_t
+parse_opt (int key, char *arg, struct argp_state *state)
+{
+  switch (key)
+    {
+    case 'p':
+      {
+	string prefix = canonPath (arg);
+	settings.nixStore = prefix + NIX_STORE_DIR;
+	settings.nixDataDir = prefix + NIX_DATA_DIR;
+	settings.nixLogDir = prefix + NIX_LOG_DIR;
+	settings.nixStateDir = prefix + NIX_STATE_DIR;
+	settings.nixDBPath = settings.nixStateDir + "/db";
+	break;
+      }
+
+    case ARGP_KEY_ARG:
+      {
+	std::ifstream *file;
+
+	if (state->arg_num >= 2)
+	  /* Too many arguments. */
+	  argp_usage (state);
+
+	file = new std::ifstream ();
+	file->open (arg);
+
+	input = file;
+      }
+      break;
+
+    default:
+      return (error_t) ARGP_ERR_UNKNOWN;
+    }
+
+  return (error_t) 0;
+}
+
+/* Argument parsing.  */
+static struct argp argp = { options, parse_opt, 0, doc };
+
+
+/* Read from INPUT the description of a closure, and register it as valid in
+   STORE.  The expected format on INPUT is that used by #:references-graphs:
+
+     FILE
+     DERIVER
+     NUMBER-OF-REFERENCES
+     REF1
+     ...
+     REFN
+
+   This is really meant as an internal format.  */
+static void
+register_validity (LocalStore *store, std::istream &input,
+		   bool reregister = true, bool hashGiven = false,
+		   bool canonicalise = true)
+{
+  ValidPathInfos infos;
+
+  while (1)
+    {
+      ValidPathInfo info = decodeValidPathInfo (input, hashGiven);
+      if (info.path == "")
+	break;
+      if (!store->isValidPath (info.path) || reregister)
+	{
+	  /* !!! races */
+	  if (canonicalise)
+	    canonicalisePathMetaData (info.path, -1);
+
+	  if (!hashGiven)
+	    {
+	      HashResult hash = hashPath (htSHA256, info.path);
+	      info.hash = hash.first;
+	      info.narSize = hash.second;
+	    }
+	  infos.push_back (info);
+	}
+    }
+
+  store->registerValidPaths (infos);
+}
+
+
+int
+main (int argc, char *argv[])
+{
+  try
+    {
+      argp_parse (&argp, argc, argv, 0, 0, 0);
+
+      LocalStore store;
+      register_validity (&store, *input);
+    }
+  catch (std::exception &e)
+    {
+      fprintf (stderr, "error: %s\n", e.what ());
+      return EXIT_FAILURE;
+    }
+
+  return EXIT_SUCCESS;
+}