new migrations, setting proxy and redirect profiles, setting locked profiles

core/models.py

@@ -1,5 +1,6 @@
 # core/models.py
 from django.db import models
+from django.utils.text import slugify
 
 class UserAgent(models.Model):
     user_agent_name = models.CharField(
@@ -27,35 +28,111 @@ class UserAgent(models.Model):
     def __str__(self):
         return self.user_agent_name
 
+PROXY_PROFILE = 'Proxy'
+REDIRECT_PROFILE = 'Redirect'
+
 class StreamProfile(models.Model):
     profile_name = models.CharField(max_length=255, help_text="Name of the stream profile")
     command = models.CharField(
         max_length=255,
-        help_text="Command to execute (e.g., 'yt.sh', 'streamlink', or 'vlc')"
+        help_text="Command to execute (e.g., 'yt.sh', 'streamlink', or 'vlc')",
+        blank=True
     )
     parameters = models.TextField(
-        help_text="Command-line parameters. Use {userAgent} and {streamUrl} as placeholders."
+        help_text="Command-line parameters. Use {userAgent} and {streamUrl} as placeholders.",
+        blank=True
+    )
+    locked = models.BooleanField(
+        default=False,
+        help_text="Protected - can't be deleted or modified"
     )
     is_active = models.BooleanField(default=True, help_text="Whether this profile is active")
-    user_agent = models.CharField(
-        max_length=512,
-        blank=True,
+    user_agent = models.ForeignKey(
+        "UserAgent",
+        on_delete=models.SET_NULL,
         null=True,
+        blank=True,
         help_text="Optional user agent to use. If not set, you can fall back to a default."
     )
 
     def __str__(self):
         return self.profile_name
 
-    def build_command(self, stream_url):
-        cmd = []
-        if self.command == "ffmpeg":
-            cmd = ["ffmpeg", "-i", stream_url] + self.parameters.split() + ["pipe:1"]
-        elif self.command == "streamlink":
-            cmd = ["streamlink", stream_url] + self.parameters.split()
+    def delete(self):
+        if self.locked():
+            raise ValueError("This profile is locked and cannot be deleted.")
+
+        self.delete()
+
+    def save(self, *args, **kwargs):
+        if self.pk:  # Only check existing records
+            orig = StreamProfile.objects.get(pk=self.pk)
+            if orig.is_protected:
+                allowed_fields = {"user_agent_id"}  # Only allow this field to change
+                for field in self._meta.fields:
+                    field_name = field.name
+
+                    # Convert user_agent to user_agent_id for comparison
+                    orig_value = getattr(orig, field_name)
+                    new_value = getattr(self, field_name)
+
+                    # Ensure that ForeignKey fields compare their ID values
+                    if isinstance(orig_value, models.Model):
+                        orig_value = orig_value.pk
+                    if isinstance(new_value, models.Model):
+                        new_value = new_value.pk
+
+                    if field_name not in allowed_fields and orig_value != new_value:
+                        raise ValidationError(f"Cannot modify {field_name} on a protected profile.")
+
+        super().save(*args, **kwargs)
+
+    @classmethod
+    def update(cls, pk, **kwargs):
+        instance = cls.objects.get(pk=pk)
+
+        if instance.is_protected:
+            allowed_fields = {"user_agent_id"}  # Only allow updating this field
+
+            for field_name, new_value in kwargs.items():
+                if field_name not in allowed_fields:
+                    raise ValidationError(f"Cannot modify {field_name} on a protected profile.")
+
+                # Ensure user_agent ForeignKey updates correctly
+                if field_name == "user_agent" and isinstance(new_value, cls._meta.get_field("user_agent").related_model):
+                    new_value = new_value.pk  # Convert object to ID if needed
+
+                setattr(instance, field_name, new_value)
+
+        instance.save()
+        return instance
+
+    def is_proxy(self):
+        if self.locked and self.name == PROXY_PROFILE:
+            return True
+        return False
+
+    def is_redirect(self):
+        if self.locked and self.name == REDIRECT_PROFILE:
+            return True
+        return False
+
+    def build_command(self, stream_url, user_agent):
+        if self.is_proxy():
+            return []
+
+        replacements = {
+            "{streamUrl}": stream_url,
+            "{userAgent}": user_agent,
+        }
+
+        cmd = [self.command] + [replacements.get(part, part) for part in self.parameters.split()]
 
         return cmd
 
+DEFAULT_USER_AGENT_KEY= slugify("Default User-Agent")
+DEFAULT_STREAM_PROFILE_KEY = slugify("Default Stream Profile")
+
 class CoreSettings(models.Model):
     key = models.CharField(
         max_length=255,
@@ -70,3 +147,14 @@ class CoreSettings(models.Model):
 
     def __str__(self):
         return "Core Settings"
+
+    @classmethod
+    def get_default_user_agent(cls):
+        """Retrieve a system profile by name (or return None if not found)."""
+        default_ua_id = cls.objects.get(key=DEFAULT_USER_AGENT_KEY).value
+        return UserAgent.objects.get(id=default_ua_id)
+
+    @classmethod
+    def get_default_stream_profile(cls):
+        default_sp_id = cls.objects.get(key=DEFAULT_STREAM_PROFILE_KEY).value
+        return StreamProfile.objects.get(id=default_sp_id)