import { exec } from "child_process"
import https from "https"
import os from "node:os"
import fetch from "node-fetch"
import {
	DEEP_LINK_PATH_REFRESH_DEV_EXTENSION,
	DESKTOP_SERVICE_NAME,
	KUNKUN_DESKTOP_APP_SERVER_PORTS
} from "../constants"

const httpsAgent = new https.Agent({
	rejectUnauthorized: false // Accept self-signed certificates
})

export function checkLocalKunkunService(port: number): Promise<boolean> {
	return fetch(`https://localhost:${port}/info`, { agent: httpsAgent })
		.then((res) => {
			if (!res.ok) {
				return false
			}
			return res.json()
		})
		.then((data: any) => {
			return data["service_name"].toLowerCase() === DESKTOP_SERVICE_NAME.toLowerCase()
		})
		.catch((err) => {
			// console.error("Failed to check local kunkun service", err)
			// fetch fail, i.e. server not on this port
			return false
		})
}

export async function findLocalhostKunkunPorts(): Promise<number[]> {
	const onlinePorts = []
	for (const port of KUNKUN_DESKTOP_APP_SERVER_PORTS) {
		const online = await checkLocalKunkunService(port)
		if (online) {
			onlinePorts.push(port)
		}
	}
	return onlinePorts
}

export async function refreshTemplateWorkerCommandViaServer() {
	process.env.NODE_TLS_REJECT_UNAUTHORIZED = "0"
	const ports = await findLocalhostKunkunPorts()
	console.log("Kunkun ports", ports)
	if (ports.length === 0) {
		console.warn("\x1b[33mFailed to find localhost kunkun ports\x1b[0m")
		return
	} else if (ports.length > 1) {
		console.warn("Found multiple localhost kunkun ports", ports)
		console.warn("Will Refresh Every Instance")
	}
	for (const port of ports) {
		fetch(`https://localhost:${port}/refresh-worker-extension`, {
			method: "POST",
			agent: httpsAgent
		}).catch((err) => {
			console.error("Failed to send refresh worker extension request", err)
		})
	}
}

export async function refreshTemplateWorkerCommandViaDeepLink() {
	console.log("Send Refresh Worker Extension Request")

	const platform = await os.platform()
	try {
		switch (platform) {
			case "darwin":
				exec(`open ${DEEP_LINK_PATH_REFRESH_DEV_EXTENSION}`)
				break
			case "win32":
				exec(`start ${DEEP_LINK_PATH_REFRESH_DEV_EXTENSION}`)
				break
			case "linux":
				exec(`xdg-open ${DEEP_LINK_PATH_REFRESH_DEV_EXTENSION}`)
				break
		}
	} catch (error) {
		console.error("Failed to refresh worker extension:", error)
	}
}

export const refreshTemplateWorkerCommand = refreshTemplateWorkerCommandViaServer

export function kununWorkerTemplateCommandRollupPlugin() {
	return {
		async writeBundle() {
			await refreshTemplateWorkerCommandViaDeepLink()
		}
	}
}