image_customization_lib.lua

local M = {
	customization_file = nil,
}

local function evaluate_device(env, dev)
	local selections = {
		features = {},
		packages = {},
	}
	local funcs = {}
	local device_overrides = {}

	local function add_elements(element_type, element_list)
		local error_msg = string.format(
			'incorrect use of %s(): list of %s expected as argument',
			element_type, element_type)
		assert(type(element_list) == 'table', error_msg)

		-- We depend on the fact both feature and package
		-- are already initialized as empty tables
		for _, element in ipairs(element_list) do
			assert(type(element) == 'string', error_msg)
			table.insert(selections[element_type], element)
		end
	end

	local function add_override(ovr_key, ovr_value)
		device_overrides[ovr_key] = ovr_value
	end

	function funcs.features(features)
		add_elements('features', features)
	end

	function funcs.packages(packages)
		add_elements('packages', packages)
	end

	function funcs.broken(broken)
		assert(
			type(broken) == 'boolean',
			'incorrect use of broken(): boolean argument expected')
		add_override('broken', broken)
	end

	function funcs.disable()
		add_override('disabled', true)
	end

	function funcs.disable_factory()
		add_override('disable_factory', true)
	end

	function funcs.device(device_names)
		assert(
			type(device_names) == 'table',
			'incorrect use of device(): list of device names expected as argument')

		for _, device_name in ipairs(device_names) do
			assert(
				type(device_name) == 'string',
				'incorrect use of device(): list of device names expected as argument')

			if device_name == dev.image then
				return true
			end
		end

		return false
	end