feat: implement workspace management and update user registration flow

- Added workspace model and relationships in Prisma schema.
- Updated user and team member models to include workspace references.
- Enhanced registration process to include workspace creation.
- Modified authentication responses to include workspace ID.
- Implemented workspace selection in the frontend with corresponding API updates.
- Refactored project management to support workspace-specific operations.
- Added workspace management features in the sidebar and project management components.
- Updated various components to handle workspace context and interactions.

Author: asepindrak

backend/package-lock.json

@@ -13,37 +13,6 @@ datasource db {
   provider = "postgresql"
   url      = env("DATABASE_URL")
 }
-model User {
-  id            String      @id @default(cuid())
-  email         String?     @unique
-  phone         String?
-  name          String?
-  password      String?     // kalau pake password
-  session_token String?     @unique
-  createdAt     DateTime    @default(now())
-  updatedAt     DateTime    @updatedAt
-
-  // FK ke TeamMember (one-to-one)
-  teamMemberId  String?     @unique
-  teamMember    TeamMember? @relation(fields: [teamMemberId], references: [id])
-}
-
-model TeamMember {
-  id          String   @id @default(uuid())  // pilih uuid supaya beda style dari user.id
-  clientId    String?  @unique
-  name        String
-  role        String?
-  email       String?  @unique
-  photo       String?
-  phone       String?
-  isTrash     Boolean  @default(false)
-  createdAt   DateTime @default(now())
-  updatedAt   DateTime?
-  Task        Task[]
-
-  // back relation ke User
-  user        User?    @relation
-}
 
 model aiToolCache {
   id         String   @id @default(cuid())
@@ -78,15 +47,63 @@ model chatMessage {
   @@map("chat_message")
 }
 
-model Project {
-  id          String    @id @default(uuid())
-  clientId    String?     @unique
+model User {
+  id            String        @id @default(cuid())
+  email         String?       @unique
+  phone         String?
+  name          String?
+  role          Role?         @default(USER)
+  password      String?       // kalau pake password
+  session_token String?       @unique
+  createdAt     DateTime      @default(now())
+  updatedAt     DateTime      @updatedAt
+  members       TeamMember[]  @relation("UserMembers")
+
+}
+
+model TeamMember {
+  id          String   @id @default(uuid())  // pilih uuid supaya beda style dari user.id
+  clientId    String?  @unique
+  userId      String
+  workspaceId String
+  name        String
+  role        String?
+  email       String?
+  photo       String?
+  phone       String?
+  isTrash     Boolean  @default(false)
+  createdAt   DateTime @default(now())
+  updatedAt   DateTime?
+  Task        Task[]
+
+  // FK ke Workspace
+  workspace     Workspace?  @relation("WorkspaceMembers", fields: [workspaceId], references: [id])
+  user          User?  @relation("UserMembers", fields: [userId], references: [id])
+}
+
+model Workspace {
+  id          String        @id @default(uuid())
+  clientId    String?       @unique
   name        String
   description String?
-  isTrash     Boolean   @default(false)
-  createdAt   DateTime  @default(now())
+  isTrash     Boolean       @default(false)
+  createdAt   DateTime      @default(now())
   updatedAt   DateTime?
-  tasks       Task[]    @relation("ProjectTasks")
+  projects    Project[]     @relation("WorkspaceProjects")
+  members     TeamMember[]        @relation("WorkspaceMembers")
+}
+
+model Project {
+  id              String      @id @default(uuid())
+  workspaceId     String?
+  clientId        String?     @unique
+  name            String
+  description     String?
+  isTrash         Boolean     @default(false)
+  createdAt       DateTime    @default(now())
+  updatedAt       DateTime?
+  workspace       Workspace?  @relation("WorkspaceProjects", fields: [workspaceId], references: [id])
+  tasks           Task[]      @relation("ProjectTasks")
 }
 
 model Task {
@@ -109,13 +126,18 @@ model Task {
 }
 
 model Comment {
-  id          String   @id @default(uuid())
+  id          String    @id @default(uuid())
   taskId      String
-  task        Task     @relation("TaskComments", fields: [taskId], references: [id])
+  task        Task      @relation("TaskComments", fields: [taskId], references: [id])
   author      String
   body        String
   attachments Json? // optional JSON array of attachments metadata
   isTrash     Boolean   @default(false)
-  createdAt   DateTime @default(now())
+  createdAt   DateTime  @default(now())
   updatedAt   DateTime?
+}
+
+enum Role {
+  USER
+  ADMIN
 }

backend/prisma/schema.prisma

@@ -23,6 +23,7 @@ export class AuthController {
     return {
       token: result.token,
       userId: result.user.id,
+      workspaceId: result.workspace.id,
       teamMemberId: result.teamMember.id,
       clientTempId: result.clientTempId ?? null,
     };
@@ -36,7 +37,6 @@ export class AuthController {
     return {
       token: result.token,
       userId: result?.user?.id ?? "",
-      teamMemberId: result.teamMember?.id ?? null,
     };
   }
 }

backend/src/auth/auth.controller.ts

@@ -48,40 +48,50 @@ export class AuthService {
   }
 
   async register(dto: RegisterDto) {
-    const { clientTempId, email, name, role, photo, password } = dto;
+    const { clientTempId, workspace, email, name, role, photo, password } = dto;
 
     // Basic uniqueness check
     const existing = await this.prisma.user.findUnique({ where: { email } });
     if (existing) throw new ConflictException("Email already registered");
 
     // Use transaction to create TeamMember then User referencing member.id
     try {
+      console.log(workspace);
       const result = await this.prisma.$transaction(async (tx) => {
-        // 1) create TeamMember first
-        const teamMember = await tx.teamMember.create({
+        // 1) create Workspace first
+        const createWorkspace = await tx.workspace.create({
           data: {
-            name,
-            email,
-            role: role ?? null,
-            photo: photo ?? null,
-            // optionally store clientTempId if you want mapping
-            // clientTempId: clientTempId ?? null  // uncomment if you added field
+            name: workspace,
             createdAt: new Date(),
           },
         });
 
-        // 2) create User referencing teamMember.id
+        // 2) create User referencing
         const hashed = password ? hashPassword(password) : undefined;
         const user = await tx.user.create({
           data: {
             email,
             name,
             password: hashed,
-            teamMemberId: teamMember.id, // <-- link via member id
           },
         });
 
-        return { user, teamMember };
+        // 3) create TeamMember
+        const teamMember = await tx.teamMember.create({
+          data: {
+            userId: user.id,
+            workspaceId: createWorkspace.id,
+            name,
+            email,
+            role: role ?? null,
+            photo: photo ?? null,
+            // optionally store clientTempId if you want mapping
+            // clientTempId: clientTempId ?? null  // uncomment if you added field
+            createdAt: new Date(),
+          },
+        });
+
+        return { user, teamMember, workspace: createWorkspace };
       });
 
       // Create JWT (you can include teamMemberId in payload too)
@@ -100,6 +110,7 @@ export class AuthService {
         token: jwt,
         user: result.user,
         teamMember: result.teamMember,
+        workspace: result.workspace,
         // optionally map clientTempId => memberId so FE can replace optimistic id
         clientTempId,
       };
@@ -116,9 +127,9 @@ export class AuthService {
     if (!email) throw new UnauthorizedException("Invalid credentials");
 
     // find user and include relation if exists
-    let user = await this.prisma.user.findUnique({
+    const user = await this.prisma.user.findUnique({
       where: { email },
-      include: { teamMember: true },
+      include: { members: true },
     });
 
     if (!user) throw new UnauthorizedException("Invalid credentials");
@@ -127,42 +138,16 @@ export class AuthService {
     const ok = comparePassword(password || "", user.password);
     if (!ok) throw new UnauthorizedException("Invalid credentials");
 
-    // If relation missing, try to find TeamMember by email and attach it (best-effort)
-    let teamMember = (user as any).teamMember ?? null;
-    if (!teamMember) {
-      if (user.email) {
-        const found = await this.prisma.teamMember.findUnique({
-          where: { email: user.email },
-        });
-        if (found) {
-          // attach FK so next time it's present
-          try {
-            await this.prisma.user.update({
-              where: { id: user.id },
-              data: { teamMemberId: found.id },
-            });
-            teamMember = found;
-            // also reflect in local user object
-            user = { ...user, teamMemberId: found.id } as any;
-          } catch (err) {
-            // ignore update failure (concurrency/unique) — still continue
-            console.warn("Failed to attach teamMemberId:", err);
-          }
-        }
-      }
-    }
-
     const jwt = this.jwtService.sign({
       userId: user?.id,
-      teamMemberId: teamMember?.id ?? null,
     });
 
     await this.prisma.user.update({
       where: { id: user?.id },
       data: { session_token: jwt },
     });
 
-    return { token: jwt, user, teamMember };
+    return { token: jwt, user };
   }
 
   async validateUser(token: string) {

backend/src/auth/auth.service.ts

@@ -6,6 +6,9 @@ export class RegisterDto {
   @IsString()
   clientTempId?: string; // optional, FE optimistic id
 
+  @IsString()
+  workspace: string;
+
   @IsEmail()
   email: string;
 

backend/src/auth/dto/register.dto.ts

@@ -1,20 +1,22 @@
-import { IsOptional, IsString } from 'class-validator';
+import { IsOptional, IsString } from "class-validator";
 
 export class CreateProjectDto {
-    @IsString()
-    name: string;
+  @IsString()
+  name: string;
+  @IsString()
+  workspaceId: string;
 
-    @IsOptional()
-    @IsString()
-    description?: string;
+  @IsOptional()
+  @IsString()
+  description?: string;
 }
 
 export class UpdateProjectDto {
-    @IsOptional()
-    @IsString()
-    name?: string;
+  @IsOptional()
+  @IsString()
+  name?: string;
 
-    @IsOptional()
-    @IsString()
-    description?: string;
+  @IsOptional()
+  @IsString()
+  description?: string;
 }

backend/src/project-management/dto/project.dto.ts

@@ -23,6 +23,9 @@ export class CreateTeamMemberDto {
   @IsOptional()
   @IsString()
   photo?: string;
+
+  @IsString()
+  workspaceId?: string;
 }
 
 export class UpdateTeamMemberDto {
@@ -34,10 +37,6 @@ export class UpdateTeamMemberDto {
   @IsString()
   role?: string;
 
-  @IsOptional()
-  @IsString()
-  email?: string;
-
   @IsOptional()
   @IsString()
   password?: string;

backend/src/project-management/dto/team.dto.ts

@@ -0,0 +1,20 @@
+import { IsOptional, IsString } from "class-validator";
+
+export class CreateWorkspaceDto {
+  @IsString()
+  name: string;
+
+  @IsOptional()
+  @IsString()
+  description?: string;
+}
+
+export class UpdateWorkspaceDto {
+  @IsOptional()
+  @IsString()
+  name?: string;
+
+  @IsOptional()
+  @IsString()
+  description?: string;
+}

backend/src/project-management/dto/workspace.dto.ts

@@ -1,18 +1,20 @@
-import { Test, TestingModule } from '@nestjs/testing';
-import { ProjectManagementController } from './project-management.controller';
+import { Test, TestingModule } from "@nestjs/testing";
+import { ProjectManagementController } from "./project-management.controller";
 
-describe('ProjectManagementController', () => {
+describe("ProjectManagementController", () => {
   let controller: ProjectManagementController;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       controllers: [ProjectManagementController],
     }).compile();
 
-    controller = module.get<ProjectManagementController>(ProjectManagementController);
+    controller = module.get<ProjectManagementController>(
+      ProjectManagementController
+    );
   });
 
-  it('should be defined', () => {
+  it("should be defined", () => {
     expect(controller).toBeDefined();
   });
 });