[base] introduce a command type that gets registered with the app
This commit is contained in:
16
era/commands.cc
Normal file
16
era/commands.cc
Normal file
@@ -0,0 +1,16 @@
|
||||
#include "era/commands.h"
|
||||
|
||||
using namespace base;
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
void
|
||||
era::register_era_commands(base::application &app)
|
||||
{
|
||||
app.add_cmd(command::ptr(new era_check_cmd()));
|
||||
app.add_cmd(command::ptr(new era_dump_cmd()));
|
||||
app.add_cmd(command::ptr(new era_invalidate_cmd()));
|
||||
app.add_cmd(command::ptr(new era_restore_cmd()));
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------
|
||||
@@ -6,10 +6,35 @@
|
||||
//----------------------------------------------------------------
|
||||
|
||||
namespace era {
|
||||
extern base::command era_check_cmd;
|
||||
extern base::command era_dump_cmd;
|
||||
extern base::command era_invalidate_cmd;
|
||||
extern base::command era_restore_cmd;
|
||||
class era_check_cmd : public base::command {
|
||||
public:
|
||||
era_check_cmd();
|
||||
virtual void usage(std::ostream &out) const;
|
||||
virtual int run(int argc, char **argv);
|
||||
};
|
||||
|
||||
class era_dump_cmd : public base::command {
|
||||
public:
|
||||
era_dump_cmd();
|
||||
virtual void usage(std::ostream &out) const;
|
||||
virtual int run(int argc, char **argv);
|
||||
};
|
||||
|
||||
class era_invalidate_cmd : public base::command {
|
||||
public:
|
||||
era_invalidate_cmd();
|
||||
virtual void usage(std::ostream &out) const;
|
||||
virtual int run(int argc, char **argv);
|
||||
};
|
||||
|
||||
class era_restore_cmd : public base::command {
|
||||
public:
|
||||
era_restore_cmd();
|
||||
virtual void usage(std::ostream &out) const;
|
||||
virtual int run(int argc, char **argv);
|
||||
};
|
||||
|
||||
void register_era_commands(base::application &app);
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
@@ -262,20 +262,28 @@ namespace {
|
||||
return r;
|
||||
|
||||
}
|
||||
|
||||
void usage(ostream &out, string const &cmd) {
|
||||
out << "Usage: " << cmd << " [options] {device|file}" << endl
|
||||
<< "Options:" << endl
|
||||
<< " {-q|--quiet}" << endl
|
||||
<< " {-h|--help}" << endl
|
||||
<< " {-V|--version}" << endl
|
||||
<< " {--super-block-only}" << endl;
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
int era_check_main(int argc, char **argv)
|
||||
era_check_cmd::era_check_cmd()
|
||||
: command("era_check")
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
era_check_cmd::usage(std::ostream &out) const
|
||||
{
|
||||
out << "Usage: " << get_name() << " [options] {device|file}" << endl
|
||||
<< "Options:" << endl
|
||||
<< " {-q|--quiet}" << endl
|
||||
<< " {-h|--help}" << endl
|
||||
<< " {-V|--version}" << endl
|
||||
<< " {--super-block-only}" << endl;
|
||||
}
|
||||
|
||||
int
|
||||
era_check_cmd::run(int argc, char **argv)
|
||||
{
|
||||
int c;
|
||||
flags fs;
|
||||
@@ -295,7 +303,7 @@ int era_check_main(int argc, char **argv)
|
||||
break;
|
||||
|
||||
case 'h':
|
||||
usage(cout, basename(argv[0]));
|
||||
usage(cout);
|
||||
return 0;
|
||||
|
||||
case 'q':
|
||||
@@ -307,20 +315,18 @@ int era_check_main(int argc, char **argv)
|
||||
return 0;
|
||||
|
||||
default:
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (argc == optind) {
|
||||
cerr << "No input file provided." << endl;
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
return check_with_exception_handling(argv[optind], fs);
|
||||
}
|
||||
|
||||
base::command era::era_check_cmd("era_check", era_check_main);
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
@@ -57,21 +57,29 @@ namespace {
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void usage(ostream &out, string const &cmd) {
|
||||
out << "Usage: " << cmd << " [options] {device|file}" << endl
|
||||
<< "Options:" << endl
|
||||
<< " {-h|--help}" << endl
|
||||
<< " {-o <xml file>}" << endl
|
||||
<< " {-V|--version}" << endl
|
||||
<< " {--repair}" << endl
|
||||
<< " {--logical}" << endl;
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
int era_dump_main(int argc, char **argv)
|
||||
era_dump_cmd::era_dump_cmd()
|
||||
: command("era_dump")
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
era_dump_cmd::usage(std::ostream &out) const
|
||||
{
|
||||
out << "Usage: " << get_name() << " [options] {device|file}" << endl
|
||||
<< "Options:" << endl
|
||||
<< " {-h|--help}" << endl
|
||||
<< " {-o <xml file>}" << endl
|
||||
<< " {-V|--version}" << endl
|
||||
<< " {--repair}" << endl
|
||||
<< " {--logical}" << endl;
|
||||
}
|
||||
|
||||
int
|
||||
era_dump_cmd::run(int argc, char **argv)
|
||||
{
|
||||
int c;
|
||||
flags fs;
|
||||
@@ -98,7 +106,7 @@ int era_dump_main(int argc, char **argv)
|
||||
break;
|
||||
|
||||
case 'h':
|
||||
usage(cout, basename(argv[0]));
|
||||
usage(cout);
|
||||
return 0;
|
||||
|
||||
case 'o':
|
||||
@@ -110,20 +118,18 @@ int era_dump_main(int argc, char **argv)
|
||||
return 0;
|
||||
|
||||
default:
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (argc == optind) {
|
||||
cerr << "No input file provided." << endl;
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
return dump(argv[optind], output, fs);
|
||||
}
|
||||
|
||||
base::command era::era_dump_cmd("era_dump", era_dump_main);
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
@@ -182,20 +182,28 @@ namespace {
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void usage(ostream &out, string const &cmd) {
|
||||
out << "Usage: " << cmd << " [options] --written-since <era> {device|file}\n"
|
||||
<< "Options:\n"
|
||||
<< " {-h|--help}\n"
|
||||
<< " {-o <xml file>}\n"
|
||||
<< " {--metadata-snapshot}\n"
|
||||
<< " {-V|--version}" << endl;
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
int era_invalidate_main(int argc, char **argv)
|
||||
era_invalidate_cmd::era_invalidate_cmd()
|
||||
: command("era_invalidate")
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
era_invalidate_cmd::usage(std::ostream &out) const
|
||||
{
|
||||
out << "Usage: " << get_name() << " [options] --written-since <era> {device|file}\n"
|
||||
<< "Options:\n"
|
||||
<< " {-h|--help}\n"
|
||||
<< " {-o <xml file>}\n"
|
||||
<< " {--metadata-snapshot}\n"
|
||||
<< " {-V|--version}" << endl;
|
||||
}
|
||||
|
||||
int
|
||||
era_invalidate_cmd::run(int argc, char **argv)
|
||||
{
|
||||
int c;
|
||||
flags fs;
|
||||
@@ -222,7 +230,7 @@ int era_invalidate_main(int argc, char **argv)
|
||||
break;
|
||||
|
||||
case 'h':
|
||||
usage(cout, basename(argv[0]));
|
||||
usage(cout);
|
||||
return 0;
|
||||
|
||||
case 'o':
|
||||
@@ -234,26 +242,24 @@ int era_invalidate_main(int argc, char **argv)
|
||||
return 0;
|
||||
|
||||
default:
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (argc == optind) {
|
||||
cerr << "No input file provided." << endl;
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!fs.era_threshold_) {
|
||||
cerr << "Please specify --written-since" << endl;
|
||||
usage(cerr, basename(argv[0]));
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
return invalidate(argv[optind], output, fs);
|
||||
}
|
||||
|
||||
base::command era::era_invalidate_cmd("era_invalidate", era_invalidate_main);
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
@@ -46,23 +46,32 @@ namespace {
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void usage(ostream &out, string const &cmd) {
|
||||
out << "Usage: " << cmd << " [options]" << endl
|
||||
<< "Options:" << endl
|
||||
<< " {-h|--help}" << endl
|
||||
<< " {-i|--input} <input xml file>" << endl
|
||||
<< " {-o|--output} <output device or file>" << endl
|
||||
<< " {-q|--quiet}" << endl
|
||||
<< " {-V|--version}" << endl;
|
||||
}
|
||||
}
|
||||
|
||||
int era_restore_main(int argc, char **argv)
|
||||
//----------------------------------------------------------------
|
||||
|
||||
era_restore_cmd::era_restore_cmd()
|
||||
: command("era_restore")
|
||||
{
|
||||
}
|
||||
|
||||
void
|
||||
era_restore_cmd::usage(std::ostream &out) const
|
||||
{
|
||||
out << "Usage: " << get_name() << " [options]" << endl
|
||||
<< "Options:" << endl
|
||||
<< " {-h|--help}" << endl
|
||||
<< " {-i|--input} <input xml file>" << endl
|
||||
<< " {-o|--output} <output device or file>" << endl
|
||||
<< " {-q|--quiet}" << endl
|
||||
<< " {-V|--version}" << endl;
|
||||
}
|
||||
|
||||
int
|
||||
era_restore_cmd::run(int argc, char **argv)
|
||||
{
|
||||
int c;
|
||||
flags fs;
|
||||
char const *prog_name = basename(argv[0]);
|
||||
char const *short_opts = "hi:o:qV";
|
||||
option const long_opts[] = {
|
||||
{ "help", no_argument, NULL, 'h'},
|
||||
@@ -76,7 +85,7 @@ int era_restore_main(int argc, char **argv)
|
||||
while ((c = getopt_long(argc, argv, short_opts, long_opts, NULL)) != -1) {
|
||||
switch(c) {
|
||||
case 'h':
|
||||
usage(cout, prog_name);
|
||||
usage(cout);
|
||||
return 0;
|
||||
|
||||
case 'i':
|
||||
@@ -96,31 +105,29 @@ int era_restore_main(int argc, char **argv)
|
||||
return 0;
|
||||
|
||||
default:
|
||||
usage(cerr, prog_name);
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (argc != optind) {
|
||||
usage(cerr, prog_name);
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!fs.input) {
|
||||
cerr << "No input file provided." << endl << endl;
|
||||
usage(cerr, prog_name);
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!fs.output) {
|
||||
cerr << "No output file provided." << endl << endl;
|
||||
usage(cerr, prog_name);
|
||||
usage(cerr);
|
||||
return 1;
|
||||
}
|
||||
|
||||
return restore(fs, fs.quiet);
|
||||
}
|
||||
|
||||
base::command era::era_restore_cmd("era_restore", era_restore_main);
|
||||
|
||||
//----------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user