bbe97dff8b
Most functions now take a StorePath argument rather than a Path (which is just an alias for std::string). The StorePath constructor ensures that the path is syntactically correct (i.e. it looks like <store-dir>/<base32-hash>-<name>). Similarly, functions like buildPaths() now take a StorePathWithOutputs, rather than abusing Path by adding a '!<outputs>' suffix. Note that the StorePath type is implemented in Rust. This involves some hackery to allow Rust values to be used directly in C++, via a helper type whose destructor calls the Rust type's drop() function. The main issue is the dynamic nature of C++ move semantics: after we have moved a Rust value, we should not call the drop function on the original value. So when we move a value, we set the original value to bitwise zero, and the destructor only calls drop() if the value is not bitwise zero. This should be sufficient for most types. Also lots of minor cleanups to the C++ API to make it more modern (e.g. using std::optional and std::string_view in some places).
132 lines
3.3 KiB
C++
132 lines
3.3 KiB
C++
#pragma once
|
|
|
|
#include "util.hh"
|
|
#include "args.hh"
|
|
#include "common-args.hh"
|
|
#include "path.hh"
|
|
|
|
#include <signal.h>
|
|
|
|
#include <locale>
|
|
|
|
|
|
namespace nix {
|
|
|
|
class Exit : public std::exception
|
|
{
|
|
public:
|
|
int status;
|
|
Exit() : status(0) { }
|
|
Exit(int status) : status(status) { }
|
|
virtual ~Exit();
|
|
};
|
|
|
|
int handleExceptions(const string & programName, std::function<void()> fun);
|
|
|
|
/* Don't forget to call initPlugins() after settings are initialized! */
|
|
void initNix();
|
|
|
|
void parseCmdLine(int argc, char * * argv,
|
|
std::function<bool(Strings::iterator & arg, const Strings::iterator & end)> parseArg);
|
|
|
|
void parseCmdLine(const string & programName, const Strings & args,
|
|
std::function<bool(Strings::iterator & arg, const Strings::iterator & end)> parseArg);
|
|
|
|
void printVersion(const string & programName);
|
|
|
|
/* Ugh. No better place to put this. */
|
|
void printGCWarning();
|
|
|
|
class Store;
|
|
struct StorePathWithOutputs;
|
|
|
|
void printMissing(
|
|
ref<Store> store,
|
|
const std::vector<StorePathWithOutputs> & paths,
|
|
Verbosity lvl = lvlInfo);
|
|
|
|
void printMissing(ref<Store> store, const StorePathSet & willBuild,
|
|
const StorePathSet & willSubstitute, const StorePathSet & unknown,
|
|
unsigned long long downloadSize, unsigned long long narSize, Verbosity lvl = lvlInfo);
|
|
|
|
string getArg(const string & opt,
|
|
Strings::iterator & i, const Strings::iterator & end);
|
|
|
|
template<class N> N getIntArg(const string & opt,
|
|
Strings::iterator & i, const Strings::iterator & end, bool allowUnit)
|
|
{
|
|
++i;
|
|
if (i == end) throw UsageError(format("'%1%' requires an argument") % opt);
|
|
string s = *i;
|
|
N multiplier = 1;
|
|
if (allowUnit && !s.empty()) {
|
|
char u = std::toupper(*s.rbegin());
|
|
if (std::isalpha(u)) {
|
|
if (u == 'K') multiplier = 1ULL << 10;
|
|
else if (u == 'M') multiplier = 1ULL << 20;
|
|
else if (u == 'G') multiplier = 1ULL << 30;
|
|
else if (u == 'T') multiplier = 1ULL << 40;
|
|
else throw UsageError(format("invalid unit specifier '%1%'") % u);
|
|
s.resize(s.size() - 1);
|
|
}
|
|
}
|
|
N n;
|
|
if (!string2Int(s, n))
|
|
throw UsageError(format("'%1%' requires an integer argument") % opt);
|
|
return n * multiplier;
|
|
}
|
|
|
|
|
|
struct LegacyArgs : public MixCommonArgs
|
|
{
|
|
std::function<bool(Strings::iterator & arg, const Strings::iterator & end)> parseArg;
|
|
|
|
LegacyArgs(const std::string & programName,
|
|
std::function<bool(Strings::iterator & arg, const Strings::iterator & end)> parseArg);
|
|
|
|
bool processFlag(Strings::iterator & pos, Strings::iterator end) override;
|
|
|
|
bool processArgs(const Strings & args, bool finish) override;
|
|
};
|
|
|
|
|
|
/* Show the manual page for the specified program. */
|
|
void showManPage(const string & name);
|
|
|
|
/* The constructor of this class starts a pager if stdout is a
|
|
terminal and $PAGER is set. Stdout is redirected to the pager. */
|
|
class RunPager
|
|
{
|
|
public:
|
|
RunPager();
|
|
~RunPager();
|
|
|
|
private:
|
|
Pid pid;
|
|
};
|
|
|
|
extern volatile ::sig_atomic_t blockInt;
|
|
|
|
|
|
/* GC helpers. */
|
|
|
|
string showBytes(unsigned long long bytes);
|
|
|
|
struct GCResults;
|
|
|
|
struct PrintFreed
|
|
{
|
|
bool show;
|
|
const GCResults & results;
|
|
PrintFreed(bool show, const GCResults & results)
|
|
: show(show), results(results) { }
|
|
~PrintFreed();
|
|
};
|
|
|
|
|
|
/* Install a SIGSEGV handler to detect stack overflows. */
|
|
void detectStackOverflow();
|
|
|
|
|
|
}
|