.

Author: HoangTran0410

empty_script.js

@@ -1,5 +1,3 @@
-import { runScript, runScriptInCurrentTab } from "./helpers/utils.js";
-
 export default {
   icon: "",
   name: {
@@ -13,10 +11,6 @@ export default {
   blackList: [],
   whiteList: [],
 
-  // Check if this script is on (show checkmark on UI)
-  getActive: () => {},
-  setActive: () => {},
-
   // run (if active) in background context
   backgroundScript: {
     onDocumentStart: (tab) => {},

popup/index.js

@@ -1,11 +1,17 @@
-import { GlobalBlackList, MsgType } from "../scripts/helpers/constants.js";
+import {
+  Events,
+  GlobalBlackList,
+  MsgType,
+  ScriptType,
+} from "../scripts/helpers/constants.js";
 import {
   checkBlackWhiteList,
+  getActiveScript,
   getCurrentTab,
   isFunction,
   removeAccents,
-  runScriptInCurrentTab,
   sendEventToTab,
+  toggleActiveScript,
 } from "../scripts/helpers/utils.js";
 import { allScripts } from "../scripts/index.js";
 import { checkForUpdate } from "./helpers/checkForUpdate.js";
@@ -146,12 +152,14 @@ function createScriptButton(script, isFavorite = false) {
   buttonContainer.className = "buttonContainer";
 
   // button checker
-  if (isFunction(script.getActive)) {
+  if (
+    ScriptType.contentScript in script ||
+    ScriptType.backgroundScript in script
+  ) {
     const checkmark = document.createElement("button");
     checkmark.className = "checkmark tooltip";
     checkmark.onclick = async (e) => {
-      let newValue = !(await script.getActive());
-      await script.setActive(newValue);
+      let newValue = await toggleActiveScript(script.id);
       updateButtonChecker(script, buttonContainer, newValue);
     };
 
@@ -264,7 +272,7 @@ function createScriptButton(script, isFavorite = false) {
 async function updateButtonChecker(script, button, val) {
   let checkmark = button.querySelector(".checkmark");
   if (!checkmark) return;
-  if (val ?? (await script.getActive?.())) {
+  if (val ?? (await getActiveScript(script.id))) {
     checkmark.classList.add("active");
     checkmark.title = t({
       vi: "Tắt tự động chạy",

popup/tabs.js

@@ -1,3 +1,4 @@
+import { ScriptType } from "../scripts/helpers/constants.js";
 import { isFunction } from "../scripts/helpers/utils.js";
 import { allScripts as s } from "../scripts/index.js";
 import { CATEGORY } from "./helpers/category.js";
@@ -6,7 +7,10 @@ import { favoriteScriptsSaver, recentScriptsSaver } from "./helpers/storage.js";
 const createTitle = (en, vi) => ({ name: { en, vi } });
 const canClick = (script) =>
   isFunction(script.onClick) || isFunction(script.onClickExtension);
-const isTitle = (script) => !canClick(script);
+const isTitle = (script) =>
+  !canClick(script) &&
+  !(ScriptType.contentScript in script) &&
+  !(ScriptType.backgroundScript in script);
 
 const specialTabs = [
   {

scripts/content-scripts/document_start.js

@@ -31,6 +31,7 @@
     ScriptType.contentScript,
     location.href
   );
+
   sendEventToBackground({
     type: MsgType.runScript,
     event: Events.document_start,

scripts/fb_toggleLight.js

@@ -1,12 +1,3 @@
-import {
-  getCurrentTab,
-  localStorage,
-  runScriptInCurrentTab,
-  runScriptInTab,
-} from "./helpers/utils.js";
-
-const key = "ufs-fb-toggle-light";
-
 export default {
   icon: `<i class="fa-solid fa-lightbulb"></i>`,
   name: {
@@ -19,9 +10,6 @@ export default {
   },
   whiteList: ["https://www.facebook.com"],
 
-  getActive: async () => await await localStorage.get(key),
-  setActive: async (v) => await await localStorage.set(key, v),
-
   contentScript: {
     onDocumentIdle: () => {
       shared.toggleLight(false);

scripts/helpers/utils.js

@@ -10,7 +10,7 @@ export async function runAllScriptWithEventType(eventType, scriptType, url) {
 
     let func = script?.[scriptType]?.[funcName];
     if (isFunction(func) && !isEmptyFunction(func)) {
-      let isActive = (await script.getActive?.()) ?? true;
+      let isActive = (await getActiveScript(script.id)) ?? true;
       isActive && func();
       console.log(
         `%c > Run ${script.id} ${scriptType} ${funcName}`,
@@ -37,14 +37,27 @@ export async function sendEventToTab(tabId, data) {
 // https://developer.chrome.com/docs/extensions/reference/storage/
 export const localStorage = {
   set: async (key, value) => {
-    return chrome.storage.sync.set({ [key]: value });
+    await chrome.storage.sync.set({ [key]: value });
+    return value;
   },
   get: async (key, defaultValue = "") => {
     let result = await chrome.storage.sync.get([key]);
     return result[key] || defaultValue;
   },
 };
 
+export async function setActiveScript(scriptId, isActive = true) {
+  return await localStorage.set("isactive" + scriptId, isActive);
+}
+
+export async function getActiveScript(scriptId) {
+  return await localStorage.get("isactive" + scriptId, false);
+}
+
+export async function toggleActiveScript(scriptId) {
+  return await setActiveScript(scriptId, !(await getActiveScript(scriptId)));
+}
+
 // #endregion
 
 // #region Tab Utils