feat:增加黑名单支持

leewei0923 · Nov 26, 2022 · 8d2b69d · 8d2b69d
1 parent f06ecff
commit 8d2b69d
Show file tree

Hide file tree

Showing 3 changed files with 122 additions and 2 deletions.
diff --git a/src/ttalk/dto/ttalk.dto.ts b/src/ttalk/dto/ttalk.dto.ts
@@ -50,3 +50,19 @@ export class getAndUpdateDto {
   @IsNotEmpty()
   update_time: string;
 }
+
+/**
+ * 拉入黑名单
+ */
+
+export class PullInBlacklist {
+  @ApiProperty()
+  @IsNotEmpty()
+  user_account: string; // 申请人账号
+
+  @IsNotEmpty()
+  friend_account: string; // 被申请人账号
+
+  @IsNotEmpty()
+  blacklist?: boolean;
+}
diff --git a/src/ttalk/ttalk.controller.ts b/src/ttalk/ttalk.controller.ts
@@ -5,7 +5,12 @@ import { RegisterDto } from './dto/register.dto';
 import { TtalkService } from './ttalk.service';
 import { UpdateDto } from './dto/update.dto';
 import { AuthGuard } from '@nestjs/passport';
-import { AddFriendDto, checkOnlineDto, getAndUpdateDto } from './dto/ttalk.dto';
+import {
+  AddFriendDto,
+  checkOnlineDto,
+  getAndUpdateDto,
+  PullInBlacklist,
+} from './dto/ttalk.dto';
 import { SaveMessageDto, updateFlagDto } from './dto/message.dto';
 
 @Controller('/server/ttalk')
@@ -135,4 +140,38 @@ export class TtalkController {
   updateMessageFlag(@Body() data: updateFlagDto) {
     return this.ttalkService.updateReadFlag(data);
   }
+
+  /**
+   * 更新黑名单信息
+   */
+  @UseGuards(AuthGuard('jwt'))
+  @Post('/updateConcatBlacklist')
+  updateConcatBlacklist(@Body() data: PullInBlacklist) {
+    if (typeof data.blacklist === 'boolean') {
+      return this.ttalkService.pullIntoBlacklist(data);
+    } else {
+      return {
+        code: 400,
+        status: 'fail',
+        msg: '更新状态失败',
+      };
+    }
+  }
+
+  /**
+   * 更新黑名单信息
+   */
+  @UseGuards(AuthGuard('jwt'))
+  @Post('/deleteFriend')
+  updateConcatFriendFlag(@Body() data: PullInBlacklist) {
+    if (typeof data.blacklist === 'boolean') {
+      return this.ttalkService.DeleteFriend(data);
+    } else {
+      return {
+        code: 400,
+        status: 'fail',
+        msg: '更新状态失败',
+      };
+    }
+  }
 }
diff --git a/src/ttalk/ttalk.service.ts b/src/ttalk/ttalk.service.ts
@@ -5,7 +5,12 @@ import { encryptPassword, makeSalt } from '../utils/cryptogram';
 import { ttalk_user } from './entities/ttalk.entity.mysql';
 import { AuthService } from 'src/auth/auth.service';
 import { UpdateDto } from './dto/update.dto';
-import { AddFriendDto, checkOnlineDto, getAndUpdateDto } from './dto/ttalk.dto';
+import {
+  AddFriendDto,
+  checkOnlineDto,
+  getAndUpdateDto,
+  PullInBlacklist,
+} from './dto/ttalk.dto';
 import { ttalk_user_concat } from './entities/user_concat.entity.mysql';
 import dayjs from 'dayjs';
 import { ttalk_online } from './entities/online.entity.mysql';
@@ -390,4 +395,64 @@ export class TtalkService {
       this.MessageRecordRepository.query(queryCode);
     }
   }
+
+  /**
+   * 拉进黑名单
+   */
+  async pullIntoBlacklist(data: PullInBlacklist) {
+    const { user_account, friend_account } = data;
+
+    const concatRes = await this.TTalkUserConcatRepository.findOne({
+      where: {
+        user_account,
+        friend_account,
+      },
+    });
+
+    if (typeof concatRes === 'object') {
+      const queryCode = `UPDATE ttalk_user_concat SET blacklist = ${!concatRes.blacklist} WHERE user_account = '${user_account}' and friend_account = '${friend_account}'`;
+
+      this.TTalkUserConcatRepository.query(queryCode);
+    }
+
+    return {
+      code: 200,
+      status: 'ok',
+      msg: '更新状态成功',
+      info: {
+        friend_account,
+        blacklist_status: !concatRes.blacklist,
+      },
+    };
+  }
+
+  /**
+   * 删除好友
+   */
+  async DeleteFriend(data: PullInBlacklist) {
+    const { user_account, friend_account } = data;
+
+    const concatRes = await this.TTalkUserConcatRepository.findOne({
+      where: {
+        user_account,
+        friend_account,
+      },
+    });
+
+    if (typeof concatRes === 'object') {
+      const queryCode = `DELETE ttalk_user_concat SET friend_flag = ${!concatRes.friend_flag} WHERE user_account = '${user_account}' and friend_account = '${friend_account}'`;
+
+      this.TTalkUserConcatRepository.query(queryCode);
+    }
+
+    return {
+      code: 200,
+      status: 'ok',
+      msg: '更新状态成功',
+      info: {
+        friend_account,
+        friend_status: !concatRes.friend_flag,
+      },
+    };
+  }
 }