mirror of
https://github.com/privatevoid-net/nix-super.git
synced 2024-09-22 09:18:04 +03:00
0a2fa2d684
Before this change, stdout was closed after the pager exits. This is fine for non-interactive commands where we want to exit right after the pager exits anyways, but for interactive things (e.g. nix repl) this breaks the output after we quit the pager. Keep the initial stdout fd as part of RunPager, and restore it in RunPager::~RunPager using dup2.
118 lines
2.7 KiB
C++
118 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#include "util.hh"
|
|
#include "args.hh"
|
|
#include "common-args.hh"
|
|
#include "path.hh"
|
|
#include "derived-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<DerivedPath> & paths,
|
|
Verbosity lvl = lvlInfo);
|
|
|
|
void printMissing(ref<Store> store, const StorePathSet & willBuild,
|
|
const StorePathSet & willSubstitute, const StorePathSet & unknown,
|
|
uint64_t downloadSize, uint64_t 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("'%1%' requires an argument", opt);
|
|
return string2IntWithUnitPrefix<N>(*i);
|
|
}
|
|
|
|
|
|
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;
|
|
int stdout;
|
|
};
|
|
|
|
extern volatile ::sig_atomic_t blockInt;
|
|
|
|
|
|
/* GC helpers. */
|
|
|
|
string showBytes(uint64_t 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();
|
|
|
|
|
|
}
|