Added Schedule module

Author: michu2k

src/app.module.ts

@@ -6,6 +6,7 @@ import { BookingModule } from "./booking/booking.module";
 import { validateEnvs } from "./env.validation";
 import { LocationModule } from "./location/location.module";
 import { PrismaModule } from "./prisma/prisma.module";
+import { ScheduleModule } from "./schedule/schedule.module";
 import { ServiceModule } from "./service/service.module";
 import { UserModule } from "./user/user.module";
 
@@ -20,6 +21,7 @@ import { UserModule } from "./user/user.module";
     UserModule,
     LocationModule,
     ServiceModule,
+    ScheduleModule,
     PrismaModule,
     BookingModule
   ],

src/booking/booking.controller.ts

@@ -45,7 +45,7 @@ export class BookingController {
   }
 
   /**
-   * Get a user's booking with the specified `id`
+   * Get a user's booking with the specified id
    */
   @Get(":id")
   async findOne(@Param("id", ParseIntPipe) id: number, @User() user: UserEntity) {
@@ -72,7 +72,7 @@ export class BookingController {
   }
 
   /**
-   * Update a booking with the specified `id`
+   * Update a booking with the specified id
    */
   @Patch(":id")
   async update(@Param("id", ParseIntPipe) id: number, @Body() data: UpdateBookingDto) {
@@ -85,7 +85,7 @@ export class BookingController {
   }
 
   /**
-   * Delete a booking with the specified `id`
+   * Delete a booking with the specified id
    */
   @Delete(":id")
   @UseGuards(RolesGuard)

src/location/location.controller.ts

@@ -44,7 +44,7 @@ export class LocationController {
   }
 
   /**
-   * Get a location with the specified `id`
+   * Get a location with the specified id
    */
   @Get(":id")
   async findOne(@Param("id", ParseIntPipe) id: number) {
@@ -73,7 +73,7 @@ export class LocationController {
   }
 
   /**
-   * Update a location with the specified `id`
+   * Update a location with the specified id
    */
   @Patch(":id")
   @UseGuards(JwtAuthGuard, RolesGuard)
@@ -88,7 +88,7 @@ export class LocationController {
   }
 
   /**
-   * Delete a location with the specified `id`
+   * Delete a location with the specified id
    */
   @Delete(":id")
   @UseGuards(JwtAuthGuard, RolesGuard)

src/schedule/schedule.controller.ts

@@ -0,0 +1,52 @@
+import {
+  BadRequestException,
+  Body,
+  ClassSerializerInterceptor,
+  Controller,
+  Get,
+  Param,
+  ParseIntPipe,
+  Patch,
+  UseGuards,
+  UseInterceptors
+} from "@nestjs/common";
+import { ApiTags } from "@nestjs/swagger";
+import { UserRole } from "@prisma/client";
+
+import { JwtAuthGuard } from "../auth/guards/jwt.guard";
+import { Roles } from "../decorators/roles.decorator";
+import { RolesGuard } from "../guards/roles.guard";
+import { getErrorMessage } from "../utils/get-error-message";
+import { ScheduleEntity, UpdateScheduleDto } from "./schedule.dto";
+import { ScheduleService } from "./schedule.service";
+
+@ApiTags("Schedule")
+@Controller("schedule")
+@UseInterceptors(ClassSerializerInterceptor)
+export class ScheduleController {
+  constructor(private readonly scheduleService: ScheduleService) {}
+
+  /**
+   * Get a schedule for a service with the specified id
+   */
+  @Get()
+  async findAll(@Param("serviceId", ParseIntPipe) serviceId: number) {
+    const schedule = await this.scheduleService.findSchedule({ serviceId });
+    return schedule.map((s) => new ScheduleEntity(s));
+  }
+
+  /**
+   * Update a schedule with the specified id
+   */
+  @Patch(":id")
+  @UseGuards(JwtAuthGuard, RolesGuard)
+  @Roles(UserRole.ADMIN, UserRole.MANAGER)
+  async update(@Param("id", ParseIntPipe) id: number, @Body() data: UpdateScheduleDto) {
+    try {
+      return new ScheduleEntity(await this.scheduleService.updateSchedule(id, data));
+    } catch (e) {
+      console.error(getErrorMessage(e));
+      throw new BadRequestException("Failed to update schedule.");
+    }
+  }
+}

src/schedule/schedule.module.ts

@@ -0,0 +1,13 @@
+import { Module } from "@nestjs/common";
+
+import { PrismaModule } from "../prisma/prisma.module";
+import { ScheduleController } from "./schedule.controller";
+import { ScheduleService } from "./schedule.service";
+
+@Module({
+  imports: [PrismaModule],
+  providers: [ScheduleService],
+  controllers: [ScheduleController],
+  exports: [ScheduleService]
+})
+export class ScheduleModule {}

src/schedule/schedule.service.ts

@@ -0,0 +1,40 @@
+import { Injectable } from "@nestjs/common";
+import { Prisma } from "@prisma/client";
+
+import { PrismaService } from "../prisma/prisma.service";
+import { CreateScheduleDto, UpdateScheduleDto } from "./schedule.dto";
+
+@Injectable()
+export class ScheduleService {
+  constructor(private prisma: PrismaService) {}
+
+  private buildScheduleData(serviceId: number) {
+    return new Array(7).fill(null).map(
+      (_, index): CreateScheduleDto => ({
+        day: index,
+        startTime: null,
+        endTime: null,
+        serviceId
+      })
+    );
+  }
+
+  async findSchedule(where: Prisma.ServiceScheduleWhereInput) {
+    return this.prisma.serviceSchedule.findMany({
+      where
+    });
+  }
+
+  async createScheduleForService(serviceId: number, tx?: Prisma.TransactionClient) {
+    return (tx ?? this.prisma).serviceSchedule.createManyAndReturn({
+      data: this.buildScheduleData(serviceId)
+    });
+  }
+
+  async updateSchedule(id: number, data: UpdateScheduleDto) {
+    return this.prisma.serviceSchedule.update({
+      where: { id },
+      data
+    });
+  }
+}

src/service/service.controller.ts

@@ -42,7 +42,7 @@ export class ServiceController {
   }
 
   /**
-   * Get a service with the specified `id`
+   * Get a service with the specified id
    */
   @Get(":id")
   async findOne(@Param("id", ParseIntPipe) id: number) {
@@ -71,7 +71,7 @@ export class ServiceController {
   }
 
   /**
-   * Update a service with the specified `id`
+   * Update a service with the specified id
    */
   @Patch(":id")
   @UseGuards(JwtAuthGuard, RolesGuard)
@@ -86,7 +86,7 @@ export class ServiceController {
   }
 
   /**
-   * Delete a service with the specified `id`
+   * Delete a service with the specified id
    */
   @Delete(":id")
   @UseGuards(JwtAuthGuard, RolesGuard)