import type { KVNamespaceListOptions, KVNamespaceListResult, KVNamespacePutOptions } from "@cloudflare/workers-types"
import { Resource as ResourceBase } from "sst"
import Cloudflare from "cloudflare"

export const Resource = new Proxy(
  {},
  {
    get(_target, prop: keyof typeof ResourceBase) {
      const value = ResourceBase[prop]
      // @ts-ignore
      if ("type" in value && value.type === "sst.cloudflare.Kv") {
        const client = new Cloudflare({
          apiToken: ResourceBase.CLOUDFLARE_API_TOKEN.value,
        })
        // @ts-ignore
        const namespaceId = value.namespaceId
        const accountId = ResourceBase.CLOUDFLARE_DEFAULT_ACCOUNT_ID.value
        return {
          get: (k: string | string[]) => {
            const isMulti = Array.isArray(k)
            return client.kv.namespaces
              .bulkGet(namespaceId, {
                keys: Array.isArray(k) ? k : [k],
                account_id: accountId,
              })
              .then((result) => (isMulti ? new Map(Object.entries(result?.values ?? {})) : result?.values?.[k]))
          },
          put: (k: string, v: string, opts?: KVNamespacePutOptions) =>
            client.kv.namespaces.values.update(namespaceId, k, {
              account_id: accountId,
              value: v,
              expiration: opts?.expiration,
              expiration_ttl: opts?.expirationTtl,
              metadata: opts?.metadata,
            }),
          delete: (k: string) =>
            client.kv.namespaces.values.delete(namespaceId, k, {
              account_id: accountId,
            }),
          list: (opts?: KVNamespaceListOptions): Promise<KVNamespaceListResult<unknown, string>> =>
            client.kv.namespaces.keys
              .list(namespaceId, {
                account_id: accountId,
                prefix: opts?.prefix ?? undefined,
              })
              .then((result) => {
                return {
                  keys: result.result,
                  list_complete: true,
                  cacheStatus: null,
                }
              }),
        }
      }
      return value
    },
  },
) as Record<string, any>