Configure Test RTMP URL to Event

.gitignore

@@ -1,6 +1,6 @@
+.env*
+.venv/
 .vscode/
 __pycache__/
-db.sqlite3
-.env*
+muxy.db
 static/
-.venv/

env.sample

@@ -8,6 +8,10 @@ TIME_ZONE=UTC
 
 DB_PATH=./muxy.db
 
-NGINX_RTMP_UPDATE_TIMEOUT=30
+NGINX_RTMP_UPDATE_TIMEOUT=10
 
 EMAIL_FROM=
+
+DEFAULT_PUBLIC_RTMP_URL=rtmp://<insert_your_domain>/live
+DEFAULT_RTMP_URL=rtmp://localhost/main
+DEFAULT_TEST_RTMP_URL=rtmp://localhost/test

events/migrations/0021_event_test_rtmp_url.py

@@ -0,0 +1,19 @@
+# Generated by Django 3.1.14 on 2024-02-05 15:23
+
+from django.db import migrations
+import events.models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('events', '0020_stream_tags'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='event',
+            name='test_rtmp_url',
+            field=events.models.RTMPURLField(blank=True, null=True),
+        ),
+    ]

events/migrations/0022_auto_20240205_2058.py

@@ -0,0 +1,29 @@
+# Generated by Django 3.1.14 on 2024-02-05 20:58
+
+from django.db import migrations
+import events.models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('events', '0021_event_test_rtmp_url'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='event',
+            name='public_rtmp_url',
+            field=events.models.RTMPURLField(blank=True, default=events.models.get_default_public_rtmp_url, null=True),
+        ),
+        migrations.AlterField(
+            model_name='event',
+            name='rtmp_url',
+            field=events.models.RTMPURLField(blank=True, default=events.models.get_default_rtmp_url, null=True),
+        ),
+        migrations.AlterField(
+            model_name='event',
+            name='test_rtmp_url',
+            field=events.models.RTMPURLField(blank=True, default=events.models.get_default_test_rtmp_url, null=True),
+        ),
+    ]

events/models.py

@@ -8,7 +8,6 @@
 
 from autoslug import AutoSlugField
 from django.conf import settings
-from django.contrib.auth.models import User
 from django.core import validators
 from django.core.exceptions import ValidationError
 from django.db import models
@@ -57,6 +56,18 @@ class Meta(AbstractAPIKey.Meta):
         verbose_name_plural = "API keys"
 
 
+def get_default_public_rtmp_url():
+    return settings.DEFAULT_PUBLIC_RTMP_URL
+
+
+def get_default_rtmp_url():
+    return settings.DEFAULT_RTMP_URL
+
+
+def get_default_test_rtmp_url():
+    return settings.DEFAULT_TEST_RTMP_URL
+
+
 class Event(models.Model):
     name = models.CharField(max_length=200)
     slug = AutoSlugField(null=True, default=None, populate_from="name", unique="name")
@@ -66,8 +77,13 @@ class Event(models.Model):
     ends_at = models.DateTimeField()
     active = models.BooleanField(default=True)
     preparation_time = models.PositiveIntegerField(default=5)
-    rtmp_url = RTMPURLField(blank=True, null=True)
-    public_rtmp_url = RTMPURLField(blank=True, null=True)
+    public_rtmp_url = RTMPURLField(
+        blank=True, null=True, default=get_default_public_rtmp_url
+    )
+    rtmp_url = RTMPURLField(blank=True, null=True, default=get_default_rtmp_url)
+    test_rtmp_url = RTMPURLField(
+        blank=True, null=True, default=get_default_test_rtmp_url
+    )
     contact_email = models.EmailField(blank=True, null=True)
 
     def __str__(self):
@@ -93,6 +109,11 @@ def resolved_rtmp_url(self):
         if self.rtmp_url:
             return resolve_url(self.rtmp_url)
 
+    @property
+    def resolved_test_rtmp_url(self):
+        if self.test_rtmp_url:
+            return resolve_url(self.test_rtmp_url)
+
     @property
     def duration(self):
         if self.starts_at and self.ends_at:
