use std::fs::File;
use std::os::unix::io::{FromRawFd, RawFd};
use std::process::Command;
use std::sync::Arc;
use std::thread;

use log::{error, info};
use pci_types::PciAddress;

use crate::driver_interface;
use crate::pci::cap::Capability as PciCapability;
use crate::pci::PciFunc;
use crate::State;

pub struct DriverHandler {
    addr: PciAddress,
    capabilities: Vec<PciCapability>,

    state: Arc<State>,
}

impl DriverHandler {
    pub fn spawn(
        state: Arc<State>,
        func: driver_interface::PciFunction,
        capabilities: Vec<PciCapability>,
        args: &[String],
    ) {
        let subdriver_args = driver_interface::SubdriverArguments { func };

        let mut args = args.iter();
        if let Some(program) = args.next() {
            let mut command = Command::new(program);
            for arg in args {
                if arg.starts_with("$") {
                    panic!("support for $VARIABLE has been removed. use pcid_interface instead");
                }
                command.arg(arg);
            }

            info!("PCID SPAWN {:?}", command);

            // TODO: libc wrapper?
            let [fds1, fds2] = unsafe {
                let mut fds1 = [0 as libc::c_int; 2];
                let mut fds2 = [0 as libc::c_int; 2];

                assert_eq!(
                    libc::pipe(fds1.as_mut_ptr()),
                    0,
                    "pcid: failed to create pcid->client pipe"
                );
                assert_eq!(
                    libc::pipe(fds2.as_mut_ptr()),
                    0,
                    "pcid: failed to create client->pcid pipe"
                );

                [fds1.map(|c| c as usize), fds2.map(|c| c as usize)]
            };

            let [pcid_to_client_read, pcid_to_client_write] = fds1;
            let [pcid_from_client_read, pcid_from_client_write] = fds2;

            let envs = vec![
                ("PCID_TO_CLIENT_FD", format!("{}", pcid_to_client_read)),
                ("PCID_FROM_CLIENT_FD", format!("{}", pcid_from_client_write)),
            ];

            match command.envs(envs).spawn() {
                Ok(mut child) => {
                    let driver_handler = DriverHandler {
                        addr: func.addr,
                        state,
                        capabilities,
                    };
                    let _handle = thread::spawn(move || {
                        driver_handler.handle_spawn(
                            pcid_to_client_write,
                            pcid_from_client_read,
                            subdriver_args,
                        );
                    });
                    // FIXME this currently deadlocks as pcid doesn't daemonize
                    //state.threads.lock().unwrap().push(handle);
                    match child.wait() {
                        Ok(_status) => (),
                        Err(err) => error!("pcid: failed to wait for {:?}: {}", command, err),
                    }
                }
                Err(err) => error!("pcid: failed to execute {:?}: {}", command, err),
            }
        }
    }

