/* eslint-disable @typescript-eslint/no-non-null-assertion */
/* eslint-disable no-await-in-loop */
/* eslint-disable no-constant-condition */
import { setTimeout } from 'node:timers/promises'

import { Client } from 'undici'

import { getContext } from './config'
import { pull } from './headscale'

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

	if (!process.env.HEADSCALE_CONTAINER) {
		throw new Error('HEADSCALE_CONTAINER is not set')
	}

	const client = new Client('http://localhost', {
		socketPath: '/var/run/docker.sock'
	})

	const container = process.env.HEADSCALE_CONTAINER
	const response = await client.request({
		method: 'POST',
		path: `/v1.30/containers/${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 getContext()
	if (!context.hasDockerSock) {
		return
	}

	if (!process.env.HEADSCALE_CONTAINER) {
		throw new Error('HEADSCALE_CONTAINER is not set')
	}

	const client = new Client('http://localhost', {
		socketPath: '/var/run/docker.sock'
	})

	const container = process.env.HEADSCALE_CONTAINER
	const response = await client.request({
		method: 'POST',
		path: `/v1.30/containers/${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
	while (true) {
		try {
			await pull('v1/apikey', process.env.API_KEY!)
			return
		} catch {
			if (attempts > 10) {
				throw new Error('Headscale did not restart in time')
			}

			attempts++
			await setTimeout(1000)
		}
	}
}