pass the endpoint URL

Author: yann300

libs/remix-core-plugin/src/lib/helpers/fetch-etherscan.ts

@@ -1,18 +1,26 @@
-export const fetchContractFromEtherscan = async (plugin, network, contractAddress, targetPath, shouldSetFile = true, etherscanKey?) => {
+
+export type Network = {
+  id: number
+  name: string
+}
+
+export const fetchContractFromEtherscan = async (plugin, endpoint: string | Network, contractAddress, targetPath, shouldSetFile = true, etherscanKey?) => {
   let data
   const compilationTargets = {}
   if (!etherscanKey) etherscanKey = await plugin.call('config', 'getAppParameter', 'etherscan-access-token')
   if (!etherscanKey) etherscanKey = '2HKUX5ZVASZIKWJM8MIQVCRUVZ6JAWT531'
 
   if (etherscanKey) {
-    const endpoint = network.id == 1 ? 'api.etherscan.io' : 'api-' + network.name + '.etherscan.io'
+    if (typeof endpoint === 'object' && endpoint !== null && 'id' in endpoint && 'name' in endpoint) {
+      endpoint = endpoint.id == 1 ? 'api.etherscan.io' : 'api-' + endpoint.name + '.etherscan.io'
+    }
     try {
       data = await fetch('https://' + endpoint + '/api?module=contract&action=getsourcecode&address=' + contractAddress + '&apikey=' + etherscanKey)
       data = await data.json()
       // etherscan api doc https://docs.etherscan.io/api-endpoints/contracts
       if (data.message === 'OK' && data.status === "1") {
         if (data.result.length) {
-          if (data.result[0].SourceCode === '') throw new Error(`contract not verified on Etherscan ${network.name} network`)
+          if (data.result[0].SourceCode === '') throw new Error(`contract not verified on Etherscan ${endpoint}`)
           if (data.result[0].SourceCode.startsWith('{')) {
             data.result[0].SourceCode = JSON.parse(data.result[0].SourceCode.replace(/(?:\r\n|\r|\n)/g, '').replace(/^{{/, '{').replace(/}}$/, '}'))
           }

libs/remix-ui/workspace/src/lib/actions/index.tsx

@@ -29,6 +29,7 @@ export type UrlParametersType = {
   opendir: string,
   blockscout: string,
   ghfolder: string
+  endpoint: string
 }
 
 const basicWorkspaceInit = async (workspaces: { name: string; isGitRepo: boolean; }[], workspaceProvider) => {
@@ -136,39 +137,25 @@ export const initWorkspace = (filePanelPlugin) => async (reducerDispatch: React.
         try {
           let etherscanKey = await plugin.call('config', 'getAppParameter', 'etherscan-access-token')
           if (!etherscanKey) etherscanKey = '2HKUX5ZVASZIKWJM8MIQVCRUVZ6JAWT531'
-          const networks = [
-            { id: 1, name: 'mainnet' },
-            { id: 11155111, name: 'sepolia' }
-          ]
-          let found = false
           const workspaceName = 'code-sample'
           let filePath
           const foundOnNetworks = []
-          for (const network of networks) {
-            const target = `/${network.name}/${contractAddress}`
-            try {
-              data = await fetchContractFromEtherscan(plugin, network, contractAddress, target, false, etherscanKey)
-            } catch (error) {
-              if ((error.message.startsWith('contract not verified on Etherscan') || error.message.startsWith('unable to retrieve contract data')) && network.id !== 5)
-                continue
-              else {
-                if (!found) await basicWorkspaceInit(workspaces, workspaceProvider)
-                break
-              }
-            }
-            found = true
-            foundOnNetworks.push(network.name)
-            if (await workspaceExists(workspaceName)) workspaceProvider.setWorkspace(workspaceName)
-            else await createWorkspaceTemplate(workspaceName, 'code-template')
-            plugin.setWorkspace({ name: workspaceName, isLocalhost: false })
-            dispatch(setCurrentWorkspace({ name: workspaceName, isGitRepo: false }))
-            count = count + (Object.keys(data.compilationTargets)).length
-            for (filePath in data.compilationTargets)
-              await workspaceProvider.set(filePath, data.compilationTargets[filePath]['content'])
-
-            if (data.config) {
-              await workspaceProvider.set('compiler_config.json', JSON.stringify(data.config, null, '\t'))
-            }
+          const endpoint = params.endpoint || 'api.etherscan.io'
+          try {
+            data = await fetchContractFromEtherscan(plugin, endpoint, contractAddress, '', false, etherscanKey)
+          } catch (error) {
+            await basicWorkspaceInit(workspaces, workspaceProvider)
+          }
+          if (await workspaceExists(workspaceName)) workspaceProvider.setWorkspace(workspaceName)
+          else await createWorkspaceTemplate(workspaceName, 'code-template')
+          plugin.setWorkspace({ name: workspaceName, isLocalhost: false })
+          dispatch(setCurrentWorkspace({ name: workspaceName, isGitRepo: false }))
+          count = count + (Object.keys(data.compilationTargets)).length
+          for (filePath in data.compilationTargets)
+            await workspaceProvider.set(filePath, data.compilationTargets[filePath]['content'])
+
+          if (data.config) {
+            await workspaceProvider.set('compiler_config.json', JSON.stringify(data.config, null, '\t'))
           }
 
           plugin.on('filePanel', 'workspaceInitializationCompleted', async () => {