import { loadContext } from './config/headplane';
import log from './log';

export class HeadscaleError extends Error {
	status: number;

	constructor(message: string, status: number) {
		super(message);
		this.name = 'HeadscaleError';
		this.status = status;
	}
}

export class FatalError extends Error {
	constructor() {
		super(
			'The Headscale server is not accessible or the supplied API key is invalid',
		);
		this.name = 'FatalError';
	}
}

export async function pull<T>(url: string, key: string) {
	if (!key || key === 'undefined' || key.length === 0) {
		throw new Error('Missing API key, could this be a cookie setting issue?');
	}

	const context = await loadContext();
	const prefix = context.headscaleUrl;

	log.debug('APIC', 'GET %s', `${prefix}/api/${url}`);
	const response = await fetch(`${prefix}/api/${url}`, {
		headers: {
			Authorization: `Bearer ${key}`,
		},
	});

	if (!response.ok) {
		log.debug(
			'APIC',
			'GET %s failed with status %d',
			`${prefix}/api/${url}`,
			response.status,
		);
		throw new HeadscaleError(await response.text(), response.status);
	}

	return response.json() as Promise<T>;
}

export async function post<T>(url: string, key: string, body?: unknown) {
	if (!key || key === 'undefined' || key.length === 0) {
		throw new Error('Missing API key, could this be a cookie setting issue?');
	}

	const context = await loadContext();
	const prefix = context.headscaleUrl;

	log.debug('APIC', 'POST %s', `${prefix}/api/${url}`);
	const response = await fetch(`${prefix}/api/${url}`, {
		method: 'POST',
		body: body ? JSON.stringify(body) : undefined,
		headers: {
			Authorization: `Bearer ${key}`,
		},
	});

	if (!response.ok) {
		log.debug(
			'APIC',
			'POST %s failed with status %d',
			`${prefix}/api/${url}`,
			response.status,
		);
		throw new HeadscaleError(await response.text(), response.status);
	}

	return response.json() as Promise<T>;
}

export async function put<T>(url: string, key: string, body?: unknown) {
	if (!key || key === 'undefined' || key.length === 0) {
		throw new Error('Missing API key, could this be a cookie setting issue?');
	}

	const context = await loadContext();
	const prefix = context.headscaleUrl;

	log.debug('APIC', 'PUT %s', `${prefix}/api/${url}`);
	const response = await fetch(`${prefix}/api/${url}`, {
		method: 'PUT',
		body: body ? JSON.stringify(body) : undefined,
		headers: {
			Authorization: `Bearer ${key}`,
		},
	});

	if (!response.ok) {
		log.debug(
			'APIC',
			'PUT %s failed with status %d',
			`${prefix}/api/${url}`,
			response.status,
		);
		throw new HeadscaleError(await response.text(), response.status);
	}

	return response.json() as Promise<T>;
}

export async function del<T>(url: string, key: string) {
	if (!key || key === 'undefined' || key.length === 0) {
		throw new Error('Missing API key, could this be a cookie setting issue?');
	}

	const context = await loadContext();
	const prefix = context.headscaleUrl;

	log.debug('APIC', 'DELETE %s', `${prefix}/api/${url}`);
	const response = await fetch(`${prefix}/api/${url}`, {
		method: 'DELETE',
		headers: {
			Authorization: `Bearer ${key}`,
		},
	});

	if (!response.ok) {
		log.debug(
			'APIC',
			'DELETE %s failed with status %d',
			`${prefix}/api/${url}`,
			response.status,
		);
		throw new HeadscaleError(await response.text(), response.status);
	}

	return response.json() as Promise<T>;
}
