use super::{use_filter, use_valid_fields};
use anyhow::Result;
use serde_yaml::{self, Mapping};

pub fn use_script(
  script: String,
  config: Mapping,
  valid: Vec<String>,
) -> Result<(Mapping, Vec<(String, String)>)> {
  use rquickjs::{Context, Func, Runtime};
  use std::sync::{Arc, Mutex};

  let runtime = Runtime::new().unwrap();
  let context = Context::full(&runtime).unwrap();
  let outputs = Arc::new(Mutex::new(vec![]));

  let copy_outputs = outputs.clone();
  let result = context.with(|ctx| -> Result<Mapping> {
    ctx.globals().set(
      "__verge_log__",
      Func::from(move |level: String, data: String| {
        let mut out = copy_outputs.lock().unwrap();
        out.push((level, data));
      }),
    )?;

    ctx.eval(
      r#"var console = Object.freeze({
        log(data){__verge_log__("log",JSON.stringify(data))}, 
        info(data){__verge_log__("info",JSON.stringify(data))}, 
        error(data){__verge_log__("error",JSON.stringify(data))},
        debug(data){__verge_log__("debug",JSON.stringify(data))},
      });"#,
    )?;

    let config_str = serde_json::to_string(&config)?;

    let code = format!("\n{script}\n;\nJSON.stringify(main({config_str})||'')");
    let result: String = ctx.eval(code.as_str())?;
    if result == "\"\"" {
      anyhow::bail!("main function should return object");
    }
    Ok(serde_json::from_str::<Mapping>(result.as_str())?)
  });

  let mut out = outputs.lock().unwrap();
  match result {
    Ok(config) => {
      let valid = use_valid_fields(valid);
      let config = use_filter(config, valid);
      Ok((config, out.to_vec()))
    }
    Err(err) => {
      out.push(("error".into(), err.to_string()));
      Ok((config, out.to_vec()))
    }
  }
}

#[test]
fn test_script() {
  let script = r#"
    function main(config) {
      if (Array.isArray(config.rules)) {
        config.rules = [...config.rules, "add"];
      }
      console.log(config);
      config.proxies = ["111"];
      return config;
    }
  "#;

  let config = r#"
    rules:
      - 111
      - 222
    tun:
      enable: false
    dns:
      enable: false
  "#;

  let config = serde_yaml::from_str(config).unwrap();
  let (config, results) = use_script(
    script.into(),
    config,
    vec!["tun"].iter().map(|s| s.to_string()).collect(),
  )
  .unwrap();

  let config_str = serde_yaml::to_string(&config).unwrap();

  println!("{config_str}");

  dbg!(results);
}