ryd.content-script.js 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376
  1. import {
  2. LIKED_STATE,
  3. DISLIKED_STATE,
  4. NEUTRAL_STATE,
  5. } from "./src/state";
  6. let storedData = {
  7. likes: 0,
  8. dislikes: 0,
  9. previousState: NEUTRAL_STATE,
  10. };
  11. function cLog(message, writer) {
  12. message = `[return youtube dislike]: ${message}`;
  13. if (writer) {
  14. writer(message);
  15. } else {
  16. console.log(message);
  17. }
  18. }
  19. function isMobile() {
  20. return location.hostname == "m.youtube.com";
  21. }
  22. function getButtons() {
  23. if (isMobile()) {
  24. return document.querySelector(".slim-video-action-bar-actions");
  25. }
  26. //--- If Menu Element Is Displayed: ---//
  27. if (document.getElementById("menu-container")?.offsetParent === null) {
  28. return document.querySelector(
  29. "ytd-menu-renderer.ytd-watch-metadata > div"
  30. );
  31. //--- If Menu Element Isnt Displayed: ---//
  32. } else {
  33. return document
  34. .getElementById("menu-container")
  35. ?.querySelector("#top-level-buttons-computed");
  36. }
  37. }
  38. function getLikeButton() {
  39. return getButtons().children[0];
  40. }
  41. function getDislikeButton() {
  42. return getButtons().children[1];
  43. }
  44. function isVideoLiked() {
  45. if (isMobile()) {
  46. return (
  47. getLikeButton().querySelector("button").getAttribute("aria-label") ==
  48. "true"
  49. );
  50. }
  51. return getLikeButton().classList.contains("style-default-active");
  52. }
  53. function isVideoDisliked() {
  54. if (isMobile()) {
  55. return (
  56. getDislikeButton().querySelector("button").getAttribute("aria-label") ==
  57. "true"
  58. );
  59. }
  60. return getDislikeButton().classList.contains("style-default-active");
  61. }
  62. function checkForSignInButton() {
  63. if (
  64. document.querySelector(
  65. "a[href^='https://accounts.google.com/ServiceLogin']"
  66. )
  67. ) {
  68. return true;
  69. } else {
  70. return false;
  71. }
  72. }
  73. function getState() {
  74. if (isVideoLiked()) {
  75. return { current: LIKED_STATE, previous: storedData.previousState };
  76. }
  77. if (isVideoDisliked()) {
  78. return { current: DISLIKED_STATE, previous: storedData.previousState };
  79. }
  80. return { current: NEUTRAL_STATE, previous: storedData.previousState };
  81. }
  82. //--- Sets The Likes And Dislikes Values ---//
  83. function setLikes(likesCount) {
  84. (getButtons().children[0].querySelector("#text")).innerText = likesCount;
  85. }
  86. function setDislikes(dislikesCount) {
  87. if (isMobile()) {
  88. getButtons().children[1].querySelector(
  89. ".button-renderer-text"
  90. ).innerText = dislikesCount;
  91. return;
  92. }
  93. (getButtons().children[1].querySelector("#text").innerText) = dislikesCount;
  94. }
  95. function getLikeCountFromButton() {
  96. let likesStr = getLikeButton()
  97. .querySelector("button")
  98. .getAttribute("aria-label")
  99. .replace(/\D/g, "");
  100. return likesStr.length > 0 ? parseInt(likesStr) : false;
  101. }
  102. function processResponse(response) {
  103. const formattedDislike = numberFormat(response.dislikes);
  104. setDislikes(formattedDislike);
  105. storedData.dislikes = parseInt(response.dislikes);
  106. storedData.likes = getLikeCountFromButton() || parseInt(response.likes);
  107. createRateBar(storedData.likes, storedData.dislikes);
  108. }
  109. function setState() {
  110. storedData.previousState = isVideoDisliked()
  111. ? DISLIKED_STATE
  112. : isVideoLiked()
  113. ? LIKED_STATE
  114. : NEUTRAL_STATE;
  115. let statsSet = false;
  116. getBrowser().runtime.sendMessage(
  117. {
  118. message: "set_state",
  119. videoId: getVideoId(window.location.href),
  120. state: getState(storedData).current,
  121. likeCount: getLikeCountFromButton() || null,
  122. },
  123. function (response) {
  124. cLog("response from api:");
  125. cLog(JSON.stringify(response));
  126. if (response !== undefined && !("traceId" in response) && !statsSet) {
  127. processResponse(response);
  128. } else {
  129. }
  130. }
  131. );
  132. }
  133. function sendVote(vote) {
  134. getBrowser().runtime.sendMessage({
  135. message: "send_vote",
  136. vote: vote,
  137. videoId: getVideoId(window.location.href),
  138. });
  139. }
  140. function likeClicked() {
  141. if (checkForSignInButton() === false) {
  142. if (storedData.previousState === DISLIKED_STATE) {
  143. sendVote(1);
  144. storedData.dislikes--;
  145. storedData.likes++;
  146. createRateBar(storedData.likes, storedData.dislikes);
  147. setDislikes(numberFormat(storedData.dislikes));
  148. storedData.previousState = LIKED_STATE;
  149. } else if (storedData.previousState === NEUTRAL_STATE) {
  150. sendVote(1);
  151. storedData.likes++;
  152. createRateBar(storedData.likes, storedData.dislikes);
  153. storedData.previousState = LIKED_STATE;
  154. } else if ((storedData.previousState = LIKED_STATE)) {
  155. sendVote(0);
  156. storedData.likes--;
  157. createRateBar(storedData.likes, storedData.dislikes);
  158. storedData.previousState = NEUTRAL_STATE;
  159. }
  160. }
  161. }
  162. function dislikeClicked() {
  163. if (checkForSignInButton() == false) {
  164. if (storedData.previousState === NEUTRAL_STATE) {
  165. sendVote(-1);
  166. storedData.dislikes++;
  167. setDislikes(numberFormat(storedData.dislikes));
  168. createRateBar(storedData.likes, storedData.dislikes);
  169. storedData.previousState = DISLIKED_STATE;
  170. } else if (storedData.previousState === DISLIKED_STATE) {
  171. sendVote(0);
  172. storedData.dislikes--;
  173. setDislikes(numberFormat(storedData.dislikes));
  174. createRateBar(storedData.likes, storedData.dislikes);
  175. storedData.previousState = NEUTRAL_STATE;
  176. } else if (storedData.previousState === LIKED_STATE) {
  177. sendVote(-1);
  178. storedData.likes--;
  179. storedData.dislikes++;
  180. setDislikes(numberFormat(storedData.dislikes));
  181. createRateBar(storedData.likes, storedData.dislikes);
  182. storedData.previousState = DISLIKED_STATE;
  183. }
  184. }
  185. }
  186. function setInitialState() {
  187. setState();
  188. setTimeout(() => {
  189. sendVideoIds();
  190. }, 1500);
  191. }
  192. function getVideoId(url) {
  193. const urlObject = new URL(url);
  194. const pathname = urlObject.pathname;
  195. if (pathname.startsWith("/clip")) {
  196. return document.querySelector("meta[itemprop='videoId']").content;
  197. } else {
  198. return urlObject.searchParams.get("v");
  199. }
  200. }
  201. function isVideoLoaded() {
  202. const videoId = getVideoId(window.location.href);
  203. return (
  204. document.querySelector(`ytd-watch-flexy[video-id='${videoId}']`) !== null ||
  205. // mobile: no video-id attribute
  206. document.querySelector('#player[loading="false"]:not([hidden])') !== null
  207. );
  208. }
  209. function roundDown(num) {
  210. if (num < 1000) return num;
  211. const int = Math.floor(Math.log10(num) - 2);
  212. const decimal = int + (int % 3 ? 1 : 0);
  213. const value = Math.floor(num / 10 ** decimal);
  214. return value * 10 ** decimal;
  215. }
  216. function numberFormat(numberState) {
  217. let userLocales;
  218. try {
  219. userLocales = new URL(
  220. Array.from(document.querySelectorAll("head > link[rel='search']"))
  221. ?.find((n) => n?.getAttribute("href")?.includes("?locale="))
  222. ?.getAttribute("href")
  223. )?.searchParams?.get("locale");
  224. } catch {}
  225. const formatter = Intl.NumberFormat(
  226. document.documentElement.lang || userLocales || navigator.language,
  227. {
  228. notation: "compact",
  229. }
  230. );
  231. return formatter.format(roundDown(numberState));
  232. }
  233. let jsInitChecktimer = null;
  234. function setEventListeners(evt) {
  235. function checkForJS_Finish() {
  236. if (getButtons()?.offsetParent && isVideoLoaded()) {
  237. clearInterval(jsInitChecktimer);
  238. jsInitChecktimer = null;
  239. const buttons = getButtons();
  240. if (!window.returnDislikeButtonlistenersSet) {
  241. buttons.children[0].addEventListener("click", likeClicked);
  242. buttons.children[1].addEventListener("click", dislikeClicked);
  243. window.returnDislikeButtonlistenersSet = true;
  244. }
  245. setInitialState();
  246. }
  247. }
  248. if (window.location.href.indexOf("watch?") >= 0) {
  249. jsInitChecktimer = setInterval(checkForJS_Finish, 111);
  250. }
  251. }
  252. function createRateBar(likes, dislikes) {
  253. let rateBar = document.getElementById("ryd-bar-container");
  254. const widthPx =
  255. getButtons().children[0].clientWidth +
  256. getButtons().children[1].clientWidth +
  257. 8;
  258. const widthPercent =
  259. likes + dislikes > 0 ? (likes / (likes + dislikes)) * 100 : 50;
  260. if (!rateBar) {
  261. (
  262. document.getElementById("menu-container") ||
  263. document.querySelector("ytm-slim-video-action-bar-renderer")
  264. ).insertAdjacentHTML(
  265. "beforeend",
  266. `
  267. <div class="ryd-tooltip" style="width: ${widthPx}px">
  268. <div class="ryd-tooltip-bar-container">
  269. <div
  270. id="ryd-bar-container"
  271. style="width: 100%; height: 2px;"
  272. >
  273. <div
  274. id="ryd-bar"
  275. style="width: ${widthPercent}%; height: 100%"
  276. ></div>
  277. </div>
  278. </div>
  279. <tp-yt-paper-tooltip position="top" id="ryd-dislike-tooltip" class="style-scope ytd-sentiment-bar-renderer" role="tooltip" tabindex="-1">
  280. <!--css-build:shady-->${likes.toLocaleString()}&nbsp;/&nbsp;${dislikes.toLocaleString()}
  281. </tp-yt-paper-tooltip>
  282. </div>
  283. `
  284. );
  285. } else {
  286. document.getElementById("ryd-bar-container").style.width = widthPx + "px";
  287. document.getElementById("ryd-bar").style.width = widthPercent + "%";
  288. document.querySelector(
  289. "#ryd-dislike-tooltip > #tooltip"
  290. ).innerHTML = `${likes.toLocaleString()}&nbsp;/&nbsp;${dislikes.toLocaleString()}`;
  291. }
  292. }
  293. function sendVideoIds() {
  294. let links = Array.from(
  295. document.getElementsByClassName(
  296. "yt-simple-endpoint ytd-compact-video-renderer"
  297. )
  298. ).concat(
  299. Array.from(
  300. document.getElementsByClassName("yt-simple-endpoint ytd-thumbnail")
  301. )
  302. );
  303. // Also try mobile
  304. if (links.length < 1)
  305. links = Array.from(
  306. document.querySelectorAll(
  307. ".large-media-item-metadata > a, a.large-media-item-thumbnail-container"
  308. )
  309. );
  310. const ids = links
  311. .filter((x) => x.href && x.href.indexOf("/watch?v=") > 0)
  312. .map((x) => getVideoId(x.href));
  313. getBrowser().runtime.sendMessage({
  314. message: "send_links",
  315. videoIds: ids,
  316. });
  317. }
  318. setEventListeners();
  319. document.addEventListener("yt-navigate-finish", function (event) {
  320. if (jsInitChecktimer !== null) clearInterval(jsInitChecktimer);
  321. window.returnDislikeButtonlistenersSet = false;
  322. setEventListeners();
  323. });
  324. setTimeout(() => sendVideoIds(), 2500);
  325. function getBrowser() {
  326. if (typeof chrome !== "undefined" && typeof chrome.runtime !== "undefined") {
  327. return chrome;
  328. } else if (
  329. typeof browser !== "undefined" &&
  330. typeof browser.runtime !== "undefined"
  331. ) {
  332. return browser;
  333. } else {
  334. console.log("browser is not supported");
  335. return false;
  336. }
  337. }