import { Inject, Injectable } from '@nestjs/common'; import { DI } from '@/di-symbols.js'; import { SigninsRepository } from '@/models/index.js'; import type { UsersRepository } from '@/models/index.js'; import { Config } from '@/config.js'; import { IdService } from '@/core/IdService.js'; import type { ILocalUser } from '@/models/entities/User.js'; import { GlobalEventService } from '@/core/GlobalEventService.js'; import { SigninEntityService } from '@/core/entities/SigninEntityService.js'; import type Koa from 'koa'; @Injectable() export class SigninService { constructor( @Inject(DI.config) private config: Config, @Inject(DI.signinsRepository) private signinsRepository: SigninsRepository, private signinEntityService: SigninEntityService, private idService: IdService, private globalEventService: GlobalEventService, ) { } public signin(ctx: Koa.Context, user: ILocalUser, redirect = false) { if (redirect) { //#region Cookie ctx.cookies.set('igi', user.token!, { path: '/', // SEE: https://github.com/koajs/koa/issues/974 // When using a SSL proxy it should be configured to add the "X-Forwarded-Proto: https" header secure: this.config.url.startsWith('https'), httpOnly: false, }); //#endregion ctx.redirect(this.config.url); } else { ctx.body = { id: user.id, i: user.token, }; ctx.status = 200; } (async () => { // Append signin history const record = await this.signinsRepository.insert({ id: this.idService.genId(), createdAt: new Date(), userId: user.id, ip: ctx.ip, headers: ctx.headers, success: true, }).then(x => this.signinsRepository.findOneByOrFail(x.identifiers[0])); // Publish signin event this.globalEventService.publishMainStream(user.id, 'signin', await this.signinEntityService.pack(record)); })(); } }