import { v } from 'convex/values'
import { internalMutation, query } from './_generated/server'

export const get = query({
  args: { slug: v.string() },
  handler: async (ctx, args) => {
    const plugin = await ctx.db
      .query('plugins')
      .withIndex('by_slug', (q) => q.eq('slug', args.slug))
      .unique()
    return plugin
      ? { slug: plugin.slug, flashCount: plugin.flashCount }
      : { slug: args.slug, flashCount: 0 }
  },
})

export const getAll = query({
  args: {},
  handler: async (ctx) => {
    const plugins = await ctx.db.query('plugins').collect()
    const counts: Record<string, number> = {}
    for (const plugin of plugins) {
      counts[plugin.slug] = plugin.flashCount
    }
    return counts
  },
})

export const incrementFlashCount = internalMutation({
  args: { slugs: v.array(v.string()) },
  handler: async (ctx, args) => {
    for (const pluginSpec of args.slugs) {
      // Extract slug from "slug@version" format, or use as-is if no @ present
      const slug = pluginSpec.split('@')[0]

      const existing = await ctx.db
        .query('plugins')
        .withIndex('by_slug', (q) => q.eq('slug', slug))
        .unique()

      if (existing) {
        await ctx.db.patch(existing._id, {
          flashCount: existing.flashCount + 1,
          updatedAt: Date.now(),
        })
      } else {
        await ctx.db.insert('plugins', {
          slug,
          flashCount: 1,
          updatedAt: Date.now(),
        })
      }
    }
  },
})