view rust/rhg/src/commands/files.rs @ 45367:53af26aa5951

rhg: handle broken pipe error for stderr

Differential Revision: https://phab.mercurial-scm.org/D8871

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(())
    }
}