OpenVelo Agent
64 lines
2.5 KiB
TypeScript
64 lines
2.5 KiB
TypeScript
import { Injectable } from '@nestjs/common';
|
|
import { ConfigService } from '@nestjs/config';
|
|
import { InjectRepository } from '@nestjs/typeorm';
|
|
import { Repository } from 'typeorm';
|
|
import { UserEntity } from '../../database/entities/user.entity';
|
|
import { SettingEntity } from '../../database/entities/setting.entity';
|
|
import { ThemeLoaderService } from '../../common/utils/theme-loader.service';
|
|
import { SETTINGS_KEYS } from '../../config/env.schema';
|
|
|
|
export interface PasswordPolicyInfo {
|
|
minLength: number;
|
|
requireMixed: boolean;
|
|
description: string;
|
|
}
|
|
|
|
export interface BootstrapPayload {
|
|
initialized: boolean;
|
|
pageTitle: string;
|
|
logo: string;
|
|
welcomeMarkdown: string;
|
|
theme: any;
|
|
defaultChallengeIp: string;
|
|
registrationsEnabled: boolean;
|
|
passwordPolicy: PasswordPolicyInfo;
|
|
}
|
|
|
|
@Injectable()
|
|
export class SystemService {
|
|
constructor(
|
|
@InjectRepository(UserEntity) private readonly users: Repository<UserEntity>,
|
|
@InjectRepository(SettingEntity) private readonly settings: Repository<SettingEntity>,
|
|
private readonly themes: ThemeLoaderService,
|
|
private readonly config: ConfigService,
|
|
) {}
|
|
|
|
async bootstrap(): Promise<BootstrapPayload> {
|
|
const adminCount = await this.users.count({ where: { role: 'admin' } });
|
|
const themeKey = await this.getSetting(SETTINGS_KEYS.THEME_KEY, 'classic');
|
|
return {
|
|
initialized: adminCount > 0,
|
|
pageTitle: await this.getSetting(SETTINGS_KEYS.PAGE_TITLE, 'HIPCTF'),
|
|
logo: await this.getSetting(SETTINGS_KEYS.LOGO, ''),
|
|
welcomeMarkdown: await this.getSetting(SETTINGS_KEYS.WELCOME_MARKDOWN, ''),
|
|
theme: this.themes.getTheme(themeKey),
|
|
defaultChallengeIp: await this.getSetting(SETTINGS_KEYS.DEFAULT_CHALLENGE_IP, '127.0.0.1'),
|
|
registrationsEnabled: (await this.getSetting(SETTINGS_KEYS.REGISTRATIONS_ENABLED, 'false')) === 'true',
|
|
passwordPolicy: this.passwordPolicy(),
|
|
};
|
|
}
|
|
|
|
private passwordPolicy(): PasswordPolicyInfo {
|
|
const minLength = this.config.get<number>('PASSWORD_MIN_LENGTH', 12);
|
|
const requireMixed = this.config.get<boolean>('PASSWORD_REQUIRE_MIXED', true);
|
|
const description = requireMixed
|
|
? `At least ${minLength} characters with upper, lower, digit and symbol`
|
|
: `At least ${minLength} characters`;
|
|
return { minLength, requireMixed, description };
|
|
}
|
|
|
|
private async getSetting(key: string, fallback: string): Promise<string> {
|
|
const row = await this.settings.findOne({ where: { key } });
|
|
return row?.value ?? fallback;
|
|
}
|
|
} |