import { existsSync, readFileSync, writeFileSync } from 'node:fs'
import { resolve } from 'node:path'
import { argv, exit } from 'node:process'
import { parse } from 'dotenv'

// Check if the environment file name is provided as a command-line argument
const envFileName = argv[2]
const envName = argv[3]
if (!envFileName) {
  console.error('Please provide the environment file name as the second parameter.')
  exit(1)
}
if (!envName) {
  console.error('Please provide the worker name as the third parameter.')
  exit(1)
}

// Resolve and check the existence of the .env file
const envFilePath = resolve(envFileName)
if (!existsSync(envFilePath)) {
  console.error(`Failed to read the environment file at ${envFilePath}.`)
  exit(1)
}

let envContent
try {
  envContent = readFileSync(envFilePath, 'utf8')
}
catch (error) {
  console.error(`Failed to read the environment file at ${envFilePath}:`, error)
  exit(1)
}

// Use dotenv.parse to convert the file content into an object
const customEnv = parse(envContent)

console.log('Environment file:', envFileName)
console.log('Worker Name:', envName)
// Avoid printing secret values to stdout (this file is used for local workflows too).
console.log('Environment variable keys', Object.keys(customEnv))

function escapeTomlBasicString(value) {
  let out = ''
  for (const ch of String(value)) {
    switch (ch) {
      case '\\': out += '\\\\'; break
      case '"': out += '\\"'; break
      case '\n': out += '\\n'; break
      case '\r': out += '\\r'; break
      case '\t': out += '\\t'; break
      default: {
        const code = ch.codePointAt(0)
        // Escape other control characters to keep the TOML valid.
        if (code != null && code < 0x20)
          out += `\\u${code.toString(16).padStart(4, '0')}`
        else
          out += ch
      }
    }
  }
  return out
}

function formatTomlKey(key) {
  // Wrangler expects TOML: use bare keys when safe, otherwise quote.
  if (/^[A-Za-z0-9_-]+$/.test(key))
    return key
  return `"${escapeTomlBasicString(key)}"`
}

// wrangler config file path
const wranglerPath = resolve('./wrangler.toml')

// add variable to wrangler.toml add [vars] section
const wranglerFile = readFileSync(wranglerPath, 'utf8')
const wranglerFileLines = wranglerFile.split('\n')
let varsIndex = wranglerFileLines.findIndex(line => line.includes('[vars]'))
if (varsIndex === -1) {
  wranglerFileLines.push('[vars]')
  varsIndex = wranglerFileLines.length
}
else {
  // Find the index of the next section after [vars]
  let nextSectionIndex = wranglerFileLines.findIndex((line, index) => {
    return index > varsIndex && line.startsWith('[')
  })

  if (nextSectionIndex !== -1) {
    // Remove all lines between [vars] and the next section
    wranglerFileLines.splice(varsIndex + 1, nextSectionIndex - varsIndex - 1)
  }
  else {
    // If no next section, remove everything after [vars]
    wranglerFileLines.splice(varsIndex + 1)
  }
}

for (const key in customEnv) {
  const tomlKey = formatTomlKey(key)
  const tomlValue = escapeTomlBasicString(customEnv[key])
  wranglerFileLines.splice(varsIndex + 1, 0, `${tomlKey} = "${tomlValue}"`)
  varsIndex++ // Move the index forward after each insertion
}
const newWranglerFile = wranglerFileLines.join('\n')
// write new wrangler.toml
try {
  writeFileSync(wranglerPath, newWranglerFile)
  console.log('New wrangler.toml file written successfully')
}
catch (error) {
  console.error('Failed to upload secrets:', error)
  exit(1)
}