comparison rust/rhg/src/commands/status.rs @ 48734:3e2b4bb286e7

rhg: Colorize `rhg status` output when appropriate

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

48733:39c447e03dbc

        } else {
            None
        },
    };
    if display_states.modified {
        output.display(b"M", ds_status.modified)?;
    }
    if display_states.added {
        output.display(b"A", ds_status.added)?;
    }
    if display_states.removed {
        output.display(b"R", ds_status.removed)?;
    }
    if display_states.deleted {
        output.display(b"!", ds_status.deleted)?;
    }
    if display_states.unknown {
        output.display(b"?", ds_status.unknown)?;
    }
    if display_states.ignored {
        output.display(b"I", ds_status.ignored)?;
    }
    if display_states.clean {
        output.display(b"C", ds_status.clean)?;
    }

    let mut dirstate_write_needed = ds_status.dirty;
    let filesystem_time_at_status_start =
        ds_status.filesystem_time_at_status_start;

    // Probably more elegant to use a Deref or Borrow trait rather than
    // harcode HgPathBuf, but probably not really useful at this point
    fn display(
        &self,
        status_prefix: &[u8],
        mut paths: Vec<StatusPath<'_>>,
    ) -> Result<(), CommandError> {
        paths.sort_unstable();
        for StatusPath { path, copy_source } in paths {
            let relative;
            let path = if let Some(relativize) = &self.relativize {
                relative = relativize.relativize(&path);
                &*relative
            } else {
                path.as_bytes()
            };
            // TODO optim, probably lots of unneeded copies here, especially
            // if out stream is buffered
            if self.no_status {
                self.ui.write_stdout(&format_bytes!(b"{}\n", path))?
            } else {
                self.ui.write_stdout(&format_bytes!(
                    b"{} {}\n",
                    status_prefix,
                    path
                ))?
            }
            if let Some(source) = copy_source {
                self.ui.write_stdout(&format_bytes!(
                    b"  {}\n",
                    source.as_bytes()
                ))?
            }
        }
        Ok(())
    }
}

48734:3e2b4bb286e7

        } else {
            None
        },
    };
    if display_states.modified {
        output.display(b"M ", "status.modified", ds_status.modified)?;
    }
    if display_states.added {
        output.display(b"A ", "status.added", ds_status.added)?;
    }
    if display_states.removed {
        output.display(b"R ", "status.removed", ds_status.removed)?;
    }
    if display_states.deleted {
        output.display(b"! ", "status.deleted", ds_status.deleted)?;
    }
    if display_states.unknown {
        output.display(b"? ", "status.unknown", ds_status.unknown)?;
    }
    if display_states.ignored {
        output.display(b"I ", "status.ignored", ds_status.ignored)?;
    }
    if display_states.clean {
        output.display(b"C ", "status.clean", ds_status.clean)?;
    }

    let mut dirstate_write_needed = ds_status.dirty;
    let filesystem_time_at_status_start =
        ds_status.filesystem_time_at_status_start;

    // Probably more elegant to use a Deref or Borrow trait rather than
    // harcode HgPathBuf, but probably not really useful at this point
    fn display(
        &self,
        status_prefix: &[u8],
        label: &'static str,
        mut paths: Vec<StatusPath<'_>>,
    ) -> Result<(), CommandError> {
        paths.sort_unstable();
        // TODO: get the stdout lock once for the whole loop instead of in each write
        for StatusPath { path, copy_source } in paths {
            let relative;
            let path = if let Some(relativize) = &self.relativize {
                relative = relativize.relativize(&path);
                &*relative
            } else {
                path.as_bytes()
            };
            // TODO: Add a way to use `write_bytes!` instead of `format_bytes!`
            // in order to stream to stdout instead of allocating an
            // itermediate `Vec<u8>`.
            if !self.no_status {
                self.ui.write_stdout_labelled(status_prefix, label)?
            }
            self.ui
                .write_stdout_labelled(&format_bytes!(b"{}\n", path), label)?;
            if let Some(source) = copy_source {
                let label = "status.copied";
                self.ui.write_stdout_labelled(
                    &format_bytes!(b"  {}\n", source.as_bytes()),
                    label,
                )?
            }
        }
        Ok(())
    }
}