From 4092bf9b0546f13be383937fbc052d3b9c1abdca Mon Sep 17 00:00:00 2001
From: Ajay Ramachandran <dev@ajay.app>
Date: Wed, 1 Sep 2021 20:52:40 -0400
Subject: [PATCH] Make skip notice buttons work with mute segments

---
 public/_locales/en/messages.json       |   3 +
 src/components/SkipNoticeComponent.tsx | 102 +++++++++++++++++--------
 src/content.ts                         |  30 +++++---
 3 files changed, 92 insertions(+), 43 deletions(-)

diff --git a/public/_locales/en/messages.json b/public/_locales/en/messages.json
index d579ac45..043d64e0 100644
--- a/public/_locales/en/messages.json
+++ b/public/_locales/en/messages.json
@@ -52,6 +52,9 @@
     "reskip": {
         "message": "Reskip"
     },
+    "unmute": {
+        "message": "Unmute"
+    },
     "paused": {
         "message": "Paused"
     },
diff --git a/src/components/SkipNoticeComponent.tsx b/src/components/SkipNoticeComponent.tsx
index 0af1efad..b3ca619b 100644
--- a/src/components/SkipNoticeComponent.tsx
+++ b/src/components/SkipNoticeComponent.tsx
@@ -1,7 +1,7 @@
 import * as React from "react";
 import * as CompileConfig from "../../config.json";
 import Config from "../config"
-import { Category, ContentContainer, CategoryActionType, SponsorHideType, SponsorTime, NoticeVisbilityMode } from "../types";
+import { Category, ContentContainer, CategoryActionType, SponsorHideType, SponsorTime, NoticeVisbilityMode, ActionType } from "../types";
 import NoticeComponent from "./NoticeComponent";
 import NoticeTextSelectionComponent from "./NoticeTextSectionComponent";
 
@@ -39,8 +39,8 @@ export interface SkipNoticeState {
     maxCountdownTime?: () => number;
     countdownText?: string;
 
-    unskipText?: string;
-    unskipCallback?: (index: number) => void;
+    skipButtonText?: string;
+    skipButtonCallback?: (index: number) => void;
 
     downvoting?: boolean;
     choosingCategory?: boolean;
@@ -110,8 +110,8 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
             countdownTime: Config.config.skipNoticeDuration,
             countdownText: null,
 
-            unskipText: chrome.i18n.getMessage("unskip"),
-            unskipCallback: (index) => this.unskip(index),
+            skipButtonText: this.getUnskipText(),
+            skipButtonCallback: (index) => this.unskip(index),
 
             downvoting: false,
             choosingCategory: false,
@@ -126,7 +126,7 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
 
         if (!this.autoSkip) {
             // Assume manual skip is only skipping 1 submission
-            Object.assign(this.state, this.getUnskippedModeInfo(0, chrome.i18n.getMessage("skip")));
+            Object.assign(this.state, this.getUnskippedModeInfo(0, this.getSkipText()));
         }
     }
 
@@ -306,7 +306,7 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
                         style={{marginLeft: "4px"}}
                         onClick={() => this.prepAction(SkipNoticeAction.Unskip)}>
 
-                        {this.state.unskipText + (this.state.showKeybindHint ? " (" + Config.config.skipKeybind + ")" : "")}
+                        {this.state.skipButtonText + (this.state.showKeybindHint ? " (" + Config.config.skipKeybind + ")" : "")}
                     </button>
                 </span>
             );
@@ -396,7 +396,7 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
                 this.contentContainer().vote(undefined, this.segments[index].UUID, this.categoryOptionRef.current.value as Category, this)
                 break;
             case SkipNoticeAction.Unskip:
-                this.state.unskipCallback(index);
+                this.state.skipButtonCallback(index);
                 break;
         }
 
@@ -456,7 +456,29 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
     unskip(index: number): void {
         this.contentContainer().unskipSponsorTime(this.segments[index], this.props.unskipTime);
 
-        this.unskippedMode(index, chrome.i18n.getMessage("reskip"));
+        this.unskippedMode(index, this.getReskipText());
+    }
+
+    reskip(index: number): void {
+        this.contentContainer().reskipSponsorTime(this.segments[index]);
+
+        const newState: SkipNoticeState = {
+            skipButtonText: this.getUnskipText(),
+            skipButtonCallback: this.unskip.bind(this),
+
+            maxCountdownTime: () => Config.config.skipNoticeDuration,
+            countdownTime: Config.config.skipNoticeDuration
+        };
+
+        // See if the title should be changed
+        if (!this.autoSkip) {
+            newState.noticeTitle = chrome.i18n.getMessage("noticeTitle");
+        }       
+
+        //reset countdown
+        this.setState(newState, () => {
+            this.noticeRef.current.resetCountdown();
+        });
     }
 
     /** Sets up notice to be not skipped yet */
