AFFiNE-Mirror/packages/backend/server/src/data/migrations/99999999-self-host-admin.ts

import { ModuleRef } from '@nestjs/core';
import { PrismaClient } from '@prisma/client';

import { UserService } from '../../core/user';
import { Config, CryptoHelper } from '../../fundamentals';

export class SelfHostAdmin99999999 {
  // do the migration
  static async up(_db: PrismaClient, ref: ModuleRef) {
    const config = ref.get(Config, { strict: false });
    const crypto = ref.get(CryptoHelper, { strict: false });
    const user = ref.get(UserService, { strict: false });
    if (config.isSelfhosted) {
      if (
        !process.env.AFFINE_ADMIN_EMAIL ||
        !process.env.AFFINE_ADMIN_PASSWORD
      ) {
        throw new Error(
          'You have to set AFFINE_ADMIN_EMAIL and AFFINE_ADMIN_PASSWORD environment variables to generate the initial user for self-hosted AFFiNE Server.'
        );
      }
      await user.findOrCreateUser(process.env.AFFINE_ADMIN_EMAIL, {
        name: 'AFFINE First User',
        emailVerifiedAt: new Date(),
        password: await crypto.encryptPassword(
          process.env.AFFINE_ADMIN_PASSWORD
        ),
      });
    }
  }

  // revert the migration
  static async down(db: PrismaClient) {
    await db.user.deleteMany({
      where: {
        email: process.env.AFFINE_ADMIN_EMAIL ?? 'admin@example.com',
      },
    });
  }
}