Mercurial > hg4j

comparison cmdline/org/tmatesoft/hg/console/Main.java @ 143:b9700740553a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Command line tools parse and respect most of command-line arguments
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Thu, 17 Feb 2011 22:16:25 +0100
parents
children d5268ca7715b
comparison
equal deleted inserted replaced
142:37a34044e6bd 143:b9700740553a
1 /*
2 * Copyright (c) 2011 TMate Software Ltd
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; version 2 of the License.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 * For information on how to redistribute this software under
14 * the terms of a license other than GNU General Public License
15 * contact TMate Software at support@hg4j.com
16 */
17 package org.tmatesoft.hg.console;
18
19 import static org.tmatesoft.hg.repo.HgRepository.TIP;
20
21 import java.util.ArrayList;
22 import java.util.Collections;
23 import java.util.List;
24 import java.util.Map;
25
26 import org.tmatesoft.hg.core.HgLogCommand.FileRevision;
27 import org.tmatesoft.hg.core.HgManifestCommand;
28 import org.tmatesoft.hg.core.Nodeid;
29 import org.tmatesoft.hg.internal.DigestHelper;
30 import org.tmatesoft.hg.repo.HgDataFile;
31 import org.tmatesoft.hg.repo.HgInternals;
32 import org.tmatesoft.hg.repo.HgManifest;
33 import org.tmatesoft.hg.repo.HgRepository;
34 import org.tmatesoft.hg.repo.HgStatusCollector;
35 import org.tmatesoft.hg.repo.HgStatusInspector;
36 import org.tmatesoft.hg.repo.HgWorkingCopyStatusCollector;
37 import org.tmatesoft.hg.util.Path;
38
39 /**
40 * Various debug dumps.
41 *
42 * @author Artem Tikhomirov
43 * @author TMate Software Ltd.
44 */
45 public class Main {
46
47 private Options cmdLineOpts;
48 private HgRepository hgRepo;
49
50 public Main(String[] args) throws Exception {
51 cmdLineOpts = Options.parse(args);
52 hgRepo = cmdLineOpts.findRepository();
53 if (hgRepo.isInvalid()) {
54 System.err.printf("Can't find repository in: %s\n", hgRepo.getLocation());
55 return;
56 }
57 System.out.println("REPO:" + hgRepo.getLocation());
58 }
59
60 public static void main(String[] args) throws Exception {
61 Main m = new Main(args);
62 m.dumpIgnored();
63 m.dumpDirstate();
64 m.testStatusInternals();
65 m.catCompleteHistory();
66 m.dumpCompleteManifestLow();
67 m.dumpCompleteManifestHigh();
68 m.bunchOfTests();
69 }
70
71 private void dumpIgnored() {
72 HgInternals debug = new HgInternals(hgRepo);
73 String[] toCheck = new String[] {"design.txt", "src/com/tmate/hgkit/ll/Changelog.java", "src/Extras.java", "bin/com/tmate/hgkit/ll/Changelog.class"};
74 boolean[] checkResult = debug.checkIgnored(toCheck);
75 for (int i = 0; i < toCheck.length; i++) {
76 System.out.println("Ignored " + toCheck[i] + ": " + checkResult[i]);
77 }
78 }
79
80 private void dumpDirstate() {
81 new HgInternals(hgRepo).dumpDirstate();
82 }
83
84
85 private void catCompleteHistory() {
86 DigestHelper dh = new DigestHelper();
87 for (String fname : cmdLineOpts.getList("")) {
88 System.out.println(fname);
89 HgDataFile fn = hgRepo.getFileNode(fname);
90 if (fn.exists()) {
91 int total = fn.getRevisionCount();
92 System.out.printf("Total revisions: %d\n", total);
93 for (int i = 0; i < total; i++) {
94 byte[] content = fn.content(i);
95 System.out.println("==========>");
96 System.out.println(new String(content));
97 int[] parentRevisions = new int[2];
98 byte[] parent1 = new byte[20];
99 byte[] parent2 = new byte[20];
100 fn.parents(i, parentRevisions, parent1, parent2);
101 System.out.println(dh.sha1(parent1, parent2, content).asHexString());
102 }
103 } else {
104 System.out.println(">>>Not found!");
105 }
106 }
107 }
108
109 private void dumpCompleteManifestLow() {
110 hgRepo.getManifest().walk(0, TIP, new ManifestDump());
111 }
112
113 public static final class ManifestDump implements HgManifest.Inspector {
114 public boolean begin(int revision, Nodeid nid) {
115 System.out.printf("%d : %s\n", revision, nid);
116 return true;
117 }
118
119 public boolean next(Nodeid nid, String fname, String flags) {
120 System.out.println(nid + "\t" + fname + "\t\t" + flags);
121 return true;
122 }
123
124 public boolean end(int revision) {
125 System.out.println();
126 return true;
127 }
128 }
129
130 private void dumpCompleteManifestHigh() {
131 new HgManifestCommand(hgRepo).dirs(true).execute(new HgManifestCommand.Handler() {
132
133 public void begin(Nodeid manifestRevision) {
134 System.out.println(">> " + manifestRevision);
135 }
136 public void dir(Path p) {
137 System.out.println(p);
138 }
139 public void file(FileRevision fileRevision) {
140 System.out.print(fileRevision.getRevision());;
141 System.out.print(" ");
142 System.out.println(fileRevision.getPath());
143 }
144
145 public void end(Nodeid manifestRevision) {
146 System.out.println();
147 }
148 });
149 }
150
151 private void bunchOfTests() throws Exception {
152 HgInternals debug = new HgInternals(hgRepo);
153 debug.dumpDirstate();
154 final StatusDump dump = new StatusDump();
155 dump.showIgnored = false;
156 dump.showClean = false;
157 HgStatusCollector sc = new HgStatusCollector(hgRepo);
158 final int r1 = 0, r2 = 3;
159 System.out.printf("Status for changes between revision %d and %d:\n", r1, r2);
160 sc.walk(r1, r2, dump);
161 //
162 System.out.println("\n\nSame, but sorted in the way hg status does:");
163 HgStatusCollector.Record r = sc.status(r1, r2);
164 sortAndPrint('M', r.getModified(), null);
165 sortAndPrint('A', r.getAdded(), null);
166 sortAndPrint('R', r.getRemoved(), null);
167 //
168 System.out.println("\n\nTry hg status --change <rev>:");
169 sc.change(0, dump);
170 System.out.println("\nStatus against working dir:");
171 HgWorkingCopyStatusCollector wcc = new HgWorkingCopyStatusCollector(hgRepo);
172 wcc.walk(TIP, dump);
173 System.out.println();
174 System.out.printf("Manifest of the revision %d:\n", r2);
175 hgRepo.getManifest().walk(r2, r2, new ManifestDump());
176 System.out.println();
177 System.out.printf("\nStatus of working dir against %d:\n", r2);
178 r = wcc.status(r2);
179 sortAndPrint('M', r.getModified(), null);
180 sortAndPrint('A', r.getAdded(), r.getCopied());
181 sortAndPrint('R', r.getRemoved(), null);
182 sortAndPrint('?', r.getUnknown(), null);
183 sortAndPrint('I', r.getIgnored(), null);
184 sortAndPrint('C', r.getClean(), null);
185 sortAndPrint('!', r.getMissing(), null);
186 }
187
188 private void sortAndPrint(char prefix, List<Path> ul, Map<Path, Path> copies) {
189 ArrayList<Path> sortList = new ArrayList<Path>(ul);
190 Collections.sort(sortList);
191 for (Path s : sortList) {
192 System.out.print(prefix);
193 System.out.print(' ');
194 System.out.println(s);
195 if (copies != null && copies.containsKey(s)) {
196 System.out.println(" " + copies.get(s));
197 }
198 }
199 }
200
201
202 private void testStatusInternals() {
203 HgDataFile n = hgRepo.getFileNode(Path.create("design.txt"));
204 for (String s : new String[] {"011dfd44417c72bd9e54cf89b82828f661b700ed", "e5529faa06d53e06a816e56d218115b42782f1ba", "c18e7111f1fc89a80a00f6a39d51288289a382fc"}) {
205 // expected: 359, 2123, 3079
206 byte[] b = s.getBytes();
207 final Nodeid nid = Nodeid.fromAscii(b, 0, b.length);
208 System.out.println(s + " : " + n.length(nid));
209 }
210 }
211
212 private static class StatusDump implements HgStatusInspector {
213 public boolean hideStatusPrefix = false; // hg status -n option
214 public boolean showCopied = true; // -C
215 public boolean showIgnored = true; // -i
216 public boolean showClean = true; // -c
217
218 public void modified(Path fname) {
219 print('M', fname);
220 }
221
222 public void added(Path fname) {
223 print('A', fname);
224 }
225
226 public void copied(Path fnameOrigin, Path fnameAdded) {
227 added(fnameAdded);
228 if (showCopied) {
229 print(' ', fnameOrigin);
230 }
231 }
232
233 public void removed(Path fname) {
234 print('R', fname);
235 }
236
237 public void clean(Path fname) {
238 if (showClean) {
239 print('C', fname);
240 }
241 }
242
243 public void missing(Path fname) {
244 print('!', fname);
245 }
246
247 public void unknown(Path fname) {
248 print('?', fname);
249 }
250
251 public void ignored(Path fname) {
252 if (showIgnored) {
253 print('I', fname);
254 }
255 }
256
257 private void print(char status, Path fname) {
258 if (!hideStatusPrefix) {
259 System.out.print(status);
260 System.out.print(' ');
261 }
262 System.out.println(fname);
263 }
264 }
265 }