Refresh segments when creating new segment to submit

1 files changed, 13 insertions, 0 deletions

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);
 }