aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/getopt/small/last_getopt_easy_setup.cpp
blob: c87dedf95ee20cf070fbc3692896cb09b0270fd1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#include "last_getopt_easy_setup.h"

namespace NLastGetopt {
    TEasySetup::TEasySetup(const TStringBuf& optstring)
        : TOpts(optstring)
    {
        AddHelpOption();
    }

    TOpt& TEasySetup::AdjustParam(const char* longName, const char* help, const char* argName, bool required) {
        Y_ASSERT(longName);
        TOpt& o = AddLongOption(longName);
        if (help) {
            o.Help(help);
        }
        if (argName) {
            o.RequiredArgument(argName);
        } else {
            o.HasArg(NO_ARGUMENT);
        }
        if (required) {
            o.Required();
        }
        return o;
    }

    TEasySetup& TEasySetup::operator()(char shortName, const char* longName, const char* help, bool required) {
        AdjustParam(longName, help, nullptr, required).AddShortName(shortName);
        return *this;
    }

    TEasySetup& TEasySetup::operator()(char shortName, const char* longName, const char* argName, const char* help, bool required) {
        AdjustParam(longName, help, argName, required).AddShortName(shortName);
        return *this;
    }

    TEasySetup& TEasySetup::operator()(const char* longName, const char* help, bool required) {
        AdjustParam(longName, help, nullptr, required);
        return *this;
    }

    TEasySetup& TEasySetup::operator()(const char* longName, const char* argName, const char* help, bool required) {
        AdjustParam(longName, help, argName, required);
        return *this;
    }

}