locatorjs-vim/inject.js at main · xiaoliyooo/locatorjs-vim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
console.log('[Locator Vim Injected] Script loaded in page context');

let lastHoveredElement = null;

document.addEventListener(
  'mousemove',
  function (e) {
    if (!e.altKey) {
      if (lastHoveredElement) {
        window.dispatchEvent(
          new CustomEvent('locator-vim-hover', {
            detail: { sourceInfo: null },
          })
        );
        lastHoveredElement = null;
      }
      return;
    }

    const element = e.target;
    if (element === lastHoveredElement) return;

    lastHoveredElement = element;
    const sourceInfo = getSourceInfo(element);

    window.dispatchEvent(
      new CustomEvent('locator-vim-hover', {
        detail: {
          sourceInfo,
          rect: element.getBoundingClientRect(),
        },
      })
    );
  },
  true
);

document.addEventListener(
  'click',
  function (e) {
    if (!e.altKey) return;

    const sourceInfo = getSourceInfo(e.target);
    if (sourceInfo) {
      e.preventDefault();
      e.stopPropagation();

      window.dispatchEvent(
        new CustomEvent('locator-vim-click', {
          detail: { sourceInfo },
        })
      );
    }
  },
  true
);

function getSourceInfo(element) {
  let current = element;

  while (current && current !== document.body) {
    // Check for Vue 3
    if (current.__vueParentComponent) {
      const component = current.__vueParentComponent;
      // console.log('[Locator Vim Injected] Found __vueParentComponent');

      if (component?.type?.__file) {
        // console.log('[Locator Vim Injected] Found __file:', component.type.__file);
        return {
          file: component.type.__file,
          line: 1,
          column: 1,
          framework: 'vue3',
        };
      }

      // parent
      if (component?.parent?.type?.__file) {
        return {
          file: component.parent.type.__file,
          line: 1,
          column: 1,
          framework: 'vue3',
        };
      }
    }

    current = current.parentElement;
  }

  // console.log('[Locator Vim Injected] No source info found');
  return null;
}

console.log('[Locator Vim Injected] Ready! Hold Alt and hover over components.');