use std::{
    fs::{remove_dir_all, File},
    io::prelude::*,
};

use anyhow::{Context, Result};
use serde_json::json;

mod common;

use common::{
    fetch_artifact_digest, get_json_output, init, output_to_string, set_test_file_content,
    test_dir_file, test_dir_with_subfolder, wash,
};

const ECHO_WASM: &str = "ghcr.io/wasmcloud/components/http-hello-world-rust:0.1.0";
const LOGGING_PAR: &str = "wasmcloud.azurecr.io/logging:0.9.1";
const LOCAL_REGISTRY: &str = "localhost:5001";

#[test]
#[cfg_attr(not(can_reach_ghcr_io), ignore = "ghcr.io is not reachable")]
fn integration_reg_pull_basic() {
    const SUBFOLDER: &str = "pull_basic";
    let pull_dir = test_dir_with_subfolder(SUBFOLDER);

    let basic_echo = test_dir_file(SUBFOLDER, "basic_echo.wasm");

    let pull_basic = wash()
        .args([
            "pull",
            ECHO_WASM,
            "--destination",
            basic_echo.to_str().unwrap(),
            "--allow-latest",
        ])
        .output()
        .unwrap_or_else(|_| panic!("failed to pull {ECHO_WASM}"));
    assert!(pull_basic.status.success());
    // Very important
    assert!(output_to_string(pull_basic).unwrap().contains('\u{1F6BF}'));

    remove_dir_all(pull_dir).unwrap();
}

#[test]
#[cfg_attr(not(can_reach_ghcr_io), ignore = "ghcr.io is not reachable")]
fn integration_reg_pull_comprehensive() {
    const SUBFOLDER: &str = "pull_comprehensive";
    let pull_dir = test_dir_with_subfolder(SUBFOLDER);

    let comprehensive_echo = test_dir_file(SUBFOLDER, "comprehensive_echo.wasm");
    let comprehensive_logging = test_dir_file(SUBFOLDER, "comprehensive_logging.par.gz");

    let pull_echo_comprehensive = wash()
        .args([
            "pull",
            ECHO_WASM,
            "--destination",
            comprehensive_echo.to_str().unwrap(),
            "--digest",
            "sha256:079275a324c0fcd0c201878f0c158120c4984472215ec3f64eb91ba9ee139f72",
            "--output",
            "json",
        ])
        .output()
        .unwrap_or_else(|_| panic!("failed to pull {ECHO_WASM}"));

    assert!(pull_echo_comprehensive.status.success());
    let output = get_json_output(pull_echo_comprehensive).unwrap();

    let expected_json = json!({"file": comprehensive_echo.to_str().unwrap(), "success": true});

    assert_eq!(output, expected_json);

    let pull_logging_comprehensive = wash()
        .args([
            "pull",
            LOGGING_PAR,
            "--destination",
            comprehensive_logging.to_str().unwrap(),
            "--digest",
            "sha256:169f2764e529c2b57ad20abb87e0854d67bf6f0912896865e2911dee1bf6af98",
            "--output",
            "json",
        ])
        .output()
        .unwrap_or_else(|_| panic!("failed to pull {ECHO_WASM}"));

    assert!(pull_logging_comprehensive.status.success());
    let output = get_json_output(pull_logging_comprehensive).unwrap();

    let expected_json = json!({"file": comprehensive_logging.to_str().unwrap(), "success": true});

    assert_eq!(output, expected_json);

    remove_dir_all(pull_dir).unwrap();
}

// NOTE: This test will fail without a local docker registry running
#[test]
#[cfg_attr(not(can_reach_ghcr_io), ignore = "ghcr.io is not reachable")]
fn integration_reg_push_basic() {
    const SUBFOLDER: &str = "push_basic";
    let push_dir = test_dir_with_subfolder(SUBFOLDER);

    let pull_echo_wasm = test_dir_file(SUBFOLDER, "echo.wasm");

    // Pull echo.wasm for push tests
    wash()
        .args([
            "pull",
            ECHO_WASM,
            "--destination",
            pull_echo_wasm.to_str().unwrap(),
        ])
        .stderr(std::process::Stdio::inherit())
        .stdout(std::process::Stdio::inherit())
        .output()
        .unwrap_or_else(|_| panic!("failed to pull {ECHO_WASM} for push basic"));

    // Push echo.wasm and pull from local registry
    let echo_push_basic = &format!("{LOCAL_REGISTRY}/echo:pushbasic");
    let localregistry_echo_wasm = test_dir_file(SUBFOLDER, "echo_local.wasm");
    let push_echo = wash()
        .args([
            "push",
            echo_push_basic,
            pull_echo_wasm.to_str().unwrap(),
            "--insecure",
        ])
        .stderr(std::process::Stdio::inherit())
        .stdout(std::process::Stdio::inherit())
        .output()
        .expect("failed to push echo.wasm to local registry");
    assert!(
        push_echo.status.success(),
        "failed to push to local registry"
    );

    let pull_local_registry_echo = wash()
        .args([
            "pull",
            echo_push_basic,
            "--insecure",
            "--destination",
            localregistry_echo_wasm.to_str().unwrap(),
        ])
        .stderr(std::process::Stdio::inherit())
        .stdout(std::process::Stdio::inherit())
        .output()
        .expect("failed to pull echo.wasm from local registry");

    assert!(
        pull_local_registry_echo.status.success(),
        "failed to pull echo.wasm from local registry"
    );

    remove_dir_all(push_dir).unwrap();
}

