Upload dvr.py with huggingface_hub

dvr.py

@@ -60,10 +60,12 @@ def _start_buffer(channel_id, stream_url):
         "-reconnect", "1", "-reconnect_streamed", "1", "-reconnect_delay_max", "5",
         "-i", stream_url,
         "-c", "copy",                         # copy, không transcode -> nhẹ CPU
+        "-copyts", "-start_at_zero",
         "-f", "hls",
         "-hls_time", str(SEG_SEC),
         "-hls_list_size", str(LIST_SIZE),     # giữ ~5 phút
-        "-hls_flags", "delete_segments+omit_endlist+independent_segments",
+        "-hls_flags", "delete_segments+omit_endlist+independent_segments+program_date_time",
+        "-hls_segment_type", "mpegts",
         "-hls_segment_filename", seg,
         playlist,
     ]
@@ -102,16 +104,23 @@ def buffer_playlist(channel_id: str):
     if not os.path.exists(pl):
         return JSONResponse({"error": "buffer not ready"}, status_code=404)
     txt = open(pl, "r", encoding="utf-8", errors="ignore").read()
-    # rewrite tên segment -> route phục vụ .ts
+    # rewrite segment names -> absolute URLs for HLS.js
     out = []
     for line in txt.split("\n"):
         s = line.strip()
         if s and not s.startswith("#") and s.endswith(".ts"):
+            # Dùng absolute URL để HLS.js resolve đúng
             out.append(f"/api/dvr/buffer/seg/{channel_id}/{s}")
         else:
             out.append(line)
+    # Thêm cache-bust để tránh browser cache cũ
     return Response("\n".join(out), media_type="application/vnd.apple.mpegurl",
-                    headers={"Access-Control-Allow-Origin": "*", "Cache-Control": "no-cache"})
+                    headers={
+                        "Access-Control-Allow-Origin": "*",
+                        "Cache-Control": "no-cache, no-store, must-revalidate",
+                        "Pragma": "no-cache",
+                        "Expires": "0",
+                    })
 
 @router.get("/api/dvr/buffer/seg/{channel_id}/{seg}")
 def buffer_seg(channel_id: str, seg: str):
@@ -202,15 +211,40 @@ _SCHEDULES = _load_sched()
 
 @router.post("/api/dvr/record/schedule")
 def record_schedule(payload: dict = Body(...)):
-    """body: {channel, start_iso (giờ VN), duration_sec, prebuffer?, upload_repo?}"""
+    """body: {channel, start_iso (giờ VN), end_iso (giờ VN), duration_sec, prebuffer?, upload_repo?}"""
     ch = str(payload.get("channel", "")).lower().strip()
     if not ch:
         return JSONResponse({"error": "missing channel"}, status_code=400)
+    start_iso = payload.get("start_iso")
+    end_iso = payload.get("end_iso")
+    if not start_iso:
+        return JSONResponse({"error": "missing start_iso"}, status_code=400)
+    # Parse và validate timezone
+    try:
+        start_dt = datetime.fromisoformat(start_iso)
+        if start_dt.tzinfo is None:
+            start_dt = start_dt.replace(tzinfo=VN_TZ)
+    except Exception:
+        return JSONResponse({"error": "invalid start_iso"}, status_code=400)
+    # Tính duration từ end_iso nếu có, nếu không dùng duration_sec
+    if end_iso:
+        try:
+            end_dt = datetime.fromisoformat(end_iso)
+            if end_dt.tzinfo is None:
+                end_dt = end_dt.replace(tzinfo=VN_TZ)
+            duration_from_end = int((end_dt - start_dt).total_seconds())
+            duration = max(5, min(3600, duration_from_end))
+        except Exception:
+            duration = int(payload.get("duration_sec", 600))
+    else:
+        duration = int(payload.get("duration_sec", 600))
+    duration = max(5, min(3600, duration))
     job = {
         "id": str(int(time.time() * 1000)),
         "channel": ch,
-        "start_iso": payload.get("start_iso") or datetime.now(VN_TZ).isoformat(),
-        "duration_sec": int(payload.get("duration_sec", 600)),
+        "start_iso": start_dt.isoformat(),
+        "end_iso": end_iso or (start_dt + timedelta(seconds=duration)).isoformat(),
+        "duration_sec": duration,
         "prebuffer": bool(payload.get("prebuffer", True)),
         "upload_repo": payload.get("upload_repo"),
         "status": "pending", "file": None,