From 2e1b17f8bdeba7c3242ab8575234f46908e66ad1 Mon Sep 17 00:00:00 2001
From: Ajay Ramachandran <dev@ajay.app>
Date: Mon, 31 May 2021 14:48:03 -0400
Subject: Refresh segments when creating new segment to submit

---
 src/content.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/content.ts b/src/content.ts
index 8154e99e..a9e92f10 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -589,6 +589,7 @@ async function sponsorsLookup(id: string) {
                 }
             }
 
+            const oldSegments = sponsorTimes;
             sponsorTimes = recievedSegments;
 
             // Hide all submissions smaller than the minimum duration
@@ -600,6 +601,15 @@ async function sponsorsLookup(id: string) {
                 }
             }
 
+            for (const segment of oldSegments) {
+                const otherSegment = sponsorTimes.find((other) => segment.UUID === other.UUID);
+                if (otherSegment) {
+                    // If they downvoted it, or changed the category, keep it
+                    otherSegment.hidden = segment.hidden;
+                    otherSegment.category = segment.category;
+                }
+            }
+
             startSkipScheduleCheckingForStartSponsors();
 
             // Reset skip save
@@ -1181,6 +1191,9 @@ function startOrEndTimingNewSegment() {
     // Save the newly created segment
     Config.config.segmentTimes.set(sponsorVideoID, sponsorTimesSubmitting);
 
+    // Make sure they know if someone has already submitted something it while they were watching
+    sponsorsLookup(sponsorVideoID);
+
     updateEditButtonsOnPlayer();
     updateSponsorTimesSubmitting(false);
 }
-- 
cgit v1.2.3