about summary refs log tree commit diff
path: root/src/nix/command.cc
blob: 0c2103392f8edc43296755332b1c2739f2894b1b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include "command.hh"
#include "store-api.hh"

namespace nix {

Commands * RegisterCommand::commands = 0;

MultiCommand::MultiCommand(const Commands & _commands)
    : commands(_commands)
{
    expectedArgs.push_back(ExpectedArg{"command", 1, [=](Strings ss) {
        assert(!command);
        auto i = commands.find(ss.front());
        if (i == commands.end())
            throw UsageError(format("‘%1%’ is not a recognised command") % ss.front());
        command = i->second;
    }});
}

void MultiCommand::printHelp(const string & programName, std::ostream & out)
{
    if (command) {
        command->printHelp(programName + " " + command->name(), out);
        return;
    }

    out << "Usage: " << programName << " <COMMAND> <FLAGS>... <ARGS>...\n";

    out << "\n";
    out << "Common flags:\n";
    printFlags(out);

    out << "\n";
    out << "Available commands:\n";

    Table2 table;
    for (auto & command : commands)
        table.push_back(std::make_pair(command.second->name(), command.second->description()));
    printTable(out, table);

    out << "\n";
    out << "For full documentation, run ‘man " << programName << "’ or ‘man " << programName << "-<COMMAND>’.\n";
}

bool MultiCommand::processFlag(Strings::iterator & pos, Strings::iterator end)
{
    if (Args::processFlag(pos, end)) return true;
    if (command && command->processFlag(pos, end)) return true;
    return false;
}

bool MultiCommand::processArgs(const Strings & args, bool finish)
{
    if (command)
        return command->processArgs(args, finish);
    else
        return Args::processArgs(args, finish);
}

StoreCommand::StoreCommand()
{
    storeUri = getEnv("NIX_REMOTE");

    mkFlag(0, "store", "store-uri", "URI of the Nix store to use", &storeUri);
}

void StoreCommand::run()
{
    run(openStoreAt(storeUri));
}

}