]>
Commit | Line | Data |
---|---|---|
1 | #include "builtin.h" | |
2 | #include "config.h" | |
3 | #include "dir.h" | |
4 | #include "lockfile.h" | |
5 | #include "parse-options.h" | |
6 | #include "repository.h" | |
7 | #include "commit-graph.h" | |
8 | #include "object-store.h" | |
9 | ||
10 | static char const * const builtin_commit_graph_usage[] = { | |
11 | N_("git commit-graph [--object-dir <objdir>]"), | |
12 | N_("git commit-graph read [--object-dir <objdir>]"), | |
13 | N_("git commit-graph verify [--object-dir <objdir>] [--shallow] [--[no-]progress]"), | |
14 | N_("git commit-graph write [--object-dir <objdir>] [--append|--split] [--reachable|--stdin-packs|--stdin-commits] [--[no-]progress] <split options>"), | |
15 | NULL | |
16 | }; | |
17 | ||
18 | static const char * const builtin_commit_graph_verify_usage[] = { | |
19 | N_("git commit-graph verify [--object-dir <objdir>] [--shallow] [--[no-]progress]"), | |
20 | NULL | |
21 | }; | |
22 | ||
23 | static const char * const builtin_commit_graph_read_usage[] = { | |
24 | N_("git commit-graph read [--object-dir <objdir>]"), | |
25 | NULL | |
26 | }; | |
27 | ||
28 | static const char * const builtin_commit_graph_write_usage[] = { | |
29 | N_("git commit-graph write [--object-dir <objdir>] [--append|--split] [--reachable|--stdin-packs|--stdin-commits] [--[no-]progress] <split options>"), | |
30 | NULL | |
31 | }; | |
32 | ||
33 | static struct opts_commit_graph { | |
34 | const char *obj_dir; | |
35 | int reachable; | |
36 | int stdin_packs; | |
37 | int stdin_commits; | |
38 | int append; | |
39 | int split; | |
40 | int shallow; | |
41 | int progress; | |
42 | } opts; | |
43 | ||
44 | static int graph_verify(int argc, const char **argv) | |
45 | { | |
46 | struct commit_graph *graph = NULL; | |
47 | char *graph_name; | |
48 | int open_ok; | |
49 | int fd; | |
50 | struct stat st; | |
51 | int flags = 0; | |
52 | ||
53 | static struct option builtin_commit_graph_verify_options[] = { | |
54 | OPT_STRING(0, "object-dir", &opts.obj_dir, | |
55 | N_("dir"), | |
56 | N_("The object directory to store the graph")), | |
57 | OPT_BOOL(0, "shallow", &opts.shallow, | |
58 | N_("if the commit-graph is split, only verify the tip file")), | |
59 | OPT_BOOL(0, "progress", &opts.progress, N_("force progress reporting")), | |
60 | OPT_END(), | |
61 | }; | |
62 | ||
63 | trace2_cmd_mode("verify"); | |
64 | ||
65 | opts.progress = isatty(2); | |
66 | argc = parse_options(argc, argv, NULL, | |
67 | builtin_commit_graph_verify_options, | |
68 | builtin_commit_graph_verify_usage, 0); | |
69 | ||
70 | if (!opts.obj_dir) | |
71 | opts.obj_dir = get_object_directory(); | |
72 | if (opts.shallow) | |
73 | flags |= COMMIT_GRAPH_VERIFY_SHALLOW; | |
74 | if (opts.progress) | |
75 | flags |= COMMIT_GRAPH_WRITE_PROGRESS; | |
76 | ||
77 | graph_name = get_commit_graph_filename(opts.obj_dir); | |
78 | open_ok = open_commit_graph(graph_name, &fd, &st); | |
79 | if (!open_ok && errno != ENOENT) | |
80 | die_errno(_("Could not open commit-graph '%s'"), graph_name); | |
81 | ||
82 | FREE_AND_NULL(graph_name); | |
83 | ||
84 | if (open_ok) | |
85 | graph = load_commit_graph_one_fd_st(fd, &st); | |
86 | else | |
87 | graph = read_commit_graph_one(the_repository, opts.obj_dir); | |
88 | ||
89 | /* Return failure if open_ok predicted success */ | |
90 | if (!graph) | |
91 | return !!open_ok; | |
92 | ||
93 | UNLEAK(graph); | |
94 | return verify_commit_graph(the_repository, graph, flags); | |
95 | } | |
96 | ||
97 | static int graph_read(int argc, const char **argv) | |
98 | { | |
99 | struct commit_graph *graph = NULL; | |
100 | char *graph_name; | |
101 | int open_ok; | |
102 | int fd; | |
103 | struct stat st; | |
104 | ||
105 | static struct option builtin_commit_graph_read_options[] = { | |
106 | OPT_STRING(0, "object-dir", &opts.obj_dir, | |
107 | N_("dir"), | |
108 | N_("The object directory to store the graph")), | |
109 | OPT_END(), | |
110 | }; | |
111 | ||
112 | trace2_cmd_mode("read"); | |
113 | ||
114 | argc = parse_options(argc, argv, NULL, | |
115 | builtin_commit_graph_read_options, | |
116 | builtin_commit_graph_read_usage, 0); | |
117 | ||
118 | if (!opts.obj_dir) | |
119 | opts.obj_dir = get_object_directory(); | |
120 | ||
121 | graph_name = get_commit_graph_filename(opts.obj_dir); | |
122 | ||
123 | open_ok = open_commit_graph(graph_name, &fd, &st); | |
124 | if (!open_ok) | |
125 | die_errno(_("Could not open commit-graph '%s'"), graph_name); | |
126 | ||
127 | graph = load_commit_graph_one_fd_st(fd, &st); | |
128 | if (!graph) | |
129 | return 1; | |
130 | ||
131 | FREE_AND_NULL(graph_name); | |
132 | ||
133 | printf("header: %08x %d %d %d %d\n", | |
134 | ntohl(*(uint32_t*)graph->data), | |
135 | *(unsigned char*)(graph->data + 4), | |
136 | *(unsigned char*)(graph->data + 5), | |
137 | *(unsigned char*)(graph->data + 6), | |
138 | *(unsigned char*)(graph->data + 7)); | |
139 | printf("num_commits: %u\n", graph->num_commits); | |
140 | printf("chunks:"); | |
141 | ||
142 | if (graph->chunk_oid_fanout) | |
143 | printf(" oid_fanout"); | |
144 | if (graph->chunk_oid_lookup) | |
145 | printf(" oid_lookup"); | |
146 | if (graph->chunk_commit_data) | |
147 | printf(" commit_metadata"); | |
148 | if (graph->chunk_extra_edges) | |
149 | printf(" extra_edges"); | |
150 | printf("\n"); | |
151 | ||
152 | UNLEAK(graph); | |
153 | ||
154 | return 0; | |
155 | } | |
156 | ||
157 | extern int read_replace_refs; | |
158 | static struct split_commit_graph_opts split_opts; | |
159 | ||
160 | static int graph_write(int argc, const char **argv) | |
161 | { | |
162 | struct string_list *pack_indexes = NULL; | |
163 | struct string_list *commit_hex = NULL; | |
164 | struct string_list lines; | |
165 | int result = 0; | |
166 | enum commit_graph_write_flags flags = 0; | |
167 | ||
168 | static struct option builtin_commit_graph_write_options[] = { | |
169 | OPT_STRING(0, "object-dir", &opts.obj_dir, | |
170 | N_("dir"), | |
171 | N_("The object directory to store the graph")), | |
172 | OPT_BOOL(0, "reachable", &opts.reachable, | |
173 | N_("start walk at all refs")), | |
174 | OPT_BOOL(0, "stdin-packs", &opts.stdin_packs, | |
175 | N_("scan pack-indexes listed by stdin for commits")), | |
176 | OPT_BOOL(0, "stdin-commits", &opts.stdin_commits, | |
177 | N_("start walk at commits listed by stdin")), | |
178 | OPT_BOOL(0, "append", &opts.append, | |
179 | N_("include all commits already in the commit-graph file")), | |
180 | OPT_BOOL(0, "progress", &opts.progress, N_("force progress reporting")), | |
181 | OPT_BOOL(0, "split", &opts.split, | |
182 | N_("allow writing an incremental commit-graph file")), | |
183 | OPT_INTEGER(0, "max-commits", &split_opts.max_commits, | |
184 | N_("maximum number of commits in a non-base split commit-graph")), | |
185 | OPT_INTEGER(0, "size-multiple", &split_opts.size_multiple, | |
186 | N_("maximum ratio between two levels of a split commit-graph")), | |
187 | OPT_EXPIRY_DATE(0, "expire-time", &split_opts.expire_time, | |
188 | N_("maximum number of commits in a non-base split commit-graph")), | |
189 | OPT_END(), | |
190 | }; | |
191 | ||
192 | opts.progress = isatty(2); | |
193 | split_opts.size_multiple = 2; | |
194 | split_opts.max_commits = 0; | |
195 | split_opts.expire_time = 0; | |
196 | ||
197 | trace2_cmd_mode("write"); | |
198 | ||
199 | argc = parse_options(argc, argv, NULL, | |
200 | builtin_commit_graph_write_options, | |
201 | builtin_commit_graph_write_usage, 0); | |
202 | ||
203 | if (opts.reachable + opts.stdin_packs + opts.stdin_commits > 1) | |
204 | die(_("use at most one of --reachable, --stdin-commits, or --stdin-packs")); | |
205 | if (!opts.obj_dir) | |
206 | opts.obj_dir = get_object_directory(); | |
207 | if (opts.append) | |
208 | flags |= COMMIT_GRAPH_WRITE_APPEND; | |
209 | if (opts.split) | |
210 | flags |= COMMIT_GRAPH_WRITE_SPLIT; | |
211 | if (opts.progress) | |
212 | flags |= COMMIT_GRAPH_WRITE_PROGRESS; | |
213 | ||
214 | read_replace_refs = 0; | |
215 | ||
216 | if (opts.reachable) { | |
217 | if (write_commit_graph_reachable(opts.obj_dir, flags, &split_opts)) | |
218 | return 1; | |
219 | return 0; | |
220 | } | |
221 | ||
222 | string_list_init(&lines, 0); | |
223 | if (opts.stdin_packs || opts.stdin_commits) { | |
224 | struct strbuf buf = STRBUF_INIT; | |
225 | ||
226 | while (strbuf_getline(&buf, stdin) != EOF) | |
227 | string_list_append(&lines, strbuf_detach(&buf, NULL)); | |
228 | ||
229 | if (opts.stdin_packs) | |
230 | pack_indexes = &lines; | |
231 | if (opts.stdin_commits) { | |
232 | commit_hex = &lines; | |
233 | flags |= COMMIT_GRAPH_WRITE_CHECK_OIDS; | |
234 | } | |
235 | ||
236 | UNLEAK(buf); | |
237 | } | |
238 | ||
239 | if (write_commit_graph(opts.obj_dir, | |
240 | pack_indexes, | |
241 | commit_hex, | |
242 | flags, | |
243 | &split_opts)) | |
244 | result = 1; | |
245 | ||
246 | UNLEAK(lines); | |
247 | return result; | |
248 | } | |
249 | ||
250 | int cmd_commit_graph(int argc, const char **argv, const char *prefix) | |
251 | { | |
252 | static struct option builtin_commit_graph_options[] = { | |
253 | OPT_STRING(0, "object-dir", &opts.obj_dir, | |
254 | N_("dir"), | |
255 | N_("The object directory to store the graph")), | |
256 | OPT_END(), | |
257 | }; | |
258 | ||
259 | if (argc == 2 && !strcmp(argv[1], "-h")) | |
260 | usage_with_options(builtin_commit_graph_usage, | |
261 | builtin_commit_graph_options); | |
262 | ||
263 | git_config(git_default_config, NULL); | |
264 | argc = parse_options(argc, argv, prefix, | |
265 | builtin_commit_graph_options, | |
266 | builtin_commit_graph_usage, | |
267 | PARSE_OPT_STOP_AT_NON_OPTION); | |
268 | ||
269 | save_commit_buffer = 0; | |
270 | ||
271 | if (argc > 0) { | |
272 | if (!strcmp(argv[0], "read")) | |
273 | return graph_read(argc, argv); | |
274 | if (!strcmp(argv[0], "verify")) | |
275 | return graph_verify(argc, argv); | |
276 | if (!strcmp(argv[0], "write")) | |
277 | return graph_write(argc, argv); | |
278 | } | |
279 | ||
280 | usage_with_options(builtin_commit_graph_usage, | |
281 | builtin_commit_graph_options); | |
282 | } |