import { Elysia, error, t } from "elysia";
import { authPlugin } from "$api/authPlugin";
import { collections } from "$lib/server/database";
import { ObjectId } from "mongodb";
import { authCondition } from "$lib/server/auth";
import { validModelIdSchema } from "$lib/server/models";
import { convertLegacyConversation } from "$lib/utils/tree/convertLegacyConversation";
import type { Conversation } from "$lib/types/Conversation";
import type { MetacognitiveEvent } from "$lib/types/Message";
import { logger } from "$lib/server/logger";
import { getMetacognitiveConfig, selectRandomFrequency } from "$lib/server/metacognitiveConfig";

import { CONV_NUM_PER_PAGE } from "$lib/constants/pagination";

export const conversationGroup = new Elysia().use(authPlugin).group("/conversations", (app) => {
	return (
		app
			.guard({
				as: "scoped",
				beforeHandle: async ({ locals }) => {
					if (!locals.user?._id && !locals.sessionId) {
						return error(401, "Must have a valid session or user");
					}
				},
			})
			.get(
				"",
				async ({ locals, query }) => {
					const convs = await collections.conversations
						.find(authCondition(locals))
						.project<Pick<Conversation, "_id" | "title" | "updatedAt" | "model">>({
							title: 1,
							updatedAt: 1,
							model: 1,
						})
						.sort({ updatedAt: -1 })
						.skip((query.p ?? 0) * CONV_NUM_PER_PAGE)
						.limit(CONV_NUM_PER_PAGE)
						.toArray();

					const nConversations = await collections.conversations.countDocuments(
						authCondition(locals)
					);

					const res = convs.map((conv) => ({
						_id: conv._id,
						id: conv._id, // legacy param iOS
						title: conv.title,
						updatedAt: conv.updatedAt,
						model: conv.model,
						modelId: conv.model, // legacy param iOS
					}));

					return { conversations: res, nConversations };
				},
				{
					query: t.Object({
						p: t.Optional(t.Number()),
					}),
				}
			)
			.delete("", async ({ locals }) => {
				const res = await collections.conversations.deleteMany({
					...authCondition(locals),
				});
				return res.deletedCount;
			})
			// search endpoint removed
			.group(
				"/:id",
				{
					params: t.Object({
						id: t.String(),
					}),
				},
				(app) => {
					return app
						.derive(async ({ locals, params }) => {
							let conversation;
							let shared = false;
							const metacognitiveConfig = getMetacognitiveConfig();

							// if the conver
							if (params.id.length === 7) {
								// shared link of length 7
								conversation = await collections.sharedConversations.findOne({
									_id: params.id,
								});
								shared = true;

								if (!conversation) {
									throw new Error("Conversation not found");
								}
							} else {
								// todo: add validation on params.id
								try {
									new ObjectId(params.id);
								} catch {
									throw new Error("Invalid conversation ID format");
								}
								conversation = await collections.conversations.findOne({
									_id: new ObjectId(params.id),
									...authCondition(locals),
								});

								if (!conversation) {
									const conversationExists =
										(await collections.conversations.countDocuments({
											_id: new ObjectId(params.id),
										})) !== 0;

									if (conversationExists) {
										throw new Error(
											"You don't have access to this conversation. If someone gave you this link, ask them to use the 'share' feature instead."
										);
									}

									throw new Error("Conversation not found.");
								}

								// Initialize metacognitiveState server-side so targetFrequency is consistent across refresh/devices.
								if (metacognitiveConfig.enabled) {
									const existingTarget = conversation.metacognitiveState?.targetFrequency;
									if (!existingTarget) {
										const metacognitiveState = {
											targetFrequency: selectRandomFrequency(),
											lastPromptedAtMessageId: null,
											updatedAt: new Date(),
										};
										await collections.conversations.updateOne(
											{ _id: new ObjectId(params.id), ...authCondition(locals) },
											{ $set: { metacognitiveState } }
										);
										conversation.metacognitiveState = metacognitiveState;
									}
								}
							}

							const convertedConv = {
								...conversation,
								...convertLegacyConversation(conversation),
								shared,
							};

							return { conversation: convertedConv };
						})
						.get("", async ({ conversation }) => {
							return {
								messages: conversation.messages,
								title: conversation.title,
								model: conversation.model,
								preprompt: conversation.preprompt,
								rootMessageId: conversation.rootMessageId,
								id: conversation._id.toString(),
								updatedAt: conversation.updatedAt,
								modelId: conversation.model,
								shared: conversation.shared,
								metacognitiveState: (conversation as Conversation).metacognitiveState,
							};
						})
						.post("", () => {
							// todo: post new message
							throw new Error("Not implemented");
						})
						.delete("", async ({ locals, params }) => {
							const res = await collections.conversations.deleteOne({
								_id: new ObjectId(params.id),
								...authCondition(locals),
							});

							if (res.deletedCount === 0) {
								throw new Error("Conversation not found");
							}

							return { success: true };
						})
						.get("/output/:sha256", () => {
							// todo: get output
							throw new Error("Not implemented");
						})
						.post("/share", () => {
							// todo: share conversation
							throw new Error("Not implemented");
						})
						.post("/stop-generating", () => {
							// todo: stop generating
							throw new Error("Not implemented");
						})
						.patch(
							"",
							async ({ locals, params, body }) => {
								if (body.model) {
									if (!validModelIdSchema.safeParse(body.model).success) {
										throw new Error("Invalid model ID");
									}
								}

								// Only include defined values in the update (sanitize title)
								const updateValues = {
									...(body.title !== undefined && {
										title: body.title.replace(/<\/?think>/gi, "").trim(),
									}),
									...(body.model !== undefined && { model: body.model }),
								};

								const res = await collections.conversations.updateOne(
									{
										_id: new ObjectId(params.id),
										...authCondition(locals),
									},
									{
										$set: updateValues,
									}
								);

								if (res.modifiedCount === 0) {
									throw new Error("Conversation not found");
								}

								return { success: true };
							},
							{
								body: t.Object({
									title: t.Optional(
										t.String({
											minLength: 1,
											maxLength: 100,
										})
									),
									model: t.Optional(t.String()),
								}),
							}
						)
						.delete(
							"/message/:messageId",
							async ({ locals, params, conversation }) => {
								if (!conversation.messages.map((m) => m.id).includes(params.messageId)) {
									throw new Error("Message not found");
								}

								const filteredMessages = conversation.messages
									.filter(
										(message) =>
											// not the message AND the message is not in ancestors
											!(message.id === params.messageId) &&
											message.ancestors &&
											!message.ancestors.includes(params.messageId)
									)
									.map((message) => {
										// remove the message from children if it's there
										if (message.children && message.children.includes(params.messageId)) {
											message.children = message.children.filter(
												(child) => child !== params.messageId
											);
										}
										return message;
									});

								const res = await collections.conversations.updateOne(
									{ _id: new ObjectId(conversation._id), ...authCondition(locals) },
									{ $set: { messages: filteredMessages } }
								);

								if (res.modifiedCount === 0) {
									throw new Error("Deleting message failed");
								}

								return { success: true };
							},
							{
								params: t.Object({
									id: t.String(),
									messageId: t.String(),
								}),
							}
						)
						.post(
							"/message/:messageId/metacognitive-event",
							async ({ locals, params, body, conversation }) => {
								// Find the message in the conversation
								const messageIndex = conversation.messages.findIndex(
									(m) => m.id === params.messageId
								);
								if (messageIndex === -1) {
									throw new Error("Message not found");
								}

								// Create the metacognitive event
								const event: MetacognitiveEvent = {
									type: body.type as "comprehension" | "perspective",
									promptText: body.promptText,
									triggerFrequency: body.triggerFrequency,
									timestamp: new Date(),
									suggestedPersonaId: body.suggestedPersonaId,
									suggestedPersonaName: body.suggestedPersonaName,
									accepted: body.accepted,
								};

								// Log the event for research tracking
								logger.info(
									{
										conversationId: params.id,
										messageId: params.messageId,
										eventType: event.type,
										triggerFrequency: event.triggerFrequency,
										accepted: event.accepted,
										suggestedPersonaId: event.suggestedPersonaId,
									},
									"Metacognitive prompt event"
								);

								// Update the message with the new event
								const updatePath = `messages.${messageIndex}.metacognitiveEvents`;
								// eslint-disable-next-line @typescript-eslint/no-explicit-any
								const updateDoc: Record<string, any> = {
									$push: { [updatePath]: event },
								};

								// If this is a shown event, persist metacognitive state so
								// targetFrequency stays consistent across refresh/devices and the counter can reset.
								let nextTargetFrequency: number | undefined;
								if (body.accepted === false) {
									const metacognitiveConfig = getMetacognitiveConfig();
									if (metacognitiveConfig.enabled) {
										nextTargetFrequency = selectRandomFrequency();
										updateDoc.$set = {
											"metacognitiveState.lastPromptedAtMessageId": params.messageId,
											"metacognitiveState.targetFrequency": nextTargetFrequency,
											"metacognitiveState.updatedAt": new Date(),
										};
									}
								}

								const res = await collections.conversations.updateOne(
									{
										_id: new ObjectId(params.id),
										...authCondition(locals),
									},
									updateDoc
								);

								if (res.modifiedCount === 0) {
									throw new Error("Failed to log metacognitive event");
								}

								return {
									success: true,
									event,
									nextTargetFrequency,
								};
							},
							{
								params: t.Object({
									id: t.String(),
									messageId: t.String(),
								}),
								body: t.Object({
									type: t.Union([t.Literal("comprehension"), t.Literal("perspective")]),
									promptText: t.String(),
									triggerFrequency: t.Number(),
									suggestedPersonaId: t.Optional(t.String()),
									suggestedPersonaName: t.Optional(t.String()),
									accepted: t.Boolean(),
								}),
							}
						);
				}
			)
	);
});