improvement: all demo bugs fixed. give me more baby.

backend/src/modules/cases/cases.controller.ts

@@ -14,7 +14,7 @@ import { ApiBearerAuth, ApiOperation, ApiTags } from '@nestjs/swagger';
 import { LabOrgGuard } from '../../common/guards/lab-org.guard';
 import { JwtAuthGuard } from '../auth/guards/jwt-auth.guard';
 import { CasesService } from './cases.service';
-import { ListLabCasesDto, UpdateLabCaseTaskDto } from './dto/cases.dto';
+import { ListLabCasesDto, UpdateLabCaseImportantDto } from './dto/cases.dto';
 
 @ApiTags('cases')
 @ApiBearerAuth('JWT-auth')
@@ -64,15 +64,14 @@ export class CasesController {
     file.stream.pipe(res);
   }
 
-  @Patch(':id/tasks/:taskId')
-  @ApiOperation({ summary: 'Toggle task important flag' })
-  updateTask(
+  @Patch(':id/important')
+  @ApiOperation({ summary: 'Toggle the important flag for a whole case' })
+  setCaseImportant(
     @Param('id') id: string,
-    @Param('taskId') taskId: string,
-    @Body() dto: UpdateLabCaseTaskDto,
+    @Body() dto: UpdateLabCaseImportantDto,
     @Req() req,
   ) {
     const organizationId = this.casesService.getOrganizationIdFromUser(req.user);
-    return this.casesService.updateTask(id, taskId, dto, organizationId, req.user.id, req.user.language);
+    return this.casesService.setCaseImportant(id, dto, organizationId, req.user.id, req.user.language);
   }
 }

backend/src/modules/cases/cases.service.ts

@@ -14,7 +14,7 @@ import {
 } from '../catalog/catalog-label.service';
 import { TreatmentCatalogService } from '../treatment-catalog/treatment-catalog.service';
 import { normalizeTeeth } from '../treatments/treatment.utils';
-import { ListLabCasesDto, UpdateLabCaseTaskDto } from './dto/cases.dto';
+import { ListLabCasesDto, UpdateLabCaseImportantDto } from './dto/cases.dto';
 import { normalizeTaskTeeth } from './lab-case-task.util';
 
 const labCaseListInclude = {
@@ -335,51 +335,39 @@ export class CasesService {
     };
   }
 
