import yargs from 'yargs/yargs'

void yargs(process.argv.slice(2))
  .scriptName('frcsdata')
  .command({
    command: 'check <file>',
    describe: 'check survey file for errors or warnings',
    builder: (yargs) =>
      yargs.positional('file', {
        type: 'string',
        demandOption: true,
      }),
    handler: async ({ file }) => {
      const { checkSurvey } = await import('./cli/check')
      await checkSurvey(file)
    },
  })
  .command({
    command: 'parse <file>',
    describe: 'parse survey file and output JSON parse tree',
    builder: (yargs) =>
      yargs.positional('file', {
        type: 'string',
        demandOption: true,
      }),
    handler: async ({ file }) => {
      const { parseSurvey } = await import('./cli/parse-survey')
      await parseSurvey(file)
    },
  })
  .command({
    command: 'summarize <file>',
    describe: 'parse survey file and output trip summaries',
    builder: (yargs) =>
      yargs.positional('file', {
        type: 'string',
        demandOption: true,
      }),
    handler: async ({ file }) => {
      const { summarizeSurvey } = await import('./cli/summarize')
      await summarizeSurvey(file)
    },
  })
  .command({
    command: 'list-names <file>',
    describe: 'parse survey file and output surveyor name/count table',
    builder: (yargs) =>
      yargs
        .positional('file', {
          type: 'string',
          demandOption: true,
        })
        .option('counts', {
          alias: 'c',
          type: 'boolean',
          describe: 'count the number of occurrences of each name',
          demandOption: false,
        })
        .option('suggest-replacements', {
          alias: 's',
          type: 'boolean',
          describe: 'output suggested replacements',
          demandOption: false,
        }),
    handler: async ({ file, counts, suggestReplacements }) => {
      const { listSurveyNames: listNames } = await import('./cli/list-names')
      await listNames(file, { includeCounts: counts, suggestReplacements })
    },
  })
  .command({
    command: 'replace-names <surveyFile> <replacementsFile>',
    describe: 'replace names survey file',
    builder: (yargs) =>
      yargs
        .positional('surveyFile', {
          type: 'string',
          demandOption: true,
        })
        .positional('replacementsFile', {
          type: 'string',
          demandOption: true,
        })
        .option('verbose', {
          alias: 'v',
          type: 'boolean',
        })
        .option('write', {
          alias: 'w',
          describe: 'overwrite survey file with replacements',
          type: 'boolean',
        }),
    handler: async ({ surveyFile, replacementsFile, verbose, write }) => {
      const { replaceSurveyNames: replaceNames } = await import(
        './cli/replace-names'
      )
      await replaceNames(surveyFile, replacementsFile, { verbose, write })
    },
  })
  .command({
    command: 'merge-names-files <namesFiles..>',
    describe: 'merge two or more names files',
    builder: (yargs) =>
      yargs.positional('namesFiles', {
        type: 'string',
        array: true,
        demandOption: true,
      }),
    handler: async ({ namesFiles }) => {
      const { mergeNamesFiles } = await import('./cli/merge-names-files')
      await mergeNamesFiles(...namesFiles)
    },
  })
  .command({
    command: 'check-correspondence <surveyFile> <summaryFile>',
    describe: 'parse survey file and output trip summaries',
    builder: (yargs) =>
      yargs
        .positional('surveyFile', {
          type: 'string',
          demandOption: true,
        })
        .positional('summaryFile', {
          type: 'string',
          demandOption: true,
        }),
    handler: async ({ surveyFile, summaryFile }) => {
      const { checkSurveyCorrespondence } = await import(
        './cli/check-correspondence'
      )
      await checkSurveyCorrespondence(surveyFile, summaryFile)
    },
  })
  .demandCommand().argv