nix-super/src/libmain/shared.hh

87 lines
2.1 KiB
C++
Raw Normal View History

#pragma once
#include "util.hh"
#include <signal.h>
#include <locale>
2014-08-13 04:50:44 +03:00
namespace nix {
2014-08-13 04:50:44 +03:00
MakeError(UsageError, nix::Error);
2014-08-13 04:50:44 +03:00
class Exit : public std::exception
{
public:
int status;
Exit() : status(0) { }
Exit(int status) : status(status) { }
};
2014-08-13 04:50:44 +03:00
class StoreAPI;
2014-08-13 04:50:44 +03:00
int handleExceptions(const string & programName, std::function<void()> fun);
2014-08-13 04:50:44 +03:00
void initNix();
2014-08-13 04:50:44 +03:00
void parseCmdLine(int argc, char * * argv,
std::function<bool(Strings::iterator & arg, const Strings::iterator & end)> parseArg);
2009-11-24 14:26:25 +02:00
2014-08-13 04:50:44 +03:00
void printVersion(const string & programName);
/* Ugh. No better place to put this. */
void printGCWarning();
void printMissing(StoreAPI & store, const PathSet & paths);
void printMissing(const PathSet & willBuild,
const PathSet & willSubstitute, const PathSet & unknown,
unsigned long long downloadSize, unsigned long long narSize);
2014-08-13 04:50:44 +03:00
string getArg(const string & opt,
Strings::iterator & i, const Strings::iterator & end);
2009-11-24 14:26:25 +02:00
template<class N> N getIntArg(const string & opt,
Strings::iterator & i, const Strings::iterator & end, bool allowUnit)
2009-11-24 14:26:25 +02:00
{
++i;
2014-08-20 18:00:17 +03:00
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;
2014-08-20 18:00:17 +03:00
else throw UsageError(format("invalid unit specifier %1%") % u);
s.resize(s.size() - 1);
}
}
2009-11-24 14:26:25 +02:00
N n;
if (!string2Int(s, n))
2014-08-20 18:00:17 +03:00
throw UsageError(format("%1% requires an integer argument") % opt);
return n * multiplier;
2009-11-24 14:26:25 +02:00
}
/* 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;
}