script.js 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. (function(securityToken, extensionId) {
  2. function getButtons() {
  3. return document
  4. .getElementById("menu-container")
  5. ?.querySelector("#top-level-buttons-computed");
  6. }
  7. function getLikeButton() {
  8. return getButtons().children[0];
  9. }
  10. function getDislikeButton() {
  11. return getButtons().children[1];
  12. }
  13. function isVideoLiked() {
  14. return getLikeButton().classList.contains("style-default-active");
  15. }
  16. function isVideoDisliked() {
  17. return getDislikeButton().classList.contains("style-default-active");
  18. }
  19. function isVideoNotLiked() {
  20. return getLikeButton().classList.contains("style-text");
  21. }
  22. function isVideoNotDisliked() {
  23. return getDislikeButton().classList.contains("style-text");
  24. }
  25. function getState() {
  26. if (isVideoLiked()) {
  27. return "liked";
  28. }
  29. if (isVideoDisliked()) {
  30. return "disliked";
  31. }
  32. return "neutral";
  33. }
  34. function setLikes(likesCount) {
  35. getButtons().children[0].querySelector("#text").innerText = likesCount;
  36. }
  37. function setDislikes(dislikesCount) {
  38. getButtons().children[1].querySelector("#text").innerText = dislikesCount;
  39. }
  40. function setState() {
  41. chrome.runtime.sendMessage(extensionId, {
  42. securityToken: securityToken,
  43. message: "set_state",
  44. videoId: getVideoId(),
  45. state: getState(),
  46. },
  47. function(response) {
  48. if (response != undefined) {
  49. const formattedDislike = numberFormat(response.dislikes);
  50. // setLikes(response.likes);
  51. console.log(response);
  52. setDislikes(formattedDislike);
  53. } else {}
  54. }
  55. );
  56. }
  57. function likeClicked() {
  58. console.log("like" + getState());
  59. setState();
  60. }
  61. function dislikeClicked() {
  62. console.log("dislike" + getState());
  63. setState();
  64. }
  65. function setInitalState() {
  66. setState();
  67. }
  68. function getVideoId() {
  69. const urlParams = new URLSearchParams(window.location.search);
  70. const videoId = urlParams.get("v");
  71. return videoId;
  72. }
  73. function isVideoLoaded() {
  74. const videoId = getVideoId();
  75. return (
  76. document.querySelector(`ytd-watch-flexy[video-id='${videoId}']`) !== null
  77. );
  78. }
  79. function numberFormat(numberState) {
  80. const userLocales = navigator.language;
  81. const formatter = Intl.NumberFormat(userLocales, { notation: "compact" });
  82. return formatter.format(numberState);
  83. }
  84. function setEventListeners(evt) {
  85. function checkForJS_Finish() {
  86. if (getButtons()?.offsetParent && isVideoLoaded()) {
  87. clearInterval(jsInitChecktimer);
  88. const buttons = getButtons();
  89. if (!window.returnDislikeButtonlistenersSet) {
  90. buttons.children[0].addEventListener("click", likeClicked);
  91. buttons.children[1].addEventListener("click", dislikeClicked);
  92. window.returnDislikeButtonlistenersSet = true;
  93. }
  94. setInitalState();
  95. }
  96. }
  97. if (window.location.href.indexOf("watch?") >= 0) {
  98. var jsInitChecktimer = setInterval(checkForJS_Finish, 111);
  99. }
  100. }
  101. setEventListeners();
  102. })(document.currentScript.getAttribute("security-token"), document.currentScript.getAttribute("extension-id"));