import axios from 'axios'

import { CHAIN_IDS } from './chain'

export const SUBGRAPH_URL: {
  [chain in CHAIN_IDS]: string
} = {
  [CHAIN_IDS.CLOBER_TESTNET]:
    'https://subgraph.satsuma-prod.com/f6a8c4889b7b/clober/v2-subgraph-arbitrum-sepolia/api',
  [CHAIN_IDS.ARBITRUM_SEPOLIA]:
    'https://subgraph.satsuma-prod.com/f6a8c4889b7b/clober/v2-subgraph-arbitrum-sepolia/api',
  // [CHAIN_IDS.BASE]:
  //   'https://subgraph.satsuma-prod.com/f6a8c4889b7b/clober/v2-core-subgraph-base/api',
  [CHAIN_IDS.BERACHAIN_MAINNET]:
    'https://subgraph.satsuma-prod.com/f6a8c4889b7b/clober/v2-subgraph-berachain-mainnet/api',
  [CHAIN_IDS.RISE_SEPOLIA]:
    'https://api.goldsky.com/api/public/project_clsljw95chutg01w45cio46j0/subgraphs/v2-subgraph-rise-sepolia/latest/gn',
  [CHAIN_IDS.GIWA_SEPOLIA]:
    'https://subgraph.giwadex.io/subgraphs/name/v2-subgraph-giwa-sepolia',
  [CHAIN_IDS.MONAD_TESTNET]:
    'https://api.goldsky.com/api/public/project_clsljw95chutg01w45cio46j0/subgraphs/v2-subgraph-monad-testnet/latest/gn',
  [CHAIN_IDS.MONAD_MAINNET]:
    'https://api.goldsky.com/api/public/project_clsljw95chutg01w45cio46j0/subgraphs/v2-subgraph-monad/latest/gn',
  // [CHAIN_IDS.SONIC_MAINNET]:
  //   'https://subgraph.satsuma-prod.com/f6a8c4889b7b/clober/v2-core-subgraph-sonic-mainnet/api',
}

export const FALLBACK_SUBGRAPH_URL: {
  [chain in CHAIN_IDS]: string | undefined
} = {
  [CHAIN_IDS.MONAD_MAINNET]:
    'https://api.subgraph.ormilabs.com/api/public/27ad58eb-e12c-4b7b-8df8-0560d8e26b37/subgraphs/v2-subgraph-monad/latest/gn',
}

export class Subgraph {
  public static async get<T>(
    chainId: CHAIN_IDS,
    operationName: string,
    query: string,
    variables: {},
  ): Promise<T> {
    const timeout = 5000
    const primary = SUBGRAPH_URL[chainId]
    const fallback = FALLBACK_SUBGRAPH_URL[chainId]

    if (!primary) {
      throw new Error('Unsupported chain for subgraph')
    }

    const safePost = async (url: string) => {
      try {
        const res = await axios.post(
          url,
          { query, variables, operationName },
          {
            timeout,
            validateStatus: () => true,
          },
        )
        if (res.data?.errors) {
          return {
            ok: false,
            status: res.status,
            data: res.data,
            error: res.data.errors,
          }
        }
        return { ok: true, status: res.status, data: res.data }
      } catch (err: any) {
        console.error(
          `[${operationName}] Error while fetching from subgraph: ${err.message}`,
        )
        return {
          ok: false,
          status: err.response?.status ?? null,
          data: err.response?.data,
          error: err.message,
        }
      }
    }

    const primaryRes = await safePost(primary)

    if (primaryRes.ok && primaryRes.status === 200) {
      return primaryRes.data
    }

    console.warn(
      `[${operationName}] Primary subgraph endpoint failed with status: ${primaryRes.status}`,
    )

    if ((!primaryRes.ok || primaryRes.status === 429) && fallback) {
      const fallbackRes = await safePost(fallback)

      if (fallbackRes.ok && fallbackRes.status === 200) {
        return fallbackRes.data
      }

      throw new Error(
        fallbackRes.data?.errors ??
          `Fallback failed (status: ${fallbackRes.status})`,
      )
    }

    throw new Error(
      (primaryRes.data as any)?.errors ??
        `Failed with status ${primaryRes.status}`,
    )
  }
}