From 1a6ec7034a96ae08ac319d35cc7047f2b24ca1d2 Mon Sep 17 00:00:00 2001
From: Nikhil <118773738+pablohashescobar@users.noreply.github.com>
Date: Fri, 22 Nov 2024 16:05:58 +0530
Subject: [PATCH] chore: management command to add user to a project (#6084)

---
 .../commands/create_project_member.py         | 118 ++++++++++++++++++
 1 file changed, 118 insertions(+)
 create mode 100644 apiserver/plane/db/management/commands/create_project_member.py

diff --git a/apiserver/plane/db/management/commands/create_project_member.py b/apiserver/plane/db/management/commands/create_project_member.py
new file mode 100644
index 0000000000..ec1d616a19
--- /dev/null
+++ b/apiserver/plane/db/management/commands/create_project_member.py
@@ -0,0 +1,118 @@
+# Django imports
+from typing import Any
+from django.core.management import BaseCommand, CommandError
+
+# Module imports
+from plane.db.models import (
+    User,
+    WorkspaceMember,
+    ProjectMember,
+    Project,
+    IssueUserProperty,
+)
+
+
+class Command(BaseCommand):
+
+    help = "Add a member to a project. If present in the workspace"
+
+    def add_arguments(self, parser):
+        # Positional argument
+        parser.add_argument(
+            "--project_id",
+            type=str,
+            nargs="?",
+            help="Project ID",
+        )
+        parser.add_argument(
+            "--user_email",
+            type=str,
+            nargs="?",
+            help="User Email",
+        )
+        parser.add_argument(
+            "--role",
+            type=int,
+            nargs="?",
+            help="Role of the user in the project",
+        )
+
+    def handle(self, *args: Any, **options: Any):
+        try:
+            if not options["project_id"]:
+                raise CommandError("Project ID is required")
+            if not options["user_email"]:
+                raise CommandError("User Email is required")
+
+            project_id = options["project_id"]
+            user_email = options["user_email"]
+            role = options.get("role", 20)
+
+            print(f"Role: {role}")
+
+            user = User.objects.filter(email=user_email).first()
+            if not user:
+                raise CommandError("User not found")
+
+            # Check if the project exists
+            project = Project.objects.filter(pk=project_id).first()
+            if not project:
+                raise CommandError("Project not found")
+
+            # Check if the user exists in the workspace
+            if not WorkspaceMember.objects.filter(
+                workspace=project.workspace, member=user, is_active=True
+            ).exists():
+                raise CommandError("User not member in workspace")
+
+            # Check if the user is already a member of the project
+            if ProjectMember.objects.filter(
+                project=project, member=user, is_active=True
+            ).exists():
+                raise CommandError("User already a member of the project")
+
+            # Get the smallest sort order
+            smallest_sort_order = (
+                ProjectMember.objects.filter(
+                    workspace_id=project.workspace_id,
+                )
+                .order_by("sort_order")
+                .first()
+            )
+
+            if smallest_sort_order:
+                sort_order = smallest_sort_order.sort_order - 1000
+            else:
+                sort_order = 65535
+
+            if ProjectMember.objects.filter(
+                project=project,
+                member=user,
+            ).exists():
+                # Update the project member
+                ProjectMember.objects.filter(
+                    project=project,
+                    member=user,
+                ).update(is_active=True, sort_order=sort_order, role=role)
+            else:
+                # Create the project member
+                ProjectMember.objects.create(
+                    project=project,
+                    member=user,
+                    role=role,
+                    sort_order=sort_order,
+                )
+
+            # Issue Property
+            IssueUserProperty.objects.get_or_create(user=user, project=project)
+
+            # Success message
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"User {user_email} added to project {project_id}"
+                )
+            )
+            return
+        except CommandError as e:
+            self.stdout.write(self.style.ERROR(e))
+            return