ryd.content-script.js 10.0 KB

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