import { query, mutation } from "./_generated/server";
import { paginationOptsValidator } from "convex/server";
import { v } from "convex/values";

export const list = query({
  args: {
    paginationOpts: paginationOptsValidator,
    channelId: v.id("channels"),
    userId: v.optional(v.id("userProfiles")),
  },
  returns: v.any(),
  handler: async (ctx, args) => {
    const result = await ctx.db
      .query("messages")
      .withIndex("by_channel", (q) => q.eq("channelId", args.channelId))
      .order("desc")
      .paginate(args.paginationOpts);

    const enrichedPage = await Promise.all(
      result.page.map(async (msg) => {
        const sender = await ctx.db.get(msg.senderId);

        const reactionDocs = await ctx.db
          .query("messageReactions")
          .withIndex("by_message", (q) => q.eq("messageId", msg._id))
          .collect();

        const reactions: Record<string, { count: number; me: boolean }> = {};
        for (const r of reactionDocs) {
          const entry = (reactions[r.emoji] ??= { count: 0, me: false });
          entry.count++;
          if (args.userId && r.userId === args.userId) {
            entry.me = true;
          }
        }

        return {
          id: msg._id,
          channel_id: msg.channelId,
          sender_id: msg.senderId,
          ciphertext: msg.ciphertext,
          nonce: msg.nonce,
          signature: msg.signature,
          key_version: msg.keyVersion,
          created_at: new Date(msg._creationTime).toISOString(),
          username: sender?.username || "Unknown",
          public_signing_key: sender?.publicSigningKey || "",
          reactions: Object.keys(reactions).length > 0 ? reactions : null,
        };
      })
    );

    return { ...result, page: enrichedPage };
  },
});

export const send = mutation({
  args: {
    channelId: v.id("channels"),
    senderId: v.id("userProfiles"),
    ciphertext: v.string(),
    nonce: v.string(),
    signature: v.string(),
    keyVersion: v.number(),
  },
  returns: v.object({ id: v.id("messages") }),
  handler: async (ctx, args) => {
    const id = await ctx.db.insert("messages", {
      channelId: args.channelId,
      senderId: args.senderId,
      ciphertext: args.ciphertext,
      nonce: args.nonce,
      signature: args.signature,
      keyVersion: args.keyVersion,
    });
    return { id };
  },
});

export const remove = mutation({
  args: { id: v.id("messages") },
  returns: v.null(),
  handler: async (ctx, args) => {
    const reactions = await ctx.db
      .query("messageReactions")
      .withIndex("by_message", (q) => q.eq("messageId", args.id))
      .collect();
    for (const r of reactions) {
      await ctx.db.delete(r._id);
    }

    await ctx.db.delete(args.id);
    return null;
  },
});