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

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

@@ -5,6 +5,7 @@ import { UserRole } from '@mp-pilates/shared'
 import { AuthService, RANDOM_FN_TOKEN } from '../auth.service'
 import { WechatService } from '../wechat.service'
 import { PrismaService } from '../../prisma/prisma.service'
+import { InviteService } from '../../invite/invite.service'
 
 // ─── Fixtures ────────────────────────────────────────────────────────────────
 
@@ -46,6 +47,10 @@ const mockJwtService = {
   sign: jest.fn(),
 }
 
+const mockInviteService = {
+  bindInviterToUser: jest.fn(),
+}
+
 // ─── Tests ───────────────────────────────────────────────────────────────────
 
 describe('AuthService', () => {
@@ -58,6 +63,7 @@ describe('AuthService', () => {
         { provide: PrismaService, useValue: mockPrismaService },
         { provide: WechatService, useValue: mockWechatService },
         { provide: JwtService, useValue: mockJwtService },
+        { provide: InviteService, useValue: mockInviteService },
         { provide: RANDOM_FN_TOKEN, useValue: () => 0 }, // deterministic nickname
       ],
     }).compile()
@@ -91,10 +97,20 @@ describe('AuthService', () => {
         where: { openid: OPENID },
       })
       expect(mockPrismaService.user.create).toHaveBeenCalledWith({
-        data: { openid: OPENID, nickname: TEST_NICKNAME },
+        data: { openid: OPENID, nickname: TEST_NICKNAME, adminBookingSubscriptionCount: 0 },
       })
       expect(result.user).toEqual(mockUser)
       expect(result.isNewUser).toBe(true)
+      expect(mockInviteService.bindInviterToUser).toHaveBeenCalledWith(USER_ID, undefined)
+    })
+
+    it('binds inviter for new users when inviterId is present', async () => {
+      mockPrismaService.user.findUnique.mockResolvedValue(null)
+      mockPrismaService.user.create.mockResolvedValue(mockUser)
+
+      await authService.login(loginCode, undefined, undefined, 'inviter-001')
+
+      expect(mockInviteService.bindInviterToUser).toHaveBeenCalledWith(USER_ID, 'inviter-001')
     })
 
     it('creates user with unionid when present', async () => {
@@ -110,7 +126,7 @@ describe('AuthService', () => {
       await authService.login(loginCode)
 
       expect(mockPrismaService.user.create).toHaveBeenCalledWith({
-        data: { openid: OPENID, unionid, nickname: TEST_NICKNAME },
+        data: { openid: OPENID, unionid, nickname: TEST_NICKNAME, adminBookingSubscriptionCount: 0 },
       })
     })
 

packages/server/src/auth/auth.controller.ts

@@ -29,6 +29,7 @@ export class AuthController {
       loginDto.code,
       loginDto.nickname,
       loginDto.avatarUrl,
+      loginDto.inviterId,
     )
   }
 

packages/server/src/auth/auth.module.ts

@@ -8,10 +8,12 @@ import { WechatService } from './wechat.service'
 import { JwtStrategy } from './jwt.strategy'
 import { JwtAuthGuard } from './jwt-auth.guard'
 import { RolesGuard } from './roles.guard'
+import { InviteModule } from '../invite/invite.module'
 
 @Module({
   imports: [
     PassportModule,
+    InviteModule,
     JwtModule.registerAsync({
       imports: [ConfigModule],
       inject: [ConfigService],

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)
 

packages/server/src/auth/dto/login.dto.ts

@@ -12,4 +12,8 @@ export class LoginDto {
   @IsString()
   @IsOptional()
   avatarUrl?: string
+
+  @IsString()
+  @IsOptional()
+  inviterId?: string
 }