diff options
Diffstat (limited to '')
-rw-r--r-- | third_party/nix/src/nix-daemon/nix-daemon.cc (renamed from third_party/nix/src/nix-daemon/nix-daemon-main.cc) | 50 |
1 files changed, 19 insertions, 31 deletions
diff --git a/third_party/nix/src/nix-daemon/nix-daemon-main.cc b/third_party/nix/src/nix-daemon/nix-daemon.cc index e492c343b900..02b350c3d87b 100644 --- a/third_party/nix/src/nix-daemon/nix-daemon-main.cc +++ b/third_party/nix/src/nix-daemon/nix-daemon.cc @@ -1,5 +1,8 @@ #include <filesystem> +#include <absl/flags/flag.h> +#include <absl/flags/parse.h> +#include <absl/flags/usage_config.h> #include <absl/strings/str_format.h> #include <fcntl.h> #include <glog/logging.h> @@ -9,7 +12,7 @@ #include <sys/socket.h> #include <sys/un.h> -#include "libmain/shared.hh" +#include "libmain/shared.hh" // TODO(tazjin): can this be removed? #include "libstore/globals.hh" #include "libstore/store-api.hh" #include "libutil/util.hh" @@ -17,6 +20,8 @@ #include "nix-daemon/nix-daemon-proto.hh" #include "nix/legacy.hh" +ABSL_FLAG(bool, pipe, false, "Use pipes for daemon communication"); + namespace nix::daemon { using grpc::Server; @@ -105,39 +110,22 @@ int RunServer() { } } -static int main_(int argc, char** argv) { - auto pipe = false; - - // TODO(grfn): Replace with absl::flags - parseCmdLine(argc, argv, - [&](Strings::iterator& arg, const Strings::iterator& end) { - if (*arg == "--help") { - showManPage("nix-daemon"); - } else if (*arg == "--version") { - printVersion("nix-daemon"); - } else if (*arg == "--pipe") { - // Causes the daemon to forward stdin and stdout to and from - // the actual daemon socket - pipe = true; - } else { - return false; - } - return true; - }); - - if (pipe) { - if (getStoreType() == tDaemon) { - return ForwardToSocket(settings.nixDaemonSocketFile); +} // namespace nix::daemon + +int main(int argc, char** argv) { + absl::SetFlagsUsageConfig({.version_string = [] { return nix::nixVersion; }}); + absl::ParseCommandLine(argc, argv); + + if (absl::GetFlag(FLAGS_pipe)) { + if (nix::getStoreType() == nix::tDaemon) { + return nix::daemon::ForwardToSocket(nix::settings.nixDaemonSocketFile); } else { // TODO(grfn): Need to launch a server on stdin here - upstream calls // processConnection(true, "root", 0); - throw "Not implemented"; + LOG(ERROR) << "not implemented"; + return 1; } } - return RunServer(); -} - -// TODO(grfn): Replace this with something less magical -static RegisterLegacyCommand s1("nix-daemon", main_); -} // namespace nix::daemon + return nix::daemon::RunServer(); +} |