diff --git a/api/src/config.js b/api/src/config.js index b8fa97d..d207431 100644 --- a/api/src/config.js +++ b/api/src/config.js @@ -2,119 +2,139 @@ const fss = require('fs'); const Logger = require('logplease'); const logger = Logger.create('config'); -const options = { - log_level: { +const options = [ + { + key: 'log_level', desc: 'Level of data to log', default: 'INFO', + options: Object.values(Logger.LogLevels), validators: [ x => Object.values(Logger.LogLevels).includes(x) || `Log level ${x} does not exist`, ], }, - bind_address: { + { + key: 'bind_address', desc: 'Address to bind REST API on', default: `0.0.0.0:${process.env['PORT'] || 2000}`, validators: [], }, - data_directory: { + { + key: 'data_directory', desc: 'Absolute path to store all piston related data at', default: '/piston', validators: [ x => fss.exists_sync(x) || `Directory ${x} does not exist`, ], }, - runner_uid_min: { + { + key: 'runner_uid_min', desc: 'Minimum uid to use for runner', default: 1001, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - runner_uid_max: { + { + key: 'runner_uid_max', desc: 'Maximum uid to use for runner', default: 1500, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - runner_gid_min: { + { + key: 'runner_gid_min', desc: 'Minimum gid to use for runner', default: 1001, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - runner_gid_max: { + { + key: 'runner_gid_max', desc: 'Maximum gid to use for runner', default: 1500, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - disable_networking: { + { + key: 'disable_networking', desc: 'Set to true to disable networking', default: true, parser: x => x === 'true', validators: [x => typeof x === 'boolean' || `${x} is not a boolean`], }, - output_max_size: { + { + key: 'output_max_size', desc: 'Max size of each stdio buffer', default: 1024, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - max_process_count: { + { + key: 'max_process_count', desc: 'Max number of processes per job', default: 64, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - max_open_files: { + { + key: 'max_open_files', desc: 'Max number of open files per job', default: 2048, parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - max_file_size: { + { + key: 'max_file_size', desc: 'Max file size in bytes for a file', default: 10000000, //10MB parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - compile_timeout: { + { + key: 'compile_timeout', desc: 'Max time allowed for compile stage in milliseconds', default: 10000, // 10 seconds parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - run_timeout: { + { + key: 'run_timeout', desc: 'Max time allowed for run stage in milliseconds', default: 3000, // 3 seconds parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - compile_memory_limit: { + { + key: 'compile_memory_limit', desc: 'Max memory usage for compile stage in bytes (set to -1 for no limit)', default: -1, // no limit parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - run_memory_limit: { + { + key: 'run_memory_limit', desc: 'Max memory usage for run stage in bytes (set to -1 for no limit)', default: -1, // no limit parser: parse_int, validators: [(x, raw) => !is_nan(x) || `${raw} is not a number`], }, - repo_url: { + { + key: 'repo_url', desc: 'URL of repo index', default: 'https://github.com/engineer-man/piston/releases/download/pkgs/index', validators: [], }, - max_concurrent_jobs: { + { + key: 'max_concurrent_jobs', desc: 'Maximum number of concurrent jobs to run at one time', default: 64, parser: parse_int, validators: [x => x > 0 || `${x} cannot be negative`], }, - limit_overrides: { + { + key: 'limit_overrides', desc: 'Per-language exceptions in JSON format for each of:\ max_process_count, max_open_files, max_file_size, compile_memory_limit,\ run_memory_limit, compile_timeout, run_timeout, output_max_size', @@ -125,7 +145,7 @@ const options = { validate_overrides, ], }, -}; +]; Object.freeze(options); @@ -149,7 +169,7 @@ function parse_overrides(overrides_string) { } const overrides = get_parsed_json_or_null(overrides_string); - if (overrides === null) { + if (typeof overrides === null) { return null; } const parsed_overrides = {}; @@ -171,11 +191,11 @@ function parse_overrides(overrides_string) { return null; } // Find the option for the override - const option = options[key]; + const option = options.find(o => o.key === key); const parser = option.parser; - const raw_value = overrides[language][key]; - const parsed_value = parser(raw_value); - parsed_overrides[language][key] = parsed_value; + const raw = overrides[language][key]; + const value = parser(raw); + parsed_overrides[language][key] = value; } } return parsed_overrides; @@ -185,7 +205,7 @@ function validate_overrides(overrides) { for (const language in overrides) { for (const key in overrides[language]) { const value = overrides[language][key]; - const option = options[key]; + const option = options.find(o => o.key === key); const validators = option.validators; const validation_response = apply_validators(validators, [ value, @@ -203,9 +223,8 @@ logger.info(`Loading Configuration from environment`); let config = {}; -for (const option_name in options) { - const env_key = 'PISTON_' + option_name.to_upper_case(); - const option = options[option_name]; +options.forEach(option => { + const env_key = 'PISTON_' + option.key.to_upper_case(); const parser = option.parser || (x => x); const env_val = process.env[env_key]; const parsed_val = parser(env_val); @@ -218,13 +237,13 @@ for (const option_name in options) { ); if (validation_response !== true) { logger.error( - `Config option ${option_name} failed validation:`, + `Config option ${option.key} failed validation:`, validation_response ); process.exit(1); } - config[option_name] = value; -} + config[option.key] = value; +}); logger.info('Configuration successfully loaded');