]>
Commit | Line | Data |
---|---|---|
1 | #!/bin/bash | |
2 | g() { git "$@"; } | |
3 | err() { printf '%b\n' "$*" 1>&2; exit 1; } | |
4 | vr() { echo "$@"; "$@"; } | |
5 | ||
6 | case $1 in | |
7 | -x) set -x; shift;; | |
8 | esac | |
9 | ||
10 | case $1 in | |
11 | rb-all) | |
12 | cmd=rebase | |
13 | shift | |
14 | set -- all "$@" | |
15 | ;; | |
16 | esac | |
17 | ||
18 | mj_init() { | |
19 | pipe=$(mktemp) | |
20 | rm -f "${pipe}" | |
21 | mkfifo "${pipe}" | |
22 | exec {ctlfd}<>"${pipe}" | |
23 | rm -f "${pipe}" | |
24 | jobs=0 | |
25 | jobs_max=${1:-$(getconf _NPROCESSORS_ONLN)} | |
26 | } | |
27 | _mj_child() { echo ${BASHPID} $? >&${ctlfd} ; } | |
28 | mj_child() { | |
29 | ( | |
30 | "$@" | |
31 | _mj_child | |
32 | ) & | |
33 | mj_post_child | |
34 | } | |
35 | mj_post_child() { | |
36 | : $(( ++jobs )) | |
37 | if [[ ${jobs} -eq ${jobs_max} ]] ; then | |
38 | read -r -u ${ctlfd} pid ret | |
39 | : $(( --jobs )) | |
40 | fi | |
41 | } | |
42 | mj_finish() { | |
43 | wait | |
44 | } | |
45 | ||
46 | repo_root() { | |
47 | local root=${PWD} | |
48 | while [[ ! -d ${root}/.repo && ${root:-/} != "/" ]] ; do | |
49 | root=${root%/*} | |
50 | done | |
51 | echo "${root}" | |
52 | } | |
53 | ||
54 | process_reviewers() { | |
55 | local r arr=() | |
56 | for r in ${*//,/ } ; do | |
57 | case ${r} in | |
58 | OWNERS*|*/OWNERS*) | |
59 | local owners=$( | |
60 | awk -F'@' ' | |
61 | ($2 == "chromium.org" || $2 == "google.com") {list = list "," $1} | |
62 | END {print substr(list, 2)} | |
63 | ' "${r}" | |
64 | ) | |
65 | if [[ -z ${owners} ]] ; then | |
66 | err "cannot find OWNERS list" | |
67 | else | |
68 | echo "Auto setting reviewers to: ${owners}" | |
69 | fi | |
70 | arr+=( ${owners} ) | |
71 | ;; | |
72 | *) | |
73 | arr+=( "${r}" ) | |
74 | ;; | |
75 | esac | |
76 | done | |
77 | reviewers=$(printf '%s,' "${arr[@]}") | |
78 | reviewers=${reviewers%,} | |
79 | } | |
80 | ||
81 | find_repo() { | |
82 | # Diff projects have diff versions of repo. Find a compatible one. | |
83 | local root=$(repo_root) | |
84 | ||
85 | # Use the manifest repo URL. | |
86 | local manifest_dir="${root}/.repo/manifests.git" | |
87 | local d | |
88 | ||
89 | # Default to the local repo if it's there. | |
90 | local search=( | |
91 | "${root}/.repo/repo" | |
92 | ) | |
93 | ||
94 | case $(g --git-dir="${manifest_dir}" config remote.origin.url) in | |
95 | *android*) | |
96 | search+=( | |
97 | /usr/local/src/repo | |
98 | ~/src/repo | |
99 | ) | |
100 | ;; | |
101 | *chromium*|*chrome*) | |
102 | search+=( | |
103 | ~/depot_tools | |
104 | ~/chromiumos/depot_tools | |
105 | /usr/local/src/depot_tools | |
106 | ~/src/depot_tools | |
107 | ) | |
108 | ;; | |
109 | esac | |
110 | for d in "${search[@]}" ; do | |
111 | if [[ -x ${d}/repo ]] ; then | |
112 | echo "${d}"/repo | |
113 | return | |
114 | fi | |
115 | done | |
116 | ||
117 | # Fallback: use $PATH. | |
118 | type -P repo | |
119 | } | |
120 | ||
121 | case ${acmd:-${cmd}} in | |
122 | rebase) | |
123 | if [[ $1 == "all" ]] ; then | |
124 | shift | |
125 | if [[ $# -eq 0 ]] ; then | |
126 | # This bash version is still slightly faster than repo! | |
127 | #exec r forall -j$(getconf _NPROCESSORS_ONLN) -p -c git rb-all | |
128 | ||
129 | eval "$(bash-colors --env | sed 's:^:export :')" | |
130 | root=$(repo_root) | |
131 | mj_init | |
132 | while read -a line ; do | |
133 | dir=${line[0]} | |
134 | proj=${line[2]} | |
135 | if ! cd "${root}/${dir}" ; then | |
136 | echo "bad ${proj}" | |
137 | continue | |
138 | fi | |
139 | ( | |
140 | out=$(r rb-all . 2>&1) | |
141 | if [[ -n ${out} ]] ; then | |
142 | head=$(printf "%-40s" "${proj}") | |
143 | echo "${out}" | sed "s:^:${head} :" | |
144 | fi | |
145 | _mj_child | |
146 | ) & | |
147 | mj_post_child | |
148 | done < <(r l) | |
149 | mj_finish | |
150 | exit 0 | |
151 | fi | |
152 | ||
153 | exec git rb-all -q | |
154 | fi | |
155 | ;; | |
156 | clean) | |
157 | root=$(repo_root) | |
158 | cd "${root}" || exit 1 | |
159 | mj_init | |
160 | while read -a line ; do | |
161 | dir=${line[0]} | |
162 | proj=${line[2]} | |
163 | cd "${root}/${dir}" | |
164 | ( | |
165 | out=$(g clean "$@" 2>&1) | |
166 | if [[ -n ${out} ]] ; then | |
167 | echo "### ${proj}" | |
168 | echo "${out}" | |
169 | fi | |
170 | _mj_child | |
171 | ) & | |
172 | mj_post_child | |
173 | done < <(r l) | |
174 | mj_finish | |
175 | exit | |
176 | ;; | |
177 | sb) | |
178 | sb_cmd=$1 | |
179 | case ${sb_cmd} in | |
180 | pull) ;; | |
181 | push) ;; | |
182 | f|fetch) sb_cmd="fetch" ;; | |
183 | *) err "unknown sandbox command: $1" | |
184 | esac | |
185 | ||
186 | sync_branch="v" | |
187 | ||
188 | root=$(repo_root) | |
189 | cd "${root}" || exit 1 | |
190 | ||
191 | if [[ ! -e .repo/sandbox-url ]] ; then | |
192 | err "Please configure remote url base in ${root}/.repo/sandbox-url" | |
193 | fi | |
194 | remote=$(<.repo/sandbox-url) || exit 1 | |
195 | ||
196 | echo "pushing projects from ${root}" | |
197 | ||
198 | # ssh servers do not like it when you hammer them :) | |
199 | # Received disconnect from 74.125.248.80: 7: Too many concurrent connections | |
200 | # fatal: The remote end hung up unexpectedly | |
201 | mj_init 16 | |
202 | ||
203 | rlist=$(r l) | |
204 | tcnt=$(echo "${rlist}" | wc -l) | |
205 | cnt=1 | |
206 | while read line ; do | |
207 | line=( ${line} ) | |
208 | path=${line[0]} | |
209 | export GIT_DIR=${path}/.git | |
210 | proj=${line[2]} | |
211 | printf '### (%*i/%i %3i%%) %s\n' \ | |
212 | ${#tcnt} $((cnt++)) ${tcnt} $(( cnt * 100 / tcnt )) ${proj} | |
213 | src="${sync_branch}" | |
214 | case ${sb_cmd} in | |
215 | push) | |
216 | g l -1 ${src} >& /dev/null || src= | |
217 | mj_child g push --force ${remote}/${proj} ${src}:refs/sandbox/${USER}/${sync_branch} >/dev/null | |
218 | ;; | |
219 | pull) | |
220 | ;; | |
221 | fetch) | |
222 | mj_child g fetch ${remote}/${proj} refs/sandbox/${USER}/${sync_branch}:refs/remotes/sb/${sync_branch} >/dev/null | |
223 | ;; | |
224 | esac | |
225 | done < <(echo "${rlist}") | |
226 | mj_finish | |
227 | ||
228 | exit 0 | |
229 | ;; | |
230 | upload) | |
231 | args=() | |
232 | while [[ $# -gt 0 ]] ; do | |
233 | case $1 in | |
234 | --re) | |
235 | process_reviewers "$2" | |
236 | args+=( --re "${reviewers}" ) | |
237 | shift 2 | |
238 | continue | |
239 | ;; | |
240 | esac | |
241 | args+=( "$1" ) | |
242 | shift | |
243 | done | |
244 | set -- "${args[@]}" | |
245 | ;; | |
246 | email) | |
247 | email=${1:-${USER}@chromium.org} | |
248 | ||
249 | root=$(repo_root) | |
250 | git --git-dir="${root}/.repo/manifests.git" cfg user.email "${email}" | |
251 | git --git-dir="${root}/.repo/repo/.git" cfg user.email "${email}" | |
252 | ||
253 | cmd='forall' | |
254 | set -- -c "git cfg user.email '${email}'" | |
255 | ;; | |
256 | esac | |
257 | ||
258 | exec $(find_repo) ${acmd:-${cmd}} "$@" |