use std::{env, path::Path};

use base64::{prelude::BASE64_STANDARD, Engine};
use oxc_allocator::Allocator;
use oxc_codegen::{Codegen, CodegenOptions};
use oxc_parser::Parser;
use oxc_span::SourceType;

// Instruction:
// 1. create a `test.js`
// 2. run `cargo run -p oxc_codegen --example sourcemap`

fn main() -> std::io::Result<()> {
    let name = env::args().nth(1).unwrap_or_else(|| "test.js".to_string());
    let path = Path::new(&name);
    let source_text = std::fs::read_to_string(path)?;
    let source_type = SourceType::from_path(path).unwrap();
    let allocator = Allocator::default();
    let ret = Parser::new(&allocator, &source_text, source_type).parse();

    if !ret.errors.is_empty() {
        for error in ret.errors {
            let error = error.with_source_code(source_text.clone());
            println!("{error:?}");
        }
        return Ok(());
    }

    let codegen_options = CodegenOptions::default();
    let mut codegen = Codegen::<false>::new(source_text.len(), codegen_options);
    let content = codegen.with_sourcemap(&source_text, "").build(&ret.program);
    let map = codegen.into_sourcemap();

    let mut buff = vec![];
    map.to_writer(&mut buff).unwrap();
    let result = String::from_utf8(buff).unwrap();
    let hash = BASE64_STANDARD.encode(format!(
        "{}\0{}{}\0{}",
        content.len(),
        content,
        result.len(),
        result
    ));
    println!("https://evanw.github.io/source-map-visualization/#{hash}");
    Ok(())
}