mender_artifact 0.1.3

use flate2::read::GzDecoder;
use log::{debug, info, trace, warn};
use serde::{Deserialize, Serialize};
// use serde_json::{Result, Value};
use std::fs::File;
use std::io::prelude::*;
use tar::Archive;

#[derive(Debug)]
pub enum ParseError {
    IOError(std::io::Error),
    ParseError(String),
    JSONParseError(serde_json::Error),
}

impl std::fmt::Display for ParseError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "SuperError is here!")
    }
}

impl std::error::Error for ParseError {
    fn description(&self) -> &str {
        match &self {
            &IOError => "IOError",
            &ParseError => "ParseError",
            &JSONParseError => "JSONParrseError",
        }
    }

    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
        match &self {
            &IOError => None,
            &ParseError => None,
            &JSONParseError => None,
        }
    }
}
impl From<std::io::Error> for ParseError {
    fn from(error: std::io::Error) -> Self {
        ParseError::IOError(error)
    }
}

impl From<serde_json::Error> for ParseError {
    fn from(error: serde_json::Error) -> Self {
        ParseError::JSONParseError(error)
    }
}

#[derive(Deserialize, Debug)]
pub struct Version {
    format: String,
    version: i32,
}

#[derive(Debug)]
pub struct Manifest {
    data: std::collections::HashMap<String, String>, // Name : Hash
}

use std::io::{BufRead, BufReader};
impl Manifest {
    fn parse<R>(r: R) -> std::io::Result<Manifest>
    where
        R: Read,
    {
        let mut m = Manifest {
            data: std::collections::HashMap::new(),
        };
        for line in BufReader::new(r).lines() {
            let line = line?.clone();
            debug!("Manifest line: {}", line);
            let mut line_it = line.split_ascii_whitespace();
            m.data.insert(
                line_it.next().unwrap().to_string(),
                line_it.next().unwrap().to_string(),
            );
        }
        Ok(m)
    }
}

#[derive(Deserialize, Debug)]
pub struct ArtifactDepends {
    artifact_name: Option<Vec<String>>,
    device_type: Vec<String>,
    artifact_group: Option<String>,
}

#[derive(Deserialize, Debug)]
pub struct ArtifactProvides {
    artifact_name: String,
    artifact_group: Option<String>,
}

#[derive(Deserialize, Debug)]
pub struct HeaderInfo {
    payloads: Vec<std::collections::HashMap<String, String>>,
    artifact_provides: ArtifactProvides,
    artifact_depends: ArtifactDepends,
}

// {
//     "type": "rootfs-image"
//         "artifact_provides": {
//             "rootfs_image_checksum": "4d480539cdb23a4aee6330ff80673a5af92b7793eb1c57c4694532f96383b619"
//         },
//     "artifact_depends": {
//         "rootfs_image_checksum": "4d480539cdb23a4aee6330ff80673a5af92b7793eb1c57c4694532f96383b619"
//     },
// }
#[derive(Deserialize, Debug)]
pub struct TypeInfo {
    r#type: String,
    artifact_provides: Option<TypeInfoArtifactProvides>,
    artifact_depends: Option<TypeInfoArtifactDepends>,
}

#[derive(Deserialize, Debug)]
pub struct TypeInfoArtifactProvides {
    rootfs_image_checksum: String,
}

#[derive(Deserialize, Debug)]
pub struct TypeInfoArtifactDepends {
    rootfs_image_checksum: String,
}

#[derive(Deserialize, Debug)]
pub struct SubHeader {
    type_info: TypeInfo,
    meta_data: Option<std::collections::HashMap<String, String>>,
}

#[derive(Debug)]
pub struct Header {
    header_info: HeaderInfo,
    // Scripts are ignored for now
    headers: Vec<SubHeader>,
}