@@ -230,12 +251,14 @@ def __str__(self):
 
 
 class StreamArchiveURL(models.Model):
-    stream = models.ForeignKey(Stream, on_delete=models.CASCADE, related_name="archive_urls")
+    stream = models.ForeignKey(
+        Stream, on_delete=models.CASCADE, related_name="archive_urls"
+    )
     url = models.URLField()
     name = models.CharField(max_length=255, blank=True)
 
     def __str__(self):
         return self.url
 
     class Meta:
-        unique_together = ("stream", "url")
+        unique_together = ("stream", "url")

events/tests.py

@@ -207,8 +207,8 @@ def test_update_stream_set_archive_urls(self):
 
         data = {
             "archive_urls": [
-                { "url": "https://archive.org/foo", "name": "archive.org" },
-                { "url": "https://youtube.com/foo", "name": "youtube.com" },
+                {"url": "https://archive.org/foo", "name": "archive.org"},
+                {"url": "https://youtube.com/foo", "name": "youtube.com"},
             ]
         }
         response = self.client.patch(

events/views.py

@@ -82,24 +82,37 @@ def on_publish(request):
 
     now = timezone.now()
 
-    # Check if stream is valid
+    # If stream is not valid (i.e. not preparing or inactive)
     if not stream.is_valid_at(now):
-        print("[PUBLISH] Stream is not valid at %s" % (now))
-        return HttpResponseForbidden("Stream is not valid now")
+        # If event has a test RTMP URL, redirect to it
+        if stream.event.test_rtmp_url:
+            print(
+                "[PUBLISH] Stream is not valid now (%s). Redirect to Test RTMP URL."
+                % (now)
+            )
+            return RtmpRedirect(stream.resolved_test_rtmp_url)
+        else:
+            # Otherwise, deny the stream
+            print("[PUBLISH] Stream is not valid now (%s)" % (now))
+            return HttpResponseForbidden("Stream is not valid now")
 
     # If event has a custom RTMP URL, redirect to it
     if stream.event.rtmp_url:
         if stream.is_preparing_at(now):
-            print(
-                "[PUBLISH] Stream is preparing and not active yet. Allow but do not redirect."
-            )
-            return HttpResponse("OK")
+            if stream.event.test_rtmp_url:
+                print(
+                    "[PUBLISH] Stream is preparing and not active yet. Redirect to Test RTMP URL."
+                )
+                return RtmpRedirect(stream.resolved_test_rtmp_url)
+            else:
+                print("[PUBLISH] Stream is preparing and not active yet. Allow.")
+                return HttpResponse("OK")
         else:
             # Set the stream live
             stream.live_at = now
             stream.save()
 
-            print("[PUBLISH] Stream is active. Allow and redirect.")
+            print("[PUBLISH] Stream is active. Allow and redirect to custom RTMP URL.")
             return RtmpRedirect(stream.resolved_rtmp_url)
     else:
         print("[PUBLISH] Stream is active. Allow.")

muxy/settings.py

@@ -136,7 +136,7 @@
     "PAGE_SIZE": 1000,
 }
 
-NGINX_RTMP_UPDATE_TIMEOUT = int(os.getenv("NGINX_RTMP_UPDATE_TIMEOUT", "30"))
+NGINX_RTMP_UPDATE_TIMEOUT = int(os.getenv("NGINX_RTMP_UPDATE_TIMEOUT", "10"))
 
 EMAIL_BACKEND = os.getenv(
     "EMAIL_BACKEND", "django.core.mail.backends.console.EmailBackend"
@@ -200,3 +200,7 @@
 CORS_ALLOW_HEADERS = list(default_headers) + [
     STREAM_KEY_HEADER,
 ]
+
+DEFAULT_PUBLIC_RTMP_URL = os.getenv("DEFAULT_PUBLIC_RTMP_URL")
+DEFAULT_RTMP_URL = os.getenv("DEFAULT_RTMP_URL")
+DEFAULT_TEST_RTMP_URL = os.getenv("DEFAULT_TEST_RTMP_URL")