From fb0516506e84d4c4b96fbf0a530fa384ec8349ba Mon Sep 17 00:00:00 2001
From: xu_yanfeng <xu_yanfeng@126.com>
Date: Sat, 25 Jan 2025 14:12:33 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=8F=E5=8A=A9=E6=89=8B=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=E4=BA=86=E4=B8=80=E4=BA=9B=E5=8F=B3=E9=94=AE=E7=9A=84=E8=AE=BE?=
 =?UTF-8?q?=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cc-inspector/src/scripts/inject-view/app.vue  | 127 ++++++++++++++++--
 cc-inspector/src/scripts/inject-view/store.ts |  31 +++++
 .../src/scripts/inject-view/web-test.ts       |   3 +-
 .../src/scripts/inject/inject-view.ts         |   2 +
 .../src/scripts/inject/inspect-list.ts        |  38 ++++++
 cc-inspector/src/scripts/inject/inspector.ts  |  10 +-
 6 files changed, 197 insertions(+), 14 deletions(-)
 create mode 100644 cc-inspector/src/scripts/inject-view/store.ts
 create mode 100644 cc-inspector/src/scripts/inject/inspect-list.ts

diff --git a/cc-inspector/src/scripts/inject-view/app.vue b/cc-inspector/src/scripts/inject-view/app.vue
index ca7a552..276b225 100644
--- a/cc-inspector/src/scripts/inject-view/app.vue
+++ b/cc-inspector/src/scripts/inject-view/app.vue
@@ -1,12 +1,12 @@
 <template>
-  <div class="ad" ref="rootEl" v-show="!picking" @mouseleave="onMouseLeaveRoot" @mousedown="onMouseDown" @mouseenter="onMouseEnterCocosLogo">
+  <div class="ad" ref="rootEl" v-show="!picking" @contextmenu.prevent="onContextMenuRoot" @mouseleave="onMouseLeaveRoot" @mouseenter="onMouseEnterCocosLogo">
     <div class="title">
       <div class="btns" v-show="showBtns">
-        <div v-for="(item, index) in listArray" :key="index" class="list" @click="item.cb" :title="item.txt">
-          <i class="iconfont icon" :class="item.icon"></i>
+        <div v-for="(item, index) in listArray" :key="index" class="list" @click="item.click($event, item)" :title="item.txt" v-show="item.visible">
+          <i class="iconfont icon" :class="item.icon" @contextmenu.prevent.stop="item.contextmenu"></i>
         </div>
       </div>
-      <i class="iconfont icon_cocos cocos"></i>
+      <i class="iconfont icon_cocos cocos" @mousedown="onMouseDown"></i>
     </div>
     <!-- <Memory></Memory> -->
     <CCDialog></CCDialog>
@@ -15,30 +15,43 @@
 </template>
 <script lang="ts">
 import ccui from "@xuyanfeng/cc-ui";
+import { IUiMenuItem } from "@xuyanfeng/cc-ui/types/cc-menu/const";
+import { storeToRefs } from "pinia";
 import { defineComponent, onMounted, ref, toRaw } from "vue";
 import { GA_EventName } from "../../ga/type";
 import { DocumentEvent } from "../const";
+import { inspectTarget } from "../inject/inspect-list";
 import Ad from "./ad.vue";
 import Banner from "./banner.vue";
 import Memory from "./memory.vue";
+import { appStore } from "./store";
 import { ga } from "./util";
