summaryrefslogtreecommitdiff
path: root/src/options.py
blob: 4c419f2880598b987273260b8343ad96335ce07d (plain)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import sys

options = {
        "h": {
                "name": "help",
                "flag" : True,
                "desc" : "prints the command usage and exists the program",
            },
        "y" : {
                "name" : "no-confirm",
                "flag" : True,
                "desc": "will not prompt the user"
            },
        "r" : {
                "name" : "root",
                "flag" : False,
                "desc" : "specify the directory to use as the system root",
                "default" : "/"
            },
        "l": {
                "name" : "no-sync",
                "flag" : True,
                "desc" : "skip syncing with repo sources (not recommended)"
            },
        "v": {
                "name" : "verbose",
                "flag" : True,
                "desc" : "print more"
            },
        "c": {
                "name" : "config",
                "flag" : False,
                "desc" : "specify the configuration file to use",
                "default" : "/etc/xipkg.conf"
            }
        }

def parse_args():

    # re-organise the options by name rather than by single letter
    # a dict with "name": option_leter
    names = { v["name"] if v["name"] else k : k for k,v in options.items()}

    args = sys.argv
    index = 1

    # save all of the options into a "parsed" dictionary
    parsed = {"args" : []}

    while index < len(args):
        arg = args[index]

        if len(arg) > 1 and arg[0] == "-":
            option = None

            # is a named argument with a --
            if arg[1] == "-" and len(arg) > 2 and arg[2:].split("=")[0] in names:
                option = names[arg[2:].split("=")[0]]
            # is a single letter argument with a -
            elif arg[1] in options:
                option = arg[1]
            else:
                parsed["args"].append(arg)

            # add the option and any values ot the parsed dict
            if option is not None:
                if options[option]["flag"]:
                    parsed[option] = True
                else:
                    if "=" in arg:
                        parsed[option] = arg.split("=")[1]
                    else:
                        index += 1
                        parsed[option] = args[index]
        else:
            parsed["args"].append(arg)
    

        index += 1

    # add all default values to the parsed options
    for option in options:
        if not option in parsed:
            if options[option]["flag"]:
                parsed[option] = False
            else:
                parsed[option] = options[option]["default"]

    return parsed

def print_usage():
    for option,o in options.items():
        name = o["name"]
        description = o["desc"]
        d = ("[default=" + o["default"] + "]") if not o["flag"] else ""

        print(f"\t-{option}, --{name}\t{d}")
        print(f"\t\t{description}\n")

        if "verbs" in globals():
            print("Available actions:")
            for verb in verbs:
                print(f"\t{verb}")