From 650fda3c677def0e354eb9f9feb9806400be6cb2 Mon Sep 17 00:00:00 2001
From: hanyd <hanyd@bsbsb.top>
Date: Fri, 13 Sep 2024 23:50:45 +0800
Subject: [PATCH] Refresh video label cache after submitting

---
 src/content.ts              |  3 +++
 src/requests/videoLabels.ts | 10 +++++-----
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/content.ts b/src/content.ts
index 647dbe2e..4e1a6545 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -13,6 +13,7 @@ import { PlayerButton } from "./render/PlayerButton";
 import SkipNotice from "./render/SkipNotice";
 import SubmissionNotice from "./render/SubmissionNotice";
 import { asyncRequestToServer } from "./requests/requests";
+import { getVideoLabel } from "./requests/videoLabels";
 import { setupThumbnailListener, updateAll } from "./thumbnail-utils/thumbnailManagement";
 import {
     ActionType,
@@ -2374,6 +2375,8 @@ async function sendSubmitMessage(): Promise<boolean> {
         const fullVideoSegment = sponsorTimes.filter((time) => time.actionType === ActionType.Full)[0];
         if (fullVideoSegment) {
             categoryPill?.setSegment(fullVideoSegment);
+            // refresh the video labels cache
+            getVideoLabel(getVideoID(), true);
         }
 
         return true;
diff --git a/src/requests/videoLabels.ts b/src/requests/videoLabels.ts
index 41dacadf..1e074eba 100644
--- a/src/requests/videoLabels.ts
+++ b/src/requests/videoLabels.ts
@@ -14,14 +14,14 @@ export interface LabelCacheEntry {
 const labelCache: Record<string, LabelCacheEntry> = {};
 const cacheLimit = 1000;
 
-async function getLabelHashBlock(hashPrefix: string): Promise<LabelCacheEntry | null> {
+async function getLabelHashBlock(hashPrefix: string, refreshCache: boolean = false): Promise<LabelCacheEntry | null> {
     // Check cache
     const cachedEntry = labelCache[hashPrefix];
-    if (cachedEntry) {
+    if (cachedEntry && !refreshCache) {
         return cachedEntry;
     }
 
-    const response = await asyncRequestToServer("GET", `/api/videoLabels/${hashPrefix}`);
+    const response = await asyncRequestToServer("GET", `/api/videoLabels/${hashPrefix}`, {}, refreshCache);
     if (response.status !== 200) {
         // No video labels or server down
         labelCache[hashPrefix] = {
@@ -54,9 +54,9 @@ async function getLabelHashBlock(hashPrefix: string): Promise<LabelCacheEntry |
     }
 }
 
-export async function getVideoLabel(videoID: VideoID): Promise<Category | null> {
+export async function getVideoLabel(videoID: VideoID, refreshCache: boolean = false): Promise<Category | null> {
     const prefix = (await getVideoIDHash(videoID)).slice(0, 3);
-    const result = await getLabelHashBlock(prefix);
+    const result = await getLabelHashBlock(prefix, refreshCache);
 
     if (result) {
         const category = result.videos[videoID];