-  async updateTask(
+  async setCaseImportant(
     labCaseId: string,
-    taskId: string,
-    dto: UpdateLabCaseTaskDto,
+    dto: UpdateLabCaseImportantDto,
     labOrganizationId: string,
     actorUserId: string,
     localeInput?: string | null,
   ) {
     await this.assertCanEditCases(actorUserId, labOrganizationId);
 
-    const task = await this.prisma.labCaseTask.findFirst({
+    const existing = await this.prisma.labCase.findFirst({
       where: {
-        id: taskId,
-        labCaseId,
-        labCase: {
-          sentAt: { not: null },
-          sends: { some: { organizationId: labOrganizationId } },
-        },
+        id: labCaseId,
+        sentAt: { not: null },
+        sends: { some: { organizationId: labOrganizationId } },
       },
+      select: { id: true },
     });
 
-    if (!task) {
-      throw new NotFoundException('Task not found');
+    if (!existing) {
+      throw new NotFoundException('Case not found');
     }
 
-    const updated = await this.prisma.labCaseTask.update({
-      where: { id: taskId },
+    await this.prisma.labCase.update({
+      where: { id: labCaseId },
       data: { isImportant: dto.isImportant },
-      include: {
-        lastStatusChangedBy: { select: { id: true, name: true } },
-        statusEvents: {
-          orderBy: { changedAt: 'asc' },
-          include: { changedBy: { select: { id: true, name: true } } },
-        },
-      },
     });
 
-    const locale = normalizeCatalogLocale(localeInput);
-    const prosthesisLabels = await this.catalogLabels.resolveLabels(
-      CatalogEntityKind.PROSTHESIS_TYPE,
-      [updated.prosthesisTypeCode],
-      locale,
-    );
+    const labCase = await this.prisma.labCase.findFirstOrThrow({
+      where: { id: labCaseId },
+      include: labCaseListInclude,
+    });
 
-    return { success: true, data: this.mapTask(updated, prosthesisLabels) };
+    return { success: true, data: await this.mapLabCaseDetail(labCase, localeInput) };
   }
 
   private buildListWhere(
@@ -499,6 +487,7 @@ export class CasesService {
     return {
       id: lc.id,
       sentAt: lc.sentAt?.toISOString() ?? null,
+      isImportant: lc.isImportant,
       clinic: lc.treatment.organization,
       patient: lc.treatment.patient,
       appointmentStartAt: lc.treatment.appointment?.startAt.toISOString() ?? null,
@@ -585,7 +574,6 @@ export class CasesService {
       stepOrder: task.stepOrder,
       stepLabel: task.stepLabel,
       status: task.status,
-      isImportant: task.isImportant,
       createdAt: task.createdAt.toISOString(),
       lastStatusChangedAt: task.lastStatusChangedAt?.toISOString() ?? null,
       lastStatusChangedBy: task.lastStatusChangedBy

backend/src/modules/cases/dto/cases.dto.ts

@@ -1,7 +1,7 @@
 import { Transform } from 'class-transformer';
 import { IsBoolean, IsDateString, IsInt, IsOptional, IsString, IsUUID, Max, Min } from 'class-validator';
 
-export class UpdateLabCaseTaskDto {
+export class UpdateLabCaseImportantDto {
   @IsBoolean()
   isImportant: boolean;
 }

backend/src/modules/tasks/dto/tasks.dto.ts

@@ -25,7 +25,14 @@ export class UpdateLabTaskDto {
   status: LabTaskStatus;
 }
 
-export type TaskSortField = 'date' | 'status' | 'clinic' | 'patient' | 'important';
+export type TaskSortField =
+  | 'date'
+  | 'status'
+  | 'clinic'
+  | 'patient'
+  | 'important'
+  | 'prosthesis'
+  | 'taskType';
 
 export class ListLabTasksDto {
   @IsOptional()
@@ -59,7 +66,7 @@ export class ListLabTasksDto {
   sentTo?: string;
 
   @IsOptional()
-  @IsIn(['date', 'status', 'clinic', 'patient', 'important'])
+  @IsIn(['date', 'status', 'clinic', 'patient', 'important', 'prosthesis', 'taskType'])
   sortBy?: TaskSortField;
 
   @IsOptional()

backend/src/modules/tasks/tasks.service.ts

@@ -188,9 +188,9 @@ export class TasksService {
           ? { treatment: { organizationId: query.clinicOrganizationId } }
           : {}),
         ...(query.q?.trim() ? { treatment: this.buildSearchWhere(query.q.trim()) } : {}),
+        ...(query.important !== undefined ? { isImportant: query.important } : {}),
       },
       ...(status !== undefined ? { status } : {}),
-      ...(query.important !== undefined ? { isImportant: query.important } : {}),
     };
   }
 
@@ -229,15 +229,24 @@ export class TasksService {
           { id: 'asc' },
         ];
       case 'important':
-        return [{ isImportant: dir }, { createdAt: 'desc' }, { id: 'asc' }];
+        return [{ labCase: { isImportant: dir } }, { createdAt: 'desc' }, { id: 'asc' }];
+      case 'prosthesis':
+        return [{ prosthesisTypeCode: dir }, { createdAt: 'desc' }, { id: 'asc' }];
+      case 'taskType':
+        return [
+          { workflowStepCode: dir },
+          { stepOrder: 'asc' },
+          { createdAt: 'desc' },
+          { id: 'asc' },
+        ];
       case 'date':
       default:
+        // date / caseId / taskId / stepId — newest first by default.
         return [
           { labCase: { sentAt: dir } },
-          { labCaseId: 'asc' },
-          { treatmentDetailId: 'asc' },
-          { stepOrder: 'asc' },
-          { id: 'asc' },
+          { labCaseId: dir },
+          { id: dir },
+          { stepOrder: dir },
         ];
     }
   }
@@ -259,7 +268,7 @@ export class TasksService {
       stepOrder: task.stepOrder,
       stepLabel: task.stepLabel,
       status: task.status,
-      isImportant: task.isImportant,
+      isImportant: task.labCase.isImportant,
       lastStatusChangedAt: task.lastStatusChangedAt?.toISOString() ?? null,
       lastStatusChangedBy: task.lastStatusChangedBy
         ? { id: task.lastStatusChangedBy.id, name: task.lastStatusChangedBy.name }