import fs from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';

let loaded = false;
let env_map = {};

function unquote(value) {
  const v = String(value);
  if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'"))) {
    return v.slice(1, -1);
  }
  return v;
}

function parse_env_text(text) {
  const out = {};
  const lines = String(text).split(/\r?\n/);

  for (const raw_line of lines) {
    const line = raw_line.trim();
    if (!line) continue;
    if (line.startsWith('#')) continue;

    const idx = line.indexOf('=');
    if (idx <= 0) continue;

    const key = line.slice(0, idx).trim();
    let value = line.slice(idx + 1).trim();

    // 去掉行尾注释：仅在未被引号包裹时生效
    const quoted = (value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"));
    if (!quoted) {
      const sharp = value.indexOf('#');
      if (sharp >= 0) {
        value = value.slice(0, sharp).trim();
      }
    }

    out[key] = unquote(value);
  }

  return out;
}

export function load_env() {
  if (loaded) {
    return;
  }

  const __filename = fileURLToPath(import.meta.url);
  const __dirname = path.dirname(__filename);
  const env_path = path.resolve(__dirname, '../.env');

  const text = fs.readFileSync(env_path, 'utf8');
  env_map = parse_env_text(text);
  loaded = true;
}

export function get_env(key) {
  if (!loaded) {
    load_env();
  }
  return env_map[key];
}

export function get_all_env() {
  if (!loaded) {
    load_env();
  }
  return { ...env_map };
}

load_env();