Merge pull request #32 from makeplane/chore/instance-admin

chore: instance script

apiserver/plane/app/management/commands/cloud_instance.py

@@ -0,0 +1,72 @@
+# Python imports
+import json
+import secrets
+
+# Django imports
+from django.core.management.base import BaseCommand, CommandError
+from django.utils import timezone
+
+# Module imports
+from plane.license.models import Instance, InstanceAdmin
+from plane.db.models import User
+
+
+class Command(BaseCommand):
+    help = "Check if instance in registered else register"
+
+    def add_arguments(self, parser):
+        # Positional argument
+        parser.add_argument("admin_email", type=str, help="Admin Email")
+
+    def handle(self, *args, **options):
+        with open("package.json", "r") as file:
+            # Load JSON content from the file
+            data = json.load(file)
+
+        admin_email = options.get("admin_email", False)
+
+        if not admin_email:
+            raise CommandError("admin email is required")
+
+        user_count = User.objects.filter(is_bot=False).count()
+
+        user = User.objects.filter(email=admin_email).first()
+        if user is None:
+            raise CommandError("User not found")
+
+        try:
+            # Check if the instance is registered
+            instance = Instance.objects.first()
+
+            if instance is None:
+                instance = Instance.objects.create(
+                    instance_name="Plane Cloud US",
+                    instance_id=secrets.token_hex(12),
+                    license_key=None,
+                    api_key=secrets.token_hex(8),
+                    version=data.get("version"),
+                    last_checked_at=timezone.now(),
+                    user_count=user_count,
+                    is_verified=True,
+                    is_setup_done=True,
+                    is_signup_screen_visited=True,
+                )
+
+            # Get or create an instance admin
+            _ , created = InstanceAdmin.objects.get_or_create(
+                user=user, instance=instance, role=20, is_verified=True
+            )
+
+            if not created:
+                raise CommandError("given email is already an instance admin")
+
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"Successful"
+                )
+            )
+        except Exception as e:
+            print(e)
+            raise CommandError(
+                "Failure"
+            )