// NOTE: This test will fail without a local docker registry running
#[tokio::test]
#[cfg_attr(
    not(can_reach_wasmcloud_azurecr_io),
    ignore = "wasmcloud.azurecr.io is not reachable"
)]
async fn integration_reg_push_comprehensive() -> Result<()> {
    const SUBFOLDER: &str = "push_comprehensive";
    let push_dir = test_dir_with_subfolder(SUBFOLDER);

    let pull_echo_wasm = test_dir_file(SUBFOLDER, "echo.wasm");
    let pull_logging_par = test_dir_file(SUBFOLDER, "logging.par.gz");

    // Pull echo.wasm and logging.par.gz for push tests
    wash()
        .args([
            "pull",
            ECHO_WASM,
            "--destination",
            pull_echo_wasm.to_str().unwrap(),
        ])
        .output()
        .unwrap_or_else(|_| panic!("failed to pull {ECHO_WASM} for push basic"));
    wash()
        .args([
            "pull",
            LOGGING_PAR,
            "--destination",
            pull_logging_par.to_str().unwrap(),
        ])
        .output()
        .unwrap_or_else(|_| panic!("failed to pull {LOGGING_PAR} for push basic"));

    let config_json = test_dir_file(SUBFOLDER, "config.json");
    let mut config = File::create(config_json.clone()).unwrap();
    config.write_all(b"{}").unwrap();

    let logging_push_all_options = format!("{LOCAL_REGISTRY}/logging:alloptions");
    let push_all_options = wash()
        .args([
            "push",
            &logging_push_all_options,
            pull_logging_par.to_str().unwrap(),
            "--allow-latest",
            "--insecure",
            "--config",
            config_json.to_str().unwrap(),
            "--output",
            "json",
            "--password",
            "supers3cr3t",
            "--user",
            "localuser",
        ])
        .output()
        .unwrap_or_else(|_| panic!("failed to push {LOGGING_PAR} for push comprehensive"));
    assert!(push_all_options.status.success());

    let output = get_json_output(push_all_options).unwrap();

    let expected_digest = fetch_artifact_digest(&logging_push_all_options).await?;
    let expected_json = json!({"url": logging_push_all_options, "digest": expected_digest, "success": true, "tag": "alloptions"});

    assert_eq!(output, expected_json);

    remove_dir_all(push_dir).unwrap();

    Ok(())
}

