import { execa } from 'execa';
import { z } from 'zod/v4';
import { MiseRegistryJson } from '../../lib/modules/manager/mise/schema.ts';
import { Json } from '../../lib/util/schema-utils/index.ts';
import { updateJsonFile } from './utils.mjs';

const MiseVersion = z.object({ version: z.string() });
const MiseRegistry = z.array(
  z.object({ short: z.string(), backends: z.array(z.string()) }),
);

// use the JSON output, as the default output includes a banner
const versionOutput = await execa('mise', ['version', '--json'], {
  encoding: 'utf8',
});
const version = Json.pipe(MiseVersion).parse(versionOutput.stdout);
console.log(`Generating mise registry using mise version ${version.version}`);
const output = await execa('mise', ['registry', '--json'], {
  encoding: 'utf8',
});
const tools = Json.pipe(MiseRegistry).parse(output.stdout);

const registry = MiseRegistryJson.parse({
  meta: {
    version: version.version,
  },
  tools: Object.fromEntries(
    tools
      .map(({ short, backends }): [string, Record<string, string>] => {
        const result: Record<string, string> = {};
        for (const backend of backends.sort()) {
          const [type, url] = backend.split(':');
          result[type] = url;
        }

        return [short, result];
      })
      .sort(([a], [b]) => a.localeCompare(b)),
  ),
});

await updateJsonFile(
  'lib/data/mise-registry.json',
  `${JSON.stringify(registry, null, 2)}\n`,
);