remove duplicates (#186)

* refactor checkRestakeOperatorsManagerAccess

* update subgraphUrl logic

Author: dfkadyr

codegen.ts

@@ -16,10 +16,12 @@ if (process.env.NETWORK === 'gnosis') {
 const config = configs[network]
 const subgraphIndex = Number(process.env.SUBGRAPH_INDEX || 0)
 
-const subgraphUrl = Array.isArray(config.api.subgraph)
+let subgraphUrl = Array.isArray(config.api.subgraph)
   ? config.api.subgraph[subgraphIndex]
   : config.api.subgraph
 
+subgraphUrl = subgraphUrl.replace('prod', 'stage')
+
 const urls: Record<string, string> = {
   backend: config.api.backend,
   subgraph: subgraphUrl,

src/StakeWiseSDK.ts

@@ -44,6 +44,16 @@ class StakeWiseSDK {
     this.provider = provider
     this.contracts = contracts
 
+    if (options.endpoints?.subgraph) {
+      // @ts-ignore: It's better to just overwrite
+      this.config.api.subgraph = options.endpoints.subgraph
+    }
+
+    if (options.endpoints?.api) {
+      // @ts-ignore: It's better to just overwrite
+      this.config.api.backend = options.endpoints.api
+    }
+
     const argsForMethods = { options, contracts, provider }
 
     this.utils = methods.createUtils(argsForMethods)

src/methods/vault/transactions/createEigenPod/checkAccess.ts

@@ -1,9 +1,10 @@
 import { commonLogic } from './common'
-import checkAccess from './checkAccess'
 import type { CreateEigenPod } from './types'
 import createEigenPodGas from './createEigenPodGas'
 import createEigenPodEncode from './createEigenPodEncode'
 
+import { checkRestakeOperatorsManagerAccess as checkAccess } from '../util'
+
 
 const createEigenPod: CreateEigenPod = async (values) => {
   const { provider, userAddress } = values

src/methods/vault/transactions/createEigenPod/index.ts

@@ -1,9 +1,10 @@
 import { commonLogic } from './common'
-import checkAccess from './checkAccess'
 import type { SetEigenPodOperator } from './types'
 import setEigenPodOperatorGas from './setEigenPodOperatorGas'
 import setEigenPodOperatorEncode from './setEigenPodOperatorEncode'
 
+import { checkRestakeOperatorsManagerAccess as checkAccess } from '../util'
+
 
 const setEigenPodOperator: SetEigenPodOperator = async (values) => {
   const { provider, userAddress, operatorAddress } = values

src/methods/vault/transactions/setEigenPodOperator/checkAccess.ts

@@ -1,8 +1,10 @@
 import { commonLogic } from './common'
-import checkAccess from './checkAccess'
 import updateEigenPodOperatorGas from './updateEigenPodOperatorGas'
 import updateEigenPodOperatorEncode from './updateEigenPodOperatorEncode'
 import type { UpdateEigenPodOperator } from './types'
+
+import { checkRestakeOperatorsManagerAccess as checkAccess } from '../util'
+
 import { eigenPodOwnerMulticall } from '../../../../contracts'
 
 

src/methods/vault/transactions/setEigenPodOperator/index.ts

@@ -1,7 +1,9 @@
 import type { CheckInput } from './types'
 
 
-const checkRestakeOperatorsManagerAccess = async ({ userAddress, vaultAddress, contracts }: CheckInput) => {
+type Action<Input, Output> = (props: Input) => Promise<Output>
+
+const checkAccess = async ({ userAddress, vaultAddress, contracts }: CheckInput) => {
   try {
     const restakingVaultContract = contracts.helpers.createVault({
       options: { isRestake: true },
@@ -18,5 +20,22 @@ const checkRestakeOperatorsManagerAccess = async ({ userAddress, vaultAddress, c
   catch {}
 }
 
+const checkRestakeOperatorsManagerAccess = <Input, Output>(action: Action<Input, Output>) => (
+  async (values: Input) => {
+    try {
+      const result = await action(values)
+
+      return result
+    }
+    catch (actionError) {
+      return checkAccess(values as unknown as CheckInput)
+        .then(
+          () => Promise.reject(actionError),
+          (error) => Promise.reject(error)
+        )
+    }
+  }
+)
+
 
 export default checkRestakeOperatorsManagerAccess