// NOTE: This test will fail without a local docker registry running
#[tokio::test]
#[cfg_attr(
    not(can_reach_wasmcloud_azurecr_io),
    ignore = "wasmcloud.azurecr.io is not reachable"
)]
async fn integration_reg_config() -> Result<()> {
    //===== Inital project setup and build component artifact
    let test_setup = init(
        /* component_name= */ "hello",
        /* template_name= */ "hello-world-rust",
    )
    .await?;
    let project_dir = test_setup.project_dir.clone();

    let status = test_setup
        .base_command()
        .args(["build"])
        .kill_on_drop(true)
        .status()
        .await
        .context("Failed to build project")?;

    assert!(status.success());
    let unsigned_file = project_dir.join("build/http_hello_world.wasm");
    assert!(unsigned_file.exists(), "unsigned file not found!");
    let signed_file = project_dir.join("build/http_hello_world_s.wasm");
    assert!(signed_file.exists(), "signed file not found!");

    //===== Setup registry config
    let envs = vec![
        ("WASH_REG_URL", LOCAL_REGISTRY),
        ("WASH_REG_USER", "iambatman"),
        ("WASH_REG_PASSWORD", "iamvengeance"),
    ];
    set_test_file_content(
        &project_dir.join("wasmcloud.toml").clone(),
        r#"
        name = "Hello World"
        language = "rust"
        type = "component"
        
        [component]
        claims = ["wasmcloud:httpserver"]

        [registry]
        url = "localhost:5001"
        credentials = "./registry_creds.json"
        "#,
    )
    .await?;

    set_test_file_content(
        &project_dir.join("registry_creds.json").clone(),
        r#"
        {
            "localhost:5001": {
                "username": "iambatman",
                "password": "iamvengeance"
            }
        }
        "#,
    )
    .await?;

    //===== case: Push (with a full artifact url) to test cli args
    let push_url = format!("{LOCAL_REGISTRY}/hello:0.1.0");
    let cmd = wash()
        .args([
            "push",
            push_url.as_str(),
            signed_file.to_str().unwrap(),
            "--allow-latest",
            "--insecure",
            // Even though we're passing a registry url, it should be ignored since we're also specifying a full artifact url
            "--registry",
            "this-is-ignored",
            "--output",
            "json",
            "--password",
            "iamvengeance",
            "--user",
            "iambatman",
        ])
        .envs(envs.clone().into_iter())
        .output()
        .unwrap_or_else(|e| panic!("failed to push artifact {e}"));

    assert!(cmd.status.success());
    // let output = output_to_string(cmd)?;
    // println!("{}", output);
    let output = get_json_output(cmd).unwrap();
    let expected_digest = fetch_artifact_digest(&push_url).await?;
    let expected_json =
        json!({"url": push_url, "digest": expected_digest, "success": true, "tag": "0.1.0"});
    assert_eq!(output, expected_json);

    //===== case: Push (with a repository url) to test cli args
    let push_url = "hello:0.2.0";
    let cmd = wash()
        .args([
            "push",
            push_url,
            signed_file.to_str().unwrap(),
            "--allow-latest",
            "--registry",
            LOCAL_REGISTRY,
            "--insecure",
            "--output",
            "json",
            "--password",
            "iamvengeance",
            "--user",
            "iambatman",
        ])
        .envs(envs.clone().into_iter())
        .stderr(std::process::Stdio::inherit())
        .output()
        .unwrap_or_else(|e| panic!("failed to push artifact {e}"));
    assert!(cmd.status.success());
    let output = get_json_output(cmd).unwrap();
    let expected_url = format!("{LOCAL_REGISTRY}/{push_url}");
    let expected_digest = fetch_artifact_digest(&expected_url).await?;
    let expected_json = json!({"url": expected_url, "digest": expected_digest, "success": true, "tag": "0.2.0", "url": "localhost:5001/hello:0.2.0"});
    assert_eq!(output, expected_json);

    //===== case: Push (with a repository url) to test env vars
    let push_url = "hello:0.3.0";
    let cmd = wash()
        .args([
            "push",
            push_url,
            signed_file.to_str().unwrap(),
            "--allow-latest",
            "--insecure",
            "--output",
            "json",
        ])
        .envs(envs.into_iter())
        .output()
        .unwrap_or_else(|e| panic!("failed to push artifact {e}"));

    assert!(cmd.status.success());
    let output = get_json_output(cmd).unwrap();
    let expected_url = format!("{LOCAL_REGISTRY}/{push_url}");
    let expected_digest = fetch_artifact_digest(&expected_url).await?;
    let expected_json = json!({"url": expected_url, "digest": expected_digest, "success": true, "tag": "0.3.0", "url": "localhost:5001/hello:0.3.0"});
    assert_eq!(output, expected_json);

    //===== case: Push (with a repository url) to test file configuration
    let push_url = "hello:0.4.0";
    let cmd = test_setup
        .base_command()
        .args([
            "push",
            push_url,
            signed_file.to_str().unwrap(),
            "--allow-latest",
            "--insecure",
            "--output",
            "json",
        ])
        .stderr(std::process::Stdio::inherit())
        .output()
        .await
        .unwrap_or_else(|e| panic!("failed to push artifact {e}"));

    assert!(cmd.status.success());
    let output = get_json_output(cmd).unwrap();
    let expected_url = format!("{LOCAL_REGISTRY}/{push_url}");
    let expected_digest = fetch_artifact_digest(&expected_url).await?;
    let expected_json = json!({"url": expected_url, "digest": expected_digest, "success": true, "tag": "0.4.0", "url": "localhost:5001/hello:0.4.0"});
    assert_eq!(output, expected_json);

    Ok(())
}