169 lines
4.0 KiB
C++
169 lines
4.0 KiB
C++
/* 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;
|
||
}
|