    fn respond(
        &mut self,
        request: driver_interface::PcidClientRequest,
        args: &driver_interface::SubdriverArguments,
    ) -> driver_interface::PcidClientResponse {
        use crate::pci::cap::{MsiCapability, MsixCapability};
        use driver_interface::*;

        let func = PciFunc {
            pci: &self.state.pcie,
            addr: self.addr,
        };

        match request {
            PcidClientRequest::RequestCapabilities => PcidClientResponse::Capabilities(
                self.capabilities
                    .iter()
                    .map(|capability| capability.clone())
                    .collect::<Vec<_>>(),
            ),
            PcidClientRequest::RequestConfig => PcidClientResponse::Config(args.clone()),
            PcidClientRequest::RequestFeatures => PcidClientResponse::AllFeatures(
                self.capabilities
                    .iter()
                    .filter_map(|capability| match capability {
                        PciCapability::Msi(_) => Some(PciFeature::Msi),
                        PciCapability::MsiX(_) => Some(PciFeature::MsiX),
                        _ => None,
                    })
                    .collect(),
            ),
            PcidClientRequest::EnableFeature(feature) => match feature {
                PciFeature::Msi => {
                    if let Some(msix_capability) = self
                        .capabilities
                        .iter_mut()
                        .find_map(|capability| capability.as_msix_mut())
                    {
                        // If MSI-X is supported disable it before enabling MSI as they can't be
                        // active at the same time.
                        unsafe {
                            msix_capability.set_msix_enabled(false);
                            msix_capability.write_a(&func);
                        }
                    }

                    let capability: &mut MsiCapability = match self
                        .capabilities
                        .iter_mut()
                        .find_map(|capability| capability.as_msi_mut())
                    {
                        Some(capability) => capability,
                        None => {
                            return PcidClientResponse::Error(
                                PcidServerResponseError::NonexistentFeature(feature),
                            )
                        }
                    };
                    unsafe {
                        capability.set_enabled(true);
                        capability.write_message_control(&func);
                    }
                    PcidClientResponse::FeatureEnabled(feature)
                }
                PciFeature::MsiX => {
                    if let Some(msi_capability) = self
                        .capabilities
                        .iter_mut()
                        .find_map(|capability| capability.as_msi_mut())
                    {
                        // If MSI is supported disable it before enabling MSI-X as they can't be
                        // active at the same time.
                        unsafe {
                            msi_capability.set_enabled(false);
                            msi_capability.write_message_control(&func);
                        }
                    }

                    let capability: &mut MsixCapability = match self
                        .capabilities
                        .iter_mut()
                        .find_map(|capability| capability.as_msix_mut())
                    {
                        Some(capability) => capability,
                        None => {
                            return PcidClientResponse::Error(
                                PcidServerResponseError::NonexistentFeature(feature),
                            )
                        }
                    };
                    unsafe {
                        capability.set_msix_enabled(true);
                        capability.write_a(&func);
                    }
                    PcidClientResponse::FeatureEnabled(feature)
                }
            },
            PcidClientRequest::FeatureInfo(feature) => PcidClientResponse::FeatureInfo(
                feature,
                match feature {
                    PciFeature::Msi => {
                        if let Some(info) = self
                            .capabilities
                            .iter()
                            .find_map(|capability| capability.as_msi())
                        {
                            PciFeatureInfo::Msi(msi::MsiInfo {
                                log2_multiple_message_capable: info.multi_message_capable(),
                                is_64bit: info.has_64_bit_addr(),
                                has_per_vector_masking: info.is_pvt_capable(),
                            })
                        } else {
                            return PcidClientResponse::Error(
                                PcidServerResponseError::NonexistentFeature(feature),
                            );
                        }
                    }
                    PciFeature::MsiX => {
                        if let Some(info) = self
                            .capabilities
                            .iter()
                            .find_map(|capability| capability.as_msix())
                        {
                            PciFeatureInfo::MsiX(msi::MsixInfo {
                                table_bar: info.table_bir(),
                                table_offset: info.table_offset(),
                                table_size: info.table_size(),
                                pba_bar: info.pba_bir(),
                                pba_offset: info.pba_offset(),
                            })
                        } else {
                            return PcidClientResponse::Error(
                                PcidServerResponseError::NonexistentFeature(feature),
                            );
                        }
                    }
                },
            ),
            PcidClientRequest::SetFeatureInfo(info_to_set) => match info_to_set {
                SetFeatureInfo::Msi(info_to_set) => {
                    if let Some(info) = self
                        .capabilities
                        .iter_mut()
                        .find_map(|capability| capability.as_msi_mut())
                    {
                        if let Some(mme) = info_to_set.multi_message_enable {
                            if info.multi_message_capable() < mme || mme > 0b101 {
                                return PcidClientResponse::Error(
                                    PcidServerResponseError::InvalidBitPattern,
                                );
                            }
                            info.set_multi_message_enable(mme);
                        }
                        if let Some(message_addr_and_data) = info_to_set.message_address_and_data {
                            let message_addr = message_addr_and_data.addr;
                            if message_addr & 0b11 != 0 {
                                return PcidClientResponse::Error(
                                    PcidServerResponseError::InvalidBitPattern,
                                );
                            }
                            info.set_message_address(message_addr as u32);
                            info.set_message_upper_address((message_addr >> 32) as u32);
                            if message_addr_and_data.data & ((1 << info.multi_message_enable()) - 1)
                                != 0
                            {
                                return PcidClientResponse::Error(
                                    PcidServerResponseError::InvalidBitPattern,
                                );
                            }
                            info.set_message_data(
                                message_addr_and_data
                                    .data
                                    .try_into()
                                    .expect("pcid: MSI message data too big"),
                            );
                        }
                        if let Some(mask_bits) = info_to_set.mask_bits {
                            info.set_mask_bits(mask_bits);
                        }
                        unsafe {
                            info.write_all(&func);
                        }
                        PcidClientResponse::SetFeatureInfo(PciFeature::Msi)
                    } else {
                        return PcidClientResponse::Error(
                            PcidServerResponseError::NonexistentFeature(PciFeature::Msi),
                        );
                    }
                }
                SetFeatureInfo::MsiX { function_mask } => {
                    if let Some(info) = self
                        .capabilities
                        .iter_mut()
                        .find_map(|capability| capability.as_msix_mut())
                    {
                        if let Some(mask) = function_mask {
                            info.set_function_mask(mask);
                            unsafe {
                                info.write_a(&func);
                            }
                        }
                        PcidClientResponse::SetFeatureInfo(PciFeature::MsiX)
                    } else {
                        return PcidClientResponse::Error(
                            PcidServerResponseError::NonexistentFeature(PciFeature::MsiX),
                        );
                    }
                }
            },
            PcidClientRequest::ReadConfig(offset) => {
                let value = unsafe { func.read_u32(offset) };
                return PcidClientResponse::ReadConfig(value);
            }
            PcidClientRequest::WriteConfig(offset, value) => {
                unsafe {
                    func.write_u32(offset, value);
                }
                return PcidClientResponse::WriteConfig;
            }
        }
    }
    fn handle_spawn(
        mut self,
        pcid_to_client_write: usize,
        pcid_from_client_read: usize,
        args: driver_interface::SubdriverArguments,
    ) {
        use driver_interface::*;

        let mut pcid_to_client = unsafe { File::from_raw_fd(pcid_to_client_write as RawFd) };
        let mut pcid_from_client = unsafe { File::from_raw_fd(pcid_from_client_read as RawFd) };

        while let Ok(msg) = recv(&mut pcid_from_client) {
            let response = self.respond(msg, &args);
            send(&mut pcid_to_client, &response).unwrap();
        }
    }
}