use io::Io;

use self::disk::Disk;
use self::hba::{HbaMem, HbaPortType};

pub mod disk;
pub mod fis;
pub mod hba;

pub fn disks(base: usize, irq: u8) -> Vec<Disk> {
    println!(" + AHCI on: {:X} IRQ: {}", base as usize, irq);

    let pi = unsafe { &mut *(base as *mut HbaMem) }.pi.read();
    let ret: Vec<Disk> = (0..32)
          .filter(|&i| pi & 1 << i as i32 == 1 << i as i32)
          .filter_map(|i| {
              let port = &mut unsafe { &mut *(base as *mut HbaMem) }.ports[i];
              let port_type = port.probe();
              println!("{}: {:?}", i, port_type);
              match port_type {
                  HbaPortType::SATA => {
                      match Disk::new(i, port) {
                          Ok(disk) => Some(disk),
                          Err(err) => {
                              println!("{}: {}", i, err);
                              None
                          }
                      }
                  }
                  _ => None,
              }
          })
          .collect();

    ret
}