use crate::commands::Command;

use crate::error::{CommandError, CommandErrorKind};

use crate::ui::utf8_to_local;

use crate::ui::Ui;

use hg::operations::FindRoot;

use hg::operations::{CatRev, CatRevError, CatRevErrorKind};

use hg::utils::hg_path::HgPathBuf;

use micro_timer::timed;

use std::convert::TryFrom;

pub const HELP_TEXT: &str = "

Output the current or given revision of files

";

pub struct CatCommand<'a> {

    rev: Option<&'a str>,

    files: Vec<&'a str>,

}

impl<'a> CatCommand<'a> {

    pub fn new(rev: Option<&'a str>, files: Vec<&'a str>) -> Self {

        Self { rev, files }

    }

    fn display(&self, ui: &Ui, data: &[u8]) -> Result<(), CommandError> {

        ui.write_stdout(data)?;

        Ok(())

    }

}

impl<'a> Command for CatCommand<'a> {

    #[timed]

    fn run(&self, ui: &Ui) -> Result<(), CommandError> {

        let root = FindRoot::new().run()?;

        let cwd = std::env::current_dir()

            .or_else(|e| Err(CommandErrorKind::CurrentDirNotFound(e)))?;

        let mut files = vec![];

        for file in self.files.iter() {

            let normalized = cwd.join(&file);

            let stripped = normalized

                .strip_prefix(&root)

                .map_err(|_| CommandErrorKind::Abort(None))?;

            let hg_file = HgPathBuf::try_from(stripped.to_path_buf())

                .map_err(|_| CommandErrorKind::Abort(None))?;

            files.push(hg_file);

        }

        match self.rev {

            Some(rev) => {

                let mut operation = CatRev::new(&root, rev, &files)

                    .map_err(|e| map_rev_error(rev, e))?;

                let data =

                    operation.run().map_err(|e| map_rev_error(rev, e))?;

                self.display(ui, &data)

            }

            None => Err(CommandErrorKind::Unimplemented.into()),

        }

    }

}

/// Convert `CatRevErrorKind` to `CommandError`

fn map_rev_error(rev: &str, err: CatRevError) -> CommandError {

    CommandError {

        kind: match err.kind {

            CatRevErrorKind::IoError(err) => CommandErrorKind::Abort(Some(

                utf8_to_local(&format!("abort: {}\n", err)).into(),

            )),

            CatRevErrorKind::InvalidRevision => CommandErrorKind::Abort(Some(

                utf8_to_local(&format!(

                    "abort: invalid revision identifier{}\n",

                    rev

                ))

                .into(),

            )),

            CatRevErrorKind::UnsuportedRevlogVersion(version) => {

                CommandErrorKind::Abort(Some(

                    utf8_to_local(&format!(

                        "abort: unsupported revlog version {}\n",

                        version

                    ))

                    .into(),

                ))

            }

            CatRevErrorKind::CorruptedRevlog => CommandErrorKind::Abort(Some(

                "abort: corrupted revlog\n".into(),

            )),

            CatRevErrorKind::UnknowRevlogDataFormat(format) => {

                CommandErrorKind::Abort(Some(

                    utf8_to_local(&format!(

                        "abort: unknow revlog dataformat {:?}\n",

                        format

                    ))

                    .into(),

                ))

            }

        },

    }

}