tailchat/server/services/core/user/friend.service.ts

import { TcCacheCleaner } from '../../../mixins/cache.cleaner.mixin';
import type { FriendDocument, FriendModel } from '../../../models/user/friend';
import { TcService, TcDbService, TcContext } from 'tailchat-server-sdk';

interface FriendService
  extends TcService,
    TcDbService<FriendDocument, FriendModel> {}
class FriendService extends TcService {
  get serviceName(): string {
    return 'friend';
  }
  onInit(): void {
    this.registerLocalDb(require('../../../models/user/friend').default);
    // this.registerMixin(TcCacheCleaner(['cache.clean.friend']));

    this.registerAction('getAllFriends', this.getAllFriends);
    this.registerAction('buildFriendRelation', this.buildFriendRelation, {
      params: {
        user1: 'string',
        user2: 'string',
      },
    });
    this.registerAction('removeFriend', this.removeFriend, {
      params: {
        friendUserId: 'string',
      },
    });
    this.registerAction('checkIsFriend', this.checkIsFriend, {
      params: {
        targetId: 'string',
      },
    });
  }

  /**
   * 获取所有好友
   */
  async getAllFriends(ctx: TcContext<{}>) {
    const userId = ctx.meta.userId;

    const list = await this.adapter.find({
      query: {
        from: userId,
      },
    });

    const records = await this.transformDocuments(ctx, {}, list);
    const ids = records.map((r) => r.to);

    return ids;
  }

  /**
   * 构建好友关系
   */
  async buildFriendRelation(ctx: TcContext<{ user1: string; user2: string }>) {
    const { user1, user2 } = ctx.params;
    await this.adapter.model.buildFriendRelation(user1, user2);

    this.unicastNotify(ctx, user1, 'add', {
      userId: user2,
    });
    this.unicastNotify(ctx, user2, 'add', {
      userId: user1,
    });
  }

  /**
   * 移除单项好友关系
   */
  async removeFriend(ctx: TcContext<{ friendUserId: string }>) {
    const { friendUserId } = ctx.params;
    const { userId } = ctx.meta;

    await this.adapter.model.findOneAndRemove({
      from: userId,
      to: friendUserId,
    });
  }

  /**
   * 检查对方是否为自己好友
   */
  async checkIsFriend(ctx: TcContext<{ targetId: string }>) {
    const { targetId } = ctx.params;
    const userId = ctx.meta.userId;

    const isFriend = await this.adapter.model.exists({
      from: userId,
      to: targetId,
    });

    return isFriend;
  }
}
export default FriendService;