From 4a19fececf4be6cad6b7b5c3b4c8a9faeb0aa850 Mon Sep 17 00:00:00 2001
From: Ajay <dev@ajay.app>
Date: Sun, 26 Dec 2021 00:17:49 -0500
Subject: [PATCH] Always set segment source

---
 src/content.ts | 8 ++++++--
 src/types.ts   | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/content.ts b/src/content.ts
index e2057499..c0045fb7 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -708,7 +708,11 @@ async function sponsorsLookup(id: string, keepOldSubmissions = true) {
     if (response?.ok) {
         const recievedSegments: SponsorTime[] = JSON.parse(response.responseText)
                     ?.filter((video) => video.videoID === id)
-                    ?.map((video) => video.segments)[0];
+                    ?.map((video) => video.segments)?.[0]
+                    ?.map((segment) => ({
+                        ...segment,
+                        source: SponsorSourceType.Server
+                    }));
         if (!recievedSegments || !recievedSegments.length) { 
             // return if no video found
             retryFetch();
@@ -1666,7 +1670,7 @@ function vote(type: number, UUID: SegmentUUID, category?: Category, skipNotice?:
     const sponsorIndex = utils.getSponsorIndexFromUUID(sponsorTimes, UUID);
 
     // Don't vote for preview sponsors
-    if (sponsorIndex == -1 || sponsorTimes[sponsorIndex].source === SponsorSourceType.Local) return;
+    if (sponsorIndex == -1 || sponsorTimes[sponsorIndex].source !== SponsorSourceType.Server) return;
 
     // See if the local time saved count and skip count should be saved
     if (type === 0 && sponsorSkipped[sponsorIndex] || type === 1 && !sponsorSkipped[sponsorIndex]) {
diff --git a/src/types.ts b/src/types.ts
index 0c1a7d87..7409e3ff 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -87,7 +87,7 @@ export interface SponsorTime extends SegmentContainer {
     description?: string;
 
     hidden?: SponsorHideType;
-    source?: SponsorSourceType;
+    source: SponsorSourceType;
     videoDuration?: number;
 }