89a9b13a925ba1a00ca416c10a5c8b6c0ee161df
[idea/community.git] / plugins / git4idea / src / git4idea / push / GitPushTargetPanel.java
1 /*
2  * Copyright 2000-2014 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package git4idea.push;
17
18 import com.intellij.dvcs.push.PushTargetPanel;
19 import com.intellij.dvcs.push.ui.*;
20 import com.intellij.openapi.diagnostic.Logger;
21 import com.intellij.openapi.ui.ValidationInfo;
22 import com.intellij.openapi.ui.popup.JBPopupFactory;
23 import com.intellij.openapi.ui.popup.ListPopup;
24 import com.intellij.openapi.ui.popup.PopupStep;
25 import com.intellij.openapi.ui.popup.util.BaseListPopupStep;
26 import com.intellij.ui.ColoredTreeCellRenderer;
27 import com.intellij.ui.SimpleTextAttributes;
28 import com.intellij.ui.awt.RelativePoint;
29 import com.intellij.ui.components.JBLabel;
30 import com.intellij.util.Function;
31 import com.intellij.util.ObjectUtils;
32 import com.intellij.util.containers.ContainerUtil;
33 import git4idea.GitRemoteBranch;
34 import git4idea.repo.GitRemote;
35 import git4idea.repo.GitRepository;
36 import org.jetbrains.annotations.NotNull;
37 import org.jetbrains.annotations.Nullable;
38
39 import javax.swing.*;
40 import javax.swing.tree.DefaultMutableTreeNode;
41 import java.awt.*;
42 import java.awt.event.MouseEvent;
43 import java.text.ParseException;
44 import java.util.Comparator;
45 import java.util.List;
46
47 public class GitPushTargetPanel extends PushTargetPanel<GitPushTarget> {
48
49   private static final Logger LOG = Logger.getInstance(GitPushTargetPanel.class);
50
51   private static final Comparator<GitRemoteBranch> REMOTE_BRANCH_COMPARATOR = new MyRemoteBranchComparator();
52   private static final String SEPARATOR = " : ";
53
54   @NotNull private final GitRepository myRepository;
55   @NotNull private final VcsEditableTextComponent myTargetRenderedComponent;
56   @NotNull private final PushTargetTextField myTargetTextField;
57   @NotNull private final VcsLinkedTextComponent myRemoteRenderedComponent;
58
59   @Nullable private GitPushTarget myCurrentTarget;
60   @Nullable private String myError;
61   @Nullable private Runnable myFireOnChangeAction;
62
63   public GitPushTargetPanel(@NotNull GitRepository repository, @Nullable GitPushTarget defaultTarget) {
64     myRepository = repository;
65
66     myCurrentTarget = defaultTarget;
67
68     String initialBranch = "";
69     String initialRemote = "";
70     if (defaultTarget == null) {
71       if (repository.getCurrentBranch() == null) {
72         myError = "Detached HEAD";
73       }
74       else if (repository.getRemotes().isEmpty()) {
75         myError = "No remotes";
76       }
77       else if (repository.isFresh()) {
78         myError = "Empty repository";
79       }
80       else {
81         myError = "Can't push";
82       }
83     }
84     else {
85       initialBranch = getTextFieldText(defaultTarget);
86       initialRemote = defaultTarget.getBranch().getRemote().getName();
87     }
88     myTargetRenderedComponent = new VcsEditableTextComponent("<a href=''>" + initialBranch + "</a>", null);
89     myTargetTextField = new PushTargetTextField(repository.getProject(), getTargetNames(myRepository), initialBranch);
90     myRemoteRenderedComponent = new VcsLinkedTextComponent("<a href=''>" + initialRemote + "</a>", new VcsLinkListener() {
91       @Override
92       public void hyperlinkActivated(@NotNull DefaultMutableTreeNode sourceNode, @NotNull MouseEvent event) {
93         showRemoteSelector(event);
94       }
95     });
96
97     setLayout(new BorderLayout());
98     setOpaque(false);
99     JPanel remoteAndSeparator = new JPanel(new BorderLayout());
100     remoteAndSeparator.setOpaque(false);
101     remoteAndSeparator.add(myRemoteRenderedComponent, BorderLayout.CENTER);
102     remoteAndSeparator.add(new JBLabel(SEPARATOR), BorderLayout.EAST);
103
104     add(remoteAndSeparator, BorderLayout.WEST);
105     add(myTargetTextField, BorderLayout.CENTER);
106     updateTextField();
107   }
108
109   private void updateTextField() {
110     myTargetTextField.setVisible(!myRepository.getRemotes().isEmpty());
111   }
112
113   private void showRemoteSelector(@NotNull MouseEvent event) {
114     final List<String> remotes = getRemotes();
115     if (remotes.size() <= 1) {
116       return;
117     }
118
119     ListPopup popup = JBPopupFactory.getInstance().createListPopup(new BaseListPopupStep<String>(null, remotes) {
120       @Override
121       public PopupStep onChosen(String selectedValue, boolean finalChoice) {
122         myRemoteRenderedComponent.updateLinkText(selectedValue);
123         if (myFireOnChangeAction != null) {
124           myFireOnChangeAction.run();
125         }
126         return super.onChosen(selectedValue, finalChoice);
127       }
128     });
129     popup.show(new RelativePoint(event));
130   }
131
132   @NotNull
133   private List<String> getRemotes() {
134     return ContainerUtil.map(myRepository.getRemotes(), new Function<GitRemote, String>() {
135       @Override
136       public String fun(GitRemote remote) {
137         return remote.getName();
138       }
139     });
140   }
141
142   @Override
143   public void render(@NotNull ColoredTreeCellRenderer renderer, boolean isSelected, boolean isActive) {
144     SimpleTextAttributes targetTextAttributes = PushLogTreeUtil.addTransparencyIfNeeded(SimpleTextAttributes.REGULAR_ATTRIBUTES, isActive);
145     if (myError != null) {
146       renderer.append(myError, PushLogTreeUtil.addTransparencyIfNeeded(SimpleTextAttributes.ERROR_ATTRIBUTES, isActive));
147     }
148     else {
149       String currentRemote = myRemoteRenderedComponent.getText();
150       if (getRemotes().size() > 1) {
151         myRemoteRenderedComponent.setSelected(isSelected);
152         myRemoteRenderedComponent.setTransparent(!isActive);
153         myRemoteRenderedComponent.render(renderer);
154       }
155       else {
156         renderer.append(currentRemote, targetTextAttributes);
157       }
158       renderer.append(SEPARATOR, targetTextAttributes);
159
160       GitPushTarget target = getValue();
161       if (target.isNewBranchCreated()) {
162         renderer.append("+", PushLogTreeUtil.addTransparencyIfNeeded(SimpleTextAttributes.SYNTHETIC_ATTRIBUTES, isActive), this);
163       }
164       myTargetRenderedComponent.setSelected(isSelected);
165       myTargetRenderedComponent.setTransparent(!isActive);
166       myTargetRenderedComponent.render(renderer);
167     }
168   }
169
170   @NotNull
171   @Override
172   public GitPushTarget getValue() {
173     return ObjectUtils.assertNotNull(myCurrentTarget);
174   }
175
176   @NotNull
177   private static String getTextFieldText(@Nullable GitPushTarget target) {
178     return (target != null ? target.getBranch().getNameForRemoteOperations() : "");
179   }
180
181   @Override
182   public void fireOnCancel() {
183     myTargetTextField.setText(getTextFieldText(myCurrentTarget));
184   }
185
186   @Override
187   public void fireOnChange() {
188     if (myError != null) {
189       return;
190     }
191     String remoteName = myRemoteRenderedComponent.getText();
192     String branchName = myTargetTextField.getText();
193     try {
194       myCurrentTarget = GitPushTarget.parse(myRepository, remoteName, branchName);
195       myTargetRenderedComponent.updateLinkText(branchName);
196     }
197     catch (ParseException e) {
198       LOG.error("Invalid remote name shouldn't be allowed. [" + remoteName + ", " + branchName + "]", e);
199     }
200   }
201
202   @Nullable
203   @Override
204   public ValidationInfo verify() {
205     if (myError != null) {
206       return new ValidationInfo(myError, myTargetTextField);
207     }
208     try {
209       GitPushTarget.parse(myRepository, myRemoteRenderedComponent.getText(), myTargetTextField.getText());
210       return null;
211     }
212     catch (ParseException e) {
213       return new ValidationInfo(e.getMessage(), myTargetTextField);
214     }
215   }
216
217   @SuppressWarnings("NullableProblems")
218   @Override
219   public void setFireOnChangeAction(@NotNull Runnable action) {
220     myFireOnChangeAction = action;
221   }
222
223   @NotNull
224   public static List<String> getTargetNames(@NotNull GitRepository repository) {
225     List<GitRemoteBranch> remoteBranches = ContainerUtil.sorted(repository.getBranches().getRemoteBranches(), REMOTE_BRANCH_COMPARATOR);
226     return ContainerUtil.map(remoteBranches, new Function<GitRemoteBranch, String>() {
227       @Override
228       public String fun(GitRemoteBranch branch) {
229         return branch.getNameForRemoteOperations();
230       }
231     });
232   }
233
234   private static class MyRemoteBranchComparator implements Comparator<GitRemoteBranch> {
235     @Override
236     public int compare(@NotNull GitRemoteBranch o1, @NotNull GitRemoteBranch o2) {
237       String remoteName1 = o1.getRemote().getName();
238       String remoteName2 = o2.getRemote().getName();
239       int remoteComparison = remoteName1.compareTo(remoteName2);
240       if (remoteComparison != 0) {
241         if (remoteName1.equals(GitRemote.ORIGIN_NAME)) {
242           return -1;
243         }
244         if (remoteName2.equals(GitRemote.ORIGIN_NAME)) {
245           return 1;
246         }
247         return remoteComparison;
248       }
249       return o1.getNameForLocalOperations().compareTo(o2.getNameForLocalOperations());
250     }
251   }
252 }