// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "proxycmd.h"
#include "methods.h"
#include <vespa/fastos/app.h>
#include <iostream>

class Application : public FastOS_Application
{
    Flags _flags;
    bool parseOpts();
public:
    void usage(void);
    int Main(void) override;

    Application() : _flags() {}
};

bool
Application::parseOpts()
{
    char c = '?';
    const char *optArg = NULL;
    int optInd = 0;
    while ((c = GetOpt("m:s:p:", optArg, optInd)) != -1) {
        switch (c) {
        case 'm':
            _flags.method = optArg;
            break;
        case 's':
            _flags.hostname = optArg;
            break;
        case 'p':
            _flags.portnumber = atoi(optArg);
            break;
        case 'h':
        default:
            return false; // triggers usage()
        }
    }
    const Method method = methods::find(_flags.method);
    if (optInd + method.args <= _argc) {
        for (int i = 0; i < method.args; ++i) {
            vespalib::string arg = _argv[optInd++];
            _flags.args.push_back(arg);
        }
    } else {
        std::cerr << "ERROR: method "<< _flags.method << " requires " << method.args
                  << " arguments, only got " << (_argc - optInd) << std::endl;
        return false;
    }
    if (optInd != _argc) {
        std::cerr << "ERROR: "<<(_argc - optInd)<<" extra arguments\n";
        return false;
    }
    _flags.method = method.rpcMethod;
    return true;
}

void
Application::usage(void)
{
    std::cerr <<
        "Usage: configproxy-cmd [options]"                                 << std::endl <<
        "    -m <method>                   method"                         << std::endl <<
        "    -s <hostname>                 hostname (default: localhost)"  << std::endl <<
        "    -p <port>                     port number (default: 19090)"   << std::endl <<
        "Available methods for -m option:"                                 << std::endl;
    methods::dump();
}

int
Application::Main(void)
{
    if (! parseOpts()) {
        usage();
        return 1;
    }
    ProxyCmd client(_flags);
    return client.action();
}

int
main(int argc, char** argv)
{
    Application app;
    return app.Entry(argc, argv);
}