2023-07-27 05:31:52 +00:00
|
|
|
/*
|
2024-02-13 15:59:27 +00:00
|
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
2023-07-27 05:31:52 +00:00
|
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
*/
|
|
|
|
|
2023-02-20 07:40:24 +00:00
|
|
|
import * as OTPAuth from 'otpauth';
|
2022-09-17 18:27:08 +00:00
|
|
|
import { Inject, Injectable } from '@nestjs/common';
|
|
|
|
import { Endpoint } from '@/server/api/endpoint-base.js';
|
2023-02-20 07:40:24 +00:00
|
|
|
import { UserEntityService } from '@/core/entities/UserEntityService.js';
|
2023-09-15 05:28:29 +00:00
|
|
|
import type { UserProfilesRepository } from '@/models/_.js';
|
2023-02-20 07:40:24 +00:00
|
|
|
import { GlobalEventService } from '@/core/GlobalEventService.js';
|
2022-09-17 18:27:08 +00:00
|
|
|
import { DI } from '@/di-symbols.js';
|
2017-12-08 13:57:58 +00:00
|
|
|
|
2018-07-16 19:36:44 +00:00
|
|
|
export const meta = {
|
2022-01-18 13:27:10 +00:00
|
|
|
requireCredential: true,
|
2018-11-02 03:49:08 +00:00
|
|
|
|
|
|
|
secure: true,
|
2024-02-16 05:27:33 +00:00
|
|
|
|
|
|
|
res: {
|
|
|
|
type: 'object',
|
|
|
|
properties: {
|
|
|
|
backupCodes: {
|
|
|
|
type: 'array',
|
|
|
|
optional: false,
|
|
|
|
items: {
|
|
|
|
type: 'string',
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2022-02-19 05:05:32 +00:00
|
|
|
} as const;
|
2018-11-02 03:49:08 +00:00
|
|
|
|
2022-02-20 04:15:40 +00:00
|
|
|
export const paramDef = {
|
2022-02-19 05:05:32 +00:00
|
|
|
type: 'object',
|
|
|
|
properties: {
|
|
|
|
token: { type: 'string' },
|
2021-12-09 14:58:30 +00:00
|
|
|
},
|
2022-02-19 05:05:32 +00:00
|
|
|
required: ['token'],
|
2022-01-18 13:27:10 +00:00
|
|
|
} as const;
|
2018-07-16 19:36:44 +00:00
|
|
|
|
2022-09-17 18:27:08 +00:00
|
|
|
@Injectable()
|
2023-08-17 12:20:58 +00:00
|
|
|
export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-disable-line import/no-default-export
|
2022-09-17 18:27:08 +00:00
|
|
|
constructor(
|
|
|
|
@Inject(DI.userProfilesRepository)
|
|
|
|
private userProfilesRepository: UserProfilesRepository,
|
2023-02-20 07:40:24 +00:00
|
|
|
|
|
|
|
private userEntityService: UserEntityService,
|
|
|
|
private globalEventService: GlobalEventService,
|
2022-09-17 18:27:08 +00:00
|
|
|
) {
|
|
|
|
super(meta, paramDef, async (ps, me) => {
|
|
|
|
const token = ps.token.replace(/\s/g, '');
|
|
|
|
|
|
|
|
const profile = await this.userProfilesRepository.findOneByOrFail({ userId: me.id });
|
|
|
|
|
|
|
|
if (profile.twoFactorTempSecret == null) {
|
|
|
|
throw new Error('二段階認証の設定が開始されていません');
|
|
|
|
}
|
|
|
|
|
2023-02-20 07:40:24 +00:00
|
|
|
const delta = OTPAuth.TOTP.validate({
|
|
|
|
secret: OTPAuth.Secret.fromBase32(profile.twoFactorTempSecret),
|
|
|
|
digits: 6,
|
|
|
|
token,
|
2023-09-22 05:12:33 +00:00
|
|
|
window: 5,
|
2022-09-17 18:27:08 +00:00
|
|
|
});
|
|
|
|
|
2023-02-20 07:40:24 +00:00
|
|
|
if (delta === null) {
|
2022-09-17 18:27:08 +00:00
|
|
|
throw new Error('not verified');
|
|
|
|
}
|
|
|
|
|
2023-08-28 09:25:31 +00:00
|
|
|
const backupCodes = Array.from({ length: 5 }, () => new OTPAuth.Secret().base32);
|
|
|
|
|
2022-09-17 18:27:08 +00:00
|
|
|
await this.userProfilesRepository.update(me.id, {
|
|
|
|
twoFactorSecret: profile.twoFactorTempSecret,
|
2023-08-28 09:25:31 +00:00
|
|
|
twoFactorBackupSecret: backupCodes,
|
2022-09-17 18:27:08 +00:00
|
|
|
twoFactorEnabled: true,
|
|
|
|
});
|
2023-02-20 07:40:24 +00:00
|
|
|
|
|
|
|
// Publish meUpdated event
|
|
|
|
this.globalEventService.publishMainStream(me.id, 'meUpdated', await this.userEntityService.pack(me.id, me, {
|
2024-01-31 06:45:35 +00:00
|
|
|
schema: 'MeDetailed',
|
2023-02-20 07:40:24 +00:00
|
|
|
includeSecrets: true,
|
|
|
|
}));
|
2023-08-28 09:25:31 +00:00
|
|
|
|
|
|
|
return {
|
|
|
|
backupCodes: backupCodes,
|
|
|
|
};
|
2022-09-17 18:27:08 +00:00
|
|
|
});
|
2017-12-08 13:57:58 +00:00
|
|
|
}
|
2022-09-17 18:27:08 +00:00
|
|
|
}
|