AstralRinth/apps/frontend/src/server/utils/api-client.ts

import {
	type AuthConfig,
	AuthFeature,
	CanaryCookieFeature,
	type FeatureConfig,
	LABRINTH_CANARY_COOKIE,
	type NuxtClientConfig,
	NuxtModrinthClient,
} from '@modrinth/api-client'
import type { H3Event } from 'h3'

async function getRateLimitKeyFromSecretsStore(): Promise<string | undefined> {
	try {
		const mod = 'cloudflare:workers'
		const { env } = await import(/* @vite-ignore */ mod)
		return await env.RATE_LIMIT_IGNORE_KEY?.get()
	} catch {
		return undefined
	}
}

export interface ServerModrinthClientOptions {
	event?: H3Event
	authToken?: string
	canaryCookie?: boolean
}

export function useServerModrinthClient(options?: ServerModrinthClientOptions): NuxtModrinthClient {
	const config = useRuntimeConfig(options?.event)
	const apiBaseUrl = (config.apiBaseUrl || config.public.apiBaseUrl).replace('/v2/', '/')

	const features = []

	if (options?.authToken) {
		features.push(
			new AuthFeature({
				token: options.authToken,
				tokenPrefix: '',
			} as AuthConfig as FeatureConfig),
		)
	}

	if (options?.canaryCookie) {
		features.push(new CanaryCookieFeature({ getCookie: () => LABRINTH_CANARY_COOKIE }))
	}

	const clientConfig: NuxtClientConfig = {
		labrinthBaseUrl: apiBaseUrl,
		rateLimitKey: config.rateLimitKey || getRateLimitKeyFromSecretsStore,
		features,
	}

	return new NuxtModrinthClient(clientConfig)
}