impl Header {
    fn parse<R>(tar: R) -> std::result::Result<Header, ParseError>
    where
        R: Read + Sized,
    {
        let mut archive = Archive::new(tar);
        let mut entries = archive.entries()?.peekable();
        let entry = entries
            .next()
            .expect("Failed to get the header-info from header.tar")
            .unwrap();
        let hdr_info = entry.header();
        println!("Header Info: {:?}", hdr_info);
        let header_info: HeaderInfo =
            serde_json::from_reader(entry).expect("Failed to parse the header-info json");
        let mut header = Header {
            header_info: header_info,
            headers: Vec::new(),
        };
        // Parse all the headers
        for entry in entries {
            let entry_owned = entry.expect("Failed to get the tar header");
            let t_hdr = entry_owned.header();
            println!("subheader: {:?}", t_hdr);
            if entry_owned.path()?.starts_with("scripts") {
                continue; // Discard scripts
            }

            // TODO -- Parse metadata
            if entry_owned.path()?.ends_with("meta-data") {
                continue; // Discard Meta-Data
            }

            // First entry should be the type-info
            let type_info: TypeInfo =
                serde_json::from_reader(entry_owned).expect("failed to parse the type-info");
            // First entry should be the type-info
            let subheader: SubHeader = SubHeader {
                type_info: type_info,
                meta_data: None,
            };
            header.headers.push(subheader);
        }
        Ok(header)
    }
}

pub struct Payload<'a> {
    pub name: String,

    // cur_archive is set to /data/0000, /data/0001... etc
    cur_archive: tar::Archive<flate2::read::GzDecoder<&'a mut tar::Entry<'a, &'a mut (dyn std::io::Read + 'a)>>>,

    // sub_entries: tar::Entries<'a, flate2::read::GzDecoder<tar::Entry<'a, &'a mut (dyn std::io::Read + 'a)>>>,
    // // read: Option<tar::Entry<'a, std::io::Read>>,
    entry: tar::Entry<'a, flate2::read::GzDecoder<&'a mut tar::Entry<'a, &'a mut (dyn std::io::Read + 'a)>>>,
}

impl<'a> Payload<'a> {

    // fn new(mut entries: tar::Entries<'a, &'a mut std::io::Read>) -> Result<Payload<'a>, ParseError> {
    //     // let mut entry = entries.next().unwrap().unwrap(); // Set it to data/0000

    //     // Unzip, and detar the sub-tar in the payload, (i.e. data/0000.tar.gz)
    //     let mut archive = Archive::new(GzDecoder::new(& mut entries.next().unwrap().unwrap()));
    //     // Unzipped and untared to /data/0000
    //     let mut sub_entries = archive.entries().unwrap();
    //     let mut entry = sub_entries
    //         .next()
    //         .expect("Failed to get the payload from data/0000.tar.gz")
    //         .unwrap();
    //     // let hdr_info = entry.header();
    //     // println!("Header Info: {:?}", hdr_info);
    //     let payload = Payload {
    //         name: String::from("foobar"),
    //         cur_archive: archive,
    //         // sub_entries: sub_entries,
    //         entry: entry,
    //         // read: None,
    //     };
    //     return Ok(payload);
    // }


    // fn next(&mut self) -> Result<(), ParseError> {
    //     let entry = self.cur_archive.entries()?.next().unwrap().unwrap();
    //     self.read = Some(entry);
    //     Ok(())
    // }

}

impl <'a>Read for Payload<'a> {

    // // Get next update, or nil
    // fn next(&mut self) {
    // }


