ip tagging

Author: enchantinggg4

src/core.controller.ts

@@ -19,6 +19,7 @@ import { PlayerAbandonedEvent } from 'gateway/events/bans/player-abandoned.event
 import { LobbyReadyEvent } from 'gateway/events/lobby-ready.event';
 import { ConfigService } from '@nestjs/config';
 import { SrcdsServerStartedEvent } from 'gateway/events/srcds-server-started.event';
+import { PlayerConnectedEvent } from 'gateway/events/srcds/player-connected.event';
 
 @Controller()
 export class CoreController {
@@ -123,4 +124,9 @@ export class CoreController {
   async MatchFailedEvent(data: MatchFailedEvent) {
     this.event(MatchFailedEvent, data);
   }
+
+  @EventPattern(PlayerConnectedEvent.name)
+  async PlayerConnectedEvent(data: PlayerConnectedEvent) {
+    this.event(PlayerConnectedEvent, data);
+  }
 }

src/gameserver/event-handler/player-connected.handler.ts

@@ -0,0 +1,31 @@
+import { EventsHandler, IEventHandler } from '@nestjs/cqrs';
+import { PlayerConnectedEvent } from 'gateway/events/srcds/player-connected.event';
+import { InjectRepository } from '@nestjs/typeorm';
+import { InteractionType, PlayerIpEntity } from 'gameserver/model/player-ip.entity';
+import { Repository } from 'typeorm';
+import { Logger } from '@nestjs/common';
+
+@EventsHandler(PlayerConnectedEvent)
+export class PlayerConnectedHandler
+  implements IEventHandler<PlayerConnectedEvent>
+{
+  private logger = new Logger(PlayerConnectedHandler.name);
+
+  constructor(
+    @InjectRepository(PlayerIpEntity)
+    private readonly playerIpEntityRepository: Repository<PlayerIpEntity>,
+  ) {}
+
+  async handle(event: PlayerConnectedEvent) {
+    this.logger.log("Handling player connect with ip", {
+      steam_id: event.playerId.value,
+      match: event.matchId,
+    });
+    const e = new PlayerIpEntity(
+      event.playerId.value,
+      event.ip,
+      InteractionType.MATCH_CONNECT,
+    );
+    await this.playerIpEntityRepository.save(e);
+  }
+}

src/gameserver/index.ts

@@ -33,6 +33,7 @@ import { PlayerAbandonedHandler } from 'gameserver/event-handler/player-abandone
 import { PrepareGameHandler } from 'gameserver/command/PrepareGame/prepare-game.handler';
 import { LobbyReadyHandler } from 'gameserver/event-handler/lobby-ready.handler';
 import { SrcdsServerStartedHandler } from 'gameserver/event-handler/srcds-server-started.handler';
+import { PlayerConnectedHandler } from 'gameserver/event-handler/player-connected.handler';
 
 const CommandHandlers = [
   FindGameServerHandler,
@@ -42,6 +43,7 @@ const CommandHandlers = [
   PrepareGameHandler
 ];
 const EventHandlers = [
+  PlayerConnectedHandler,
   StartFakeMatchHandler,
   LiveMatchUpdateHandler,
   SrcdsServerStartedHandler,

src/gameserver/model/player-ip.entity.ts

@@ -0,0 +1,24 @@
+import { Column, Entity, PrimaryColumn } from 'typeorm';
+
+export enum InteractionType {
+  MATCH_CONNECT = 'MATCH_CONNECT'
+}
+
+@Entity()
+export class PlayerIpEntity {
+  @PrimaryColumn({ name: "steam_id" })
+  steamId: string;
+
+  @PrimaryColumn({ name: "ip" })
+  ip: string;
+
+  @Column({ name: 'interaction_type' })
+  interactionType: InteractionType
+
+
+  constructor(steamId: string, ip: string, interactionType: InteractionType) {
+    this.steamId = steamId;
+    this.ip = ip;
+    this.interactionType = interactionType;
+  }
+}

src/gateway

@@ -16,6 +16,7 @@ import FinishedMatchEntity from 'gameserver/model/finished-match.entity';
 import { ItemHeroView } from 'gameserver/model/item-hero.view';
 import { AchievementEntity } from 'gameserver/model/achievement.entity';
 import { MatchmakingModeMappingEntity } from 'gameserver/model/matchmaking-mode-mapping.entity';
+import { PlayerIpEntity } from 'gameserver/model/player-ip.entity';
 
 export const Entities: Function[] = [
   GameSeasonEntity,
@@ -43,4 +44,5 @@ export const Entities: Function[] = [
 
   ItemView,
   ItemHeroView,
+  PlayerIpEntity
 ];