ryd.content-script.js 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256
  1. import {
  2. getButtons,
  3. getLikeButton,
  4. getDislikeButton,
  5. checkForSignInButton,
  6. } from "./src/buttons";
  7. import {
  8. isMobile,
  9. isVideoDisliked,
  10. isVideoLiked,
  11. getState,
  12. setLikes,
  13. setDislikes,
  14. getLikeCountFromButton,
  15. LIKED_STATE,
  16. DISLIKED_STATE,
  17. NEUTRAL_STATE,
  18. } from "./src/state";
  19. import { numberFormat, getBrowser } from "./src/utils";
  20. let storedData = {
  21. likes: 0,
  22. dislikes: 0,
  23. previousState: NEUTRAL_STATE,
  24. };
  25. function cLog(message, writer) {
  26. message = `[return youtube dislike]: ${message}`;
  27. if (writer) {
  28. writer(message);
  29. } else {
  30. console.log(message);
  31. }
  32. }
  33. function processResponse(response) {
  34. const formattedDislike = numberFormat(response.dislikes);
  35. setDislikes(formattedDislike);
  36. storedData.dislikes = parseInt(response.dislikes);
  37. storedData.likes = getLikeCountFromButton() || parseInt(response.likes);
  38. createRateBar(storedData.likes, storedData.dislikes);
  39. }
  40. function setState() {
  41. storedData.previousState = isVideoDisliked()
  42. ? DISLIKED_STATE
  43. : isVideoLiked()
  44. ? LIKED_STATE
  45. : NEUTRAL_STATE;
  46. let statsSet = false;
  47. getBrowser().runtime.sendMessage(
  48. {
  49. message: "set_state",
  50. videoId: getVideoId(window.location.href),
  51. state: getState(storedData).current,
  52. likeCount: getLikeCountFromButton() || null,
  53. },
  54. function (response) {
  55. cLog("response from api:");
  56. cLog(JSON.stringify(response));
  57. if (response !== undefined && !("traceId" in response) && !statsSet) {
  58. processResponse(response);
  59. } else {
  60. }
  61. }
  62. );
  63. }
  64. function sendVote(vote) {
  65. getBrowser().runtime.sendMessage({
  66. message: "send_vote",
  67. vote: vote,
  68. videoId: getVideoId(window.location.href),
  69. });
  70. }
  71. function likeClicked() {
  72. if (checkForSignInButton() === false) {
  73. if (storedData.previousState === DISLIKED_STATE) {
  74. sendVote(1);
  75. storedData.dislikes--;
  76. storedData.likes++;
  77. createRateBar(storedData.likes, storedData.dislikes);
  78. setDislikes(numberFormat(storedData.dislikes));
  79. storedData.previousState = LIKED_STATE;
  80. } else if (storedData.previousState === NEUTRAL_STATE) {
  81. sendVote(1);
  82. storedData.likes++;
  83. createRateBar(storedData.likes, storedData.dislikes);
  84. storedData.previousState = LIKED_STATE;
  85. } else if ((storedData.previousState = LIKED_STATE)) {
  86. sendVote(0);
  87. storedData.likes--;
  88. createRateBar(storedData.likes, storedData.dislikes);
  89. storedData.previousState = NEUTRAL_STATE;
  90. }
  91. }
  92. }
  93. function dislikeClicked() {
  94. if (checkForSignInButton() == false) {
  95. if (storedData.previousState === NEUTRAL_STATE) {
  96. sendVote(-1);
  97. storedData.dislikes++;
  98. setDislikes(numberFormat(storedData.dislikes));
  99. createRateBar(storedData.likes, storedData.dislikes);
  100. storedData.previousState = DISLIKED_STATE;
  101. } else if (storedData.previousState === DISLIKED_STATE) {
  102. sendVote(0);
  103. storedData.dislikes--;
  104. setDislikes(numberFormat(storedData.dislikes));
  105. createRateBar(storedData.likes, storedData.dislikes);
  106. storedData.previousState = NEUTRAL_STATE;
  107. } else if (storedData.previousState === LIKED_STATE) {
  108. sendVote(-1);
  109. storedData.likes--;
  110. storedData.dislikes++;
  111. setDislikes(numberFormat(storedData.dislikes));
  112. createRateBar(storedData.likes, storedData.dislikes);
  113. storedData.previousState = DISLIKED_STATE;
  114. }
  115. }
  116. }
  117. function setInitialState() {
  118. setState();
  119. setTimeout(() => {
  120. sendVideoIds();
  121. }, 1500);
  122. }
  123. function getVideoId(url) {
  124. const urlObject = new URL(url);
  125. const pathname = urlObject.pathname;
  126. if (pathname.startsWith("/clip")) {
  127. return document.querySelector("meta[itemprop='videoId']").content;
  128. } else {
  129. return urlObject.searchParams.get("v");
  130. }
  131. }
  132. function isVideoLoaded() {
  133. const videoId = getVideoId(window.location.href);
  134. return (
  135. document.querySelector(`ytd-watch-flexy[video-id='${videoId}']`) !== null ||
  136. // mobile: no video-id attribute
  137. document.querySelector('#player[loading="false"]:not([hidden])') !== null
  138. );
  139. }
  140. let jsInitChecktimer = null;
  141. function setEventListeners(evt) {
  142. function checkForJS_Finish() {
  143. if (getButtons()?.offsetParent && isVideoLoaded()) {
  144. clearInterval(jsInitChecktimer);
  145. jsInitChecktimer = null;
  146. const buttons = getButtons();
  147. if (!window.returnDislikeButtonlistenersSet) {
  148. buttons.children[0].addEventListener("click", likeClicked);
  149. buttons.children[1].addEventListener("click", dislikeClicked);
  150. window.returnDislikeButtonlistenersSet = true;
  151. }
  152. setInitialState();
  153. }
  154. }
  155. if (window.location.href.indexOf("watch?") >= 0) {
  156. jsInitChecktimer = setInterval(checkForJS_Finish, 111);
  157. }
  158. }
  159. function createRateBar(likes, dislikes) {
  160. let rateBar = document.getElementById("ryd-bar-container");
  161. const widthPx =
  162. getButtons().children[0].clientWidth +
  163. getButtons().children[1].clientWidth +
  164. 8;
  165. const widthPercent =
  166. likes + dislikes > 0 ? (likes / (likes + dislikes)) * 100 : 50;
  167. if (!rateBar) {
  168. (
  169. document.getElementById("menu-container") ||
  170. document.querySelector("ytm-slim-video-action-bar-renderer")
  171. ).insertAdjacentHTML(
  172. "beforeend",
  173. `
  174. <div class="ryd-tooltip" style="width: ${widthPx}px">
  175. <div class="ryd-tooltip-bar-container">
  176. <div
  177. id="ryd-bar-container"
  178. style="width: 100%; height: 2px;"
  179. >
  180. <div
  181. id="ryd-bar"
  182. style="width: ${widthPercent}%; height: 100%"
  183. ></div>
  184. </div>
  185. </div>
  186. <tp-yt-paper-tooltip position="top" id="ryd-dislike-tooltip" class="style-scope ytd-sentiment-bar-renderer" role="tooltip" tabindex="-1">
  187. <!--css-build:shady-->${likes.toLocaleString()}&nbsp;/&nbsp;${dislikes.toLocaleString()}
  188. </tp-yt-paper-tooltip>
  189. </div>
  190. `
  191. );
  192. } else {
  193. document.getElementById("ryd-bar-container").style.width = widthPx + "px";
  194. document.getElementById("ryd-bar").style.width = widthPercent + "%";
  195. document.querySelector(
  196. "#ryd-dislike-tooltip > #tooltip"
  197. ).innerHTML = `${likes.toLocaleString()}&nbsp;/&nbsp;${dislikes.toLocaleString()}`;
  198. }
  199. }
  200. function sendVideoIds() {
  201. let links = Array.from(
  202. document.getElementsByClassName(
  203. "yt-simple-endpoint ytd-compact-video-renderer"
  204. )
  205. ).concat(
  206. Array.from(
  207. document.getElementsByClassName("yt-simple-endpoint ytd-thumbnail")
  208. )
  209. );
  210. // Also try mobile
  211. if (links.length < 1)
  212. links = Array.from(
  213. document.querySelectorAll(
  214. ".large-media-item-metadata > a, a.large-media-item-thumbnail-container"
  215. )
  216. );
  217. const ids = links
  218. .filter((x) => x.href && x.href.indexOf("/watch?v=") > 0)
  219. .map((x) => getVideoId(x.href));
  220. getBrowser().runtime.sendMessage({
  221. message: "send_links",
  222. videoIds: ids,
  223. });
  224. }
  225. setEventListeners();
  226. document.addEventListener("yt-navigate-finish", function (event) {
  227. if (jsInitChecktimer !== null) clearInterval(jsInitChecktimer);
  228. window.returnDislikeButtonlistenersSet = false;
  229. setEventListeners();
  230. });
  231. setTimeout(() => sendVideoIds(), 2500);