import { fetchNpmDistTags } from "./npm-dist-tags"

const DEFAULT_PACKAGE_NAME = "oh-my-opencode"
const PRIORITIZED_TAGS = ["latest", "beta", "next"] as const

function getFallbackEntry(version: string, packageName: string): string {
  const prereleaseMatch = version.match(/-([a-zA-Z][a-zA-Z0-9-]*)(?:\.|$)/)
  if (prereleaseMatch) {
    return `${packageName}@${prereleaseMatch[1]}`
  }

  return packageName
}

export async function getPluginNameWithVersion(
  currentVersion: string,
  packageName: string = DEFAULT_PACKAGE_NAME
): Promise<string> {
  const distTags = await fetchNpmDistTags(packageName)


  if (distTags) {
    const allTags = new Set([...PRIORITIZED_TAGS, ...Object.keys(distTags)])
    for (const tag of allTags) {
      if (distTags[tag] === currentVersion) {
        return `${packageName}@${tag}`
      }
    }
  }

  return getFallbackEntry(currentVersion, packageName)
}