AFFiNE/packages/backend/server/src/data/commands/create.ts

import { appendFileSync, writeFileSync } from 'node:fs';
import { join, parse } from 'node:path';
import { fileURLToPath } from 'node:url';

import { Logger } from '@nestjs/common';
import { camelCase, kebabCase, upperFirst } from 'lodash-es';
import {
  Command,
  CommandRunner,
  InquirerService,
  Question,
  QuestionSet,
} from 'nest-commander';

@QuestionSet({ name: 'name-questions' })
export class NameQuestion {
  @Question({
    name: 'name',
    message: 'Name of the data migration script:',
  })
  parseName(val: string) {
    return val.trim();
  }
}

@Command({
  name: 'create',
  arguments: '[name]',
  description: 'create a data migration script',
})
export class CreateCommand extends CommandRunner {
  logger = new Logger(CreateCommand.name);
  constructor(private readonly inquirer: InquirerService) {
    super();
  }

  override async run(inputs: string[]): Promise<void> {
    let name = inputs[0];

    if (!name) {
      name = (
        await this.inquirer.ask<{ name: string }>('name-questions', undefined)
      ).name;
    }

    const timestamp = Date.now();
    const content = this.createScript(upperFirst(camelCase(name)) + timestamp);
    const migrationDir = join(
      fileURLToPath(import.meta.url),
      '../../migrations'
    );
    const fileName = `${timestamp}-${kebabCase(name)}.ts`;
    const filePath = join(migrationDir, fileName);

    this.logger.log(`Creating ${fileName}...`);
    writeFileSync(filePath, content);
    const indexFile = join(migrationDir, 'index.ts');
    appendFileSync(
      indexFile,
      `export * from './${parse(fileName).name}';`,
      'utf-8'
    );
    this.logger.log(`Migration file created at ${filePath}`);
    this.logger.log('Done');
  }

  private createScript(name: string) {
    const contents = ["import { PrismaClient } from '@prisma/client';", ''];
    contents.push(`export class ${name} {`);
    contents.push('  // do the migration');
    contents.push('  static async up(db: PrismaClient) {}');
    contents.push('');
    contents.push('  // revert the migration');
    contents.push('  static async down(db: PrismaClient) {}');

    contents.push('}');

    return contents.join('\n');
  }
}