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

use crate_compile_test::prelude::*;
use crate_compile_test::steps::{TestStep, TestStepFactory};

use super::LinkOutputCheckStep;

pub struct StepFactory;
pub struct Step {
    crate_path: PathBuf,
}

impl StepFactory {
    pub fn new() -> Self {
        StepFactory {}
    }
}

impl TestStepFactory for StepFactory {
    fn initialize(&self, _config: &Config, crate_path: &Path) -> Result<Box<dyn TestStep>> {
        Ok(Box::new(Step {
            crate_path: crate_path.into(),
        }))
    }
}

impl LinkOutputCheckStep for Step {
    fn get_crate_name(&self) -> String {
        self.crate_path.to_string_lossy().into()
    }

    fn get_content(&self, profile: &Profile, path: &str) -> Option<(&[&str], &[&str])> {
        match (profile, path.replace("\\", "/").as_str()) {
            (Profile::Release, "examples/intrinsics") => Some((
                &[
                    "tail call signext i32 @llvm.nvvm.read.ptx.sreg.ntid.y()",
                    "tail call signext i32 @llvm.nvvm.read.ptx.sreg.ctaid.y()",
                    "tail call signext i32 @llvm.nvvm.read.ptx.sreg.tid.y()",
                    "tail call signext i32 @llvm.nvvm.read.ptx.sreg.ntid.x()",
                    "tail call signext i32 @llvm.nvvm.read.ptx.sreg.ctaid.x()",
                    "tail call signext i32 @llvm.nvvm.read.ptx.sreg.tid.x()",
                    "declare signext i32 @vprintf(i8* nocapture readonly, i8*)",
                ],
                &["_ZN7example5image", "example..image"],
            )),
            (Profile::Debug, "examples/intrinsics") => Some((
                &[
                    "call signext i32 @llvm.nvvm.read.ptx.sreg.ntid.y()",
                    "call signext i32 @llvm.nvvm.read.ptx.sreg.ctaid.y()",
                    "call signext i32 @llvm.nvvm.read.ptx.sreg.tid.y()",
                    "call signext i32 @llvm.nvvm.read.ptx.sreg.ntid.x()",
                    "call signext i32 @llvm.nvvm.read.ptx.sreg.ctaid.x()",
                    "call signext i32 @llvm.nvvm.read.ptx.sreg.tid.x()",
                    "_ZN7example5image14Image",
                    "declare signext i32 @vprintf(i8*, i8*)",
                    "declare i8* @malloc(i64)",
                    "declare void @free(i8*)",
                ],
                &["example..image"],
            )),

            (Profile::Release, "examples/depenencies") => Some((
                &[
                    "define ptx_kernel void @dummy_math_kernel(",
                    "define ptx_kernel void @dummy_utils_kernel(",
                    "define ptx_kernel void @top_level_kernel(",
                    "double @dummy_square(double)",
                    "double @dummy_mul(double, double)",
                    "double @dummy_mul_inner(double, double)",
                    "double @dummy_mul_2(double)",
                ],
                &["constant <{ [8 x i8] }>"],
            )),
            (Profile::Debug, "examples/depenencies") => Some((
                &[
                    "define ptx_kernel void @dummy_math_kernel(",
                    "define ptx_kernel void @dummy_utils_kernel(",
                    "define ptx_kernel void @top_level_kernel(",
                    "double @dummy_square(double)",
                    "double @dummy_mul(double, double)",
                    "double @dummy_mul_inner(double, double)",
                    "double @dummy_mul_2(double)",
                ],
                &[],
            )),

            _ => None,
        }
    }
}

impl TestStep for Step {
    fn execute(&self, config: &Config, build_path: &Path) -> Result<()> {
        self.check_output(
            config,
            &match config.profile {
                Profile::Release => build_path
                    .join("nvptx64-nvidia-cuda")
                    .join("release")
                    .join("deps")
                    .join("example.ll"),

                Profile::Debug => build_path
                    .join("nvptx64-nvidia-cuda")
                    .join("debug")
                    .join("deps")
                    .join("example.ll"),
            },
        )
    }
}