use crate::commands::Command;

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

use crate::ui::Ui;

use hg::operations::{ListTrackedFiles, ListTrackedFilesErrorKind};

pub const HELP_TEXT: &str = "

List tracked files.

Returns 0 on success.

";

pub struct FilesCommand<'a> {

    ui: &'a Ui,

}

impl<'a> FilesCommand<'a> {

    pub fn new(ui: &'a Ui) -> Self {

        FilesCommand { ui }

    }

}

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

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

        let operation_builder = ListTrackedFiles::new()?;

        let operation = operation_builder.load().map_err(|err| {

            CommandErrorKind::Abort(Some(

                [b"abort: ", err.to_string().as_bytes(), b"\n"]

                    .concat()

                    .to_vec(),

            ))

        })?;

        let files = operation.run().map_err(|err| match err.kind {

            ListTrackedFilesErrorKind::ParseError(_) => {

                CommandErrorKind::Abort(Some(

                    // TODO find a better error message

                    b"abort: parse error\n".to_vec(),

                ))

            }

        })?;

        let mut stdout = self.ui.stdout_buffer();

        for file in files {

            stdout.write_all(file.as_bytes())?;

            stdout.write_all(b"\n")?;

        }

        stdout.flush()?;

        Ok(())

    }

}