commit.rs

  1use crate::{
  2    BuildCommitPermalinkParams, GitHostingProviderRegistry, GitRemote, Oid, parse_git_remote_url,
  3    repository::GitBinary, status::StatusCode,
  4};
  5use anyhow::{Context as _, Result};
  6use collections::HashMap;
  7use gpui::SharedString;
  8use std::sync::Arc;
  9
 10#[derive(Clone, Debug, Default)]
 11pub struct ParsedCommitMessage {
 12    pub message: SharedString,
 13    pub permalink: Option<url::Url>,
 14    pub pull_request: Option<crate::hosting_provider::PullRequest>,
 15    pub remote: Option<GitRemote>,
 16}
 17
 18impl ParsedCommitMessage {
 19    pub fn parse(
 20        sha: String,
 21        message: String,
 22        remote_url: Option<&str>,
 23        provider_registry: Option<Arc<GitHostingProviderRegistry>>,
 24    ) -> Self {
 25        if let Some((hosting_provider, remote)) = provider_registry
 26            .and_then(|reg| remote_url.and_then(|url| parse_git_remote_url(reg, url)))
 27        {
 28            let pull_request = hosting_provider.extract_pull_request(&remote, &message);
 29            Self {
 30                message: message.into(),
 31                permalink: Some(
 32                    hosting_provider
 33                        .build_commit_permalink(&remote, BuildCommitPermalinkParams { sha: &sha }),
 34                ),
 35                pull_request,
 36                remote: Some(GitRemote {
 37                    host: hosting_provider,
 38                    owner: remote.owner.into(),
 39                    repo: remote.repo.into(),
 40                }),
 41            }
 42        } else {
 43            Self {
 44                message: message.into(),
 45                ..Default::default()
 46            }
 47        }
 48    }
 49}
 50
 51pub(crate) async fn get_messages(git: &GitBinary, shas: &[Oid]) -> Result<HashMap<Oid, String>> {
 52    if shas.is_empty() {
 53        return Ok(HashMap::default());
 54    }
 55
 56    let output = if cfg!(windows) {
 57        // Windows has a maximum invocable command length, so we chunk the input.
 58        // Actual max is 32767, but we leave some room for the rest of the command as we aren't in precise control of what std might do here
 59        const MAX_CMD_LENGTH: usize = 30000;
 60        // 40 bytes of hash, 2 quotes and a separating space
 61        const SHA_LENGTH: usize = 40 + 2 + 1;
 62        const MAX_ENTRIES_PER_INVOCATION: usize = MAX_CMD_LENGTH / SHA_LENGTH;
 63
 64        let mut result = vec![];
 65        for shas in shas.chunks(MAX_ENTRIES_PER_INVOCATION) {
 66            let partial = get_messages_impl(git, shas).await?;
 67            result.extend(partial);
 68        }
 69        result
 70    } else {
 71        get_messages_impl(git, shas).await?
 72    };
 73
 74    Ok(shas
 75        .iter()
 76        .cloned()
 77        .zip(output)
 78        .collect::<HashMap<Oid, String>>())
 79}
 80
 81async fn get_messages_impl(git: &GitBinary, shas: &[Oid]) -> Result<Vec<String>> {
 82    const MARKER: &str = "<MARKER>";
 83    let output = git
 84        .build_command(["show"])
 85        .arg("-s")
 86        .arg(format!("--format=%B{}", MARKER))
 87        .args(shas.iter().map(ToString::to_string))
 88        .output()
 89        .await
 90        .context("starting git show process")?;
 91    anyhow::ensure!(
 92        output.status.success(),
 93        "'git show' failed with error {:?}",
 94        output.status
 95    );
 96    Ok(String::from_utf8_lossy(&output.stdout)
 97        .trim()
 98        .split_terminator(MARKER)
 99        .map(|str| str.trim().replace("<", "&lt;").replace(">", "&gt;"))
100        .collect::<Vec<_>>())
101}
102
103/// Parse the output of `git diff --name-status -z`
104pub fn parse_git_diff_name_status(content: &str) -> impl Iterator<Item = (&str, StatusCode)> {
105    let mut parts = content.split('\0');
106    std::iter::from_fn(move || {
107        loop {
108            let status_str = parts.next()?;
109            let path = parts.next()?;
110            let status = match status_str {
111                "M" => StatusCode::Modified,
112                "A" => StatusCode::Added,
113                "D" => StatusCode::Deleted,
114                _ => continue,
115            };
116            return Some((path, status));
117        }
118    })
119}
120
121#[cfg(test)]
122mod tests {
123
124    use super::*;
125
126    #[test]
127    fn test_parse_git_diff_name_status() {
128        let input = concat!(
129            "M\x00Cargo.lock\x00",
130            "M\x00crates/project/Cargo.toml\x00",
131            "M\x00crates/project/src/buffer_store.rs\x00",
132            "D\x00crates/project/src/git.rs\x00",
133            "A\x00crates/project/src/git_store.rs\x00",
134            "A\x00crates/project/src/git_store/git_traversal.rs\x00",
135            "M\x00crates/project/src/project.rs\x00",
136            "M\x00crates/project/src/worktree_store.rs\x00",
137            "M\x00crates/project_panel/src/project_panel.rs\x00",
138        );
139
140        let output = parse_git_diff_name_status(input).collect::<Vec<_>>();
141        assert_eq!(
142            output,
143            &[
144                ("Cargo.lock", StatusCode::Modified),
145                ("crates/project/Cargo.toml", StatusCode::Modified),
146                ("crates/project/src/buffer_store.rs", StatusCode::Modified),
147                ("crates/project/src/git.rs", StatusCode::Deleted),
148                ("crates/project/src/git_store.rs", StatusCode::Added),
149                (
150                    "crates/project/src/git_store/git_traversal.rs",
151                    StatusCode::Added,
152                ),
153                ("crates/project/src/project.rs", StatusCode::Modified),
154                ("crates/project/src/worktree_store.rs", StatusCode::Modified),
155                (
156                    "crates/project_panel/src/project_panel.rs",
157                    StatusCode::Modified
158                ),
159            ]
160        );
161    }
162}