@3wks/generator-gae-node-nestjs/generators/app/templates/server/src/migrations/migrations.service.ts

A generator for building nestjs apps on app engine node

import { Injectable } from '@nestjs/common';
import {
  CredentialRepository,
  newContext,
  DatastoreProvider,
  hashPassword,
  createLogger,
  LoginIdentifierRepository,
} from '@3wks/gae-node-nestjs';
import { ConfigurationProvider } from '../config/config.provider';
import { UserRepository } from '../users/users.repository';
import { randomBytes } from 'crypto';
import { UsersService } from '../users/users.service';

const generatePassword = (bits: number) => randomBytes(Math.ceil(bits / 8)).toString('base64');

const SUPER_USER_EMAIL = 'admin@3wks.com.au';

@Injectable()
export class MigrationService {
  private readonly logger = createLogger('migration-service');

  constructor(
    private readonly credentialsRepository: CredentialRepository,
    private readonly userRepository: UserRepository,
    private readonly loginIdentifierRepository: LoginIdentifierRepository,
    private readonly userService: UsersService,
    private readonly configurationProvider: ConfigurationProvider,
    private readonly datastoreProvider: DatastoreProvider,
  ) {}

  async bootstrap() {
    if (this.configurationProvider.isDevelopment()) {
      const context = newContext(this.datastoreProvider.datastore);

      const match = await this.userService.getByEmail(context, SUPER_USER_EMAIL);

      if (!match) {
        await this.bootstrapSystemUser('password');
      }
    }
  }

  async bootstrapSystemUser(password: string = generatePassword(256)): Promise<string> {
    const context = newContext(this.datastoreProvider.datastore);
    const userId = '12345';

    await this.credentialsRepository.save(context, {
      id: SUPER_USER_EMAIL,
      type: 'password',
      userId,
      password: await hashPassword(password),
    });

    await this.loginIdentifierRepository.save(context, {
      id: SUPER_USER_EMAIL,
      createdAt: new Date(),
      userId,
    });

    await this.userRepository.save(context, {
      id: userId,
      email: SUPER_USER_EMAIL,
      name: 'Admin',
      roles: ['super', 'admin'],
      enabled: true,
    });

    return password;
  }
}