#!/usr/bin/env node

import { codegen } from './codegen.js'
import { Aptos, AptosConfig, Network } from '@aptos-labs/ts-sdk'
import * as path from 'path'
import * as fs from 'fs'
import { Command } from 'commander'
import { createRequire } from 'module'
const require = createRequire(import.meta.url)
let pkg = undefined
try {
  pkg = require('../../package.json')
} catch (e) {
  pkg = require('../../../package.json')
}

const program = new Command()

program
  .name('typemove-aptos')
  .description('CLI to generate typescript types from Aptos ABIs')
  .showHelpAfterError()
  .version(pkg.version)
  .argument('<location>', 'Directory of ABI json files or address of account to generate types for')
  .option('-a, --abi-dir <dir>', 'Directory to store downloaded ABI. Only useful if <location> is address', './abis')
  .option('-t, --target-dir <dir>', 'Directory to output generated files', './types')
  .option(
    '-n, --network <mainnet|testnet|$url>',
    'Network to use, could be either "mainnet", "testnet" or any node URL',
    'mainnet'
  )
  .action(async (location, options) => {
    const endpoint = options.network
    let config: AptosConfig
    if (endpoint == 'mainnet') {
      config = new AptosConfig({
        network: Network.MAINNET
      })
    } else if (endpoint == 'testnet') {
      config = new AptosConfig({
        network: Network.TESTNET
      })
    } else {
      config = new AptosConfig({
        network: Network.CUSTOM,
        fullnode: endpoint
      })
    }
    const aptosClient = new Aptos(config)

    let abisDir = location
    if (location.startsWith('0x')) {
      const abiAddress = abisDir
      const abi = await aptosClient.getAccountModules({ accountAddress: abiAddress })
      abisDir = options.abiDir
      if (!fs.existsSync(abisDir)) {
        fs.mkdirSync(abisDir, { recursive: true })
      }
      fs.writeFileSync(path.join(abisDir, abiAddress + '.json'), JSON.stringify(abi, null, 2))
    }

    const num = await codegen(abisDir, options.targetDir, config, true)
    process.exit(num == 0 ? 1 : 0)
  })

program.parse()