import { metadataCustomSnippet } from '@cabloy/cli';
import { catchError } from '@cabloy/utils';
import fs from 'node:fs';

declare module '@cabloy/cli' {
  interface ICommandArgv {
    module: string;
  }
}

const __snippet_import1 = "import { $tableColumns } from 'vona-module-a-ormutils';";
const __snippet_update = "...$tableColumns('<%=argv.moduleResourceName%>', 'name'),";

export default metadataCustomSnippet({
  file: 'src/bean/meta.index.ts',
  language: 'plain',
  format: true,
  init: async ({ cli, argv, targetFile }) => {
    await catchError(() => {
      return cli.helper.invokeCli(
        [':create:bean', 'meta', 'index', `--module=${argv.module}`, '--nometadata'],
        {
          cwd: argv.projectPath,
        },
      );
    });
    return fs.readFileSync(targetFile).toString('utf8');
  },
  async transform({ cli, ast }) {
    // import1
    if (!ast.includes(__snippet_import1)) {
      const code = await cli.template.renderContent({ content: __snippet_import1 });
      ast = ast.replace(
        "import { Meta } from 'vona-module-a-meta';",
        `import { Meta } from 'vona-module-a-meta';\n${code}`,
      );
    }
    // update
    const code = await cli.template.renderContent({ content: __snippet_update });
    if (ast.includes('indexes: {}')) {
      ast = ast.replace('indexes: {', `indexes: {\n    ${code}\n  `);
    } else {
      ast = ast.replace('indexes: {', `indexes: {\n    ${code}`);
    }
    // ok
    return ast;
  },
});