/* * SPDX-FileCopyrightText: syuilo and other misskey contributors * SPDX-License-Identifier: AGPL-3.0-only */ import { Inject, Injectable } from '@nestjs/common'; import * as Redis from 'ioredis'; import { ModuleRef } from '@nestjs/core'; import * as Reversi from 'misskey-reversi'; import { IsNull, LessThan, MoreThan } from 'typeorm'; import type { MiReversiGame, ReversiGamesRepository, } from '@/models/_.js'; import type { MiUser } from '@/models/User.js'; import { DI } from '@/di-symbols.js'; import { bindThis } from '@/decorators.js'; import { CacheService } from '@/core/CacheService.js'; import { UserEntityService } from '@/core/entities/UserEntityService.js'; import { GlobalEventService } from '@/core/GlobalEventService.js'; import { IdService } from '@/core/IdService.js'; import { NotificationService } from '@/core/NotificationService.js'; import { Serialized } from '@/types.js'; import { ReversiGameEntityService } from './entities/ReversiGameEntityService.js'; import type { OnApplicationShutdown, OnModuleInit } from '@nestjs/common'; const MATCHING_TIMEOUT_MS = 1000 * 20; // 20sec @Injectable() export class ReversiService implements OnApplicationShutdown, OnModuleInit { private notificationService: NotificationService; constructor( private moduleRef: ModuleRef, @Inject(DI.redis) private redisClient: Redis.Redis, @Inject(DI.reversiGamesRepository) private reversiGamesRepository: ReversiGamesRepository, private cacheService: CacheService, private userEntityService: UserEntityService, private globalEventService: GlobalEventService, private reversiGameEntityService: ReversiGameEntityService, private idService: IdService, ) { } async onModuleInit() { this.notificationService = this.moduleRef.get(NotificationService.name); } @bindThis private async cacheGame(game: MiReversiGame) { await this.redisClient.setex(`reversi:game:cache:${game.id}`, 60 * 60, JSON.stringify(game)); } @bindThis private async deleteGameCache(gameId: MiReversiGame['id']) { await this.redisClient.del(`reversi:game:cache:${gameId}`); } @bindThis private getBakeProps(game: MiReversiGame) { return { startedAt: game.startedAt, endedAt: game.endedAt, // ゲームの途中からユーザーが変わることは無いので //user1Id: game.user1Id, //user2Id: game.user2Id, user1Ready: game.user1Ready, user2Ready: game.user2Ready, black: game.black, isStarted: game.isStarted, isEnded: game.isEnded, winnerId: game.winnerId, surrenderedUserId: game.surrenderedUserId, timeoutUserId: game.timeoutUserId, isLlotheo: game.isLlotheo, canPutEverywhere: game.canPutEverywhere, loopedBoard: game.loopedBoard, timeLimitForEachTurn: game.timeLimitForEachTurn, logs: game.logs, map: game.map, bw: game.bw, crc32: game.crc32, } satisfies Partial; } @bindThis public async matchSpecificUser(me: MiUser, targetUser: MiUser, multiple = false): Promise { if (targetUser.id === me.id) { throw new Error('You cannot match yourself.'); } if (!multiple) { // 既にマッチしている対局が無いか探す(3分以内) const games = await this.reversiGamesRepository.find({ where: [ { id: MoreThan(this.idService.gen(Date.now() - 1000 * 60 * 3)), user1Id: me.id, user2Id: targetUser.id, isStarted: false }, { id: MoreThan(this.idService.gen(Date.now() - 1000 * 60 * 3)), user1Id: targetUser.id, user2Id: me.id, isStarted: false }, ], relations: ['user1', 'user2'], order: { id: 'DESC' }, }); if (games.length > 0) { return games[0]; } } //#region 相手から既に招待されてないか確認 const invitations = await this.redisClient.zrange( `reversi:matchSpecific:${me.id}`, Date.now() - MATCHING_TIMEOUT_MS, '+inf', 'BYSCORE'); if (invitations.includes(targetUser.id)) { await this.redisClient.zrem(`reversi:matchSpecific:${me.id}`, targetUser.id); const game = await this.matched(targetUser.id, me.id); return game; } //#endregion this.redisClient.zadd(`reversi:matchSpecific:${targetUser.id}`, Date.now(), me.id); this.globalEventService.publishReversiStream(targetUser.id, 'invited', { user: await this.userEntityService.pack(me, targetUser), }); return null; } @bindThis public async matchAnyUser(me: MiUser, multiple = false): Promise { if (!multiple) { // 既にマッチしている対局が無いか探す(3分以内) const games = await this.reversiGamesRepository.find({ where: [ { id: MoreThan(this.idService.gen(Date.now() - 1000 * 60 * 3)), user1Id: me.id, isStarted: false }, { id: MoreThan(this.idService.gen(Date.now() - 1000 * 60 * 3)), user2Id: me.id, isStarted: false }, ], relations: ['user1', 'user2'], order: { id: 'DESC' }, }); if (games.length > 0) { return games[0]; } } //#region まず自分宛ての招待を探す const invitations = await this.redisClient.zrange( `reversi:matchSpecific:${me.id}`, Date.now() - MATCHING_TIMEOUT_MS, '+inf', 'BYSCORE'); if (invitations.length > 0) { const invitorId = invitations[Math.floor(Math.random() * invitations.length)]; await this.redisClient.zrem(`reversi:matchSpecific:${me.id}`, invitorId); const game = await this.matched(invitorId, me.id); return game; } //#endregion const matchings = await this.redisClient.zrange( 'reversi:matchAny', Date.now() - MATCHING_TIMEOUT_MS, '+inf', 'BYSCORE'); const userIds = matchings.filter(id => id !== me.id); if (userIds.length > 0) { // pick random const matchedUserId = userIds[Math.floor(Math.random() * userIds.length)]; await this.redisClient.zrem('reversi:matchAny', me.id, matchedUserId); const game = await this.matched(matchedUserId, me.id); return game; } else { await this.redisClient.zadd('reversi:matchAny', Date.now(), me.id); return null; } } @bindThis public async matchSpecificUserCancel(user: MiUser, targetUserId: MiUser['id']) { await this.redisClient.zrem(`reversi:matchSpecific:${targetUserId}`, user.id); } @bindThis public async matchAnyUserCancel(user: MiUser) { await this.redisClient.zrem('reversi:matchAny', user.id); } @bindThis public async cleanOutdatedGames() { await this.reversiGamesRepository.delete({ id: LessThan(this.idService.gen(Date.now() - 1000 * 60 * 10)), isStarted: false, }); } @bindThis public async gameReady(gameId: MiReversiGame['id'], user: MiUser, ready: boolean) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (game.isStarted) return; let isBothReady = false; if (game.user1Id === user.id) { const updatedGame = { ...game, user1Ready: ready, }; this.cacheGame(updatedGame); this.globalEventService.publishReversiGameStream(game.id, 'changeReadyStates', { user1: ready, user2: updatedGame.user2Ready, }); if (ready && updatedGame.user2Ready) isBothReady = true; } else if (game.user2Id === user.id) { const updatedGame = { ...game, user2Ready: ready, }; this.cacheGame(updatedGame); this.globalEventService.publishReversiGameStream(game.id, 'changeReadyStates', { user1: updatedGame.user1Ready, user2: ready, }); if (ready && updatedGame.user1Ready) isBothReady = true; } else { return; } if (isBothReady) { // 3秒後、両者readyならゲーム開始 setTimeout(async () => { const freshGame = await this.get(game.id); if (freshGame == null || freshGame.isStarted || freshGame.isEnded) return; if (!freshGame.user1Ready || !freshGame.user2Ready) return; this.startGame(freshGame); }, 3000); } } @bindThis private async matched(parentId: MiUser['id'], childId: MiUser['id']): Promise { const game = await this.reversiGamesRepository.insert({ id: this.idService.gen(), user1Id: parentId, user2Id: childId, user1Ready: false, user2Ready: false, isStarted: false, isEnded: false, logs: [], map: Reversi.maps.eighteight.data, bw: 'random', isLlotheo: false, }).then(x => this.reversiGamesRepository.findOneOrFail({ where: { id: x.identifiers[0].id }, relations: ['user1', 'user2'], })); this.cacheGame(game); const packed = await this.reversiGameEntityService.packDetail(game); this.globalEventService.publishReversiStream(parentId, 'matched', { game: packed }); return game; } @bindThis private async startGame(game: MiReversiGame) { let bw: number; if (game.bw === 'random') { bw = Math.random() > 0.5 ? 1 : 2; } else { bw = parseInt(game.bw, 10); } const engine = new Reversi.Game(game.map, { isLlotheo: game.isLlotheo, canPutEverywhere: game.canPutEverywhere, loopedBoard: game.loopedBoard, }); const crc32 = engine.calcCrc32().toString(); const updatedGame = await this.reversiGamesRepository.createQueryBuilder().update() .set({ ...this.getBakeProps(game), startedAt: new Date(), isStarted: true, black: bw, map: game.map, crc32, }) .where('id = :id', { id: game.id }) .returning('*') .execute() .then((response) => response.raw[0]); // キャッシュ効率化のためにユーザー情報は再利用 updatedGame.user1 = game.user1; updatedGame.user2 = game.user2; this.cacheGame(updatedGame); //#region 盤面に最初から石がないなどして始まった瞬間に勝敗が決定する場合があるのでその処理 if (engine.isEnded) { let winnerId; if (engine.winner === true) { winnerId = bw === 1 ? updatedGame.user1Id : updatedGame.user2Id; } else if (engine.winner === false) { winnerId = bw === 1 ? updatedGame.user2Id : updatedGame.user1Id; } else { winnerId = null; } await this.endGame(updatedGame, winnerId, null); return; } //#endregion this.redisClient.setex(`reversi:game:turnTimer:${game.id}:1`, updatedGame.timeLimitForEachTurn, ''); this.globalEventService.publishReversiGameStream(game.id, 'started', { game: await this.reversiGameEntityService.packDetail(updatedGame), }); } @bindThis private async endGame(game: MiReversiGame, winnerId: MiUser['id'] | null, reason: 'surrender' | 'timeout' | null) { const updatedGame = await this.reversiGamesRepository.createQueryBuilder().update() .set({ ...this.getBakeProps(game), isEnded: true, endedAt: new Date(), winnerId: winnerId, surrenderedUserId: reason === 'surrender' ? (winnerId === game.user1Id ? game.user2Id : game.user1Id) : null, timeoutUserId: reason === 'timeout' ? (winnerId === game.user1Id ? game.user2Id : game.user1Id) : null, }) .where('id = :id', { id: game.id }) .returning('*') .execute() .then((response) => response.raw[0]); // キャッシュ効率化のためにユーザー情報は再利用 updatedGame.user1 = game.user1; updatedGame.user2 = game.user2; this.cacheGame(updatedGame); this.globalEventService.publishReversiGameStream(game.id, 'ended', { winnerId: winnerId, game: await this.reversiGameEntityService.packDetail(updatedGame), }); } @bindThis public async getInvitations(user: MiUser): Promise { const invitations = await this.redisClient.zrange( `reversi:matchSpecific:${user.id}`, Date.now() - MATCHING_TIMEOUT_MS, '+inf', 'BYSCORE'); return invitations; } @bindThis public async updateSettings(gameId: MiReversiGame['id'], user: MiUser, key: string, value: any) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (game.isStarted) return; if ((game.user1Id !== user.id) && (game.user2Id !== user.id)) return; if ((game.user1Id === user.id) && game.user1Ready) return; if ((game.user2Id === user.id) && game.user2Ready) return; if (!['map', 'bw', 'isLlotheo', 'canPutEverywhere', 'loopedBoard', 'timeLimitForEachTurn'].includes(key)) return; // TODO: より厳格なバリデーション const updatedGame = { ...game, [key]: value, }; this.cacheGame(updatedGame); this.globalEventService.publishReversiGameStream(game.id, 'updateSettings', { userId: user.id, key: key, value: value, }); } @bindThis public async putStoneToGame(gameId: MiReversiGame['id'], user: MiUser, pos: number, id?: string | null) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (!game.isStarted) return; if (game.isEnded) return; if ((game.user1Id !== user.id) && (game.user2Id !== user.id)) return; const myColor = ((game.user1Id === user.id) && game.black === 1) || ((game.user2Id === user.id) && game.black === 2) ? true : false; const engine = Reversi.Serializer.restoreGame({ map: game.map, isLlotheo: game.isLlotheo, canPutEverywhere: game.canPutEverywhere, loopedBoard: game.loopedBoard, logs: game.logs, }); if (engine.turn !== myColor) return; if (!engine.canPut(myColor, pos)) return; engine.putStone(pos); const logs = Reversi.Serializer.deserializeLogs(game.logs); const log = { time: Date.now(), player: myColor, operation: 'put', pos, } as const; logs.push(log); const serializeLogs = Reversi.Serializer.serializeLogs(logs); const crc32 = engine.calcCrc32().toString(); const updatedGame = { ...game, crc32, logs: serializeLogs, }; this.cacheGame(updatedGame); this.globalEventService.publishReversiGameStream(game.id, 'log', { ...log, id: id ?? null, }); if (engine.isEnded) { let winnerId; if (engine.winner === true) { winnerId = game.black === 1 ? game.user1Id : game.user2Id; } else if (engine.winner === false) { winnerId = game.black === 1 ? game.user2Id : game.user1Id; } else { winnerId = null; } await this.endGame(updatedGame, winnerId, null); } else { this.redisClient.setex(`reversi:game:turnTimer:${game.id}:${engine.turn ? '1' : '0'}`, updatedGame.timeLimitForEachTurn, ''); } } @bindThis public async surrender(gameId: MiReversiGame['id'], user: MiUser) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (game.isEnded) return; if ((game.user1Id !== user.id) && (game.user2Id !== user.id)) return; const winnerId = game.user1Id === user.id ? game.user2Id : game.user1Id; await this.endGame(game, winnerId, 'surrender'); } @bindThis public async checkTimeout(gameId: MiReversiGame['id']) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (game.isEnded) return; const engine = Reversi.Serializer.restoreGame({ map: game.map, isLlotheo: game.isLlotheo, canPutEverywhere: game.canPutEverywhere, loopedBoard: game.loopedBoard, logs: game.logs, }); if (engine.turn == null) return; const timer = await this.redisClient.exists(`reversi:game:turnTimer:${game.id}:${engine.turn ? '1' : '0'}`); if (timer === 0) { const winnerId = engine.turn ? (game.black === 1 ? game.user2Id : game.user1Id) : (game.black === 1 ? game.user1Id : game.user2Id); await this.endGame(game, winnerId, 'timeout'); } } @bindThis public async cancelGame(gameId: MiReversiGame['id'], user: MiUser) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (game.isStarted) return; if ((game.user1Id !== user.id) && (game.user2Id !== user.id)) return; await this.reversiGamesRepository.delete(game.id); this.deleteGameCache(game.id); this.globalEventService.publishReversiGameStream(game.id, 'canceled', { userId: user.id, }); } @bindThis public async get(id: MiReversiGame['id']): Promise { const cached = await this.redisClient.get(`reversi:game:cache:${id}`); if (cached != null) { // TODO: この辺りのデシリアライズ処理をどこか別のサービスに切り出したい const parsed = JSON.parse(cached) as Serialized; return { ...parsed, startedAt: parsed.startedAt != null ? new Date(parsed.startedAt) : null, endedAt: parsed.endedAt != null ? new Date(parsed.endedAt) : null, user1: parsed.user1 != null ? { ...parsed.user1, avatar: null, banner: null, updatedAt: parsed.user1.updatedAt != null ? new Date(parsed.user1.updatedAt) : null, lastActiveDate: parsed.user1.lastActiveDate != null ? new Date(parsed.user1.lastActiveDate) : null, lastFetchedAt: parsed.user1.lastFetchedAt != null ? new Date(parsed.user1.lastFetchedAt) : null, movedAt: parsed.user1.movedAt != null ? new Date(parsed.user1.movedAt) : null, } : null, user2: parsed.user2 != null ? { ...parsed.user2, avatar: null, banner: null, updatedAt: parsed.user2.updatedAt != null ? new Date(parsed.user2.updatedAt) : null, lastActiveDate: parsed.user2.lastActiveDate != null ? new Date(parsed.user2.lastActiveDate) : null, lastFetchedAt: parsed.user2.lastFetchedAt != null ? new Date(parsed.user2.lastFetchedAt) : null, movedAt: parsed.user2.movedAt != null ? new Date(parsed.user2.movedAt) : null, } : null, }; } else { const game = await this.reversiGamesRepository.findOne({ where: { id }, relations: ['user1', 'user2'], }); if (game == null) return null; this.cacheGame(game); return game; } } @bindThis public async checkCrc(gameId: MiReversiGame['id'], crc32: string | number) { const game = await this.get(gameId); if (game == null) throw new Error('game not found'); if (crc32.toString() !== game.crc32) { return game; } else { return null; } } @bindThis public dispose(): void { } @bindThis public onApplicationShutdown(signal?: string | undefined): void { this.dispose(); } }