import crypto from 'crypto'; import { Logger } from '../logger'; import { IUnleashConfig, IUnleashStores } from '../types'; import { minutesToMilliseconds } from 'date-fns'; import { IPublicSignupTokenStore } from '../types/stores/public-signup-token-store'; import { PublicSignupTokenSchema } from '../openapi/spec/public-signup-token-schema'; import { IRoleStore } from '../types/stores/role-store'; import { IPublicSignupTokenCreate } from '../types/models/public-signup-token'; import { PublicSignupTokenCreateSchema } from '../openapi/spec/public-signup-token-create-schema'; import { RoleName } from '../types/model'; import { IEventStore } from '../types/stores/event-store'; import { PublicSignupTokenCreatedEvent, PublicSignupTokenManuallyExpiredEvent, PublicSignupTokenUserAddedEvent, } from '../types/events'; import UserService, { ICreateUser } from './user-service'; import { IUser } from '../types/user'; export class PublicSignupTokenService { private store: IPublicSignupTokenStore; private roleStore: IRoleStore; private eventStore: IEventStore; private userService: UserService; private logger: Logger; private timer: NodeJS.Timeout; private activeTokens: PublicSignupTokenSchema[] = []; constructor( { publicSignupTokenStore, roleStore, eventStore, }: Pick< IUnleashStores, 'publicSignupTokenStore' | 'roleStore' | 'eventStore' >, config: Pick, userService: UserService, ) { this.store = publicSignupTokenStore; this.userService = userService; this.roleStore = roleStore; this.eventStore = eventStore; this.logger = config.getLogger( '/services/public-signup-token-service.ts', ); this.fetchActiveTokens(); this.timer = setInterval( () => this.fetchActiveTokens(), minutesToMilliseconds(1), ).unref(); } async fetchActiveTokens(): Promise { this.activeTokens = await this.getAllActiveTokens(); } public async get(secret: string): Promise { return this.store.get(secret); } public async getAllTokens(): Promise { return this.store.getAll(); } public async getAllActiveTokens(): Promise { return this.store.getAllActive(); } public async validate(secret: string): Promise { return this.store.isValid(secret); } public async setExpiry( secret: string, expireAt: Date, ): Promise { return this.store.setExpiry(secret, expireAt); } public async addTokenUser( secret: string, createUser: ICreateUser, ): Promise { const user = await this.userService.createUser(createUser); await this.store.addTokenUser(secret, user.id); await this.eventStore.store( new PublicSignupTokenUserAddedEvent({ createdBy: 'userId', data: { secret, userId: user.id }, }), ); return user; } public async delete(secret: string, expiredBy: string): Promise { await this.expireToken(secret); await this.eventStore.store( new PublicSignupTokenManuallyExpiredEvent({ createdBy: expiredBy, data: { secret }, }), ); } private async expireToken( secret: string, ): Promise { return this.store.setExpiry(secret, new Date()); } public async createNewPublicSignupToken( tokenCreate: PublicSignupTokenCreateSchema, createdBy: string, ): Promise { const viewerRole = await this.roleStore.getRoleByName(RoleName.VIEWER); const newToken: IPublicSignupTokenCreate = { name: tokenCreate.name, expiresAt: new Date(tokenCreate.expiresAt), secret: this.generateSecretKey(), roleId: viewerRole ? viewerRole.id : -1, createdBy: createdBy, }; const token = await this.store.insert(newToken); this.activeTokens.push(token); await this.eventStore.store( new PublicSignupTokenCreatedEvent({ createdBy: createdBy, data: newToken, }), ); return token; } private generateSecretKey(): string { return crypto.randomBytes(32).toString('hex'); } destroy(): void { clearInterval(this.timer); this.timer = null; } }