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

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