51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
# Copyright (c) Kuba Szczodrzyński 2022-06-13.
|
|
|
|
from os.path import isfile
|
|
|
|
from SCons.Script import DefaultEnvironment
|
|
|
|
env = DefaultEnvironment()
|
|
|
|
|
|
def env_load_config(env, path):
|
|
path = env.subst(path)
|
|
if not isfile(path):
|
|
raise FileNotFoundError("Config file not found")
|
|
STRIP_CHARS = "\t "
|
|
config = {}
|
|
f = open(path, "r", encoding="utf-8")
|
|
for line in f:
|
|
line: str
|
|
if not line.startswith("#define"):
|
|
continue
|
|
line = line[7:].strip(STRIP_CHARS)
|
|
(key, value) = line.split(None, 2)
|
|
value = value.strip(STRIP_CHARS)
|
|
if value.isnumeric():
|
|
value = int(value, 0)
|
|
elif value.startswith('"') and value.endswith('"'):
|
|
value = value[1:-1]
|
|
else:
|
|
# store defines as bytes
|
|
value = value.encode()
|
|
config[key] = value
|
|
env.Append(
|
|
CONFIG=config,
|
|
)
|
|
f.close()
|
|
|
|
|
|
def env_get_config(env, key):
|
|
config: dict = env["CONFIG"]
|
|
if not config:
|
|
return None
|
|
value = config.get(key, None)
|
|
while isinstance(value, bytes):
|
|
value = config.get(value.decode(), None)
|
|
return value
|
|
|
|
|
|
env.AddMethod(env_load_config, "LoadConfig")
|
|
env.AddMethod(env_get_config, "Cfg")
|
|
env.AddMethod(env_get_config, "GetConfig")
|