927da438c7
* fix: enable github api to fetch latest version information * chore: typo fixes * chore: add timeout to request
88 lines
3 KiB
Python
88 lines
3 KiB
Python
# Python imports
|
|
import json
|
|
import secrets
|
|
import os
|
|
import requests
|
|
|
|
# Django imports
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils import timezone
|
|
|
|
|
|
# Module imports
|
|
from plane.license.models import Instance, InstanceEdition
|
|
from plane.license.bgtasks.tracer import instance_traces
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Check if instance in registered else register"
|
|
|
|
def add_arguments(self, parser):
|
|
# Positional argument
|
|
parser.add_argument("machine_signature", type=str, help="Machine signature")
|
|
|
|
def check_for_current_version(self):
|
|
if os.environ.get("APP_VERSION", False):
|
|
return os.environ.get("APP_VERSION")
|
|
|
|
try:
|
|
with open("package.json", "r") as file:
|
|
data = json.load(file)
|
|
return data.get("version", "v0.1.0")
|
|
except Exception:
|
|
self.stdout.write("Error checking for current version")
|
|
return "v0.1.0"
|
|
|
|
def check_for_latest_version(self, fallback_version):
|
|
try:
|
|
response = requests.get(
|
|
"https://api.github.com/repos/makeplane/plane/releases/latest",
|
|
timeout=10,
|
|
)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
return data.get("tag_name", fallback_version)
|
|
except Exception:
|
|
self.stdout.write("Error checking for latest version")
|
|
return fallback_version
|
|
|
|
def handle(self, *args, **options):
|
|
# Check if the instance is registered
|
|
instance = Instance.objects.first()
|
|
|
|
current_version = self.check_for_current_version()
|
|
latest_version = self.check_for_latest_version(current_version)
|
|
|
|
# If instance is None then register this instance
|
|
if instance is None:
|
|
machine_signature = options.get("machine_signature", "machine-signature")
|
|
|
|
if not machine_signature:
|
|
raise CommandError("Machine signature is required")
|
|
|
|
instance = Instance.objects.create(
|
|
instance_name="Plane Community Edition",
|
|
instance_id=secrets.token_hex(12),
|
|
current_version=current_version,
|
|
latest_version=latest_version,
|
|
last_checked_at=timezone.now(),
|
|
is_test=os.environ.get("IS_TEST", "0") == "1",
|
|
edition=InstanceEdition.PLANE_COMMUNITY.value,
|
|
)
|
|
|
|
self.stdout.write(self.style.SUCCESS("Instance registered"))
|
|
else:
|
|
self.stdout.write(self.style.SUCCESS("Instance already registered"))
|
|
|
|
# Update the instance details
|
|
instance.last_checked_at = timezone.now()
|
|
instance.current_version = current_version
|
|
instance.latest_version = latest_version
|
|
instance.is_test = os.environ.get("IS_TEST", "0") == "1"
|
|
instance.edition = InstanceEdition.PLANE_COMMUNITY.value
|
|
instance.save()
|
|
|
|
# Call the instance traces task
|
|
instance_traces.delay()
|
|
|
|
return
|