Files
scale-build/scale_build/config.py
2025-09-10 21:51:02 +02:00

65 lines
2.6 KiB
Python

from os import environ, cpu_count
from time import time
from datetime import datetime
IDENTITY_FILE_PATH_OVERRIDE_SUFFIX = '_OVERRIDE_IDENTITY_FILE_PATH'
_VERS = '26.04.0-MASTER'
def get_env_variable(key, _type, default_value=None):
return get_normalized_value(environ.get(key), _type, default_value)
def get_normalized_value(value, _type, default_value=None):
if value:
if _type == bool:
if value.isdigit():
return _type(int(value))
elif value.lower()[0] == 'y':
return True
elif value.lower()[0] == 'n':
return False
else:
# the variable is set but is something randon (i.e. VAR='please')
# so just assume it's set
return True
else:
return _type(value)
else:
return _type(default_value) if default_value else _type()
APT_BASE_CUSTOM = get_env_variable('APT_BASE_CUSTOM', str)
APT_INTERNAL_BUILD = get_env_variable('APT_INTERNAL_BUILD', bool, False)
BUILD_TIME = int(time())
BUILD_TIME_OBJ = datetime.fromtimestamp(BUILD_TIME)
BUILDER_DIR = get_env_variable('BUILDER_DIR', str, './')
BRANCH_OUT_NAME = get_env_variable('NEW_BRANCH_NAME', str)
BRANCH_OVERRIDES = {}
CCACHE_ENABLED = get_env_variable('CCACHE', bool, 0)
FORCE_CLEANUP_WITH_EPOCH_CHANGE = get_env_variable('FORCE_CLEANUP_WITH_EPOCH_CHANGE', bool)
GITHUB_TOKEN = get_env_variable('GITHUB_TOKEN', str)
PACKAGE_IDENTITY_FILE_PATH_OVERRIDES = {}
PARALLEL_BUILD = get_env_variable('PARALLEL_BUILDS', int, (max(cpu_count(), 8) / 4))
PKG_DEBUG = get_env_variable('PKG_DEBUG', bool, 0)
SECRET_ENV_VARS = {}
SIGNING_KEY = get_env_variable('SIGNING_KEY', str)
SIGNING_PASSWORD = get_env_variable('SIGNING_PASSWORD', str)
SKIP_SOURCE_REPO_VALIDATION = get_env_variable('SKIP_SOURCE_REPO_VALIDATION', bool)
TRAIN = get_env_variable('TRUENAS_TRAIN', str)
TRUENAS_BRANCH_OVERRIDE = get_env_variable('TRUENAS_BRANCH_OVERRIDE', str)
TRY_BRANCH_OVERRIDE = get_env_variable('TRY_BRANCH_OVERRIDE', str)
VERSION = get_env_variable('TRUENAS_VERSION', str, f'{_VERS}-{BUILD_TIME_OBJ.strftime("%Y%m%d-%H%M%S")}')
TRUENAS_VENDOR = get_env_variable('TRUENAS_VENDOR', str)
PRESERVE_ISO = get_env_variable('PRESERVE_ISO', bool, False)
# We will get branch overrides and identity file path overrides from here
for k, v in environ.items():
if k.endswith('_OVERRIDE'):
BRANCH_OVERRIDES[k[:-(len('_OVERRIDE'))]] = v
elif k.endswith(IDENTITY_FILE_PATH_OVERRIDE_SUFFIX):
PACKAGE_IDENTITY_FILE_PATH_OVERRIDES[k[:-(len(IDENTITY_FILE_PATH_OVERRIDE_SUFFIX))]] = v
elif k.startswith('SECRET_'):
SECRET_ENV_VARS[k[len('SECRET_'):]] = v