896 lines · 29.7 KB · Rust
RawDownload
use anyhow::anyhow;use commit_modal::CommitModal;use editor::{Editor, actions::DiffClipboardWithSelectionData}; use project::ProjectPath;use ui::{    Headline, HeadlineSize, Icon, IconName, IconSize, IntoElement, ParentElement, Render, Styled,    StyledExt, div, h_flex, rems, v_flex,}; mod blame_ui;pub mod clone; use git::{    repository::{Branch, Upstream, UpstreamTracking, UpstreamTrackingStatus},    status::{FileStatus, StatusCode, UnmergedStatus, UnmergedStatusCode},};use gpui::{    App, Context, DismissEvent, Entity, EventEmitter, FocusHandle, Focusable, SharedString, Window,};use menu::{Cancel, Confirm};use project::git_store::Repository;use project_diff::ProjectDiff;use ui::prelude::*;use workspace::{ModalView, Workspace, notifications::DetachAndPromptErr};use zed_actions; use crate::{git_panel::GitPanel, text_diff_view::TextDiffView}; mod askpass_modal;pub mod branch_picker;mod commit_modal;pub mod commit_tooltip;pub mod commit_view;mod conflict_view;pub mod file_diff_view;pub mod file_history_view;pub mod git_panel;mod git_panel_settings;pub mod git_picker;pub mod multi_diff_view;pub mod picker_prompt;pub mod project_diff;pub(crate) mod remote_output;pub mod repository_selector;pub mod stash_picker;pub mod text_diff_view;pub mod worktree_picker; pub fn init(cx: &mut App) {    editor::set_blame_renderer(blame_ui::GitBlameRenderer, cx);    commit_view::init(cx);     cx.observe_new(|editor: &mut Editor, _, cx| {        conflict_view::register_editor(editor, editor.buffer().clone(), cx);    })    .detach();     cx.observe_new(|workspace: &mut Workspace, _, cx| {        ProjectDiff::register(workspace, cx);        CommitModal::register(workspace);        git_panel::register(workspace);        repository_selector::register(workspace);        git_picker::register(workspace);        conflict_view::register_conflict_notification(workspace, cx);         let project = workspace.project().read(cx);        if project.is_read_only(cx) {            return;        }        if !project.is_via_collab() {            workspace.register_action(                |workspace, _: &zed_actions::git::CreatePullRequest, window, cx| {                    if let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) {                        panel.update(cx, |panel, cx| {                            panel.create_pull_request(window, cx);                        });                    }                },            );            workspace.register_action(|workspace, _: &git::Fetch, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.fetch(true, window, cx);                });            });            workspace.register_action(|workspace, _: &git::FetchFrom, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.fetch(false, window, cx);                });            });            workspace.register_action(|workspace, _: &git::Push, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.push(false, false, window, cx);                });            });            workspace.register_action(|workspace, _: &git::PushTo, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.push(false, true, window, cx);                });            });            workspace.register_action(|workspace, _: &git::ForcePush, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.push(true, false, window, cx);                });            });            workspace.register_action(|workspace, _: &git::Pull, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.pull(false, window, cx);                });            });            workspace.register_action(|workspace, _: &git::PullRebase, window, cx| {                let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                    return;                };                panel.update(cx, |panel, cx| {                    panel.pull(true, window, cx);                });            });        }        workspace.register_action(|workspace, action: &git::StashAll, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.stash_all(action, window, cx);            });        });        workspace.register_action(|workspace, action: &git::StashPop, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.stash_pop(action, window, cx);            });        });        workspace.register_action(|workspace, action: &git::StashApply, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.stash_apply(action, window, cx);            });        });        workspace.register_action(|workspace, action: &git::StageAll, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.stage_all(action, window, cx);            });        });        workspace.register_action(|workspace, action: &git::UnstageAll, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.unstage_all(action, window, cx);            });        });        workspace.register_action(|workspace, _: &git::Uncommit, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.uncommit(window, cx);            })        });        workspace.register_action(|workspace, _action: &git::Init, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };            panel.update(cx, |panel, cx| {                panel.git_init(window, cx);            });        });        workspace.register_action(|workspace, _action: &git::Clone, window, cx| {            let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {                return;            };             workspace.toggle_modal(window, cx, |window, cx| {                GitCloneModal::show(panel, window, cx)            });        });        workspace.register_action(|workspace, _: &git::OpenModifiedFiles, window, cx| {            open_modified_files(workspace, window, cx);        });        workspace.register_action(|workspace, _: &git::RenameBranch, window, cx| {            rename_current_branch(workspace, window, cx);        });        workspace.register_action(            |workspace, action: &DiffClipboardWithSelectionData, window, cx| {                if let Some(task) = TextDiffView::open(action, workspace, window, cx) {                    task.detach();                };            },        );        workspace.register_action(|workspace, _: &git::FileHistory, window, cx| {            let Some(active_item) = workspace.active_item(cx) else {                return;            };            let Some(editor) = active_item.downcast::<Editor>() else {                return;            };            let Some(buffer) = editor.read(cx).buffer().read(cx).as_singleton() else {                return;            };            let Some(file) = buffer.read(cx).file() else {                return;            };            let worktree_id = file.worktree_id(cx);            let project_path = ProjectPath {                worktree_id,                path: file.path().clone(),            };            let project = workspace.project();            let git_store = project.read(cx).git_store();            let Some((repo, repo_path)) = git_store                .read(cx)                .repository_and_path_for_project_path(&project_path, cx)            else {                return;            };            file_history_view::FileHistoryView::open(                repo_path,                git_store.downgrade(),                repo.downgrade(),                workspace.weak_handle(),                window,                cx,            );        });    })    .detach();} fn open_modified_files(    workspace: &mut Workspace,    window: &mut Window,    cx: &mut Context<Workspace>,) {    let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {        return;    };    let modified_paths: Vec<_> = panel.update(cx, |panel, cx| {        let Some(repo) = panel.active_repository.as_ref() else {            return Vec::new();        };        let repo = repo.read(cx);        repo.cached_status()            .filter_map(|entry| {                if entry.status.is_modified() {                    repo.repo_path_to_project_path(&entry.repo_path, cx)                } else {                    None                }            })            .collect()    });    for path in modified_paths {        workspace.open_path(path, None, true, window, cx).detach();    }} /// Resolves the repository for git operations, respecting the workspace's/// active worktree override from the project dropdown.pub fn resolve_active_repository(workspace: &Workspace, cx: &App) -> Option<Entity<Repository>> {    let project = workspace.project().read(cx);    workspace        .active_worktree_override()        .and_then(|override_id| {            project                .worktree_for_id(override_id, cx)                .and_then(|worktree| {                    let worktree_abs_path = worktree.read(cx).abs_path();                    let git_store = project.git_store().read(cx);                    git_store                        .repositories()                        .values()                        .find(|repo| {                            let repo_path = &repo.read(cx).work_directory_abs_path;                            *repo_path == worktree_abs_path                                || worktree_abs_path.starts_with(repo_path.as_ref())                        })                        .cloned()                })        })        .or_else(|| project.active_repository(cx))} pub fn git_status_icon(status: FileStatus) -> impl IntoElement {    GitStatusIcon::new(status)} struct RenameBranchModal {    current_branch: SharedString,    editor: Entity<Editor>,    repo: Entity<Repository>,} impl RenameBranchModal {    fn new(        current_branch: String,        repo: Entity<Repository>,        window: &mut Window,        cx: &mut Context<Self>,    ) -> Self {        let editor = cx.new(|cx| {            let mut editor = Editor::single_line(window, cx);            editor.set_text(current_branch.clone(), window, cx);            editor        });        Self {            current_branch: current_branch.into(),            editor,            repo,        }    }     fn cancel(&mut self, _: &Cancel, _window: &mut Window, cx: &mut Context<Self>) {        cx.emit(DismissEvent);    }     fn confirm(&mut self, _: &Confirm, window: &mut Window, cx: &mut Context<Self>) {        let new_name = self.editor.read(cx).text(cx);        if new_name.is_empty() || new_name == self.current_branch.as_ref() {            cx.emit(DismissEvent);            return;        }         let repo = self.repo.clone();        let current_branch = self.current_branch.to_string();        cx.spawn(async move |_, cx| {            match repo                .update(cx, |repo, _| {                    repo.rename_branch(current_branch, new_name.clone())                })                .await            {                Ok(Ok(_)) => Ok(()),                Ok(Err(error)) => Err(error),                Err(_) => Err(anyhow!("Operation was canceled")),            }        })        .detach_and_prompt_err("Failed to rename branch", window, cx, |_, _, _| None);        cx.emit(DismissEvent);    }} impl EventEmitter<DismissEvent> for RenameBranchModal {}impl ModalView for RenameBranchModal {}impl Focusable for RenameBranchModal {    fn focus_handle(&self, cx: &App) -> FocusHandle {        self.editor.focus_handle(cx)    }} impl Render for RenameBranchModal {    fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {        v_flex()            .key_context("RenameBranchModal")            .on_action(cx.listener(Self::cancel))            .on_action(cx.listener(Self::confirm))            .elevation_2(cx)            .w(rems(34.))            .child(                h_flex()                    .px_3()                    .pt_2()                    .pb_1()                    .w_full()                    .gap_1p5()                    .child(Icon::new(IconName::GitBranch).size(IconSize::XSmall))                    .child(                        Headline::new(format!("Rename Branch ({})", self.current_branch))                            .size(HeadlineSize::XSmall),                    ),            )            .child(div().px_3().pb_3().w_full().child(self.editor.clone()))    }} fn rename_current_branch(    workspace: &mut Workspace,    window: &mut Window,    cx: &mut Context<Workspace>,) {    let Some(panel) = workspace.panel::<git_panel::GitPanel>(cx) else {        return;    };    let current_branch: Option<String> = panel.update(cx, |panel, cx| {        let repo = panel.active_repository.as_ref()?;        let repo = repo.read(cx);        repo.branch.as_ref().map(|branch| branch.name().to_string())    });     let Some(current_branch_name) = current_branch else {        return;    };     let repo = panel.read(cx).active_repository.clone();    let Some(repo) = repo else {        return;    };     workspace.toggle_modal(window, cx, |window, cx| {        RenameBranchModal::new(current_branch_name, repo, window, cx)    });} fn render_remote_button(    id: impl Into<SharedString>,    branch: &Branch,    keybinding_target: Option<FocusHandle>,    show_fetch_button: bool,) -> Option<impl IntoElement> {    let id = id.into();    let upstream = branch.upstream.as_ref();    match upstream {        Some(Upstream {            tracking: UpstreamTracking::Tracked(UpstreamTrackingStatus { ahead, behind }),            ..        }) => match (*ahead, *behind) {            (0, 0) if show_fetch_button => {                Some(remote_button::render_fetch_button(keybinding_target, id))            }            (0, 0) => None,            (ahead, 0) => Some(remote_button::render_push_button(                keybinding_target,                id,                ahead,            )),            (ahead, behind) => Some(remote_button::render_pull_button(                keybinding_target,                id,                ahead,                behind,            )),        },        Some(Upstream {            tracking: UpstreamTracking::Gone,            ..        }) => Some(remote_button::render_republish_button(            keybinding_target,            id,        )),        None => Some(remote_button::render_publish_button(keybinding_target, id)),    }} mod remote_button {    use gpui::{Action, AnyView, ClickEvent, Corner, FocusHandle};    use ui::{        App, ButtonCommon, Clickable, ContextMenu, ElementId, FluentBuilder, Icon, IconName,        IconSize, IntoElement, Label, LabelCommon, LabelSize, LineHeightStyle, ParentElement,        PopoverMenu, SharedString, SplitButton, Styled, Tooltip, Window, div, h_flex, rems,    };     pub fn render_fetch_button(        keybinding_target: Option<FocusHandle>,        id: SharedString,    ) -> SplitButton {        split_button(            id,            "Fetch",            0,            0,            Some(IconName::ArrowCircle),            keybinding_target.clone(),            move |_, window, cx| {                window.dispatch_action(Box::new(git::Fetch), cx);            },            move |_window, cx| {                git_action_tooltip(                    "Fetch updates from remote",                    &git::Fetch,                    "git fetch",                    keybinding_target.clone(),                    cx,                )            },        )

Lines 1–500 of 896.