fileutil_main.cc

#include <string>
#include "absl/flags/flag.h"
#include "glog/logging.h"
#include "cpp/core/galaxy_flag.h"
#include "cpp/tool/fileutil.h"
#include "cpp/util/galaxy_util.h"

int main(int argc, char* argv[]) {
    absl::StatusOr<galaxy_schema::CellConfig> config = galaxy::util::ParseCellConfig(absl::GetFlag(FLAGS_fs_cell));
    if (config.ok()) {
        FLAGS_colorlogtostderr = true;
        FLAGS_log_dir = config->fs_log_dir();;
        FLAGS_alsologtostderr = config->fs_alsologtostderr();
        FLAGS_stderrthreshold = 3;
        google::EnableLogCleaner(config->fs_log_ttl());
    } else {
        LOG(WARNING) << "Calling from a server outside galaxy.";
    }
    google::InitGoogleLogging(argv[0]);
    CHECK_GT(argc, 1) << "Need more than 1 arguments";
    LOG(INFO) << "Getting cmd:";
    for (int i = 1; i < argc; i++) {
        LOG(INFO) << "\tArg[" << i << "]: " << argv[i];
    }
    if (strcmp(argv[1], "ls") == 0) {
        CHECK_EQ(argc,  3) << "Need 2 arguments for ls cmd.";
        galaxy::LsCmd(argv[2]);
    } else if (strcmp(argv[1], "cp_file") == 0) {
        CHECK_GE(argc,  4) << "Need at least 3 arguments for cp cmd.";
        bool overwrite = false;
        if (argc == 5 && strcmp(argv[4], "--f") == 0) {
            overwrite = true;
        }
        galaxy::CopyFileCmd(argv[2], argv[3], overwrite);
    } else if (strcmp(argv[1], "mv_file") == 0) {
        CHECK_GE(argc,  4) << "Need at least 3 arguments for mv cmd.";
        bool overwrite = false;
        if (argc == 5 && strcmp(argv[4], "--f") == 0) {
            overwrite = true;
        }
        galaxy::MoveFileCmd(argv[2], argv[3], overwrite);
    } else if (strcmp(argv[1], "cp_dir") == 0) {
        CHECK_GE(argc,  4) << "Need at least 3 arguments for cp cmd.";
        bool overwrite = false;
        if (argc == 5 && strcmp(argv[4], "--f") == 0) {
            overwrite = true;
        }
        galaxy::CopyDirCmd(argv[2], argv[3], overwrite);
    } else if (strcmp(argv[1], "mv_dir") == 0) {
        CHECK_GE(argc,  4) << "Need at least 3 arguments for mv cmd.";
        bool overwrite = false;
        if (argc == 5 && strcmp(argv[4], "--f") == 0) {
            overwrite = true;
        }
        galaxy::MoveDirCmd(argv[2], argv[3], overwrite);
    } else if (strcmp(argv[1], "rm") == 0) {
        CHECK_GE(argc,  3) << "Need at least 2 arguments for cp cmd.";
        bool recursive = false;
        if (argc == 4 && strcmp(argv[3], "--r") == 0) {
            recursive = true;
        }
        galaxy::RmCmd(argv[2], recursive);
    } else if (strcmp(argv[1], "lscells") == 0) {
        galaxy::ListCellsCmd();
    } else if (strcmp(argv[1], "enable_cell") == 0) {
        galaxy::ModifyCellCmd(argv[2], true);
    } else if (strcmp(argv[1], "disable_cell") == 0) {
        galaxy::ModifyCellCmd(argv[2], false);
    } else {
        LOG(FATAL) << "Wrong cmd.";
    }
}