import { type } from 'arktype';
import log from '~server/utils/log';

export type HeadplaneConfig = typeof headplaneConfig.infer;
const stringToBool = type('string | boolean').pipe((v) => Boolean(v));
const serverConfig = type({
	host: 'string.ip',
	port: type('string | number.integer').pipe((v) => Number(v)),
	cookie_secret: '32 <= string <= 32',
	cookie_secure: stringToBool,
});

const oidcConfig = type({
	issuer: 'string.url',
	client_id: 'string',
	client_secret: 'string',
	token_endpoint_auth_method:
		'"client_secret_basic" | "client_secret_post" | "client_secret_jwt"',
	redirect_uri: 'string.url?',
	disable_api_key_login: stringToBool,
	headscale_api_key: 'string',
	strict_validation: stringToBool.default(true),
}).onDeepUndeclaredKey('reject');

const headscaleConfig = type({
	url: 'string.url',
	public_url: 'string.url?',
	config_path: 'string?',
	config_strict: stringToBool,
}).onDeepUndeclaredKey('reject');

const headplaneConfig = type({
	debug: stringToBool,
	server: serverConfig,
	'oidc?': oidcConfig,
	headscale: headscaleConfig,
}).onDeepUndeclaredKey('reject');

const partialHeadplaneConfig = type({
	debug: stringToBool,
	server: serverConfig.partial(),
	'oidc?': oidcConfig.partial(),
	headscale: headscaleConfig.partial(),
}).partial();

export function validateConfig(config: unknown) {
	log.debug('CFGX', 'Validating Headplane configuration...');
	const out = headplaneConfig(config);
	if (out instanceof type.errors) {
		log.error('CFGX', 'Error parsing Headplane configuration:');
		for (const [number, error] of out.entries()) {
			log.error('CFGX', ` (${number}): ${error.toString()}`);
		}

		return;
	}

	log.debug('CFGX', 'Headplane configuration is valid.');
	return out;
}

export function coalesceConfig(config: unknown) {
	log.debug('CFGX', 'Validating coalescing vars for configuration...');
	const out = partialHeadplaneConfig(config);
	if (out instanceof type.errors) {
		log.error('CFGX', 'Error parsing variables:');
		for (const [number, error] of out.entries()) {
			log.error('CFGX', ` (${number}): ${error.toString()}`);
		}

		return;
	}

	log.debug('CFGX', 'Coalescing variables is valid.');
	return out;
}