import {mkdir, writeFile} from 'node:fs/promises'
import path from 'node:path'
import {defineCommand} from 'citty'
import {openApp} from '../app/context'
import {workflows} from '../db/schema'
import {
  ProjectConfigResolutionError,
  resolveProjectConfig,
} from '../project/discoverConfig'

export const factoryCmd = defineCommand({
  meta: {name: 'factory', description: '[advanced] Manage factories'},
  subCommands: {
    create: defineCommand({
      meta: {
        name: 'create',
        description: 'Create a new TypeScript factory scaffold',
      },
      args: {
        id: {type: 'string', required: true},
        config: {type: 'string', required: false},
        skipNotionBoard: {
          type: 'boolean',
          required: false,
          alias: 'skip-notion-board',
        },
        parentPage: {type: 'string', required: false, alias: 'parent-page'},
      },
      async run({args}) {
        let resolvedProject
        try {
          resolvedProject = await resolveProjectConfig({
            startDir: process.cwd(),
            configPath: args.config ? String(args.config) : undefined,
          })
        } catch (error) {
          if (!(error instanceof ProjectConfigResolutionError)) {
            throw error
          }

          console.error(`[error] ${error.message}`)
          console.error(`Start directory: ${error.startDir}`)
          if (error.attemptedPath) {
            console.error(`Attempted config path: ${error.attemptedPath}`)
          }
          console.error(
            'Run `notionflow init` in your project root first, or pass --config <path>.',
          )
          process.exitCode = 1
          return
        }

        const id = String(args.id)
        const targetDir = path.join(resolvedProject.projectRoot, 'factories')
        const targetPath = path.join(targetDir, `${id}.ts`)
        const template = `const doWork = async ({ ctx }) => ({\n  status: "done",\n  data: { ...ctx, result: "ok" },\n});\n\nexport default {\n  id: "${id}",\n  start: "start",\n  context: {},\n  states: {\n    start: {\n      type: "action",\n      agent: doWork,\n      on: { done: "done", failed: "failed" },\n    },\n    done: { type: "done" },\n    failed: { type: "failed" },\n  },\n};\n`
        await mkdir(targetDir, {recursive: true})
        await writeFile(targetPath, template, 'utf8')

        console.log(`Factory scaffold created: ${id}`)
        console.log(`Path: ${targetPath}`)

        if (!args.skipNotionBoard) {
          console.log(
            '[warn] Notion board provisioning is not yet supported for local-only factory scaffolds.',
          )
        }
      },
    }),
    list: defineCommand({
      meta: {name: 'list', description: 'List installed factories'},
      async run() {
        const {db} = await openApp()
        const rows = await db.select().from(workflows)
        if (rows.length === 0) {
          console.log('No factories configured')
          return
        }
        for (const row of rows) console.log(`${row.id}  v${row.version}`)
      },
    }),
  },
})