import "../../_dnt.polyfills.js";
import { ConnectionSpec } from "../../nets/mod.js"
import { Connect, WsConnection } from "../../rpc/mod.js"
import { tryGetEnv } from "../../util/mod.js"
import { DevnetConnection } from "./DevnetConnection.js"

export function detectConnect(
  defaultConnectionSpec: ConnectionSpec,
  targets: Record<string, ConnectionSpec>,
): Connect {
  let connectionSpec = defaultConnectionSpec
  const targetName = tryGetEnv("CAPI_TARGET")
  if (targetName) {
    const target = targets[targetName]
    if (target) connectionSpec = target
  }
  switch (connectionSpec.type) {
    case "WsConnection":
      return WsConnection.bind(connectionSpec.discovery)
    case "DevnetConnection":
      return DevnetConnection.bind(connectionSpec.discovery)
  }
}