blob: c2e7c4b9db286726670b86be2b2357c45134b33a [file] [log] [blame]
Shawn O. Pearce27b07322009-04-10 16:02:48 -07001#
2# Copyright (C) 2009 The Android Open Source Project
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
Sarah Owenscecd1d82012-11-01 22:59:27 -070016from __future__ import print_function
Shawn O. Pearce27b07322009-04-10 16:02:48 -070017import sys
18from color import Coloring
19from command import Command
20
21class BranchColoring(Coloring):
22 def __init__(self, config):
23 Coloring.__init__(self, config, 'branch')
24 self.current = self.printer('current', fg='green')
25 self.local = self.printer('local')
26 self.notinproject = self.printer('notinproject', fg='red')
27
28class BranchInfo(object):
29 def __init__(self, name):
30 self.name = name
31 self.current = 0
32 self.published = 0
33 self.published_equal = 0
34 self.projects = []
35
36 def add(self, b):
37 if b.current:
38 self.current += 1
39 if b.published:
40 self.published += 1
41 if b.revision == b.published:
42 self.published_equal += 1
43 self.projects.append(b)
44
45 @property
46 def IsCurrent(self):
47 return self.current > 0
48
49 @property
50 def IsPublished(self):
51 return self.published > 0
52
53 @property
54 def IsPublishedEqual(self):
55 return self.published_equal == len(self.projects)
56
57
58class Branches(Command):
59 common = True
60 helpSummary = "View current topic branches"
61 helpUsage = """
62%prog [<project>...]
63
64Summarizes the currently available topic branches.
Shawn O. Pearce7da73d62009-06-12 17:35:43 -070065
66Branch Display
67--------------
68
69The branch display output by this command is organized into four
70columns of information; for example:
71
72 *P nocolor | in repo
73 repo2 |
74
75The first column contains a * if the branch is the currently
76checked out branch in any of the specified projects, or a blank
77if no project has the branch checked out.
78
79The second column contains either blank, p or P, depending upon
80the upload status of the branch.
81
82 (blank): branch not yet published by repo upload
83 P: all commits were published by repo upload
84 p: only some commits were published by repo upload
85
86The third column contains the branch name.
87
88The fourth column (after the | separator) lists the projects that
89the branch appears in, or does not appear in. If no project list
90is shown, then the branch appears in all projects.
91
Shawn O. Pearce27b07322009-04-10 16:02:48 -070092"""
93
Shawn O. Pearce27b07322009-04-10 16:02:48 -070094 def Execute(self, opt, args):
95 projects = self.GetProjects(args)
96 out = BranchColoring(self.manifest.manifestProject.config)
David Pursehouse5c6eeac2012-10-11 16:44:48 +090097 all_branches = {}
Shawn O. Pearce27b07322009-04-10 16:02:48 -070098 project_cnt = len(projects)
99
100 for project in projects:
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530101 for name, b in project.GetBranches().items():
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700102 b.project = project
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900103 if name not in all_branches:
104 all_branches[name] = BranchInfo(name)
105 all_branches[name].add(b)
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700106
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530107 names = list(sorted(all_branches))
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700108
Shawn O. Pearce4e3d6732009-04-18 15:18:35 -0700109 if not names:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700110 print(' (no branches)', file=sys.stderr)
Shawn O. Pearce4e3d6732009-04-18 15:18:35 -0700111 return
112
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700113 width = 25
114 for name in names:
115 if width < len(name):
116 width = len(name)
117
118 for name in names:
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900119 i = all_branches[name]
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700120 in_cnt = len(i.projects)
121
122 if i.IsCurrent:
123 current = '*'
124 hdr = out.current
125 else:
126 current = ' '
127 hdr = out.local
128
129 if i.IsPublishedEqual:
130 published = 'P'
131 elif i.IsPublished:
132 published = 'p'
133 else:
134 published = ' '
135
136 hdr('%c%c %-*s' % (current, published, width, name))
137 out.write(' |')
138
Pär Åsfältff6929d2009-09-05 23:10:56 +0200139 if in_cnt < project_cnt:
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700140 fmt = out.write
141 paths = []
142 if in_cnt < project_cnt - in_cnt:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900143 in_type = 'in'
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700144 for b in i.projects:
145 paths.append(b.project.relpath)
146 else:
147 fmt = out.notinproject
David Pursehouse8a68ff92012-09-24 12:15:13 +0900148 in_type = 'not in'
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700149 have = set()
150 for b in i.projects:
151 have.add(b.project)
152 for p in projects:
Pär Åsfältff6929d2009-09-05 23:10:56 +0200153 if not p in have:
154 paths.append(p.relpath)
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700155
David Pursehouse8a68ff92012-09-24 12:15:13 +0900156 s = ' %s %s' % (in_type, ', '.join(paths))
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700157 if width + 7 + len(s) < 80:
158 fmt(s)
159 else:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900160 fmt(' %s:' % in_type)
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700161 for p in paths:
162 out.nl()
Pär Åsfältff6929d2009-09-05 23:10:56 +0200163 fmt(width*' ' + ' %s' % p)
164 else:
165 out.write(' in all projects')
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700166 out.nl()