[Implement][Provider] Login API for Provider

src/app.module.ts

@@ -13,6 +13,7 @@ import { CustomerModule } from '@customer/customer.module'
 import { AuthModule } from '@auth/auth.module'
 import { ProductModule } from '@product/product.module'
 import { CartModule } from '@cart/cart.module'
+import { StaffModule } from '@staff/staff.module'
 
 @Module({
   imports: [
@@ -61,7 +62,8 @@ import { CartModule } from '@cart/cart.module'
     CustomerModule,
     AuthModule,
     ProductModule,
-    CartModule
+    CartModule,
+    StaffModule
   ],
   controllers: [AppController],
   providers: [AppService]

src/auth/auth.module.ts

@@ -3,19 +3,22 @@ import { JwtAccessStrategy } from '@auth/strategies/jwt-access.strategy'
 import { PassportModule } from '@nestjs/passport'
 import { CustomerModule } from '@customer/customer.module'
 import { AuthService } from '@auth/services/auth.service'
-import { AuthController } from '@auth/controllers/auth.controller'
+import { AuthCustomerController } from '@auth/controllers/customer.controller'
 import { JwtModule } from '@nestjs/jwt'
 import { ConfigModule } from '@nestjs/config'
 import { JwtRefreshStrategy } from '@auth/strategies/jwt-refresh.strategy'
+import { StaffModule } from '@staff/staff.module'
+import { AuthProviderController } from '@auth/controllers/provider.controller'
 
 @Module({
   imports: [
     ConfigModule,
     CustomerModule,
+    StaffModule,
     PassportModule,
     JwtModule
   ],
-  controllers: [AuthController],
+  controllers: [AuthCustomerController, AuthProviderController],
   providers: [AuthService, JwtAccessStrategy, JwtRefreshStrategy],
   exports: [AuthService]
 })

src/auth/controllers/auth.controller.ts → src/auth/controllers/customer.controller.ts

@@ -8,12 +8,12 @@ import { UserSide } from '@common/contracts/constant'
 import { JwtAuthGuard } from '@auth/guards/jwt-auth.guard'
 import { RegisterReqDto } from '@auth/dto/register.dto'
 
-@ApiTags('Auth')
-@Controller()
-export class AuthController {
+@ApiTags('Auth - Customer')
+@Controller('customer')
+export class AuthCustomerController {
   constructor(private readonly authService: AuthService) {}
 
-  @Post('customer/login')
+  @Post('login')
   @ApiBody({ type: LoginReqDto })
   @ApiOkResponse({ type: ResponseTokenDto })
   @ApiBadRequestResponse({ type: ErrorResponse })
@@ -23,7 +23,7 @@ export class AuthController {
     return res
   }
 
-  @Post('customer/register')
+  @Post('register')
   @ApiBody({ type: RegisterReqDto })
   @ApiOkResponse({ type: ResponseSuccessDto })
   @ApiBadRequestResponse({ type: ErrorResponse })
@@ -32,7 +32,7 @@ export class AuthController {
   }
 
   @UseGuards(JwtAuthGuard.REFRESH_TOKEN)
-  @Post('customer/refresh')
+  @Post('refresh')
   @ApiBearerAuth()
   @ApiOkResponse({ type: ResponseTokenDto })
   @ApiBadRequestResponse({ type: ErrorResponse })

src/auth/controllers/provider.controller.ts

@@ -0,0 +1,21 @@
+import { Body, Controller, Post } from "@nestjs/common"
+import { ApiBadRequestResponse, ApiBody, ApiOkResponse, ApiTags } from "@nestjs/swagger"
+import { UserSide } from "@common/contracts/constant"
+import { ErrorResponse } from "@common/contracts/dto"
+import { LoginReqDto } from "@auth/dto/login.dto"
+import { ResponseTokenDto, TokenResDto } from "@auth/dto/token.dto"
+import { AuthService } from "@auth/services/auth.service"
+
+@ApiTags('Auth - Provider')
+@Controller('provider')
+export class AuthProviderController {
+  constructor(private readonly authService: AuthService) {}
+
+  @Post('login')
+  @ApiBody({ type: LoginReqDto })
+  @ApiOkResponse({ type: ResponseTokenDto })
+  @ApiBadRequestResponse({ type: ErrorResponse })
+  async login(@Body() loginReqDto: LoginReqDto): Promise<TokenResDto> {
+    return await this.authService.login(loginReqDto, UserSide.PROVIDER)
+  }
+}

src/auth/services/auth.service.ts

@@ -11,17 +11,20 @@ import { RefreshTokenPayload } from '@auth/strategies/jwt-refresh.strategy'
 import { TokenResDto } from '@auth/dto/token.dto'
 import { ConfigService } from '@nestjs/config'
 import { RegisterReqDto } from '@auth/dto/register.dto'
+import { StaffRepository } from '@staff/repositories/staff.repository'
+import { Staff } from '@staff/schemas/staff.schema'
 
 @Injectable()
 export class AuthService {
   constructor(
     private readonly customerRepository: CustomerRepository,
+    private readonly staffRepository: StaffRepository,
     private readonly jwtService: JwtService,
     private readonly configService: ConfigService
   ) {}
 
   public async login(loginReqDto: LoginReqDto, side: UserSide): Promise<TokenResDto> {
-    let user: Customer | null
+    let user: Customer | Staff
     let userRole: UserRole
 
     if (side === UserSide.CUSTOMER) {
@@ -34,6 +37,16 @@ export class AuthService {
       userRole = UserRole.CUSTOMER
     }
 
+    if (side === UserSide.PROVIDER) {
+      user = (await this.staffRepository.findOne({
+        conditions: {
+          email: loginReqDto.email
+        }
+      }))
+
+      userRole = user?.role
+    }
+
     if (!user) throw new BadRequestException(Errors.WRONG_EMAIL_OR_PASSWORD.message)
 
     const isPasswordMatch = await this.comparePassword(loginReqDto.password, user.password)

src/staff/repositories/staff.repository.ts

@@ -0,0 +1,12 @@
+import { PaginateModel } from 'mongoose'
+import { Injectable } from '@nestjs/common'
+import { InjectModel } from '@nestjs/mongoose'
+import { AbstractRepository } from '@common/repositories'
+import { Staff, StaffDocument } from '@staff/schemas/staff.schema'
+
+@Injectable()
+export class StaffRepository extends AbstractRepository<StaffDocument> {
+  constructor(@InjectModel(Staff.name) model: PaginateModel<StaffDocument>) {
+    super(model)
+  }
+}

src/staff/schemas/staff.schema.ts

@@ -0,0 +1,83 @@
+import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose'
+import { ApiProperty } from '@nestjs/swagger'
+import { Status, UserRole } from '@src/common/contracts/constant'
+import { Transform } from 'class-transformer'
+import { HydratedDocument } from 'mongoose'
+import * as paginate from 'mongoose-paginate-v2'
+
+export type StaffDocument = HydratedDocument<Staff>
+
+@Schema({
+  collection: 'staffs',
+  timestamps: {
+    createdAt: true,
+    updatedAt: true
+  },
+  toJSON: {
+    transform(doc, ret) {
+      delete ret.__v
+    }
+  }
+})
+export class Staff {
+  constructor(id?: string) {
+    this._id = id
+  }
+
+  @ApiProperty()
+  @Transform(({ value }) => value?.toString())
+  _id: string
+
+  @ApiProperty()
+  @Prop({ type: String, maxlength: 30, required: true })
+  firstName: string
+
+  @ApiProperty()
+  @Prop({ type: String, maxlength: 30, required: true })
+  lastName: string
+
+  @ApiProperty()
+  @Prop({
+    type: String,
+    required: true
+  })
+  email: string
+
+  @ApiProperty()
+  @Prop({ type: String, required: true })
+  password: string
+
+  @ApiProperty()
+  @Prop({ type: String, required: true })
+  staffCode: string
+
+  @ApiProperty()
+  @Prop({
+    type: String,
+    required: true
+  })
+  phone: string
+
+  @ApiProperty()
+  @Prop({
+    type: String
+  })
+  avatar: string
+
+  @ApiProperty()
+  @Prop({
+    enum: UserRole,
+    default: UserRole.STAFF
+  })
+  role: UserRole
+
+  @Prop({
+    enum: Status,
+    default: Status.ACTIVE
+  })
+  status: Status
+}
+
+export const StaffSchema = SchemaFactory.createForClass(Staff)
+
+StaffSchema.plugin(paginate)

src/staff/staff.module.ts

@@ -0,0 +1,12 @@
+import { Module } from '@nestjs/common'
+import { MongooseModule } from '@nestjs/mongoose'
+import { StaffRepository } from '@staff/repositories/staff.repository'
+import { Staff, StaffSchema } from '@staff/schemas/staff.schema'
+
+@Module({
+  imports: [MongooseModule.forFeature([{ name: Staff.name, schema: StaffSchema }])],
+  controllers: [],
+  providers: [StaffRepository],
+  exports: [StaffRepository]
+})
+export class StaffModule {}

tsconfig.json

@@ -29,6 +29,7 @@
       "@booking/*": ["src/booking/*"],
       "@product/*": ["src/product/*"],
       "@cart/*": ["src/cart/*"],
+      "@staff/*": ["src/staff/*"],
     }
   }
 }