diff --git a/src/render/SkipNotice.tsx b/src/render/SkipNotice.tsx index 81f4c026..d01e08fa 100644 --- a/src/render/SkipNotice.tsx +++ b/src/render/SkipNotice.tsx @@ -2,19 +2,19 @@ import * as React from "react"; import * as ReactDOM from "react-dom"; import SkipNoticeComponent from "../components/SkipNoticeComponent"; -import { SponsorTime } from "../types"; +import { SponsorTime, ContentContainer } from "../types"; class SkipNotice { segments: SponsorTime[]; autoSkip: boolean; // Contains functions and variables from the content script needed by the skip notice - contentContainer: () => any; + contentContainer: ContentContainer; noticeElement: HTMLDivElement; skipNoticeRef: React.MutableRefObject; - constructor(segments: SponsorTime[], autoSkip: boolean = false, contentContainer) { + constructor(segments: SponsorTime[], autoSkip: boolean = false, contentContainer: ContentContainer) { this.segments = segments; this.autoSkip = autoSkip; this.contentContainer = contentContainer; @@ -35,6 +35,10 @@ class SkipNotice { index++; } } + // YouTube Music + if (new URL(document.URL).host === "music.youtube.com") { + referenceNode = document.querySelector("#main-panel.ytmusic-player-page"); + } let amountOfPreviousNotices = document.getElementsByClassName("sponsorSkipNotice").length; //this is the suffix added at the end of every id