from rest_framework import viewsets, status from rest_framework.response import Response from rest_framework.views import APIView from apps.accounts.permissions import Authenticated, permission_classes_by_action from django.http import JsonResponse, HttpResponseForbidden, HttpResponse from drf_yasg.utils import swagger_auto_schema from drf_yasg import openapi from django.shortcuts import get_object_or_404 from apps.channels.models import Channel from .models import HDHRDevice from .serializers import HDHRDeviceSerializer from django.contrib.auth.decorators import login_required from django.shortcuts import render from django.views import View from django.utils.decorators import method_decorator from django.contrib.auth.decorators import login_required from django.views.decorators.csrf import csrf_exempt @login_required def hdhr_dashboard_view(request): """Render the HDHR management page.""" hdhr_devices = HDHRDevice.objects.all() return render(request, "hdhr/hdhr.html", {"hdhr_devices": hdhr_devices}) # 🔹 1) HDHomeRun Device API class HDHRDeviceViewSet(viewsets.ModelViewSet): """Handles CRUD operations for HDHomeRun devices""" queryset = HDHRDevice.objects.all() serializer_class = HDHRDeviceSerializer def get_permissions(self): try: return [perm() for perm in permission_classes_by_action[self.action]] except KeyError: return [Authenticated()] # 🔹 2) Discover API class DiscoverAPIView(APIView): """Returns device discovery information""" @swagger_auto_schema( operation_description="Retrieve HDHomeRun device discovery information", responses={200: openapi.Response("HDHR Discovery JSON")}, ) def get(self, request): base_url = request.build_absolute_uri("/hdhr/").rstrip("/") device = HDHRDevice.objects.first() if not device: data = { "FriendlyName": "Dispatcharr HDHomeRun", "ModelNumber": "HDTC-2US", "FirmwareName": "hdhomerun3_atsc", "FirmwareVersion": "20200101", "DeviceID": "12345678", "DeviceAuth": "test_auth_token", "BaseURL": base_url, "LineupURL": f"{base_url}/lineup.json", "TunerCount": 10, } else: data = { "FriendlyName": device.friendly_name, "ModelNumber": "HDTC-2US", "FirmwareName": "hdhomerun3_atsc", "FirmwareVersion": "20200101", "DeviceID": device.device_id, "DeviceAuth": "test_auth_token", "BaseURL": base_url, "LineupURL": f"{base_url}/lineup.json", "TunerCount": 10, } return JsonResponse(data) # 🔹 3) Lineup API class LineupAPIView(APIView): """Returns available channel lineup""" @swagger_auto_schema( operation_description="Retrieve the available channel lineup", responses={200: openapi.Response("Channel Lineup JSON")}, ) def get(self, request): channels = Channel.objects.all().order_by("channel_number") lineup = [ { "GuideNumber": str(ch.channel_number), "GuideName": ch.name, "URL": request.build_absolute_uri(f"/proxy/ts/stream/{ch.uuid}"), } for ch in channels ] return JsonResponse(lineup, safe=False) # 🔹 4) Lineup Status API class LineupStatusAPIView(APIView): """Returns the current status of the HDHR lineup""" @swagger_auto_schema( operation_description="Retrieve the HDHomeRun lineup status", responses={200: openapi.Response("Lineup Status JSON")}, ) def get(self, request): data = { "ScanInProgress": 0, "ScanPossible": 0, "Source": "Cable", "SourceList": ["Cable"], } return JsonResponse(data) # 🔹 5) Device XML API class HDHRDeviceXMLAPIView(APIView): """Returns HDHomeRun device configuration in XML""" @swagger_auto_schema( operation_description="Retrieve the HDHomeRun device XML configuration", responses={200: openapi.Response("HDHR Device XML")}, ) def get(self, request): base_url = request.build_absolute_uri("/hdhr/").rstrip("/") xml_response = f""" 12345678 Dispatcharr HDHomeRun HDTC-2US hdhomerun3_atsc 20200101 test_auth_token {base_url} {base_url}/lineup.json """ return HttpResponse(xml_response, content_type="application/xml")