e38814773a1f60da4c97c8673833cfe2d705f341
[idea/community.git] / plugins / hg4idea / src / org / zmlx / hg4idea / command / HgResolveCommand.java
1 // Copyright 2008-2010 Victor Iacoban
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software distributed under
10 // the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
11 // either express or implied. See the License for the specific language governing permissions and
12 // limitations under the License.
13 package org.zmlx.hg4idea.command;
14
15 import com.intellij.openapi.project.Project;
16 import com.intellij.openapi.vcs.FilePath;
17 import com.intellij.openapi.vfs.VirtualFile;
18 import com.intellij.util.Consumer;
19 import org.apache.commons.lang.StringUtils;
20 import org.jetbrains.annotations.Nullable;
21 import org.zmlx.hg4idea.HgFile;
22 import org.zmlx.hg4idea.execution.HgCommandExecutor;
23 import org.zmlx.hg4idea.execution.HgCommandResult;
24 import org.zmlx.hg4idea.execution.HgCommandResultHandler;
25
26 import java.io.File;
27 import java.util.Arrays;
28 import java.util.Collections;
29 import java.util.HashMap;
30 import java.util.Map;
31
32 public class HgResolveCommand {
33
34   private static final int ITEM_COUNT = 3;
35
36   private final Project myProject;
37
38   public HgResolveCommand(Project project) {
39     myProject = project;
40   }
41
42
43   public Map<HgFile, HgResolveStatusEnum> getListSynchronously(VirtualFile repo) {
44     if (repo == null) {
45       return Collections.emptyMap();
46     }
47     final HgCommandExecutor executor = new HgCommandExecutor(myProject);
48     executor.setSilent(true);
49     final HgCommandResult result = executor.executeInCurrentThread(repo, "resolve", Arrays.asList("--list"));
50     if (result == null) {
51       return Collections.emptyMap();
52     }
53     return handleResult(repo, result);
54   }
55
56   public void list(final VirtualFile repo, final Consumer<Map<HgFile, HgResolveStatusEnum>> resultHandler) {
57     if (repo == null) {
58       resultHandler.consume(Collections.<HgFile, HgResolveStatusEnum>emptyMap());
59     }
60     final HgCommandExecutor executor = new HgCommandExecutor(myProject);
61     executor.setSilent(true);
62     executor.execute(repo, "resolve", Arrays.asList("--list"), new HgCommandResultHandler() {
63       @Override
64       public void process(@Nullable HgCommandResult result) {
65         if (result == null) {
66           resultHandler.consume(Collections.<HgFile, HgResolveStatusEnum>emptyMap());
67         }
68
69         final Map<HgFile, HgResolveStatusEnum> resolveStatus = handleResult(repo, result);
70         resultHandler.consume(resolveStatus);
71       }
72     });
73   }
74
75   private static Map<HgFile, HgResolveStatusEnum> handleResult(VirtualFile repo, HgCommandResult result) {
76     final Map<HgFile, HgResolveStatusEnum> resolveStatus = new HashMap<HgFile, HgResolveStatusEnum>();
77     for (String line : result.getOutputLines()) {
78       if (StringUtils.isBlank(line) || line.length() < ITEM_COUNT) {
79         continue;
80       }
81       final HgResolveStatusEnum status = HgResolveStatusEnum.valueOf(line.charAt(0));
82       if (status != null) {
83         File ioFile = new File(repo.getPath(), line.substring(2));
84         resolveStatus.put(new HgFile(repo, ioFile), status);
85       }
86     }
87     return resolveStatus;
88   }
89
90   public void markResolved(VirtualFile repo, VirtualFile path) {
91     new HgCommandExecutor(myProject).execute(repo, "resolve", Arrays.asList("--mark", path.getPath()), null);
92   }
93
94   public void markResolved(VirtualFile repo, FilePath path) {
95     new HgCommandExecutor(myProject).execute(repo, "resolve", Arrays.asList("--mark", path.getPath()), null);
96   }
97
98 }