From 286b1cf9b67f2ff0080a38cd18f2286b03624e4b Mon Sep 17 00:00:00 2001
From: simon <simobilleter@gmail.com>
Date: Tue, 5 Apr 2022 21:51:45 +0700
Subject: [PATCH] simulate sponsorblock post request

---
 tubearchivist/api/urls.py  |  6 ++++++
 tubearchivist/api/views.py | 30 ++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/tubearchivist/api/urls.py b/tubearchivist/api/urls.py
index 8c1a5f8..0cdf11a 100644
--- a/tubearchivist/api/urls.py
+++ b/tubearchivist/api/urls.py
@@ -11,6 +11,7 @@ from api.views import (
     VideoApiListView,
     VideoApiView,
     VideoProgressView,
+    VideoSponsorView,
 )
 from django.urls import path
 
@@ -32,6 +33,11 @@ urlpatterns = [
         VideoProgressView.as_view(),
         name="api-video-progress",
     ),
+    path(
+        "video/<slug:video_id>/sponsor/",
+        VideoSponsorView.as_view(),
+        name="api-video-sponsor",
+    ),
     path(
         "channel/",
         ChannelApiListView.as_view(),
diff --git a/tubearchivist/api/views.py b/tubearchivist/api/views.py
index bf391a9..f590d43 100644
--- a/tubearchivist/api/views.py
+++ b/tubearchivist/api/views.py
@@ -3,6 +3,7 @@
 from api.src.search_processor import SearchProcess
 from home.src.download.thumbnails import ThumbManager
 from home.src.es.connect import ElasticWrap
+from home.src.index.video import SponsorBlock
 from home.src.ta.config import AppConfig
 from home.src.ta.helper import UrlListParser
 from home.src.ta.ta_redis import RedisArchivist
@@ -144,6 +145,35 @@ class VideoProgressView(ApiBaseView):
         return Response(self.response)
 
 
+class VideoSponsorView(ApiBaseView):
+    """resolves to /api/video/<video_id>/
+    handle sponsor block integration
+    """
+
+    search_base = "ta_video/_doc/"
+
+    def get(self, request, video_id):
+        """get sponsor info"""
+        # pylint: disable=unused-argument
+
+        self.get_document(video_id)
+        sponsorblock = self.response["data"].get("sponsorblock")
+
+        return Response(sponsorblock)
+
+    @staticmethod
+    def post(request, video_id):
+        """post verification and timestamps"""
+        start_time = request.data.get("startTime")
+        end_time = request.data.get("endTime")
+
+        response, status_code = SponsorBlock(request.user.id).post_timestamps(
+            video_id, start_time, end_time
+        )
+
+        return Response(response, status=status_code)
+
+
 class ChannelApiView(ApiBaseView):
     """resolves to /api/channel/<channel_id>/
     GET: returns metadata dict of channel