import { setTimeout } from 'node:timers/promises'

import { Client } from 'undici'

import { loadContext } from './config/headplane'
import { HeadscaleError, pull } from './headscale'

export async function sighupHeadscale() {
	const context = await loadContext()
	if (!context.docker) {
		return
	}

	// Supports the DOCKER_SOCK environment variable
	const client = context.docker.sock
		? new Client('http://localhost', {
			socketPath: context.docker.url,
		})
		: new Client(context.docker.url)

	const response = await client.request({
		method: 'POST',
		path: `/v1.30/containers/${context.docker.container}/kill?signal=SIGHUP`,
	})

	if (!response.statusCode || response.statusCode !== 204) {
		throw new Error('Failed to send SIGHUP to Headscale')
	}
}

export async function restartHeadscale() {
	const context = await loadContext()
	if (!context.docker) {
		return
	}

	// Supports the DOCKER_SOCK environment variable
	const client = context.docker.sock
		? new Client('http://localhost', {
			socketPath: context.docker.url,
		})
		: new Client(context.docker.url)

	const response = await client.request({
		method: 'POST',
		path: `/v1.30/containers/${context.docker.container}/restart`,
	})

	if (!response.statusCode || response.statusCode !== 204) {
		throw new Error('Failed to restart Headscale')
	}

	// Wait for Headscale to restart before continuing
	let attempts = 0
	// eslint-disable-next-line
	while (true) {
		try {
			// Acceptable blank because ROOT_API_KEY is not required
			await pull('v1/apikey', context.oidc?.rootKey ?? '')
			return
		} catch (error) {
			// This means the server is up but the API key is invalid
			// This can happen if the user only uses ROOT_API_KEY via cookies
			if (error instanceof HeadscaleError && error.status === 401) {
				break
			}

			if (attempts > 10) {
				throw new Error('Headscale did not restart in time')
			}

			attempts++
			await setTimeout(1000)
		}
	}
}