tlink/apps/extension/entrypoints/background.ts at 6bff3d46baa1a787cfa729d5e7fd4cbb897e029a · TokenScript-Framework/tlink · 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
96
97
98
99
100
101
102
103
104

export default defineBackground(() => {
  // never mark the function here async
  chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
    if (!sender.tab || !sender.tab.id) {
      return null
    }

    let rpcMethod: string = ""
    let params: any = []

    if (msg.type === "rpc") {
      rpcMethod = msg.data.method
      params = msg.data.params
    } else {
      rpcMethod = msg.type
      if (msg.type === "getConnectedAccount") {
        rpcMethod = "eth_accounts"
      } else if (msg.type === "connect") {
        rpcMethod = "eth_requestAccounts"
      }
      params = [msg?.payload?.txData]
    }

    const targetChainId = msg?.payload?.chainId || "0"

    handleWalletCommunication({
      tabId: sender.tab!.id!,
      rpcMethod,
      params,
      targetChainId
    })
      .then((res) => {
        if (["connect", "getConnectedAccount"].includes(msg.type)) {
          sendResponse(res.result?.[0] || "")
        } else {
          sendResponse(res)
        }
      })
      .catch((err) => {
        console.error("error handling message", err, msg)
      })

    return true
  })

  async function handleWalletCommunication({
    tabId,
    rpcMethod,
    params,
    targetChainId
  }: {
    tabId: number
    rpcMethod: string
    params: any
    targetChainId: string
  }) {
    const resp = await chrome.scripting.executeScript({
      world: "MAIN",
      target: { tabId },
      args: [rpcMethod, params, targetChainId],
      func: async (rpcMethod: string, params: any, targetChainId: string) => {
        try {
          const provider = window.tlink || window.ethereum

          // switch chain if needed
          if (rpcMethod === "eth_sendTransaction") {
            const currentChainId = await provider.request({
              method: "eth_chainId"
            })

            if (!targetChainId.startsWith("0x")) {
              targetChainId = "0x" + parseInt(targetChainId).toString(16)
            }

            if (targetChainId !== "0x0" && currentChainId !== targetChainId) {
              await provider.request({
                method: "wallet_switchEthereumChain",
                params: [{ chainId: targetChainId }]
              })
            }
          }

          const result = await provider.request({
            method: rpcMethod,
            params: ["eth_accounts", "eth_requestAccounts"].includes(rpcMethod)
              ? undefined
              : params
          })

          return { result }
        } catch (e: any) {
          if (e.walk) e = e.walk()

          return {
            error: e
          }
        }
      }
    })

    return resp[0].result
  }
})