import { getAuthUserId } from "@convex-dev/auth/server"
import { v } from "convex/values"
import { api } from "./_generated/api"
import { query } from "./_generated/server"
import { computeFlagsFromConfig } from "./builds"
import { adminMutation, adminQuery } from "./helpers"

export const isAdmin = query({
  args: {},
  handler: async ctx => {
    const userId = await getAuthUserId(ctx)
    if (!userId) return false

    const userSettings = await ctx.db
      .query("userSettings")
      .withIndex("by_user", q => q.eq("userId", userId))
      .first()

    return userSettings?.isAdmin === true
  },
})

export const listFailedBuilds = adminQuery({
  args: {},
  handler: async ctx => {
    const failedBuilds = await ctx.db
      .query("builds")
      .withIndex("by_status_updatedAt", q => q.eq("status", "failure"))
      .order("desc")
      .collect()

    return failedBuilds
  },
})

export const listAllBuilds = adminQuery({
  args: {},
  handler: async ctx => {
    const allBuilds = await ctx.db.query("builds").withIndex("by_updatedAt").order("desc").collect()

    return allBuilds
  },
})

export const retryBuild = adminMutation({
  args: {
    buildId: v.id("builds"),
  },
  handler: async (ctx, args) => {
    const build = await ctx.db.get(args.buildId)
    if (!build) {
      throw new Error("Build not found")
    }

    // Compute flags from config
    const flags = computeFlagsFromConfig(build.config)

    // Dispatch new GitHub build with same config
    // This will use the latest YAML from the branch
    await ctx.scheduler.runAfter(0, api.actions.dispatchGithubBuild, {
      target: build.config.target,
      version: build.config.version,
      buildId: args.buildId,
      flags,
      buildHash: build.buildHash,
      plugins: build.config.pluginsEnabled ?? [],
    })

    // Update build status to queued and clear artifact paths
    await ctx.db.patch(args.buildId, {
      status: "queued",
      updatedAt: Date.now(),
      firmwarePath: undefined,
      sourcePath: undefined,
    })

    return { success: true }
  },
})