iso8211 0.1.5

use std::io::{Read, Seek};

use crate::{EntryMap, Iso8211Error, Reader, Result};

/*
RP      Len     Entry name                          Content
=================================================================================
0       5       Record length                       number of bytes in record
5       1       Interchange level                   "3"
6       1       Leader identifier                   "L"
7       1       In line code extension indicator    "E"
8       1       Version number                      "1"
9       1       Application indicator               SPACE
10      2       Field control length                "09"
12      5       Base address of field area          Start address of field area (number of bytes in leader and directory)
17      3       Extended character set indicator    " ! " (SPACE,!,SPACE)
20      4       Entry map                           (see table below)

RP      Sub-entry name                  Len     Content
=================================================================================
20      Size of field length field      1       Variable 1-9 (defined by encoder)
21      Size of field position field    1       Variable 1-9 (defined by encoder)
22      Reserved                        1       "0"
23      Size of field tag field         1       "4"
*/

/// The structure of the DR leader
#[derive(Debug)]
pub struct Leader {
    /// Record Length
    record_length: u64,
    /// Interchange Level
    interchange_level: char,
    /// Leader Identifier
    leader_identifier: char,
    /// In Line Code Extension Indicator
    code_extension: char,
    /// Version Number
    version_number: char,
    /// Application Indicator
    application_indicator: char,
    /// Field Control Length
    field_control_length: u8,
    /// Base Address Of Field Area
    base_address: u64,
    /// Extended Character Set Indicator
    character_set: String,
    /// Entry Map
    entry_map: EntryMap,
}

impl Leader {
    pub fn record_length(&self) -> u64 {
        self.record_length
    }

    pub fn interchange_level(&self) -> char {
        self.interchange_level
    }

    pub fn leader_identifier(&self) -> char {
        self.leader_identifier
    }

    pub fn code_extension(&self) -> char {
        self.code_extension
    }

    pub fn version_number(&self) -> char {
        self.version_number
    }

    pub fn application_indicator(&self) -> char {
        self.application_indicator
    }

    pub fn field_control_length(&self) -> u8 {
        self.field_control_length
    }

    pub fn base_address(&self) -> u64 {
        self.base_address
    }

    pub fn character_set(&self) -> &str {
        &self.character_set
    }

    pub fn entry_map(&self) -> &EntryMap {
        &self.entry_map
    }
}

impl Leader {
    pub fn read_ddr<T: Read + Seek>(reader: &mut Reader<T>) -> Result<Leader> {
        Leader::read(reader, true)
    }

    pub fn read_dr<T: Read + Seek>(reader: &mut Reader<T>) -> Result<Leader> {
        Leader::read(reader, false)
    }

    fn read<T: Read + Seek>(reader: &mut Reader<T>, is_ddr: bool) -> Result<Leader> {
        let record_length = reader.read_u64_str(5)?;
        let interchange_level = reader.read_char()?;
        if (is_ddr && interchange_level != '3') || (!is_ddr && interchange_level != ' ') {
            return Err(Iso8211Error::Parse(format!(
                "Invalid Interchange Level: {}",
                interchange_level
            )));
        }
        let leader_identifier = reader.read_char()?;
        if (is_ddr && leader_identifier != 'L') || (!is_ddr && leader_identifier != 'D') {
            return Err(Iso8211Error::Parse(format!(
                "Invalid Leader Identifier: {}",
                leader_identifier
            )));
        }

        let code_extension = reader.read_char()?;
        if (is_ddr && code_extension != 'E') || (!is_ddr && code_extension != ' ') {
            return Err(Iso8211Error::Parse(format!(
                "Invalid In Line Code Extension Indicator: {}",
                code_extension
            )));
        }

        let version_number = reader.read_char()?;
        if (is_ddr && version_number != '1') || (!is_ddr && version_number != ' ') {
            return Err(Iso8211Error::Parse(format!(
                "Invalid Version Number: {}",
                version_number
            )));
        }

        let application_indicator = reader.read_char()?;
        if application_indicator != ' ' {
            return Err(Iso8211Error::Parse(format!(
                "Invalid Application Indicator: {}",
                application_indicator
            )));
        }

        let field_control_length_value = reader.read_str(2)?;
        if (is_ddr && field_control_length_value != "09")
            || (!is_ddr && field_control_length_value != "  ")
        {
            return Err(Iso8211Error::Parse(format!(
                "Invalid Field Control Length: {}",
                field_control_length_value
            )));
        }
        let field_control_length = if is_ddr {
            field_control_length_value.parse::<u8>()?
        } else {
            0
        };

        let base_address = reader.read_u64_str(5)?;

        let character_set = reader.read_str(3)?;
        if (is_ddr && character_set != " ! ") || (!is_ddr && character_set != "   ") {
            return Err(Iso8211Error::Parse(format!(
                "Invalid Extended Character Set Indicator: {}",
                character_set
            )));
        }

        let entry_map = EntryMap::read(reader)?;

        Ok(Leader {
            record_length,
            interchange_level,
            leader_identifier,
            code_extension,
            version_number,
            application_indicator,
            field_control_length,
            base_address,
            character_set,
            entry_map,
        })
    }
}

#[cfg(test)]
pub(crate) mod tests {
    use crate::{leader::Leader, Reader, Result};
    use std::io::{BufReader, Cursor};

    pub fn ascii_ddr_leader() -> Result<Leader> {
        let bytes = "019003LE1 0900319 ! 5504".as_bytes();
        let buffer = Cursor::new(bytes);
        let bufreader = BufReader::new(buffer);
        let mut reader = Reader::new(bufreader);
        Leader::read_ddr(&mut reader)
    }

    #[test]
    fn test_ddr_leader() {
        let target = ascii_ddr_leader();

        assert!(target.is_ok());

        let target = target.unwrap();
        assert_eq!(target.record_length, 1900);
        assert_eq!(target.interchange_level, '3');
        assert_eq!(target.leader_identifier, 'L');
        assert_eq!(target.entry_map.field_length(), 5);
        assert_eq!(target.entry_map.field_position(), 5);
        assert_eq!(target.entry_map.field_tag(), 4);
    }

    pub fn ascii_dr_leader(index: usize) -> Result<Leader> {
        let bytes = [
            "00197 D     00109   5504".as_bytes(),
            "00088 D     00067   5504".as_bytes(),
        ];
        let buffer = Cursor::new(bytes[index]);
        let bufreader = BufReader::new(buffer);
        let mut reader = Reader::new(bufreader);
        Leader::read_dr(&mut reader)
    }

    #[test]
    fn test_dr_leader() {
        for i in 0..2 {
            let target = ascii_dr_leader(i);

            assert!(target.is_ok());

            let target = target.unwrap();
            assert_eq!(target.interchange_level, ' ');
            assert_eq!(target.leader_identifier, 'D');
        }
    }
}