import type { Plugin } from "@opencode-ai/plugin"

type TurnState = {
	summaryCalled: boolean
}

const states = new Map<string, TurnState>()

function sessionState(sessionID: string) {
	let state = states.get(sessionID)
	if (!state) {
		state = { summaryCalled: false }
		states.set(sessionID, state)
	}
	return state
}

function eventSessionID(event: any) {
	return String(
		event?.properties?.sessionID ||
			event?.properties?.session?.id ||
			event?.properties?.info?.sessionID ||
			"",
	).trim()
}

export const RequireWorkSummaryPlugin: Plugin = async () => {
	return {
		"chat.message": async (input: any) => {
			const sessionID = String(input?.sessionID || "").trim()
			if (!sessionID) {
				return
			}
			sessionState(sessionID).summaryCalled = false
		},

		"command.execute.before": async (input: any) => {
			const sessionID = String(input?.sessionID || "").trim()
			if (!sessionID) {
				return
			}
			sessionState(sessionID).summaryCalled = false
		},

		"tool.execute.before": async (input: any) => {
			const sessionID = String(input?.sessionID || "").trim()
			if (!sessionID) {
				return
			}

			const tool = String(input?.tool || "").trim()
			if (!tool) {
				return
			}

			const state = sessionState(sessionID)
			if (tool === "set_work_summary") {
				return
			}

			if (!state.summaryCalled) {
				throw new Error(
					"Call set_work_summary first with a specific theme and next-step now label before using other tools.",
				)
			}
		},

		"tool.execute.after": async (input: any) => {
			const sessionID = String(input?.sessionID || "").trim()
			if (!sessionID) {
				return
			}

			const tool = String(input?.tool || "").trim()
			if (tool === "set_work_summary") {
				sessionState(sessionID).summaryCalled = true
			}
		},

		event: async ({ event }) => {
			if (event?.type === "session.deleted") {
				const sessionID = eventSessionID(event)
				if (sessionID) {
					states.delete(sessionID)
				}
				return
			}

		},
	}
}

export default RequireWorkSummaryPlugin