[git] refs parser can recognize lines like HEAD -> refs/heads/master in git log output
[idea/community.git] / plugins / git4idea / tests / git4idea / test / RefParser.java
1 /*
2  * Copyright 2000-2015 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.test;
17
18 import com.intellij.openapi.util.text.StringUtil;
19 import com.intellij.openapi.vfs.VirtualFile;
20 import com.intellij.vcs.log.Hash;
21 import com.intellij.vcs.log.VcsLogObjectsFactory;
22 import com.intellij.vcs.log.VcsRef;
23 import com.intellij.vcs.log.VcsRefType;
24 import com.intellij.vcs.log.impl.HashImpl;
25 import git4idea.branch.GitBranchUtil;
26 import git4idea.log.GitRefManager;
27 import org.jetbrains.annotations.NotNull;
28 import org.jetbrains.annotations.Nullable;
29
30 import java.util.ArrayList;
31 import java.util.Collections;
32 import java.util.List;
33
34 class RefParser {
35
36   private final VcsLogObjectsFactory myFactory;
37
38   public RefParser(@NotNull VcsLogObjectsFactory factory) {
39     myFactory = factory;
40   }
41
42   // e25b7d8f (HEAD, refs/remotes/origin/master, refs/remotes/origin/HEAD, refs/heads/master)
43   public List<VcsRef> parseCommitRefs(@NotNull String input, @NotNull VirtualFile root) {
44     int firstSpaceIndex = input.indexOf(' ');
45     if (firstSpaceIndex < 0) {
46       return Collections.emptyList();
47     }
48     String strHash = input.substring(0, firstSpaceIndex);
49     Hash hash = HashImpl.build(strHash);
50     String refPaths = input.substring(firstSpaceIndex + 2, input.length() - 1);
51     String[] longRefPaths = refPaths.split("(, )|( -> )");
52     List<VcsRef> refs = new ArrayList<>();
53     for (String longRefPatch : longRefPaths) {
54       VcsRef ref = createRef(hash, longRefPatch, root);
55       if (ref != null) {
56         refs.add(ref);
57       }
58     }
59     return refs;
60   }
61
62   @NotNull
63   private static String getRefName(@NotNull String longRefPath) {
64     String tagPrefix = "tag: ";
65     longRefPath = StringUtil.trimStart(longRefPath, tagPrefix);
66     return longRefPath;
67   }
68
69   // example input: fb29c80 refs/tags/92.29
70   @Nullable
71   private VcsRef createRef(@NotNull Hash hash, @NotNull String longRefPath, @NotNull VirtualFile root) {
72     String name = getRefName(longRefPath);
73     VcsRefType type = GitRefManager.getRefType(name);
74     assert type != null;
75     return myFactory.createRef(hash, GitBranchUtil.stripRefsPrefix(name), type, root);
76   }
77
78 }