use embedded_hal::{
	blocking::spi::Transfer,
	digital::v2::OutputPin,
};
use crate::cpld::CPLD;
use crate::Error;

pub struct SPISlave<'a, SPI, CS0, CS1, CS2, GPIO> (
	// SPI device to be multiplexed
	&'a CPLD<SPI, CS0, CS1, CS2, GPIO>,
	// Channel of SPI slave
	u8,
	// Need I/O Update
	bool,
);

pub struct Parts<'a, SPI, CS0, CS1, CS2, GPIO> {
	pub spi1: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
	pub spi2: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
	pub spi3: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
	pub spi4: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
	pub spi5: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
	pub spi6: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
	pub spi7: SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>,
}

impl<'a, SPI, CS0, CS1, CS2, GPIO> Parts<'a, SPI, CS0, CS1, CS2, GPIO> {
	pub(crate) fn new(cpld: &'a CPLD<SPI, CS0, CS1, CS2, GPIO>) -> Self {
		Parts {
			spi1: SPISlave(&cpld, 1, false),
			spi2: SPISlave(&cpld, 2, false),
			spi3: SPISlave(&cpld, 3, true),
			spi4: SPISlave(&cpld, 4, true),
			spi5: SPISlave(&cpld, 5, true),
			spi6: SPISlave(&cpld, 6, true),
			spi7: SPISlave(&cpld, 7, true),
		}
	}
}

impl<'a, SPI, CS0, CS1, CS2, GPIO, E> Transfer<u8> for SPISlave<'a, SPI, CS0, CS1, CS2, GPIO>
where
	CS2: OutputPin,
	CS1: OutputPin,
	CS0: OutputPin,
	SPI: Transfer<u8, Error = E>,
	GPIO: OutputPin,
{
	type Error = Error<E>;

	fn transfer<'w>(&mut self, words: &'w mut[u8]) -> Result<&'w [u8], Self::Error> {
		let mut dev = self.0.data.try_borrow_mut().map_err(|_| Error::GetRefMutDataError)?;
		dev.select_chip(self.1).map_err(|_| Error::CSError)?;
		let result = dev.spi.transfer(words).map_err(Error::SPI)?;
		dev.select_chip(0).map_err(|_| Error::CSError)?;
		if self.2 {
			dev.issue_io_update().map_err(|_| Error::IOUpdateError)?;
		}
		Ok(result)
	}
}