From 91f8a51a1e6d3cd53398479c56718c32aa82b3c7 Mon Sep 17 00:00:00 2001
From: jo <ljonas@riseup.net>
Date: Tue, 5 Apr 2022 16:10:56 +0200
Subject: [PATCH] feat(api): create set_icecast_passwords command

- allow to fetch icecast password from xml config
- set default stream password
---
 .../legacy/management/__init__.py             |  0
 .../legacy/management/commands/__init__.py    |  0
 .../commands/set_icecast_passwords.py         | 77 +++++++++++++++++++
 3 files changed, 77 insertions(+)
 create mode 100644 api/libretime_api/legacy/management/__init__.py
 create mode 100644 api/libretime_api/legacy/management/commands/__init__.py
 create mode 100644 api/libretime_api/legacy/management/commands/set_icecast_passwords.py

diff --git a/api/libretime_api/legacy/management/__init__.py b/api/libretime_api/legacy/management/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/api/libretime_api/legacy/management/commands/__init__.py b/api/libretime_api/legacy/management/commands/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/api/libretime_api/legacy/management/commands/set_icecast_passwords.py b/api/libretime_api/legacy/management/commands/set_icecast_passwords.py
new file mode 100644
index 000000000..35bda87c8
--- /dev/null
+++ b/api/libretime_api/legacy/management/commands/set_icecast_passwords.py
@@ -0,0 +1,77 @@
+import os
+from pathlib import Path
+from typing import Optional, Tuple
+from xml.etree import ElementTree
+
+from django.core.management.base import BaseCommand, CommandParser
+
+from libretime_api.core.models import StreamSetting
+from libretime_api.core.models.preference import Preference
+
+
+def get_passwords_from_config(
+    config_filepath: Path,
+) -> Tuple[Optional[str], Optional[str]]:
+    root = ElementTree.fromstring(config_filepath.read_text(encoding="utf-8"))  # nosec
+
+    def _get_text(path) -> Optional[str]:
+        element = root.find(path)
+        return element.text if element is not None else None
+
+    return (
+        _get_text("./authentication/admin-password"),
+        _get_text("./authentication/source-password"),
+    )
+
+
+class Command(BaseCommand):
+    help = "Configure icecast passwords in the database."
+
+    def add_arguments(self, parser: CommandParser):
+        parser.add_argument(
+            "--from-icecast-config",
+            help="Get passwords from the Icecast configuration file",
+            nargs="?",
+            const="/etc/icecast2/icecast.xml",
+            type=str,
+        )
+        parser.add_argument(
+            "--admin-password",
+            help="Icecast admin password",
+            type=str,
+        )
+        parser.add_argument(
+            "--source-password",
+            help="Icecast source password",
+            type=str,
+        )
+
+    def handle(self, *args, **options):
+        icecast_config = options.get("from_icecast_config")
+        admin_password = options.get("admin_password")
+        source_password = options.get("source_password")
+
+        if icecast_config is not None:
+            config_passwords = get_passwords_from_config(Path(icecast_config))
+            if admin_password is None and config_passwords[0] is not None:
+                admin_password = config_passwords[0]
+            if source_password is None and config_passwords[1] is not None:
+                source_password = config_passwords[1]
+
+        if source_password is not None:
+            Preference.objects.update_or_create(
+                keystr="default_icecast_password",
+                valstr=source_password,
+            )
+
+        for key in ["s1", "s2", "s3", "s4"]:
+            if admin_password is not None:
+                StreamSetting.objects.update_or_create(
+                    keyname=f"{key}_admin_pass",
+                    defaults={"value": admin_password},
+                )
+            if source_password is not None:
+                StreamSetting.objects.update_or_create(
+                    keyname=f"{key}_pass",
+                    defaults={"value": source_password},
+                )