64 lines
1.8 KiB
TypeScript
64 lines
1.8 KiB
TypeScript
import { Inject, Injectable } from '@nestjs/common';
|
|
import { DI } from '@/di-symbols.js';
|
|
import type { SigninsRepository } from '@/models/index.js';
|
|
import type { UsersRepository } from '@/models/index.js';
|
|
import type { 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));
|
|
})();
|
|
}
|
|
}
|
|
|