fix: load env files for cli (#3503)

server/src/admin-cli/app.module.ts

@@ -0,0 +1,18 @@
+import { DomainModule } from '@app/domain';
+import { InfraModule } from '@app/infra';
+import { Module } from '@nestjs/common';
+import { ListUsersCommand } from './commands/list-users.command';
+import { DisablePasswordLoginCommand, EnablePasswordLoginCommand } from './commands/password-login';
+import { PromptPasswordQuestions, ResetAdminPasswordCommand } from './commands/reset-admin-password.command';
+
+@Module({
+  imports: [DomainModule.register({ imports: [InfraModule] })],
+  providers: [
+    ResetAdminPasswordCommand,
+    PromptPasswordQuestions,
+    EnablePasswordLoginCommand,
+    DisablePasswordLoginCommand,
+    ListUsersCommand,
+  ],
+})
+export class AppModule {}

server/src/admin-cli/commands/list-users.command.ts

@@ -0,0 +1,23 @@
+import { UserService } from '@app/domain';
+import { Command, CommandRunner } from 'nest-commander';
+import { CLI_USER } from '../constants';
+
+@Command({
+  name: 'list-users',
+  description: 'List Immich users',
+})
+export class ListUsersCommand extends CommandRunner {
+  constructor(private userService: UserService) {
+    super();
+  }
+
+  async run(): Promise<void> {
+    try {
+      const users = await this.userService.getAllUsers(CLI_USER, true);
+      console.dir(users);
+    } catch (error) {
+      console.error(error);
+      console.error('Unable to load users');
+    }
+  }
+}

server/src/admin-cli/commands/password-login.ts

@@ -0,0 +1,39 @@
+import { SystemConfigService } from '@app/domain';
+import axios from 'axios';
+import { Command, CommandRunner } from 'nest-commander';
+
+@Command({
+  name: 'enable-password-login',
+  description: 'Enable password login',
+})
+export class EnablePasswordLoginCommand extends CommandRunner {
+  constructor(private configService: SystemConfigService) {
+    super();
+  }
+
+  async run(): Promise<void> {
+    const config = await this.configService.getConfig();
+    config.passwordLogin.enabled = true;
+    await this.configService.updateConfig(config);
+    await axios.post('http://localhost:3001/refresh-config');
+    console.log('Password login has been enabled.');
+  }
+}
+
+@Command({
+  name: 'disable-password-login',
+  description: 'Disable password login',
+})
+export class DisablePasswordLoginCommand extends CommandRunner {
+  constructor(private configService: SystemConfigService) {
+    super();
+  }
+
+  async run(): Promise<void> {
+    const config = await this.configService.getConfig();
+    config.passwordLogin.enabled = false;
+    await this.configService.updateConfig(config);
+    await axios.post('http://localhost:3001/refresh-config');
+    console.log('Password login has been disabled.');
+  }
+}

server/src/admin-cli/commands/reset-admin-password.command.ts

@@ -0,0 +1,49 @@
+import { UserResponseDto, UserService } from '@app/domain';
+import { Command, CommandRunner, InquirerService, Question, QuestionSet } from 'nest-commander';
+
+@Command({
+  name: 'reset-admin-password',
+  description: 'Reset the admin password',
+})
+export class ResetAdminPasswordCommand extends CommandRunner {
+  constructor(private userService: UserService, private readonly inquirer: InquirerService) {
+    super();
+  }
+
+  async run(): Promise<void> {
+    const ask = (admin: UserResponseDto) => {
+      const { id, oauthId, email, firstName, lastName } = admin;
+      console.log(`Found Admin:
+- ID=${id}
+- OAuth ID=${oauthId}
+- Email=${email}
+- Name=${firstName} ${lastName}`);
+
+      return this.inquirer.ask<{ password: string }>('prompt-password', undefined).then(({ password }) => password);
+    };
+
+    try {
+      const { password, provided } = await this.userService.resetAdminPassword(ask);
+
+      if (provided) {
+        console.log(`The admin password has been updated.`);
+      } else {
+        console.log(`The admin password has been updated to:\n${password}`);
+      }
+    } catch (error) {
+      console.error(error);
+      console.error('Unable to reset admin password');
+    }
+  }
+}
+
+@QuestionSet({ name: 'prompt-password' })
+export class PromptPasswordQuestions {
+  @Question({
+    message: 'Please choose a new password (optional)',
+    name: 'password',
+  })
+  parsePassword(val: string) {
+    return val;
+  }
+}

server/src/admin-cli/constants.ts

@@ -0,0 +1,9 @@
+import { AuthUserDto } from '@app/domain';
+
+export const CLI_USER: AuthUserDto = {
+  id: 'cli',
+  email: 'cli@immich.app',
+  isAdmin: true,
+  isPublicUser: false,
+  isAllowUpload: true,
+};

server/src/admin-cli/main.ts

@@ -0,0 +1,6 @@
+import { CommandFactory } from 'nest-commander';
+import { AppModule } from './app.module';
+
+export async function bootstrap() {
+  await CommandFactory.run(AppModule, ['warn', 'error']);
+}