aboutsummaryrefslogtreecommitdiff
path: root/scripts/gen-default-cfg.py
blob: afcaed2fb7a111d5d2253c48924ab03b83804117 (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
#!/usr/bin/env python3

import sys
import argparse
import kconfiglib

def node_to_str(node):
    if(node.item.type == kconfiglib.STRING):
        return "std::string(\"" + node.item.str_value + "\")"
    elif(node.item.type == kconfiglib.INT):
        return node.item.str_value
    elif(node.item.type == kconfiglib.BOOL):
        tri_val_str = ("false", "unknown", "true")[node.item.tri_value]
        return tri_val_str

    return None

def writeItem(node, file):
    while node:
        if isinstance(node.item, kconfiglib.Symbol):
            name = node.item.name.lower().replace('_', '.')
            print("\t{ \"" + name + "\", " + node_to_str(node) + " },", file=file)

        if node.list:
            writeItem(node.list, file)

        node = node.next

def writeConfigPaths(file):
    # -1: invalid
    #  0: $HOME
    #  1: $XDG_CONFIG_HOME

    locations = [
            "$HOME/.poirc",
            "$XDG_CONFIG_HOME/smolbote.conf",
            "$XDG_CONFIG_HOME/smolbote/config",
            "$HOME/.config/smolbote.conf",
            "$HOME/.config/smolbote/config",
            "/etc/smolbote/config"]

    for item in locations:
        if item.startswith("$HOME/"):
            print('\t\tstd::make_pair(0, "{}"),'.format(item[len("$HOME/"):]), file=file)
        elif item.startswith("$XDG_CONFIG_HOME/"):
            print('\t\tstd::make_pair(1, "{}"),'.format(item[len("$XDG_CONFIG_HOME/"):]), file=file)
        else:
            print('\t\tstd::make_pair(-1, "{}"),'.format(item), file=file)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--kconfig", nargs="?", default="Kconfig", help="Top-level Kconfig")
    parser.add_argument("--dotconfig", nargs="?", help=".config")
    parser.add_argument("--input", type=argparse.FileType('r'), help="settings.h.in")
    parser.add_argument("--output", type=argparse.FileType('w'), default=sys.stdout, help="Output location")
    args = parser.parse_args()

    kconf = kconfiglib.Kconfig(args.kconfig)
    if args.dotconfig is not None:
        kconf.load_config(args.dotconfig)

    print("/* Autogenerated file", file=args.output)
    print(" * kconfig: {}".format(args.kconfig), file=args.output)
    print(" * dotconfig: {}".format(args.dotconfig), file=args.output)
    print(" */", file=args.output)

    for line in args.input:
        if "@__DEFAULT_CFG__" in line:
            writeItem(kconf.top_node.list, file=args.output)
        elif "@__CONFIG_PATHS__" in line:
            writeConfigPaths(file=args.output)
        else:
            print(line, end='', file=args.output)