2 f42b1b34 2018-03-12 stsp * Copyright (c) 2017 Martin Pieuchot <mpi@openbsd.org>
3 f42b1b34 2018-03-12 stsp * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
5 5c860e29 2018-03-12 stsp * Permission to use, copy, modify, and distribute this software for any
6 5c860e29 2018-03-12 stsp * purpose with or without fee is hereby granted, provided that the above
7 5c860e29 2018-03-12 stsp * copyright notice and this permission notice appear in all copies.
9 5c860e29 2018-03-12 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 5c860e29 2018-03-12 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 5c860e29 2018-03-12 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 5c860e29 2018-03-12 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 5c860e29 2018-03-12 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 5c860e29 2018-03-12 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 5c860e29 2018-03-12 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 f42b1b34 2018-03-12 stsp #include <sys/queue.h>
20 5c860e29 2018-03-12 stsp #include <err.h>
21 5c860e29 2018-03-12 stsp #include <errno.h>
22 5c860e29 2018-03-12 stsp #include <locale.h>
23 5c860e29 2018-03-12 stsp #include <stdio.h>
24 5c860e29 2018-03-12 stsp #include <stdlib.h>
25 5c860e29 2018-03-12 stsp #include <string.h>
26 5c860e29 2018-03-12 stsp #include <unistd.h>
27 c09a553d 2018-03-12 stsp #include <libgen.h>
29 f42b1b34 2018-03-12 stsp #include "got_error.h"
30 f42b1b34 2018-03-12 stsp #include "got_object.h"
31 f42b1b34 2018-03-12 stsp #include "got_refs.h"
32 f42b1b34 2018-03-12 stsp #include "got_repository.h"
33 c09a553d 2018-03-12 stsp #include "got_worktree.h"
35 5c860e29 2018-03-12 stsp #ifndef nitems
36 5c860e29 2018-03-12 stsp #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
39 5c860e29 2018-03-12 stsp struct cmd {
40 5c860e29 2018-03-12 stsp const char *cmd_name;
41 d7d4f210 2018-03-12 stsp const struct got_error *(*cmd_main)(int, char *[]);
42 1b6b95a8 2018-03-12 stsp void (*cmd_usage)(void);
43 46a0db7d 2018-03-12 stsp const char *cmd_descr;
46 5c860e29 2018-03-12 stsp __dead void usage(void);
47 c09a553d 2018-03-12 stsp __dead void usage_checkout(void);
48 6f3d1eb0 2018-03-12 stsp __dead void usage_log(void);
50 c09a553d 2018-03-12 stsp const struct got_error* cmd_checkout(int, char *[]);
51 d7d4f210 2018-03-12 stsp const struct got_error* cmd_log(int, char *[]);
52 d7d4f210 2018-03-12 stsp const struct got_error* cmd_status(int, char *[]);
54 5c860e29 2018-03-12 stsp struct cmd got_commands[] = {
55 c09a553d 2018-03-12 stsp { "checkout", cmd_checkout, usage_checkout,
56 c09a553d 2018-03-12 stsp "check out a work tree from a repository" },
57 1b6b95a8 2018-03-12 stsp { "log", cmd_log, usage_log,
58 1b6b95a8 2018-03-12 stsp "show repository history" },
59 f42b1b34 2018-03-12 stsp #ifdef notyet
60 1b6b95a8 2018-03-12 stsp { "status", cmd_status, usage_status,
61 1b6b95a8 2018-03-12 stsp "show modification status of files" },
66 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
68 5c860e29 2018-03-12 stsp struct cmd *cmd;
69 5c860e29 2018-03-12 stsp unsigned int i;
71 1b6b95a8 2018-03-12 stsp int hflag = 0;
73 5c860e29 2018-03-12 stsp setlocale(LC_ALL, "");
75 1b6b95a8 2018-03-12 stsp while ((ch = getopt(argc, argv, "h")) != -1) {
76 5c860e29 2018-03-12 stsp switch (ch) {
82 5c860e29 2018-03-12 stsp /* NOTREACHED */
86 5c860e29 2018-03-12 stsp argc -= optind;
87 5c860e29 2018-03-12 stsp argv += optind;
89 5c860e29 2018-03-12 stsp if (argc <= 0)
92 5c860e29 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
93 d7d4f210 2018-03-12 stsp const struct got_error *error;
95 5c860e29 2018-03-12 stsp cmd = &got_commands[i];
97 5c860e29 2018-03-12 stsp if (strncmp(cmd->cmd_name, argv[0], strlen(argv[0])))
101 1b6b95a8 2018-03-12 stsp got_commands[i].cmd_usage();
103 d7d4f210 2018-03-12 stsp error = got_commands[i].cmd_main(argc, argv);
104 d7d4f210 2018-03-12 stsp if (error) {
105 d7d4f210 2018-03-12 stsp fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
112 20ecf764 2018-03-12 stsp fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
116 5c860e29 2018-03-12 stsp __dead void
117 5c860e29 2018-03-12 stsp usage(void)
121 987e94ba 2018-03-12 stsp fprintf(stderr, "usage: %s [-h] command [arg ...]\n\n"
122 987e94ba 2018-03-12 stsp "Available commands:\n", getprogname());
123 46a0db7d 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
124 46a0db7d 2018-03-12 stsp struct cmd *cmd = &got_commands[i];
125 46a0db7d 2018-03-12 stsp fprintf(stderr, " %s: %s\n", cmd->cmd_name, cmd->cmd_descr);
130 c09a553d 2018-03-12 stsp __dead void
131 c09a553d 2018-03-12 stsp usage_checkout(void)
133 c09a553d 2018-03-12 stsp fprintf(stderr, "usage: %s checkout REPO_PATH [WORKTREE_PATH]\n",
134 c09a553d 2018-03-12 stsp getprogname());
138 92a684f4 2018-03-12 stsp static void
139 92a684f4 2018-03-12 stsp checkout_progress(void *arg, const char *path)
141 92a684f4 2018-03-12 stsp char *worktree_path = arg;
143 92a684f4 2018-03-12 stsp while (path[0] == '/')
146 92a684f4 2018-03-12 stsp printf("A %s/%s\n", worktree_path, path);
149 c09a553d 2018-03-12 stsp const struct got_error *
150 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
152 c09a553d 2018-03-12 stsp const struct got_error *error = NULL;
153 c09a553d 2018-03-12 stsp struct got_repository *repo = NULL;
154 c09a553d 2018-03-12 stsp struct got_reference *head_ref = NULL;
155 c09a553d 2018-03-12 stsp struct got_worktree *worktree = NULL;
156 c09a553d 2018-03-12 stsp char *repo_path = NULL;
157 c09a553d 2018-03-12 stsp char *worktree_path = NULL;
159 c09a553d 2018-03-12 stsp if (pledge("stdio rpath wpath cpath flock", NULL) == -1)
160 c09a553d 2018-03-12 stsp err(1, "pledge");
162 c09a553d 2018-03-12 stsp if (argc == 2) {
163 c09a553d 2018-03-12 stsp char *cwd, *base, *dotgit;
164 c09a553d 2018-03-12 stsp repo_path = argv[1];
165 c09a553d 2018-03-12 stsp cwd = getcwd(NULL, 0);
166 c09a553d 2018-03-12 stsp if (cwd == NULL)
167 c09a553d 2018-03-12 stsp err(1, "getcwd");
168 c09a553d 2018-03-12 stsp base = basename(repo_path);
169 c09a553d 2018-03-12 stsp if (base == NULL)
170 c09a553d 2018-03-12 stsp err(1, "basename");
171 c09a553d 2018-03-12 stsp dotgit = strstr(base, ".git");
172 c09a553d 2018-03-12 stsp if (dotgit)
173 c09a553d 2018-03-12 stsp *dotgit = '\0';
174 c09a553d 2018-03-12 stsp if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
176 c09a553d 2018-03-12 stsp return got_error(GOT_ERR_NO_MEM);
179 c09a553d 2018-03-12 stsp } else if (argc == 3) {
180 c09a553d 2018-03-12 stsp repo_path = argv[1];
181 c09a553d 2018-03-12 stsp worktree_path = strdup(argv[2]);
182 c09a553d 2018-03-12 stsp if (worktree_path == NULL)
183 c09a553d 2018-03-12 stsp return got_error(GOT_ERR_NO_MEM);
185 c09a553d 2018-03-12 stsp usage_checkout();
187 c09a553d 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
188 c09a553d 2018-03-12 stsp if (error != NULL)
190 c09a553d 2018-03-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
191 c09a553d 2018-03-12 stsp if (error != NULL)
194 c09a553d 2018-03-12 stsp error = got_worktree_init(worktree_path, head_ref, "/", repo);
195 c09a553d 2018-03-12 stsp if (error != NULL)
198 c09a553d 2018-03-12 stsp error = got_worktree_open(&worktree, worktree_path);
199 c09a553d 2018-03-12 stsp if (error != NULL)
202 92a684f4 2018-03-12 stsp error = got_worktree_checkout_files(worktree, head_ref, repo,
203 92a684f4 2018-03-12 stsp checkout_progress, worktree_path);
204 c09a553d 2018-03-12 stsp if (error != NULL)
207 c09a553d 2018-03-12 stsp printf("checked out %s\n", worktree_path);
210 c09a553d 2018-03-12 stsp free(worktree_path);
211 c09a553d 2018-03-12 stsp return error;
214 f42b1b34 2018-03-12 stsp static const struct got_error *
215 f42b1b34 2018-03-12 stsp print_commit_object(struct got_object *, struct got_object_id *,
216 f42b1b34 2018-03-12 stsp struct got_repository *);
218 f42b1b34 2018-03-12 stsp static const struct got_error *
219 f42b1b34 2018-03-12 stsp print_parent_commits(struct got_commit_object *commit,
220 f42b1b34 2018-03-12 stsp struct got_repository *repo)
222 f42b1b34 2018-03-12 stsp struct got_parent_id *pid;
223 f42b1b34 2018-03-12 stsp const struct got_error *err = NULL;
224 f42b1b34 2018-03-12 stsp struct got_object *obj;
226 f42b1b34 2018-03-12 stsp SIMPLEQ_FOREACH(pid, &commit->parent_ids, entry) {
227 f42b1b34 2018-03-12 stsp err = got_object_open(&obj, repo, pid->id);
228 f42b1b34 2018-03-12 stsp if (err != NULL)
229 f42b1b34 2018-03-12 stsp return err;
230 f42b1b34 2018-03-12 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_COMMIT)
231 f42b1b34 2018-03-12 stsp return got_error(GOT_ERR_OBJ_TYPE);
232 f42b1b34 2018-03-12 stsp err = print_commit_object(obj, pid->id, repo);
233 f42b1b34 2018-03-12 stsp got_object_close(obj);
236 f42b1b34 2018-03-12 stsp return err;
239 f42b1b34 2018-03-12 stsp static const struct got_error *
240 f42b1b34 2018-03-12 stsp print_commit_object(struct got_object *obj, struct got_object_id *id,
241 f42b1b34 2018-03-12 stsp struct got_repository *repo)
243 f42b1b34 2018-03-12 stsp struct got_commit_object *commit;
245 f42b1b34 2018-03-12 stsp const struct got_error *err;
247 f42b1b34 2018-03-12 stsp err = got_object_commit_open(&commit, repo, obj);
249 f42b1b34 2018-03-12 stsp return err;
251 f42b1b34 2018-03-12 stsp err = got_object_id_str(&buf, id);
253 f42b1b34 2018-03-12 stsp return err;
255 f42b1b34 2018-03-12 stsp printf("-----------------------------------------------\n");
256 f42b1b34 2018-03-12 stsp printf("commit: %s\n", buf);
257 f42b1b34 2018-03-12 stsp printf("Author: %s\n", commit->author);
258 f42b1b34 2018-03-12 stsp printf("\n%s\n", commit->logmsg);
262 f42b1b34 2018-03-12 stsp err = print_parent_commits(commit, repo);
263 f42b1b34 2018-03-12 stsp got_object_commit_close(commit);
264 f42b1b34 2018-03-12 stsp return err;
267 6f3d1eb0 2018-03-12 stsp __dead void
268 6f3d1eb0 2018-03-12 stsp usage_log(void)
270 6f3d1eb0 2018-03-12 stsp fprintf(stderr, "usage: %s log [REPO_PATH]\n", getprogname());
274 d7d4f210 2018-03-12 stsp const struct got_error *
275 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
277 f42b1b34 2018-03-12 stsp const struct got_error *error;
278 f42b1b34 2018-03-12 stsp struct got_repository *repo;
279 f42b1b34 2018-03-12 stsp struct got_reference *head_ref;
280 f42b1b34 2018-03-12 stsp struct got_object_id *id;
281 f42b1b34 2018-03-12 stsp struct got_object *obj;
282 6f3d1eb0 2018-03-12 stsp char *repo_path = NULL;
284 f42b1b34 2018-03-12 stsp if (pledge("stdio rpath wpath cpath", NULL) == -1)
285 f42b1b34 2018-03-12 stsp err(1, "pledge");
287 6f3d1eb0 2018-03-12 stsp if (argc == 1) {
288 6f3d1eb0 2018-03-12 stsp repo_path = getcwd(NULL, 0);
289 6f3d1eb0 2018-03-12 stsp if (repo_path == NULL)
290 6f3d1eb0 2018-03-12 stsp err(1, "getcwd");
291 6f3d1eb0 2018-03-12 stsp } else if (argc == 2)
292 6f3d1eb0 2018-03-12 stsp repo_path = argv[1];
294 6f3d1eb0 2018-03-12 stsp usage_log();
296 f42b1b34 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
297 d7d4f210 2018-03-12 stsp if (error != NULL)
298 d7d4f210 2018-03-12 stsp return error;
299 d7d4f210 2018-03-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
300 d7d4f210 2018-03-12 stsp if (error != NULL)
301 d7d4f210 2018-03-12 stsp return error;
302 f42b1b34 2018-03-12 stsp error = got_ref_resolve(&id, repo, head_ref);
303 d7d4f210 2018-03-12 stsp if (error != NULL)
304 d7d4f210 2018-03-12 stsp return error;
306 f42b1b34 2018-03-12 stsp error = got_object_open(&obj, repo, id);
307 d7d4f210 2018-03-12 stsp if (error != NULL)
308 d7d4f210 2018-03-12 stsp return error;
309 f42b1b34 2018-03-12 stsp if (got_object_get_type(obj) == GOT_OBJ_TYPE_COMMIT) {
310 f42b1b34 2018-03-12 stsp error = print_commit_object(obj, id, repo);
312 d7d4f210 2018-03-12 stsp return error;
314 d7d4f210 2018-03-12 stsp return got_error(GOT_ERR_OBJ_TYPE);
315 f42b1b34 2018-03-12 stsp got_object_close(obj);
317 f42b1b34 2018-03-12 stsp got_ref_close(head_ref);
318 f42b1b34 2018-03-12 stsp got_repo_close(repo);
319 d7d4f210 2018-03-12 stsp return NULL;
322 f42b1b34 2018-03-12 stsp #ifdef notyet
323 d7d4f210 2018-03-12 stsp const struct got_error *
324 5c860e29 2018-03-12 stsp cmd_status(int argc __unused, char *argv[] __unused)
326 5c860e29 2018-03-12 stsp git_repository *repo = NULL;
327 5c860e29 2018-03-12 stsp git_status_list *status;
328 5c860e29 2018-03-12 stsp git_status_options statusopts;
331 5c860e29 2018-03-12 stsp git_libgit2_init();
333 5c860e29 2018-03-12 stsp if (git_repository_open_ext(&repo, ".", 0, NULL))
334 5c860e29 2018-03-12 stsp errx(1, "git_repository_open: %s", giterr_last()->message);
336 5c860e29 2018-03-12 stsp if (git_repository_is_bare(repo))
337 5c860e29 2018-03-12 stsp errx(1, "bar repository");
339 5c860e29 2018-03-12 stsp if (git_status_init_options(&statusopts, GIT_STATUS_OPTIONS_VERSION))
340 5c860e29 2018-03-12 stsp errx(1, "git_status_init_options: %s", giterr_last()->message);
342 5c860e29 2018-03-12 stsp statusopts.show = GIT_STATUS_SHOW_INDEX_AND_WORKDIR;
343 5c860e29 2018-03-12 stsp statusopts.flags = GIT_STATUS_OPT_INCLUDE_UNTRACKED |
344 5c860e29 2018-03-12 stsp GIT_STATUS_OPT_RENAMES_HEAD_TO_INDEX |
345 5c860e29 2018-03-12 stsp GIT_STATUS_OPT_SORT_CASE_SENSITIVELY;
347 5c860e29 2018-03-12 stsp if (git_status_list_new(&status, repo, &statusopts))
348 5c860e29 2018-03-12 stsp errx(1, "git_status_list_new: %s", giterr_last()->message);
350 5c860e29 2018-03-12 stsp for (i = 0; i < git_status_list_entrycount(status); i++) {
351 5c860e29 2018-03-12 stsp const git_status_entry *se;
353 5c860e29 2018-03-12 stsp se = git_status_byindex(status, i);
354 5c860e29 2018-03-12 stsp switch (se->status) {
355 5c860e29 2018-03-12 stsp case GIT_STATUS_WT_NEW:
356 5c860e29 2018-03-12 stsp printf("? %s\n", se->index_to_workdir->new_file.path);
358 5c860e29 2018-03-12 stsp case GIT_STATUS_WT_MODIFIED:
359 5c860e29 2018-03-12 stsp printf("M %s\n", se->index_to_workdir->new_file.path);
361 5c860e29 2018-03-12 stsp case GIT_STATUS_WT_DELETED:
362 5c860e29 2018-03-12 stsp printf("R %s\n", se->index_to_workdir->new_file.path);
364 5c860e29 2018-03-12 stsp case GIT_STATUS_WT_RENAMED:
365 5c860e29 2018-03-12 stsp printf("m %s -> %s\n",
366 5c860e29 2018-03-12 stsp se->index_to_workdir->old_file.path,
367 5c860e29 2018-03-12 stsp se->index_to_workdir->new_file.path);
369 5c860e29 2018-03-12 stsp case GIT_STATUS_WT_TYPECHANGE:
370 5c860e29 2018-03-12 stsp printf("t %s\n", se->index_to_workdir->new_file.path);
372 5c860e29 2018-03-12 stsp case GIT_STATUS_INDEX_NEW:
373 5c860e29 2018-03-12 stsp printf("A %s\n", se->head_to_index->new_file.path);
375 5c860e29 2018-03-12 stsp case GIT_STATUS_INDEX_MODIFIED:
376 5c860e29 2018-03-12 stsp printf("M %s\n", se->head_to_index->old_file.path);
378 5c860e29 2018-03-12 stsp case GIT_STATUS_INDEX_DELETED:
379 5c860e29 2018-03-12 stsp printf("R %s\n", se->head_to_index->old_file.path);
381 5c860e29 2018-03-12 stsp case GIT_STATUS_INDEX_RENAMED:
382 5c860e29 2018-03-12 stsp printf("m %s -> %s\n",
383 5c860e29 2018-03-12 stsp se->head_to_index->old_file.path,
384 5c860e29 2018-03-12 stsp se->head_to_index->new_file.path);
386 5c860e29 2018-03-12 stsp case GIT_STATUS_INDEX_TYPECHANGE:
387 5c860e29 2018-03-12 stsp printf("t %s\n", se->head_to_index->old_file.path);
389 5c860e29 2018-03-12 stsp case GIT_STATUS_CURRENT:
395 5c860e29 2018-03-12 stsp git_status_list_free(status);
396 5c860e29 2018-03-12 stsp git_repository_free(repo);
397 5c860e29 2018-03-12 stsp git_libgit2_shutdown();