    fn read(&mut self, buf: &mut [u8]) -> Result<usize, std::io::Error> {
        // let mut entry = self.entries.next().unwrap().unwrap(); // Set it to data/0000

        let mut n: usize = 0;

        // self.cur_archive.entries()?
        //     .filter_map(|e| e.ok())
        //     .map(|mut entry| -> Result<usize, std::io::Error> {
        //         println!("Entry name: {}", entry.path());
        //         entry.read(buf)
        //     })
        //     .filter_map(|e| e.ok())
        //     .for_each(|x| n += x);

        // for (i, file) in self.cur_archive.entries().unwrap().enumerate() {
        //     let mut file = file.unwrap();
        //     // file.unpack(format!("file-{}", i)).unwrap();
        //     n += file.read(buf).unwrap();

        // }

        // println!("bytes read from update: {}", n);
        return Ok(n);
    }
}

pub struct MenderArtifact<'a> {
    version: Option<Version>,
    manifest: Option<Manifest>,
    header: Option<Header>,
    archive: tar::Archive<&'a mut std::io::Read>,
}

impl<'a> MenderArtifact<'a> {
    pub fn new(reader: &'a mut std::io::Read) -> MenderArtifact {
        let mut archive = Archive::new(reader);
        let mut m = MenderArtifact {
            archive: archive,
            version: None,
            manifest: None,
            header: None,
        };
        m
    }

    pub fn parse(
        &'a mut self,
        filename: &str,
    ) -> Result<(), ParseError> {
        let mut entries = self.archive.entries().unwrap();

        let mut entry: tar::Entry<'a, &'a mut std::io::Read> = entries.next().unwrap().unwrap();
        // Check that the entry base path name is the same as the one we are expecting
        let path = entry.header().path().expect("Failed to get the header");
        if !path.ends_with("version") {
            return Err(ParseError::ParseError(String::from("Unexpected header")));
        }


        let version: Version =
            serde_json::from_reader(entry).expect("Failed to parse the Version header");
        println!("{:?}", version);


        entry = entries.next().unwrap().unwrap();
        // Check that the entry base path name is the same as the one we are expecting
        let path = entry.header().path().expect("Failed to get the header");
        if !path.ends_with("manifest") {
            return Err(ParseError::ParseError(String::from("Unexpected header")));
        }


        let manifest = Manifest::parse(entry);
        println!("{:?}", manifest);

        entry = entries.next().unwrap().unwrap();
        // Check that the entry base path name is the same as the one we are expecting
        let path = entry.header().path().expect("Failed to get the header");
        if !path.ends_with("header.tar.gz") {
            return Err(ParseError::ParseError(String::from("Unexpected header")));
        }


        let tar = GzDecoder::new(entry);
        let header = Header::parse(tar).expect("Failed to parse the `header.tar`");
        println!("{:?}", header);

        // TODO -- Wrap the remaining entries in a `Payload` which can be read from, and later,
        // checks the checksum for the payload once it is finished.
        // entry = entries.next().unwrap().unwrap();
        // // Check that the entry base path name is the same as the one we are expecting
        // let path = entry.header().path().expect("Failed to get the header");
        // if !path.starts_with("data") {
        //     return Err(ParseError::ParseError(String::from("Unexpected header")));
        // }

        // NOTE -- Assuming un-augmented Artifact, next should be the data sections...

        entry = entries.next().unwrap().unwrap(); // Set it to data/0000

        // Unzip, and detar the sub-tar in the payload, (i.e. data/0000.tar.gz)
        let mut decoder = GzDecoder::new(&mut entry);
        let mut archive = Archive::new(decoder);
        let mut sub_entries = archive.entries()?;
        let mut entry = sub_entries.next().unwrap().unwrap();
        let hdr_info = entry.header();
        println!("Header Info: {:?}", hdr_info);

        let mut out_file = std::fs::File::create(filename)?;
        // let stdout = std::io::stdout();
        // let mut stdout = stdout.lock();
        let read = std::io::copy(&mut entry, &mut out_file)?;

        println!("Read: {} bytes into: {}", read, filename);

        // let mut payload = Payload {
        //     name: "data/0000.tar.gz".to_string(),
        //     entries,
        //     cur_entry: entry,
        // };
        // let payload = Payload::new(entries);

        return Ok(());
    }

}