feat: 支持分享邀请好友功能

packages/server/src/auth/auth.service.ts

@@ -4,6 +4,7 @@ import { User } from '@prisma/client'
 import { UserRole } from '@mp-pilates/shared'
 import { PrismaService } from '../prisma/prisma.service'
 import { WechatService } from './wechat.service'
+import { InviteService } from '../invite/invite.service'
 
 export interface LoginResult {
   token: string
@@ -55,6 +56,7 @@ export class AuthService {
     private readonly prisma: PrismaService,
     private readonly jwtService: JwtService,
     private readonly wechatService: WechatService,
+    private readonly inviteService: InviteService,
     @Inject(RANDOM_FN_TOKEN) private readonly randomFn: () => number = Math.random,
   ) {}
 
@@ -62,6 +64,7 @@ export class AuthService {
     code: string,
     nickname?: string,
     avatarUrl?: string,
+    inviterId?: string,
   ): Promise<LoginResult> {
     const { openid, unionid, sessionKey } =
       await this.wechatService.code2Session(code)
@@ -98,6 +101,10 @@ export class AuthService {
 
     sessionKeyStore.set(user.id, sessionKey)
 
+    if (isNewUser) {
+      await this.inviteService.bindInviterToUser(user.id, inviterId)
+    }
+
     const payload: JwtPayload = { sub: user.id, role: user.role as UserRole }
     const token = this.jwtService.sign(payload)