// @ts-nocheck
import { mutation, query } from "./_generated/server";
import { v } from "convex/values";

const manualInput = v.object({
  filename: v.string(),
  path: v.string(),
  manufacturer: v.string(),
  category: v.string(),
  size: v.optional(v.number()),
  lastModified: v.optional(v.number()),
  searchTerms: v.optional(v.array(v.string())),
  commonNames: v.optional(v.array(v.string())),
  thumbnailUrl: v.optional(v.string()),
  manualUrl: v.optional(v.string()),
  hasParts: v.optional(v.boolean()),
  assetSource: v.optional(v.string()),
  sourcePath: v.optional(v.string()),
  sourceSite: v.optional(v.string()),
  sourceDomain: v.optional(v.string()),
  siteVisibility: v.optional(v.array(v.string())),
  importBatch: v.optional(v.string()),
});

export const list = query({
  args: {},
  handler: async (ctx) => {
    const manuals = await ctx.db.query("manuals").collect();
    return manuals.sort((a, b) => a.filename.localeCompare(b.filename));
  },
});

export const dashboard = query({
  args: {},
  handler: async (ctx) => {
    const manuals = await ctx.db.query("manuals").collect();
    const categories = await ctx.db.query("manualCategories").collect();

    const manufacturerMap = new Map<string, number>();
    const categoryMap = new Map<string, number>();
    for (const manual of manuals) {
      manufacturerMap.set(
        manual.manufacturer,
        (manufacturerMap.get(manual.manufacturer) ?? 0) + 1,
      );
      categoryMap.set(
        manual.category,
        (categoryMap.get(manual.category) ?? 0) + 1,
      );
    }

    return {
      missingManuals: {
        summary: {
          total_expected_models: manuals.length,
          models_missing_all: 0,
          models_partial: 0,
        },
      },
      qaData: [],
      metadata: manuals,
      structuredData: [],
      semanticIndex: {
        total_chunks: manuals.length,
      },
      acquisitionList: {
        total_items: 0,
        high_priority: 0,
        medium_priority: 0,
        low_priority: 0,
        acquisition_list: [],
      },
      nameMapping: categories,
    };
  },
});

export const upsertMany = mutation({
  args: {
    manuals: v.array(manualInput),
  },
  handler: async (ctx, args) => {
    const now = Date.now();
    const results = [];
    for (const manual of args.manuals) {
      const existing = await ctx.db
        .query("manuals")
        .withIndex("by_path", (q) => q.eq("path", manual.path))
        .unique();

      if (existing) {
        await ctx.db.patch(existing._id, {
          ...manual,
          updatedAt: now,
        });
        results.push(await ctx.db.get(existing._id));
      } else {
        const id = await ctx.db.insert("manuals", {
          ...manual,
          createdAt: now,
          updatedAt: now,
        });
        results.push(await ctx.db.get(id));
      }
    }
    return results;
  },
});