import * as core from '@actions/core'
import * as tc from '@actions/tool-cache'
import * as path from 'path'
import {OWNER, REPO, TOOL_CACHE_NAME} from '../utils/utils'
import {Architecture, Platform} from '../utils/platforms'
import {validateChecksum} from './checksum/checksum'

import * as fs from 'fs'
import * as util from 'util'

const readdir = util.promisify(fs.readdir)

export function tryGetFromToolCache(
  arch: Architecture,
  version: string
): string | undefined {
  core.debug(`Trying to get uv from tool cache for ${version}...`)
  const cachedVersions = tc.findAllVersions(TOOL_CACHE_NAME, arch)
  core.debug(`Cached versions: ${cachedVersions}`)
  return tc.find(TOOL_CACHE_NAME, version, arch)
}

export async function downloadVersion(
  platform: Platform,
  arch: Architecture,
  version: string,
  checkSum: string | undefined,
  githubToken: string | undefined
): Promise<string> {
  const artifact = `uv-${arch}-${platform}`
  let downloadUrl = `https://github.com/${OWNER}/${REPO}/releases/download/${version}/${artifact}`
  if (platform === 'pc-windows-msvc') {
    downloadUrl += '.zip'
  } else {
    downloadUrl += '.tar.gz'
  }
  core.info(`Downloading uv from "${downloadUrl}" ...`)

  const downloadPath = await tc.downloadTool(
    downloadUrl,
    undefined,
    githubToken
  )
  await validateChecksum(checkSum, downloadPath, arch, platform, version)

  let extractedDir: string
  if (platform === 'pc-windows-msvc') {
    extractedDir = await tc.extractZip(downloadPath)
    const files = await readdir(extractedDir)
    core.info(
      `Contents of extracted directory ${extractedDir}: ${files.join(', ')}`
    )
    const uvDir = path.join(extractedDir, artifact)
    const uvfiles = await readdir(uvDir)
    core.info(`Contents of directory ${uvDir}: ${uvfiles.join(', ')}`)
  } else {
    extractedDir = await tc.extractTar(downloadPath)
  }

  return await tc.cacheDir(
    path.join(extractedDir, artifact),
    TOOL_CACHE_NAME,
    version,
    arch
  )
}