feat: restart docker container

app/utils/docker.ts

@@ -0,0 +1,44 @@
+/* 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 { pull } from './headscale'
+
+export async function restartHeadscale() {
+	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)
+		}
+	}
+}

app/utils/headscale.ts

@@ -9,50 +9,42 @@ export class HeadscaleError extends Error {
 }
 
 export class FatalError extends Error {
-	constructor(message: string) {
-		super(message)
+	constructor() {
+		super('The Headscale server is not accessible or the API_KEY is invalid.')
 		this.name = 'FatalError'
 	}
 }
 
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
 export async function pull<T>(url: string, key: string) {
-	try {
-		const prefix = process.env.HEADSCALE_URL!
-		const response = await fetch(`${prefix}/api/${url}`, {
-			headers: {
-				Authorization: `Bearer ${key}`
-			}
-		})
-
-		if (!response.ok) {
-			throw new HeadscaleError(await response.text(), response.status)
+	const prefix = process.env.HEADSCALE_URL!
+	const response = await fetch(`${prefix}/api/${url}`, {
+		headers: {
+			Authorization: `Bearer ${key}`
 		}
+	})
 
-		return await (response.json() as Promise<T>)
-	} catch {
-		throw new FatalError('The Headscale server is not reachable')
+	if (!response.ok) {
+		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) {
-	try {
-		const prefix = process.env.HEADSCALE_URL!
-		const response = await fetch(`${prefix}/api/${url}`, {
-			method: 'POST',
-			body: body ? JSON.stringify(body) : undefined,
-			headers: {
-				Authorization: `Bearer ${key}`
-			}
-		})
-
-		if (!response.ok) {
-			throw new HeadscaleError(await response.text(), response.status)
+	const prefix = process.env.HEADSCALE_URL!
+	const response = await fetch(`${prefix}/api/${url}`, {
+		method: 'POST',
+		body: body ? JSON.stringify(body) : undefined,
+		headers: {
+			Authorization: `Bearer ${key}`
 		}
+	})
 
-		return await (response.json() as Promise<T>)
-	} catch {
-		throw new FatalError('The Headscale server is not reachable')
+	if (!response.ok) {
+		throw new HeadscaleError(await response.text(), response.status)
 	}
+
+	return (response.json() as Promise<T>)
 }