[add-time-to-vault-reqest] added

stakewise · Feb 6, 2025 · 4efd645 · 4efd645
1 parent 0fab51d
commit 4efd645
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 4 deletions.
diff --git a/scripts/generateGraphqlExports/queryTemplate.ts b/scripts/generateGraphqlExports/queryTemplate.ts
@@ -6,11 +6,12 @@ const queryImport = [
 const queryTemplate = `
 
 const fetch{QueryName} = <ModifiedData = {QueryName}Payload>(
-  { variables, url, modifyResult }: FetchCodegenInput<{QueryName}Payload, {QueryName}Variables, ModifiedData>
+  { variables, url, withTime, modifyResult }: FetchCodegenInput<{QueryName}Payload, {QueryName}Variables, ModifiedData>
 ) => (
   graphqlFetch<{QueryName}Payload, {QueryName}Variables, ModifiedData>({
     url,
     query,
+    withTime,
     variables,
     modifyResult,
   })

diff --git a/src/methods/vault/requests/getVault/index.ts b/src/methods/vault/requests/getVault/index.ts
@@ -7,10 +7,11 @@ import modifyVault from './modifyVault'
 type GetVaultInput = {
   options: StakeWise.Options
   vaultAddress: string
+  withTime?: boolean
 }
 
 const getVault = (input: GetVaultInput) => {
-  const { options, vaultAddress } = input
+  const { options, vaultAddress, withTime } = input
 
   validateArgs.address({ vaultAddress })
 
@@ -19,6 +20,7 @@ const getVault = (input: GetVaultInput) => {
     variables: {
       address: vaultAddress.toLowerCase(),
     },
+    withTime,
     modifyResult: (data) => modifyVault({ data, network: options.network }),
   })
 }

diff --git a/src/modules/gql-module/graphqlFetch.ts b/src/modules/gql-module/graphqlFetch.ts
@@ -10,7 +10,7 @@ type ExtendedFetchInput<Data, Variables, ModifiedData> = FetchInput<Data, Variab
 const graphqlFetch = <Data, Variables, ModifiedData>(
   options: ExtendedFetchInput<Data, Variables, ModifiedData>
 ): AbortRequest<Data, ModifiedData> => {
-  const { url, query, variables, retryCount = 0, modifyResult } = options
+  const { url, query, variables, withTime, retryCount = 0, modifyResult } = options
 
   const operationName = query
     .replace(/^(query|mutation)\s/, '')
@@ -19,7 +19,11 @@ const graphqlFetch = <Data, Variables, ModifiedData>(
 
   const currentUrl = getRequestUrl(url)
   const opName = operationName ? `?opName=${operationName}` : ''
-  const requestUrl = `${currentUrl}${opName}`
+
+  let time = withTime ? `t=${new Date().getTime()}` : ''
+  time = opName ? `&${time}` : `?${time}`
+
+  const requestUrl = `${currentUrl}${opName}${time}`
 
   return new AbortRequest<Data, ModifiedData>(requestUrl, {
     method: 'POST',

diff --git a/src/modules/gql-module/types.ts b/src/modules/gql-module/types.ts
@@ -1,12 +1,14 @@
 export type FetchInput<Data, Variables, ModifiedData> = {
   url: string | ReadonlyArray<string>
   query: string
+  withTime?: boolean
   variables?: Variables
   modifyResult?: (data: Data) => ModifiedData
 }
 
 export type FetchCodegenInput<Data, Variables, ModifiedData> = {
   url: string | ReadonlyArray<string>
+  withTime?: boolean
   variables?: Variables
   modifyResult?: (data: Data) => ModifiedData
 }