+declare const cc: any;
 const { CCDialog, CCMenu } = ccui.components;
 interface ListItem {
   icon: string;
   txt: string;
-  cb: (event: MouseEvent) => void;
+  visible: boolean;
+  /**
+   * 点击回调
+   */
+  click: (event: MouseEvent, item: ListItem) => void;
+  contextmenu: (event: MouseEvent) => void;
 }
 export default defineComponent({
   name: "ad",
   components: { CCDialog, Banner, Memory, CCMenu },
   setup() {
-    const keyAssistant = "assistant";
-
+    const store = appStore();
+    store.init();
+    const { config } = storeToRefs(appStore());
     const listArray = ref<ListItem[]>([
       {
         icon: "icon_shop_cart ani_shop_cart",
         txt: "Recommended Plugins",
-        cb: () => {
+        contextmenu: () => {},
+        visible: true,
+        click: () => {
           ccui.dialog.showDialog({
             title: "Recommended Plugins",
             comp: Ad,
@@ -50,10 +63,44 @@ export default defineComponent({
           });
         },
       },
+      {
+        icon: "icon_do_play",
+        click: (event: MouseEvent, item: ListItem) => {},
+        visible: true,
+        txt: "game play",
+        contextmenu: () => {
+          if (typeof cc !== "undefined") {
+            cc.game.resume();
+          }
+        },
+      },
+      {
+        icon: "icon_do_pause",
+        visible: true,
+        txt: "game pause",
+        click: () => {
+          if (typeof cc !== "undefined") {
+            cc.game.pause();
+          }
+        },
+        contextmenu: () => {},
+      },
+      {
+        icon: "icon_do_step",
+        visible: true,
+        txt: "game step",
+        click: () => {
+          if (typeof cc !== "undefined") {
+            cc.game.step();
+          }
+        },
+        contextmenu: () => {},
+      },
       {
         icon: "icon_target",
         txt: "Inspect Game",
-        cb: () => {
+        visible: true,
+        click: () => {
           ga(GA_EventName.DoInspector);
           showBtns.value = false;
           picking.value = true;
@@ -64,6 +111,38 @@ export default defineComponent({
             document.dispatchEvent(event);
           }
         },
+        contextmenu: (event: MouseEvent) => {
+          const arr = [
+            { name: "Inspect Label", type: typeof cc !== "undefined" ? cc.Label : "cc.Label" }, //
+            { name: "Inspect Sprite", type: typeof cc !== "undefined" ? cc.Sprite : "cc.Sprite" },
+            { name: "Inspect Button", type: typeof cc !== "undefined" ? cc.Button : "cc.Button" },
+            { name: "Inspect RichText", type: typeof cc !== "undefined" ? cc.RichText : "cc.RichText" },
+          ];
+          const compMenu: IUiMenuItem[] = arr.map((item) => {
+            return {
+              name: item.name,
+              enabled: inspectTarget.enabled,
+              selected: inspectTarget.isContainInspectType(item.type),
+              callback: (menu: IUiMenuItem) => {
+                if (menu.selected) {
+                  inspectTarget.removeInspectType(item.type);
+                } else {
+                  inspectTarget.addInspectType(item.type);
+                }
+              },
+            };
+          });
+          ccui.menu.showMenuByMouseEvent(event, [
+            {
+              name: "Filter Enabled",
+              selected: inspectTarget.enabled,
+              callback: (menu: IUiMenuItem) => {
+                inspectTarget.enabled = !inspectTarget.enabled;
+              },
+            },
+            ...compMenu,
+          ]);
+        },
       },
     ]);
     document.addEventListener(DocumentEvent.GameInspectorEnd, () => {
@@ -80,7 +159,7 @@ export default defineComponent({
       document.addEventListener("mousedown", test, true);
     }
     function recoverAssistantTop() {
-      const top = Number(localStorage.getItem(keyAssistant) || "0");
+      const top = toRaw(config.value.pos);
       updateAssistantTop(top);
     }
 
@@ -97,7 +176,8 @@ export default defineComponent({
         top = maxTop;
       }
       root.style.top = `${top}px`;
-      localStorage.setItem(keyAssistant, top.toString());
+      config.value.pos = top;
+      appStore().save();
     }
     onMounted(async () => {
       recoverAssistantTop();
@@ -115,6 +195,7 @@ export default defineComponent({
     const rootEl = ref<HTMLDivElement>(null);
     const showBtns = ref(true);
     let autoHideTimer = null;
+    let autoHide = toRaw(config.value.autoHide);
     let isDraging = false;
     return {
       showBtns,
@@ -125,10 +206,32 @@ export default defineComponent({
         clearTimeout(autoHideTimer);
         showBtns.value = true;
       },
+      onContextMenuRoot(event: MouseEvent) {
+        const arr: IUiMenuItem[] = [
+          {
+            name: "auto hide",
+            selected: autoHide,
+            callback: () => {
+              autoHide = !autoHide;
+              config.value.autoHide = autoHide;
+              appStore().save();
+              ga(GA_EventName.MouseMenu, "auto hide");
+              if (!autoHide) {
+                clearTimeout(autoHideTimer);
+                showBtns.value = true;
+              }
+            },
+          },
+        ];
+        ccui.menu.showMenuByMouseEvent(event, arr);
+      },
       onMouseLeaveRoot(event: MouseEvent) {
         if (isDraging) {
           return;
         }
+        if (!autoHide) {
+          return;
+        }
         autoHideTimer = setTimeout(() => {
           showBtns.value = false;
         }, 500);
@@ -174,7 +277,7 @@ export default defineComponent({
 }
 .ad {
   position: fixed;
-  z-index: 99999;
+  //z-index: 99999;
   top: 0px;
   right: 0px;
   display: flex;
diff --git a/cc-inspector/src/scripts/inject-view/store.ts b/cc-inspector/src/scripts/inject-view/store.ts
new file mode 100644
index 0000000..bf930a6
--- /dev/null
+++ b/cc-inspector/src/scripts/inject-view/store.ts
@@ -0,0 +1,31 @@
+import { ref, toRaw } from "vue";
+import { defineStore } from "pinia";
+import profile from "cc-plugin/src/ccp/profile";
+import pluginConfig from "../../../cc-plugin.config";
+export class ConfigData {
+  /**
+   * 用户拖动的位置
+   */
+  pos: number = 0;
+  /**
+   * 是否自动隐藏
+   */
+  autoHide: boolean = true;
+}
+
+export const appStore = defineStore("app", () => {
+  const config = ref<ConfigData>(new ConfigData());
+  return {
+    config,
+    init() {
+      profile.init(new ConfigData(), pluginConfig);
+      const data = profile.load(`${pluginConfig.manifest.name}-assistant.json`) as ConfigData;
+      config.value.autoHide = data.autoHide;
+      config.value.pos = data.pos;
+    },
+    save() {
+      const cfg = toRaw(config.value);
+      profile.save(cfg);
+    },
+  };
+});
diff --git a/cc-inspector/src/scripts/inject-view/web-test.ts b/cc-inspector/src/scripts/inject-view/web-test.ts
index 4a49816..92a5f62 100644
--- a/cc-inspector/src/scripts/inject-view/web-test.ts
+++ b/cc-inspector/src/scripts/inject-view/web-test.ts
@@ -5,13 +5,14 @@ import ccui from "@xuyanfeng/cc-ui";
 import "@xuyanfeng/cc-ui/dist/ccui.css";
 import "@xuyanfeng/cc-ui/iconfont/iconfont.css";
 import CCP from "cc-plugin/src/ccp/entry-render";
+import { createPinia } from "pinia";
 import { createApp } from "vue";
 import pluginConfig from "../../../cc-plugin.config";
 import App from "./app.vue";
-
 export default CCP.init(pluginConfig, {
   ready: function (rootElement: any, args: any) {
     const app = createApp(App);
+    app.use(createPinia());
     app.use(ccui);
     app.mount(rootElement);
   },
diff --git a/cc-inspector/src/scripts/inject/inject-view.ts b/cc-inspector/src/scripts/inject/inject-view.ts
index 7469cfe..8952038 100644
--- a/cc-inspector/src/scripts/inject/inject-view.ts
+++ b/cc-inspector/src/scripts/inject/inject-view.ts
@@ -2,6 +2,7 @@ import ccui from "@xuyanfeng/cc-ui";
 import "@xuyanfeng/cc-ui/dist/ccui.css";
 import "@xuyanfeng/cc-ui/iconfont/iconfont.css";
 import { createApp } from "vue";
+import { createPinia } from "pinia";
 import { DocumentEvent } from "../const";
 import App from "../inject-view/app.vue";
 export class InjectView {
@@ -45,6 +46,7 @@ export class InjectView {
     this.loadCss();
     // vue
     const app = createApp(App);
+    app.use(createPinia());
     // ccui.uiElement.setDoc(document);
     app.use(ccui);
     app.mount(el);
diff --git a/cc-inspector/src/scripts/inject/inspect-list.ts b/cc-inspector/src/scripts/inject/inspect-list.ts
new file mode 100644
index 0000000..9840118
--- /dev/null
+++ b/cc-inspector/src/scripts/inject/inspect-list.ts
@@ -0,0 +1,38 @@
+export class InspectTarget {
+  private list = [];
+
+  /**
+   * 是否启用过滤
+   */
+  enabled: boolean = false;
+  addInspectType(item: any) {
+    if (!this.list.find((el) => item === el)) {
+      this.list.push(item);
+    }
+    console.log(this.list);
+  }
+
+  removeInspectType(item: any) {
+    this.list.splice(this.list.indexOf(item), 1);
+    console.log(this.list);
+  }
+  cleanInspectType() {
+    this.list.length = 0;
+  }
+  isContainInspectType(type: any) {
+    return !!this.list.find((el) => type === el);
+  }
+
+  checkNodeComponentsIsInList(node: any) {
+    const comps = node._components;
+    for (let i = 0; i < comps.length; i++) {
+      const comp = comps[i];
+      if (this.list.find((el) => comp instanceof el)) {
+        return true;
+      }
+    }
+    return false;
+  }
+}
+
+export const inspectTarget = new InspectTarget();
diff --git a/cc-inspector/src/scripts/inject/inspector.ts b/cc-inspector/src/scripts/inject/inspector.ts
index e8b2ad6..b4fc0db 100644
--- a/cc-inspector/src/scripts/inject/inspector.ts
+++ b/cc-inspector/src/scripts/inject/inspector.ts
@@ -8,6 +8,8 @@ import { injectView } from "./inject-view";
 import { getValue, trySetValueWithConfig } from "./setValue";
 import { BuildArrayOptions, BuildImageOptions, BuildObjectOptions, BuildVecOptions } from "./types";
 import { isHasProperty } from "./util";
+import { inspectTarget } from "./inspect-list";
+
 declare const cc: any;
 
 export class Inspector extends InjectEvent {
@@ -200,7 +202,13 @@ export class Inspector extends InjectEvent {
     for (let key in this.inspectorGameMemoryStorage) {
       const item = this.inspectorGameMemoryStorage[key];
       if (item && item.isValid && item instanceof cc.Node) {
-        cb(item);
+        if (inspectTarget.enabled) {
+          if (inspectTarget.checkNodeComponentsIsInList(item)) {
+            cb(item);
+          }
+        } else {
+          cb(item);
+        }
       }
     }
   }