@@ -478,36 +500,14 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
         } : this.state.maxCountdownTime;
 
         return {
-            unskipText: buttonText,
-            unskipCallback: (index) => this.reskip(index),
+            skipButtonText: buttonText,
+            skipButtonCallback: (index) => this.reskip(index),
             // change max duration to however much of the sponsor is left
             maxCountdownTime: maxCountdownTime,
             countdownTime: maxCountdownTime()
         } as SkipNoticeState;
     }
 
-    reskip(index: number): void {
-        this.contentContainer().reskipSponsorTime(this.segments[index]);
-
-        const newState: SkipNoticeState = {
-            unskipText: chrome.i18n.getMessage("unskip"),
-            unskipCallback: this.unskip.bind(this),
-
-            maxCountdownTime: () => Config.config.skipNoticeDuration,
-            countdownTime: Config.config.skipNoticeDuration
-        };
-
-        // See if the title should be changed
-        if (!this.autoSkip) {
-            newState.noticeTitle = chrome.i18n.getMessage("noticeTitle");
-        }       
-
-        //reset countdown
-        this.setState(newState, () => {
-            this.noticeRef.current.resetCountdown();
-        });
-    }
-
     afterVote(segment: SponsorTime, type: number, category: Category): void {
         this.addVoteButtonInfo(chrome.i18n.getMessage("voted"));
 
@@ -558,6 +558,42 @@ class SkipNoticeComponent extends React.Component<SkipNoticeProps, SkipNoticeSta
 
         this.props.closeListener();
     }
+
+    private getUnskipText(): string {
+        switch (this.props.segments[0].actionType) {
+            case ActionType.Mute: {
+                return chrome.i18n.getMessage("unmute");
+            }
+            case ActionType.Skip: 
+            default: {
+                return chrome.i18n.getMessage("unskip");
+            }
+        }
+    }
+
+    private getReskipText(): string {
+        switch (this.props.segments[0].actionType) {
+            case ActionType.Mute: {
+                return chrome.i18n.getMessage("mute");
+            }
+            case ActionType.Skip: 
+            default: {
+                return chrome.i18n.getMessage("reskip");
+            }
+        }
+    }
+
+    private getSkipText(): string {
+        switch (this.props.segments[0].actionType) {
+            case ActionType.Mute: {
+                return chrome.i18n.getMessage("mute");
+            }
+            case ActionType.Skip: 
+            default: {
+                return chrome.i18n.getMessage("skip");
+            }
+        }
+    }
 }
 
 export default SkipNoticeComponent;
diff --git a/src/content.ts b/src/content.ts
index 0cb95b37..45856871 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -1190,19 +1190,29 @@ function skipToTime({v, skipTime, skippingSegments, openNotice, forceAutoSkip, u
 }
 
 function unskipSponsorTime(segment: SponsorTime, unskipTime: number = null) {
-    //add a tiny bit of time to make sure it is not skipped again
-    console.log(unskipTime)
-    video.currentTime = unskipTime ?? segment.segment[0] + 0.001;
+    if (segment.actionType === ActionType.Mute) {
+        video.muted = false;
+        videoMuted = false;
+    } else {
+        //add a tiny bit of time to make sure it is not skipped again
+        video.currentTime = unskipTime ?? segment.segment[0] + 0.001;
+    }
+    
 }
 
 function reskipSponsorTime(segment: SponsorTime) {
-    const skippedTime = Math.max(segment.segment[1] - video.currentTime, 0);
-    const segmentDuration = segment.segment[1] - segment.segment[0];
-    const fullSkip = skippedTime / segmentDuration > manualSkipPercentCount;
-    
-    video.currentTime = segment.segment[1];
-    sendTelemetryAndCount([segment], skippedTime, fullSkip);
-    startSponsorSchedule(true, segment.segment[1], false);
+    if (segment.actionType === ActionType.Mute) {
+        video.muted = true;
+        videoMuted = true;
+    } else {
+        const skippedTime = Math.max(segment.segment[1] - video.currentTime, 0);
+        const segmentDuration = segment.segment[1] - segment.segment[0];
+        const fullSkip = skippedTime / segmentDuration > manualSkipPercentCount;
+        
+        video.currentTime = segment.segment[1];
+        sendTelemetryAndCount([segment], skippedTime, fullSkip);
+        startSponsorSchedule(true, segment.segment[1], false);
+    }
 }
 
 function createButton(baseID: string, title: string, callback: () => void, imageName: string, isDraggable = false): HTMLElement {