org.rstudio.studio.client.workbench.views.vcs.BranchToolbarButton.java Source code

Introduction

Here is the source code for org.rstudio.studio.client.workbench.views.vcs.BranchToolbarButton.java

Source

/*
 * BranchToolbarButton.java
 *
 * Copyright (C) 2009-12 by RStudio, Inc.
 *
 * Unless you have received this program directly from RStudio pursuant
 * to the terms of a commercial license agreement with RStudio, then
 * this program is licensed to you under the terms of version 3 of the
 * GNU Affero General Public License. This program is distributed WITHOUT
 * ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,
 * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the
 * AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.
 *
 */
package org.rstudio.studio.client.workbench.views.vcs;

import com.google.gwt.core.client.JsArrayString;
import com.google.gwt.event.logical.shared.HasValueChangeHandlers;
import com.google.gwt.event.logical.shared.ValueChangeEvent;
import com.google.gwt.event.logical.shared.ValueChangeHandler;
import com.google.gwt.event.shared.HandlerRegistration;
import com.google.gwt.user.client.Command;
import com.google.gwt.user.client.ui.MenuItem;
import com.google.inject.Inject;
import com.google.inject.Provider;
import org.rstudio.core.client.StringUtil;
import org.rstudio.core.client.WidgetHandlerRegistration;
import org.rstudio.core.client.widget.ScrollableToolbarPopupMenu;
import org.rstudio.core.client.widget.ToolbarButton;
import org.rstudio.core.client.widget.ToolbarPopupMenu;
import org.rstudio.studio.client.common.icons.StandardIcons;
import org.rstudio.studio.client.workbench.views.vcs.common.events.VcsRefreshEvent;
import org.rstudio.studio.client.workbench.views.vcs.common.events.VcsRefreshHandler;
import org.rstudio.studio.client.workbench.views.vcs.git.model.GitState;

public class BranchToolbarButton extends ToolbarButton
        implements HasValueChangeHandlers<String>, VcsRefreshHandler {
    protected class SwitchBranchCommand implements Command {
        public SwitchBranchCommand(String branchLabel, String branchValue) {
            branchLabel_ = branchLabel;
            branchValue_ = branchValue;
        }

        @Override
        public void execute() {
            setBranchCaption(branchLabel_);
            ValueChangeEvent.fire(BranchToolbarButton.this, branchValue_);
        }

        private final String branchLabel_;
        private final String branchValue_;
    }

    @Inject
    public BranchToolbarButton(final Provider<GitState> pVcsState) {
        super("", StandardIcons.INSTANCE.empty_command(), new ScrollableToolbarPopupMenu());
        pVcsState_ = pVcsState;

        setTitle("Switch branch");

        new WidgetHandlerRegistration(this) {
            @Override
            protected HandlerRegistration doRegister() {
                return pVcsState.get().addVcsRefreshHandler(BranchToolbarButton.this, true);
            }
        };
    }

    public void setBranchCaption(String caption) {
        if (StringUtil.isNullOrEmpty(caption))
            caption = NO_BRANCH;

        setText(caption);
    }

    @Override
    public HandlerRegistration addValueChangeHandler(ValueChangeHandler<String> handler) {
        return addHandler(handler, ValueChangeEvent.getType());
    }

    @Override
    public void onVcsRefresh(VcsRefreshEvent event) {
        ToolbarPopupMenu rootMenu = getMenu();
        rootMenu.clearItems();
        JsArrayString branches = pVcsState_.get().getBranchInfo().getBranches();

        onBeforePopulateMenu(rootMenu);
        for (int i = 0; i < branches.length(); i++) {
            String branch = branches.get(i);
            final String branchLabel = branch.replaceFirst("^remotes/", "");
            final String branchValue = branch.replaceFirst(" ->.*", "");
            rootMenu.addItem(new MenuItem(branchLabel, new SwitchBranchCommand(branchLabel, branchValue)));
        }
    }

    protected void onBeforePopulateMenu(ToolbarPopupMenu rootMenu) {
    }

    protected final Provider<GitState> pVcsState_;

    private static final String NO_BRANCH = "(No branch)";
}