2 * Copyright (c) 2017 Martin Pieuchot <mpi@openbsd.org>
3 * Copyright (c) 2018, 2019, 2020 Stefan Sperling <stsp@openbsd.org>
4 * Copyright (c) 2020 Ori Bernstein <ori@openbsd.org>
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <sys/queue.h>
20 #include <sys/types.h>
43 #include "got_version.h"
44 #include "got_error.h"
45 #include "got_object.h"
46 #include "got_reference.h"
47 #include "got_repository.h"
49 #include "got_cancel.h"
50 #include "got_worktree.h"
52 #include "got_commit_graph.h"
53 #include "got_fetch.h"
55 #include "got_blame.h"
56 #include "got_privsep.h"
57 #include "got_opentemp.h"
58 #include "got_gotconfig.h"
60 #include "got_patch.h"
63 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
66 static volatile sig_atomic_t sigint_received;
67 static volatile sig_atomic_t sigpipe_received;
70 catch_sigint(int signo)
76 catch_sigpipe(int signo)
84 const struct got_error *(*cmd_main)(int, char *[]);
85 void (*cmd_usage)(void);
86 const char *cmd_alias;
89 __dead static void usage(int, int);
90 __dead static void usage_init(void);
91 __dead static void usage_import(void);
92 __dead static void usage_clone(void);
93 __dead static void usage_fetch(void);
94 __dead static void usage_checkout(void);
95 __dead static void usage_update(void);
96 __dead static void usage_log(void);
97 __dead static void usage_diff(void);
98 __dead static void usage_blame(void);
99 __dead static void usage_tree(void);
100 __dead static void usage_status(void);
101 __dead static void usage_ref(void);
102 __dead static void usage_branch(void);
103 __dead static void usage_tag(void);
104 __dead static void usage_add(void);
105 __dead static void usage_remove(void);
106 __dead static void usage_patch(void);
107 __dead static void usage_revert(void);
108 __dead static void usage_commit(void);
109 __dead static void usage_send(void);
110 __dead static void usage_cherrypick(void);
111 __dead static void usage_backout(void);
112 __dead static void usage_rebase(void);
113 __dead static void usage_histedit(void);
114 __dead static void usage_integrate(void);
115 __dead static void usage_merge(void);
116 __dead static void usage_stage(void);
117 __dead static void usage_unstage(void);
118 __dead static void usage_cat(void);
119 __dead static void usage_info(void);
121 static const struct got_error* cmd_init(int, char *[]);
122 static const struct got_error* cmd_import(int, char *[]);
123 static const struct got_error* cmd_clone(int, char *[]);
124 static const struct got_error* cmd_fetch(int, char *[]);
125 static const struct got_error* cmd_checkout(int, char *[]);
126 static const struct got_error* cmd_update(int, char *[]);
127 static const struct got_error* cmd_log(int, char *[]);
128 static const struct got_error* cmd_diff(int, char *[]);
129 static const struct got_error* cmd_blame(int, char *[]);
130 static const struct got_error* cmd_tree(int, char *[]);
131 static const struct got_error* cmd_status(int, char *[]);
132 static const struct got_error* cmd_ref(int, char *[]);
133 static const struct got_error* cmd_branch(int, char *[]);
134 static const struct got_error* cmd_tag(int, char *[]);
135 static const struct got_error* cmd_add(int, char *[]);
136 static const struct got_error* cmd_remove(int, char *[]);
137 static const struct got_error* cmd_patch(int, char *[]);
138 static const struct got_error* cmd_revert(int, char *[]);
139 static const struct got_error* cmd_commit(int, char *[]);
140 static const struct got_error* cmd_send(int, char *[]);
141 static const struct got_error* cmd_cherrypick(int, char *[]);
142 static const struct got_error* cmd_backout(int, char *[]);
143 static const struct got_error* cmd_rebase(int, char *[]);
144 static const struct got_error* cmd_histedit(int, char *[]);
145 static const struct got_error* cmd_integrate(int, char *[]);
146 static const struct got_error* cmd_merge(int, char *[]);
147 static const struct got_error* cmd_stage(int, char *[]);
148 static const struct got_error* cmd_unstage(int, char *[]);
149 static const struct got_error* cmd_cat(int, char *[]);
150 static const struct got_error* cmd_info(int, char *[]);
152 static const struct got_cmd got_commands[] = {
153 { "init", cmd_init, usage_init, "" },
154 { "import", cmd_import, usage_import, "im" },
155 { "clone", cmd_clone, usage_clone, "cl" },
156 { "fetch", cmd_fetch, usage_fetch, "fe" },
157 { "checkout", cmd_checkout, usage_checkout, "co" },
158 { "update", cmd_update, usage_update, "up" },
159 { "log", cmd_log, usage_log, "" },
160 { "diff", cmd_diff, usage_diff, "di" },
161 { "blame", cmd_blame, usage_blame, "bl" },
162 { "tree", cmd_tree, usage_tree, "tr" },
163 { "status", cmd_status, usage_status, "st" },
164 { "ref", cmd_ref, usage_ref, "" },
165 { "branch", cmd_branch, usage_branch, "br" },
166 { "tag", cmd_tag, usage_tag, "" },
167 { "add", cmd_add, usage_add, "" },
168 { "remove", cmd_remove, usage_remove, "rm" },
169 { "patch", cmd_patch, usage_patch, "pa" },
170 { "revert", cmd_revert, usage_revert, "rv" },
171 { "commit", cmd_commit, usage_commit, "ci" },
172 { "send", cmd_send, usage_send, "se" },
173 { "cherrypick", cmd_cherrypick, usage_cherrypick, "cy" },
174 { "backout", cmd_backout, usage_backout, "bo" },
175 { "rebase", cmd_rebase, usage_rebase, "rb" },
176 { "histedit", cmd_histedit, usage_histedit, "he" },
177 { "integrate", cmd_integrate, usage_integrate,"ig" },
178 { "merge", cmd_merge, usage_merge, "mg" },
179 { "stage", cmd_stage, usage_stage, "sg" },
180 { "unstage", cmd_unstage, usage_unstage, "ug" },
181 { "cat", cmd_cat, usage_cat, "" },
182 { "info", cmd_info, usage_info, "" },
186 list_commands(FILE *fp)
190 fprintf(fp, "commands:");
191 for (i = 0; i < nitems(got_commands); i++) {
192 const struct got_cmd *cmd = &got_commands[i];
193 fprintf(fp, " %s", cmd->cmd_name);
199 option_conflict(char a, char b)
201 errx(1, "-%c and -%c options are mutually exclusive", a, b);
205 main(int argc, char *argv[])
207 const struct got_cmd *cmd;
210 int hflag = 0, Vflag = 0;
211 static const struct option longopts[] = {
212 { "version", no_argument, NULL, 'V' },
216 setlocale(LC_CTYPE, "");
218 while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
238 got_version_print_str();
243 usage(hflag, hflag ? 0 : 1);
245 signal(SIGINT, catch_sigint);
246 signal(SIGPIPE, catch_sigpipe);
248 for (i = 0; i < nitems(got_commands); i++) {
249 const struct got_error *error;
251 cmd = &got_commands[i];
253 if (strcmp(cmd->cmd_name, argv[0]) != 0 &&
254 strcmp(cmd->cmd_alias, argv[0]) != 0)
260 error = cmd->cmd_main(argc, argv);
261 if (error && error->code != GOT_ERR_CANCELLED &&
262 error->code != GOT_ERR_PRIVSEP_EXIT &&
263 !(sigpipe_received &&
264 error->code == GOT_ERR_ERRNO && errno == EPIPE) &&
266 error->code == GOT_ERR_ERRNO && errno == EINTR)) {
267 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
274 fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
275 list_commands(stderr);
280 usage(int hflag, int status)
282 FILE *fp = (status == 0) ? stdout : stderr;
284 fprintf(fp, "usage: %s [-h] [-V | --version] command [arg ...]\n",
291 static const struct got_error *
292 get_editor(char **abspath)
294 const struct got_error *err = NULL;
299 editor = getenv("VISUAL");
301 editor = getenv("EDITOR");
304 err = got_path_find_prog(abspath, editor);
309 if (*abspath == NULL) {
310 *abspath = strdup("/bin/ed");
311 if (*abspath == NULL)
312 return got_error_from_errno("strdup");
318 static const struct got_error *
319 apply_unveil(const char *repo_path, int repo_read_only,
320 const char *worktree_path)
322 const struct got_error *err;
325 if (unveil("gmon.out", "rwc") != 0)
326 return got_error_from_errno2("unveil", "gmon.out");
328 if (repo_path && unveil(repo_path, repo_read_only ? "r" : "rwc") != 0)
329 return got_error_from_errno2("unveil", repo_path);
331 if (worktree_path && unveil(worktree_path, "rwc") != 0)
332 return got_error_from_errno2("unveil", worktree_path);
334 if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
335 return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
337 err = got_privsep_unveil_exec_helpers();
341 if (unveil(NULL, NULL) != 0)
342 return got_error_from_errno("unveil");
350 fprintf(stderr, "usage: %s init repository-path\n", getprogname());
354 static const struct got_error *
355 cmd_init(int argc, char *argv[])
357 const struct got_error *error = NULL;
358 char *repo_path = NULL;
361 while ((ch = getopt(argc, argv, "")) != -1) {
373 if (pledge("stdio rpath wpath cpath unveil", NULL) == -1)
379 repo_path = strdup(argv[0]);
380 if (repo_path == NULL)
381 return got_error_from_errno("strdup");
383 got_path_strip_trailing_slashes(repo_path);
385 error = got_path_mkdir(repo_path);
387 !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
390 error = apply_unveil(repo_path, 0, NULL);
394 error = got_repo_init(repo_path);
403 fprintf(stderr, "usage: %s import [-b branch] [-m message] "
404 "[-r repository-path] [-I pattern] path\n", getprogname());
409 spawn_editor(const char *editor, const char *file)
412 sig_t sighup, sigint, sigquit;
415 sighup = signal(SIGHUP, SIG_IGN);
416 sigint = signal(SIGINT, SIG_IGN);
417 sigquit = signal(SIGQUIT, SIG_IGN);
419 switch (pid = fork()) {
423 execl(editor, editor, file, (char *)NULL);
427 while (waitpid(pid, &st, 0) == -1)
432 (void)signal(SIGHUP, sighup);
433 (void)signal(SIGINT, sigint);
434 (void)signal(SIGQUIT, sigquit);
436 if (!WIFEXITED(st)) {
441 return WEXITSTATUS(st);
444 static const struct got_error *
445 edit_logmsg(char **logmsg, const char *editor, const char *logmsg_path,
446 const char *initial_content, size_t initial_content_len,
447 int require_modification)
449 const struct got_error *err = NULL;
455 size_t len, logmsg_len;
456 char *initial_content_stripped = NULL, *buf = NULL, *s;
460 if (stat(logmsg_path, &st) == -1)
461 return got_error_from_errno2("stat", logmsg_path);
463 if (spawn_editor(editor, logmsg_path) == -1)
464 return got_error_from_errno("failed spawning editor");
466 if (stat(logmsg_path, &st2) == -1)
467 return got_error_from_errno("stat");
469 if (require_modification &&
470 st.st_mtime == st2.st_mtime && st.st_size == st2.st_size)
471 return got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
472 "no changes made to commit message, aborting");
475 * Set up a stripped version of the initial content without comments
476 * and blank lines. We need this in order to check if the message
477 * has in fact been edited.
479 initial_content_stripped = malloc(initial_content_len + 1);
480 if (initial_content_stripped == NULL)
481 return got_error_from_errno("malloc");
482 initial_content_stripped[0] = '\0';
484 buf = strdup(initial_content);
486 err = got_error_from_errno("strdup");
491 while ((line = strsep(&s, "\n")) != NULL) {
492 if ((line[0] == '#' || (len == 0 && line[0] == '\n')))
493 continue; /* remove comments and leading empty lines */
494 len = strlcat(initial_content_stripped, line,
495 initial_content_len + 1);
496 if (len >= initial_content_len + 1) {
497 err = got_error(GOT_ERR_NO_SPACE);
501 while (len > 0 && initial_content_stripped[len - 1] == '\n') {
502 initial_content_stripped[len - 1] = '\0';
506 logmsg_len = st2.st_size;
507 *logmsg = malloc(logmsg_len + 1);
509 return got_error_from_errno("malloc");
512 fp = fopen(logmsg_path, "re");
514 err = got_error_from_errno("fopen");
519 while ((linelen = getline(&line, &linesize, fp)) != -1) {
520 if ((line[0] == '#' || (len == 0 && line[0] == '\n')))
521 continue; /* remove comments and leading empty lines */
522 len = strlcat(*logmsg, line, logmsg_len + 1);
523 if (len >= logmsg_len + 1) {
524 err = got_error(GOT_ERR_NO_SPACE);
530 err = got_ferror(fp, GOT_ERR_IO);
533 while (len > 0 && (*logmsg)[len - 1] == '\n') {
534 (*logmsg)[len - 1] = '\0';
539 err = got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
540 "commit message cannot be empty, aborting");
543 if (require_modification &&
544 strcmp(*logmsg, initial_content_stripped) == 0)
545 err = got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
546 "no changes made to commit message, aborting");
548 free(initial_content_stripped);
550 if (fp && fclose(fp) == EOF && err == NULL)
551 err = got_error_from_errno("fclose");
559 static const struct got_error *
560 collect_import_msg(char **logmsg, char **logmsg_path, const char *editor,
561 const char *path_dir, const char *branch_name)
563 char *initial_content = NULL;
564 const struct got_error *err = NULL;
565 int initial_content_len;
568 initial_content_len = asprintf(&initial_content,
569 "\n# %s to be imported to branch %s\n", path_dir,
571 if (initial_content_len == -1)
572 return got_error_from_errno("asprintf");
574 err = got_opentemp_named_fd(logmsg_path, &fd,
575 GOT_TMPDIR_STR "/got-importmsg");
579 if (write(fd, initial_content, initial_content_len) == -1) {
580 err = got_error_from_errno2("write", *logmsg_path);
584 err = edit_logmsg(logmsg, editor, *logmsg_path, initial_content,
585 initial_content_len, 1);
587 if (fd != -1 && close(fd) == -1 && err == NULL)
588 err = got_error_from_errno2("close", *logmsg_path);
589 free(initial_content);
597 static const struct got_error *
598 import_progress(void *arg, const char *path)
600 printf("A %s\n", path);
605 valid_author(const char *author)
608 * Really dumb email address check; we're only doing this to
609 * avoid git's object parser breaking on commits we create.
611 while (*author && *author != '<')
615 while (*author && *author != '@')
619 while (*author && *author != '>')
621 return *author == '>';
624 static const struct got_error *
625 get_author(char **author, struct got_repository *repo,
626 struct got_worktree *worktree)
628 const struct got_error *err = NULL;
629 const char *got_author = NULL, *name, *email;
630 const struct got_gotconfig *worktree_conf = NULL, *repo_conf = NULL;
635 worktree_conf = got_worktree_get_gotconfig(worktree);
636 repo_conf = got_repo_get_gotconfig(repo);
639 * Priority of potential author information sources, from most
640 * significant to least significant:
641 * 1) work tree's .got/got.conf file
642 * 2) repository's got.conf file
643 * 3) repository's git config file
644 * 4) environment variables
645 * 5) global git config files (in user's home directory or /etc)
649 got_author = got_gotconfig_get_author(worktree_conf);
650 if (got_author == NULL)
651 got_author = got_gotconfig_get_author(repo_conf);
652 if (got_author == NULL) {
653 name = got_repo_get_gitconfig_author_name(repo);
654 email = got_repo_get_gitconfig_author_email(repo);
656 if (asprintf(author, "%s <%s>", name, email) == -1)
657 return got_error_from_errno("asprintf");
661 got_author = getenv("GOT_AUTHOR");
662 if (got_author == NULL) {
663 name = got_repo_get_global_gitconfig_author_name(repo);
664 email = got_repo_get_global_gitconfig_author_email(
667 if (asprintf(author, "%s <%s>", name, email)
669 return got_error_from_errno("asprintf");
672 /* TODO: Look up user in password database? */
673 return got_error(GOT_ERR_COMMIT_NO_AUTHOR);
677 *author = strdup(got_author);
679 return got_error_from_errno("strdup");
681 if (!valid_author(*author)) {
682 err = got_error_fmt(GOT_ERR_COMMIT_NO_EMAIL, "%s", *author);
689 static const struct got_error *
690 get_gitconfig_path(char **gitconfig_path)
692 const char *homedir = getenv("HOME");
694 *gitconfig_path = NULL;
696 if (asprintf(gitconfig_path, "%s/.gitconfig", homedir) == -1)
697 return got_error_from_errno("asprintf");
703 static const struct got_error *
704 cmd_import(int argc, char *argv[])
706 const struct got_error *error = NULL;
707 char *path_dir = NULL, *repo_path = NULL, *logmsg = NULL;
708 char *gitconfig_path = NULL, *editor = NULL, *author = NULL;
709 const char *branch_name = "main";
710 char *refname = NULL, *id_str = NULL, *logmsg_path = NULL;
711 struct got_repository *repo = NULL;
712 struct got_reference *branch_ref = NULL, *head_ref = NULL;
713 struct got_object_id *new_commit_id = NULL;
715 struct got_pathlist_head ignores;
716 struct got_pathlist_entry *pe;
717 int preserve_logmsg = 0;
718 int *pack_fds = NULL;
720 TAILQ_INIT(&ignores);
722 while ((ch = getopt(argc, argv, "b:m:r:I:")) != -1) {
725 branch_name = optarg;
728 logmsg = strdup(optarg);
729 if (logmsg == NULL) {
730 error = got_error_from_errno("strdup");
735 repo_path = realpath(optarg, NULL);
736 if (repo_path == NULL) {
737 error = got_error_from_errno2("realpath",
743 if (optarg[0] == '\0')
745 error = got_pathlist_insert(&pe, &ignores, optarg,
760 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
768 if (repo_path == NULL) {
769 repo_path = getcwd(NULL, 0);
770 if (repo_path == NULL)
771 return got_error_from_errno("getcwd");
773 got_path_strip_trailing_slashes(repo_path);
774 error = get_gitconfig_path(&gitconfig_path);
777 error = got_repo_pack_fds_open(&pack_fds);
780 error = got_repo_open(&repo, repo_path, gitconfig_path, pack_fds);
784 error = get_author(&author, repo, NULL);
789 * Don't let the user create a branch name with a leading '-'.
790 * While technically a valid reference name, this case is usually
791 * an unintended typo.
793 if (branch_name[0] == '-')
794 return got_error_path(branch_name, GOT_ERR_REF_NAME_MINUS);
796 if (asprintf(&refname, "refs/heads/%s", branch_name) == -1) {
797 error = got_error_from_errno("asprintf");
801 error = got_ref_open(&branch_ref, repo, refname, 0);
803 if (error->code != GOT_ERR_NOT_REF)
806 error = got_error_msg(GOT_ERR_BRANCH_EXISTS,
807 "import target branch already exists");
811 path_dir = realpath(argv[0], NULL);
812 if (path_dir == NULL) {
813 error = got_error_from_errno2("realpath", argv[0]);
816 got_path_strip_trailing_slashes(path_dir);
819 * unveil(2) traverses exec(2); if an editor is used we have
820 * to apply unveil after the log message has been written.
822 if (logmsg == NULL || strlen(logmsg) == 0) {
823 error = get_editor(&editor);
827 error = collect_import_msg(&logmsg, &logmsg_path, editor,
830 if (error->code != GOT_ERR_COMMIT_MSG_EMPTY &&
837 if (unveil(path_dir, "r") != 0) {
838 error = got_error_from_errno2("unveil", path_dir);
844 error = apply_unveil(got_repo_get_path(repo), 0, NULL);
851 error = got_repo_import(&new_commit_id, path_dir, logmsg,
852 author, &ignores, repo, import_progress, NULL);
859 error = got_ref_alloc(&branch_ref, refname, new_commit_id);
866 error = got_ref_write(branch_ref, repo);
873 error = got_object_id_str(&id_str, new_commit_id);
880 error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
882 if (error->code != GOT_ERR_NOT_REF) {
888 error = got_ref_alloc_symref(&head_ref, GOT_REF_HEAD,
896 error = got_ref_write(head_ref, repo);
904 printf("Created branch %s with commit %s\n",
905 got_ref_get_name(branch_ref), id_str);
908 const struct got_error *pack_err =
909 got_repo_pack_fds_close(pack_fds);
913 if (preserve_logmsg) {
914 fprintf(stderr, "%s: log message preserved in %s\n",
915 getprogname(), logmsg_path);
916 } else if (logmsg_path && unlink(logmsg_path) == -1 && error == NULL)
917 error = got_error_from_errno2("unlink", logmsg_path);
926 free(gitconfig_path);
928 got_ref_close(branch_ref);
930 got_ref_close(head_ref);
937 fprintf(stderr, "usage: %s clone [-a] [-b branch] [-l] [-m] [-q] [-v] "
938 "[-R reference] repository-url [directory]\n", getprogname());
942 struct got_fetch_progress_arg {
943 char last_scaled_size[FMT_SCALED_STRSIZE];
948 struct got_repository *repo;
953 struct got_pathlist_head *symrefs;
954 struct got_pathlist_head *wanted_branches;
955 struct got_pathlist_head *wanted_refs;
959 const char *remote_repo_path;
961 int fetch_all_branches;
962 int mirror_references;
966 /* XXX forward declaration */
967 static const struct got_error *
968 create_config_files(const char *proto, const char *host, const char *port,
969 const char *remote_repo_path, const char *git_url, int fetch_all_branches,
970 int mirror_references, struct got_pathlist_head *symrefs,
971 struct got_pathlist_head *wanted_branches,
972 struct got_pathlist_head *wanted_refs, struct got_repository *repo);
974 static const struct got_error *
975 fetch_progress(void *arg, const char *message, off_t packfile_size,
976 int nobj_total, int nobj_indexed, int nobj_loose, int nobj_resolved)
978 const struct got_error *err = NULL;
979 struct got_fetch_progress_arg *a = arg;
980 char scaled_size[FMT_SCALED_STRSIZE];
981 int p_indexed, p_resolved;
982 int print_size = 0, print_indexed = 0, print_resolved = 0;
985 * In order to allow a failed clone to be resumed with 'got fetch'
986 * we try to create configuration files as soon as possible.
987 * Once the server has sent information about its default branch
988 * we have all required information.
990 if (a->create_configs && !a->configs_created &&
991 !TAILQ_EMPTY(a->config_info.symrefs)) {
992 err = create_config_files(a->config_info.proto,
993 a->config_info.host, a->config_info.port,
994 a->config_info.remote_repo_path,
995 a->config_info.git_url,
996 a->config_info.fetch_all_branches,
997 a->config_info.mirror_references,
998 a->config_info.symrefs,
999 a->config_info.wanted_branches,
1000 a->config_info.wanted_refs, a->repo);
1003 a->configs_created = 1;
1006 if (a->verbosity < 0)
1009 if (message && message[0] != '\0') {
1010 printf("\rserver: %s", message);
1015 if (packfile_size > 0 || nobj_indexed > 0) {
1016 if (fmt_scaled(packfile_size, scaled_size) == 0 &&
1017 (a->last_scaled_size[0] == '\0' ||
1018 strcmp(scaled_size, a->last_scaled_size)) != 0) {
1020 if (strlcpy(a->last_scaled_size, scaled_size,
1021 FMT_SCALED_STRSIZE) >= FMT_SCALED_STRSIZE)
1022 return got_error(GOT_ERR_NO_SPACE);
1024 if (nobj_indexed > 0) {
1025 p_indexed = (nobj_indexed * 100) / nobj_total;
1026 if (p_indexed != a->last_p_indexed) {
1027 a->last_p_indexed = p_indexed;
1032 if (nobj_resolved > 0) {
1033 p_resolved = (nobj_resolved * 100) /
1034 (nobj_total - nobj_loose);
1035 if (p_resolved != a->last_p_resolved) {
1036 a->last_p_resolved = p_resolved;
1044 if (print_size || print_indexed || print_resolved)
1047 printf("%*s fetched", FMT_SCALED_STRSIZE - 2, scaled_size);
1049 printf("; indexing %d%%", p_indexed);
1051 printf("; resolving deltas %d%%", p_resolved);
1052 if (print_size || print_indexed || print_resolved)
1058 static const struct got_error *
1059 create_symref(const char *refname, struct got_reference *target_ref,
1060 int verbosity, struct got_repository *repo)
1062 const struct got_error *err;
1063 struct got_reference *head_symref;
1065 err = got_ref_alloc_symref(&head_symref, refname, target_ref);
1069 err = got_ref_write(head_symref, repo);
1070 if (err == NULL && verbosity > 0) {
1071 printf("Created reference %s: %s\n", GOT_REF_HEAD,
1072 got_ref_get_name(target_ref));
1074 got_ref_close(head_symref);
1078 static const struct got_error *
1079 list_remote_refs(struct got_pathlist_head *symrefs,
1080 struct got_pathlist_head *refs)
1082 const struct got_error *err;
1083 struct got_pathlist_entry *pe;
1085 TAILQ_FOREACH(pe, symrefs, entry) {
1086 const char *refname = pe->path;
1087 const char *targetref = pe->data;
1089 printf("%s: %s\n", refname, targetref);
1092 TAILQ_FOREACH(pe, refs, entry) {
1093 const char *refname = pe->path;
1094 struct got_object_id *id = pe->data;
1097 err = got_object_id_str(&id_str, id);
1100 printf("%s: %s\n", refname, id_str);
1107 static const struct got_error *
1108 create_ref(const char *refname, struct got_object_id *id,
1109 int verbosity, struct got_repository *repo)
1111 const struct got_error *err = NULL;
1112 struct got_reference *ref;
1115 err = got_object_id_str(&id_str, id);
1119 err = got_ref_alloc(&ref, refname, id);
1123 err = got_ref_write(ref, repo);
1126 if (err == NULL && verbosity >= 0)
1127 printf("Created reference %s: %s\n", refname, id_str);
1134 match_wanted_ref(const char *refname, const char *wanted_ref)
1136 if (strncmp(refname, "refs/", 5) != 0)
1141 * Prevent fetching of references that won't make any
1142 * sense outside of the remote repository's context.
1144 if (strncmp(refname, "got/", 4) == 0)
1146 if (strncmp(refname, "remotes/", 8) == 0)
1149 if (strncmp(wanted_ref, "refs/", 5) == 0)
1152 /* Allow prefix match. */
1153 if (got_path_is_child(refname, wanted_ref, strlen(wanted_ref)))
1156 /* Allow exact match. */
1157 return (strcmp(refname, wanted_ref) == 0);
1161 is_wanted_ref(struct got_pathlist_head *wanted_refs, const char *refname)
1163 struct got_pathlist_entry *pe;
1165 TAILQ_FOREACH(pe, wanted_refs, entry) {
1166 if (match_wanted_ref(refname, pe->path))
1173 static const struct got_error *
1174 create_wanted_ref(const char *refname, struct got_object_id *id,
1175 const char *remote_repo_name, int verbosity, struct got_repository *repo)
1177 const struct got_error *err;
1178 char *remote_refname;
1180 if (strncmp("refs/", refname, 5) == 0)
1183 if (asprintf(&remote_refname, "refs/remotes/%s/%s",
1184 remote_repo_name, refname) == -1)
1185 return got_error_from_errno("asprintf");
1187 err = create_ref(remote_refname, id, verbosity, repo);
1188 free(remote_refname);
1192 static const struct got_error *
1193 create_gotconfig(const char *proto, const char *host, const char *port,
1194 const char *remote_repo_path, const char *default_branch,
1195 int fetch_all_branches, struct got_pathlist_head *wanted_branches,
1196 struct got_pathlist_head *wanted_refs, int mirror_references,
1197 struct got_repository *repo)
1199 const struct got_error *err = NULL;
1200 char *gotconfig_path = NULL;
1201 char *gotconfig = NULL;
1202 FILE *gotconfig_file = NULL;
1203 const char *branchname = NULL;
1204 char *branches = NULL, *refs = NULL;
1207 if (!fetch_all_branches && !TAILQ_EMPTY(wanted_branches)) {
1208 struct got_pathlist_entry *pe;
1209 TAILQ_FOREACH(pe, wanted_branches, entry) {
1211 branchname = pe->path;
1212 if (strncmp(branchname, "refs/heads/", 11) == 0)
1214 if (asprintf(&s, "%s\"%s\" ",
1215 branches ? branches : "", branchname) == -1) {
1216 err = got_error_from_errno("asprintf");
1222 } else if (!fetch_all_branches && default_branch) {
1223 branchname = default_branch;
1224 if (strncmp(branchname, "refs/heads/", 11) == 0)
1226 if (asprintf(&branches, "\"%s\" ", branchname) == -1) {
1227 err = got_error_from_errno("asprintf");
1231 if (!TAILQ_EMPTY(wanted_refs)) {
1232 struct got_pathlist_entry *pe;
1233 TAILQ_FOREACH(pe, wanted_refs, entry) {
1235 const char *refname = pe->path;
1236 if (strncmp(refname, "refs/", 5) == 0)
1238 if (asprintf(&s, "%s\"%s\" ",
1239 refs ? refs : "", refname) == -1) {
1240 err = got_error_from_errno("asprintf");
1248 /* Create got.conf(5). */
1249 gotconfig_path = got_repo_get_path_gotconfig(repo);
1250 if (gotconfig_path == NULL) {
1251 err = got_error_from_errno("got_repo_get_path_gotconfig");
1254 gotconfig_file = fopen(gotconfig_path, "ae");
1255 if (gotconfig_file == NULL) {
1256 err = got_error_from_errno2("fopen", gotconfig_path);
1259 if (asprintf(&gotconfig,
1264 "\trepository \"%s\"\n"
1270 GOT_FETCH_DEFAULT_REMOTE_NAME, host, proto,
1271 port ? "\tport " : "", port ? port : "", port ? "\n" : "",
1272 remote_repo_path, branches ? "\tbranch { " : "",
1273 branches ? branches : "", branches ? "}\n" : "",
1274 refs ? "\treference { " : "", refs ? refs : "", refs ? "}\n" : "",
1275 mirror_references ? "\tmirror-references yes\n" : "",
1276 fetch_all_branches ? "\tfetch-all-branches yes\n" : "") == -1) {
1277 err = got_error_from_errno("asprintf");
1280 n = fwrite(gotconfig, 1, strlen(gotconfig), gotconfig_file);
1281 if (n != strlen(gotconfig)) {
1282 err = got_ferror(gotconfig_file, GOT_ERR_IO);
1287 if (gotconfig_file && fclose(gotconfig_file) == EOF && err == NULL)
1288 err = got_error_from_errno2("fclose", gotconfig_path);
1289 free(gotconfig_path);
1294 static const struct got_error *
1295 create_gitconfig(const char *git_url, const char *default_branch,
1296 int fetch_all_branches, struct got_pathlist_head *wanted_branches,
1297 struct got_pathlist_head *wanted_refs, int mirror_references,
1298 struct got_repository *repo)
1300 const struct got_error *err = NULL;
1301 char *gitconfig_path = NULL;
1302 char *gitconfig = NULL;
1303 FILE *gitconfig_file = NULL;
1304 char *branches = NULL, *refs = NULL;
1305 const char *branchname;
1308 /* Create a config file Git can understand. */
1309 gitconfig_path = got_repo_get_path_gitconfig(repo);
1310 if (gitconfig_path == NULL) {
1311 err = got_error_from_errno("got_repo_get_path_gitconfig");
1314 gitconfig_file = fopen(gitconfig_path, "ae");
1315 if (gitconfig_file == NULL) {
1316 err = got_error_from_errno2("fopen", gitconfig_path);
1319 if (fetch_all_branches) {
1320 if (mirror_references) {
1321 if (asprintf(&branches,
1322 "\tfetch = refs/heads/*:refs/heads/*\n") == -1) {
1323 err = got_error_from_errno("asprintf");
1326 } else if (asprintf(&branches,
1327 "\tfetch = refs/heads/*:refs/remotes/%s/*\n",
1328 GOT_FETCH_DEFAULT_REMOTE_NAME) == -1) {
1329 err = got_error_from_errno("asprintf");
1332 } else if (!TAILQ_EMPTY(wanted_branches)) {
1333 struct got_pathlist_entry *pe;
1334 TAILQ_FOREACH(pe, wanted_branches, entry) {
1336 branchname = pe->path;
1337 if (strncmp(branchname, "refs/heads/", 11) == 0)
1339 if (mirror_references) {
1341 "%s\tfetch = refs/heads/%s:refs/heads/%s\n",
1342 branches ? branches : "",
1343 branchname, branchname) == -1) {
1344 err = got_error_from_errno("asprintf");
1347 } else if (asprintf(&s,
1348 "%s\tfetch = refs/heads/%s:refs/remotes/%s/%s\n",
1349 branches ? branches : "",
1350 branchname, GOT_FETCH_DEFAULT_REMOTE_NAME,
1351 branchname) == -1) {
1352 err = got_error_from_errno("asprintf");
1360 * If the server specified a default branch, use just that one.
1361 * Otherwise fall back to fetching all branches on next fetch.
1363 if (default_branch) {
1364 branchname = default_branch;
1365 if (strncmp(branchname, "refs/heads/", 11) == 0)
1368 branchname = "*"; /* fall back to all branches */
1369 if (mirror_references) {
1370 if (asprintf(&branches,
1371 "\tfetch = refs/heads/%s:refs/heads/%s\n",
1372 branchname, branchname) == -1) {
1373 err = got_error_from_errno("asprintf");
1376 } else if (asprintf(&branches,
1377 "\tfetch = refs/heads/%s:refs/remotes/%s/%s\n",
1378 branchname, GOT_FETCH_DEFAULT_REMOTE_NAME,
1379 branchname) == -1) {
1380 err = got_error_from_errno("asprintf");
1384 if (!TAILQ_EMPTY(wanted_refs)) {
1385 struct got_pathlist_entry *pe;
1386 TAILQ_FOREACH(pe, wanted_refs, entry) {
1388 const char *refname = pe->path;
1389 if (strncmp(refname, "refs/", 5) == 0)
1391 if (mirror_references) {
1393 "%s\tfetch = refs/%s:refs/%s\n",
1394 refs ? refs : "", refname, refname) == -1) {
1395 err = got_error_from_errno("asprintf");
1398 } else if (asprintf(&s,
1399 "%s\tfetch = refs/%s:refs/remotes/%s/%s\n",
1401 refname, GOT_FETCH_DEFAULT_REMOTE_NAME,
1403 err = got_error_from_errno("asprintf");
1411 if (asprintf(&gitconfig,
1416 "\tfetch = refs/tags/*:refs/tags/*\n",
1417 GOT_FETCH_DEFAULT_REMOTE_NAME, git_url, branches ? branches : "",
1418 refs ? refs : "") == -1) {
1419 err = got_error_from_errno("asprintf");
1422 n = fwrite(gitconfig, 1, strlen(gitconfig), gitconfig_file);
1423 if (n != strlen(gitconfig)) {
1424 err = got_ferror(gitconfig_file, GOT_ERR_IO);
1428 if (gitconfig_file && fclose(gitconfig_file) == EOF && err == NULL)
1429 err = got_error_from_errno2("fclose", gitconfig_path);
1430 free(gitconfig_path);
1435 static const struct got_error *
1436 create_config_files(const char *proto, const char *host, const char *port,
1437 const char *remote_repo_path, const char *git_url, int fetch_all_branches,
1438 int mirror_references, struct got_pathlist_head *symrefs,
1439 struct got_pathlist_head *wanted_branches,
1440 struct got_pathlist_head *wanted_refs, struct got_repository *repo)
1442 const struct got_error *err = NULL;
1443 const char *default_branch = NULL;
1444 struct got_pathlist_entry *pe;
1447 * If we asked for a set of wanted branches then use the first
1450 if (!TAILQ_EMPTY(wanted_branches)) {
1451 pe = TAILQ_FIRST(wanted_branches);
1452 default_branch = pe->path;
1454 /* First HEAD ref listed by server is the default branch. */
1455 TAILQ_FOREACH(pe, symrefs, entry) {
1456 const char *refname = pe->path;
1457 const char *target = pe->data;
1459 if (strcmp(refname, GOT_REF_HEAD) != 0)
1462 default_branch = target;
1467 /* Create got.conf(5). */
1468 err = create_gotconfig(proto, host, port, remote_repo_path,
1469 default_branch, fetch_all_branches, wanted_branches,
1470 wanted_refs, mirror_references, repo);
1474 /* Create a config file Git can understand. */
1475 return create_gitconfig(git_url, default_branch, fetch_all_branches,
1476 wanted_branches, wanted_refs, mirror_references, repo);
1479 static const struct got_error *
1480 cmd_clone(int argc, char *argv[])
1482 const struct got_error *error = NULL;
1483 const char *uri, *dirname;
1484 char *proto, *host, *port, *repo_name, *server_path;
1485 char *default_destdir = NULL, *id_str = NULL;
1486 const char *repo_path;
1487 struct got_repository *repo = NULL;
1488 struct got_pathlist_head refs, symrefs, wanted_branches, wanted_refs;
1489 struct got_pathlist_entry *pe;
1490 struct got_object_id *pack_hash = NULL;
1491 int ch, fetchfd = -1, fetchstatus;
1492 pid_t fetchpid = -1;
1493 struct got_fetch_progress_arg fpa;
1494 char *git_url = NULL;
1495 int verbosity = 0, fetch_all_branches = 0, mirror_references = 0;
1496 int list_refs_only = 0;
1497 int *pack_fds = NULL;
1500 TAILQ_INIT(&symrefs);
1501 TAILQ_INIT(&wanted_branches);
1502 TAILQ_INIT(&wanted_refs);
1504 while ((ch = getopt(argc, argv, "ab:lmvqR:")) != -1) {
1507 fetch_all_branches = 1;
1510 error = got_pathlist_append(&wanted_branches,
1519 mirror_references = 1;
1524 else if (verbosity < 3)
1531 error = got_pathlist_append(&wanted_refs,
1544 if (fetch_all_branches && !TAILQ_EMPTY(&wanted_branches))
1545 option_conflict('a', 'b');
1546 if (list_refs_only) {
1547 if (!TAILQ_EMPTY(&wanted_branches))
1548 option_conflict('l', 'b');
1549 if (fetch_all_branches)
1550 option_conflict('l', 'a');
1551 if (mirror_references)
1552 option_conflict('l', 'm');
1553 if (!TAILQ_EMPTY(&wanted_refs))
1554 option_conflict('l', 'R');
1566 error = got_dial_parse_uri(&proto, &host, &port, &server_path,
1571 if (asprintf(&git_url, "%s://%s%s%s%s%s", proto,
1572 host, port ? ":" : "", port ? port : "",
1573 server_path[0] != '/' ? "/" : "", server_path) == -1) {
1574 error = got_error_from_errno("asprintf");
1578 if (strcmp(proto, "git") == 0) {
1580 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
1581 "sendfd dns inet unveil", NULL) == -1)
1584 } else if (strcmp(proto, "git+ssh") == 0 ||
1585 strcmp(proto, "ssh") == 0) {
1587 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
1588 "sendfd unveil", NULL) == -1)
1591 } else if (strcmp(proto, "http") == 0 ||
1592 strcmp(proto, "git+http") == 0) {
1593 error = got_error_path(proto, GOT_ERR_NOT_IMPL);
1596 error = got_error_path(proto, GOT_ERR_BAD_PROTO);
1599 if (dirname == NULL) {
1600 if (asprintf(&default_destdir, "%s.git", repo_name) == -1) {
1601 error = got_error_from_errno("asprintf");
1604 repo_path = default_destdir;
1606 repo_path = dirname;
1608 if (!list_refs_only) {
1609 error = got_path_mkdir(repo_path);
1611 (!(error->code == GOT_ERR_ERRNO && errno == EISDIR) &&
1612 !(error->code == GOT_ERR_ERRNO && errno == EEXIST)))
1614 if (!got_path_dir_is_empty(repo_path)) {
1615 error = got_error_path(repo_path,
1616 GOT_ERR_DIR_NOT_EMPTY);
1621 error = got_dial_apply_unveil(proto);
1625 error = apply_unveil(repo_path, 0, NULL);
1630 printf("Connecting to %s%s%s\n", host,
1631 port ? ":" : "", port ? port : "");
1633 error = got_fetch_connect(&fetchpid, &fetchfd, proto, host, port,
1634 server_path, verbosity);
1638 if (!list_refs_only) {
1639 error = got_repo_init(repo_path);
1642 error = got_repo_pack_fds_open(&pack_fds);
1645 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
1650 fpa.last_scaled_size[0] = '\0';
1651 fpa.last_p_indexed = -1;
1652 fpa.last_p_resolved = -1;
1653 fpa.verbosity = verbosity;
1654 fpa.create_configs = 1;
1655 fpa.configs_created = 0;
1657 fpa.config_info.symrefs = &symrefs;
1658 fpa.config_info.wanted_branches = &wanted_branches;
1659 fpa.config_info.wanted_refs = &wanted_refs;
1660 fpa.config_info.proto = proto;
1661 fpa.config_info.host = host;
1662 fpa.config_info.port = port;
1663 fpa.config_info.remote_repo_path = server_path;
1664 fpa.config_info.git_url = git_url;
1665 fpa.config_info.fetch_all_branches = fetch_all_branches;
1666 fpa.config_info.mirror_references = mirror_references;
1667 error = got_fetch_pack(&pack_hash, &refs, &symrefs,
1668 GOT_FETCH_DEFAULT_REMOTE_NAME, mirror_references,
1669 fetch_all_branches, &wanted_branches, &wanted_refs,
1670 list_refs_only, verbosity, fetchfd, repo,
1671 fetch_progress, &fpa);
1675 if (list_refs_only) {
1676 error = list_remote_refs(&symrefs, &refs);
1680 if (pack_hash == NULL) {
1681 error = got_error_fmt(GOT_ERR_FETCH_FAILED, "%s",
1682 "server sent an empty pack file");
1685 error = got_object_id_str(&id_str, pack_hash);
1689 printf("\nFetched %s.pack\n", id_str);
1692 /* Set up references provided with the pack file. */
1693 TAILQ_FOREACH(pe, &refs, entry) {
1694 const char *refname = pe->path;
1695 struct got_object_id *id = pe->data;
1696 char *remote_refname;
1698 if (is_wanted_ref(&wanted_refs, refname) &&
1699 !mirror_references) {
1700 error = create_wanted_ref(refname, id,
1701 GOT_FETCH_DEFAULT_REMOTE_NAME,
1702 verbosity - 1, repo);
1708 error = create_ref(refname, id, verbosity - 1, repo);
1712 if (mirror_references)
1715 if (strncmp("refs/heads/", refname, 11) != 0)
1718 if (asprintf(&remote_refname,
1719 "refs/remotes/%s/%s", GOT_FETCH_DEFAULT_REMOTE_NAME,
1720 refname + 11) == -1) {
1721 error = got_error_from_errno("asprintf");
1724 error = create_ref(remote_refname, id, verbosity - 1, repo);
1725 free(remote_refname);
1730 /* Set the HEAD reference if the server provided one. */
1731 TAILQ_FOREACH(pe, &symrefs, entry) {
1732 struct got_reference *target_ref;
1733 const char *refname = pe->path;
1734 const char *target = pe->data;
1735 char *remote_refname = NULL, *remote_target = NULL;
1737 if (strcmp(refname, GOT_REF_HEAD) != 0)
1740 error = got_ref_open(&target_ref, repo, target, 0);
1742 if (error->code == GOT_ERR_NOT_REF) {
1749 error = create_symref(refname, target_ref, verbosity, repo);
1750 got_ref_close(target_ref);
1754 if (mirror_references)
1757 if (strncmp("refs/heads/", target, 11) != 0)
1760 if (asprintf(&remote_refname,
1761 "refs/remotes/%s/%s", GOT_FETCH_DEFAULT_REMOTE_NAME,
1763 error = got_error_from_errno("asprintf");
1766 if (asprintf(&remote_target,
1767 "refs/remotes/%s/%s", GOT_FETCH_DEFAULT_REMOTE_NAME,
1768 target + 11) == -1) {
1769 error = got_error_from_errno("asprintf");
1770 free(remote_refname);
1773 error = got_ref_open(&target_ref, repo, remote_target, 0);
1775 free(remote_refname);
1776 free(remote_target);
1777 if (error->code == GOT_ERR_NOT_REF) {
1783 error = create_symref(remote_refname, target_ref,
1784 verbosity - 1, repo);
1785 free(remote_refname);
1786 free(remote_target);
1787 got_ref_close(target_ref);
1793 * We failed to set the HEAD reference. If we asked for
1794 * a set of wanted branches use the first of one of those
1795 * which could be fetched instead.
1797 TAILQ_FOREACH(pe, &wanted_branches, entry) {
1798 const char *target = pe->path;
1799 struct got_reference *target_ref;
1801 error = got_ref_open(&target_ref, repo, target, 0);
1803 if (error->code == GOT_ERR_NOT_REF) {
1810 error = create_symref(GOT_REF_HEAD, target_ref,
1812 got_ref_close(target_ref);
1820 printf("Created %s repository '%s'\n",
1821 mirror_references ? "mirrored" : "cloned", repo_path);
1824 const struct got_error *pack_err =
1825 got_repo_pack_fds_close(pack_fds);
1830 if (kill(fetchpid, SIGTERM) == -1)
1831 error = got_error_from_errno("kill");
1832 if (waitpid(fetchpid, &fetchstatus, 0) == -1 && error == NULL)
1833 error = got_error_from_errno("waitpid");
1835 if (fetchfd != -1 && close(fetchfd) == -1 && error == NULL)
1836 error = got_error_from_errno("close");
1838 const struct got_error *close_err = got_repo_close(repo);
1842 TAILQ_FOREACH(pe, &refs, entry) {
1843 free((void *)pe->path);
1846 got_pathlist_free(&refs);
1847 TAILQ_FOREACH(pe, &symrefs, entry) {
1848 free((void *)pe->path);
1851 got_pathlist_free(&symrefs);
1852 got_pathlist_free(&wanted_branches);
1853 got_pathlist_free(&wanted_refs);
1860 free(default_destdir);
1865 static const struct got_error *
1866 update_ref(struct got_reference *ref, struct got_object_id *new_id,
1867 int replace_tags, int verbosity, struct got_repository *repo)
1869 const struct got_error *err = NULL;
1870 char *new_id_str = NULL;
1871 struct got_object_id *old_id = NULL;
1873 err = got_object_id_str(&new_id_str, new_id);
1877 if (!replace_tags &&
1878 strncmp(got_ref_get_name(ref), "refs/tags/", 10) == 0) {
1879 err = got_ref_resolve(&old_id, repo, ref);
1882 if (got_object_id_cmp(old_id, new_id) == 0)
1884 if (verbosity >= 0) {
1885 printf("Rejecting update of existing tag %s: %s\n",
1886 got_ref_get_name(ref), new_id_str);
1891 if (got_ref_is_symbolic(ref)) {
1892 if (verbosity >= 0) {
1893 printf("Replacing reference %s: %s\n",
1894 got_ref_get_name(ref),
1895 got_ref_get_symref_target(ref));
1897 err = got_ref_change_symref_to_ref(ref, new_id);
1900 err = got_ref_write(ref, repo);
1904 err = got_ref_resolve(&old_id, repo, ref);
1907 if (got_object_id_cmp(old_id, new_id) == 0)
1910 err = got_ref_change_ref(ref, new_id);
1913 err = got_ref_write(ref, repo);
1919 printf("Updated %s: %s\n", got_ref_get_name(ref),
1927 static const struct got_error *
1928 update_symref(const char *refname, struct got_reference *target_ref,
1929 int verbosity, struct got_repository *repo)
1931 const struct got_error *err = NULL, *unlock_err;
1932 struct got_reference *symref;
1933 int symref_is_locked = 0;
1935 err = got_ref_open(&symref, repo, refname, 1);
1937 if (err->code != GOT_ERR_NOT_REF)
1939 err = got_ref_alloc_symref(&symref, refname, target_ref);
1943 err = got_ref_write(symref, repo);
1948 printf("Created reference %s: %s\n",
1949 got_ref_get_name(symref),
1950 got_ref_get_symref_target(symref));
1952 symref_is_locked = 1;
1954 if (strcmp(got_ref_get_symref_target(symref),
1955 got_ref_get_name(target_ref)) == 0)
1958 err = got_ref_change_symref(symref,
1959 got_ref_get_name(target_ref));
1963 err = got_ref_write(symref, repo);
1968 printf("Updated %s: %s\n", got_ref_get_name(symref),
1969 got_ref_get_symref_target(symref));
1973 if (symref_is_locked) {
1974 unlock_err = got_ref_unlock(symref);
1975 if (unlock_err && err == NULL)
1978 got_ref_close(symref);
1985 fprintf(stderr, "usage: %s fetch [-a] [-b branch] [-d] [-l] "
1986 "[-r repository-path] [-t] [-q] [-v] [-R reference] [-X] "
1987 "[remote-repository-name]\n",
1992 static const struct got_error *
1993 delete_missing_ref(struct got_reference *ref,
1994 int verbosity, struct got_repository *repo)
1996 const struct got_error *err = NULL;
1997 struct got_object_id *id = NULL;
1998 char *id_str = NULL;
2000 if (got_ref_is_symbolic(ref)) {
2001 err = got_ref_delete(ref, repo);
2004 if (verbosity >= 0) {
2005 printf("Deleted %s: %s\n",
2006 got_ref_get_name(ref),
2007 got_ref_get_symref_target(ref));
2010 err = got_ref_resolve(&id, repo, ref);
2013 err = got_object_id_str(&id_str, id);
2017 err = got_ref_delete(ref, repo);
2020 if (verbosity >= 0) {
2021 printf("Deleted %s: %s\n",
2022 got_ref_get_name(ref), id_str);
2031 static const struct got_error *
2032 delete_missing_refs(struct got_pathlist_head *their_refs,
2033 struct got_pathlist_head *their_symrefs,
2034 const struct got_remote_repo *remote,
2035 int verbosity, struct got_repository *repo)
2037 const struct got_error *err = NULL, *unlock_err;
2038 struct got_reflist_head my_refs;
2039 struct got_reflist_entry *re;
2040 struct got_pathlist_entry *pe;
2041 char *remote_namespace = NULL;
2042 char *local_refname = NULL;
2044 TAILQ_INIT(&my_refs);
2046 if (asprintf(&remote_namespace, "refs/remotes/%s/", remote->name)
2048 return got_error_from_errno("asprintf");
2050 err = got_ref_list(&my_refs, repo, NULL, got_ref_cmp_by_name, NULL);
2054 TAILQ_FOREACH(re, &my_refs, entry) {
2055 const char *refname = got_ref_get_name(re->ref);
2056 const char *their_refname;
2058 if (remote->mirror_references) {
2059 their_refname = refname;
2061 if (strncmp(refname, remote_namespace,
2062 strlen(remote_namespace)) == 0) {
2063 if (strcmp(refname + strlen(remote_namespace),
2066 if (asprintf(&local_refname, "refs/heads/%s",
2067 refname + strlen(remote_namespace)) == -1) {
2068 err = got_error_from_errno("asprintf");
2071 } else if (strncmp(refname, "refs/tags/", 10) != 0)
2074 their_refname = local_refname;
2077 TAILQ_FOREACH(pe, their_refs, entry) {
2078 if (strcmp(their_refname, pe->path) == 0)
2084 TAILQ_FOREACH(pe, their_symrefs, entry) {
2085 if (strcmp(their_refname, pe->path) == 0)
2091 err = delete_missing_ref(re->ref, verbosity, repo);
2095 if (local_refname) {
2096 struct got_reference *ref;
2097 err = got_ref_open(&ref, repo, local_refname, 1);
2099 if (err->code != GOT_ERR_NOT_REF)
2101 free(local_refname);
2102 local_refname = NULL;
2105 err = delete_missing_ref(ref, verbosity, repo);
2108 unlock_err = got_ref_unlock(ref);
2110 if (unlock_err && err == NULL) {
2115 free(local_refname);
2116 local_refname = NULL;
2120 free(remote_namespace);
2121 free(local_refname);
2125 static const struct got_error *
2126 update_wanted_ref(const char *refname, struct got_object_id *id,
2127 const char *remote_repo_name, int verbosity, struct got_repository *repo)
2129 const struct got_error *err, *unlock_err;
2130 char *remote_refname;
2131 struct got_reference *ref;
2133 if (strncmp("refs/", refname, 5) == 0)
2136 if (asprintf(&remote_refname, "refs/remotes/%s/%s",
2137 remote_repo_name, refname) == -1)
2138 return got_error_from_errno("asprintf");
2140 err = got_ref_open(&ref, repo, remote_refname, 1);
2142 if (err->code != GOT_ERR_NOT_REF)
2144 err = create_ref(remote_refname, id, verbosity, repo);
2146 err = update_ref(ref, id, 0, verbosity, repo);
2147 unlock_err = got_ref_unlock(ref);
2148 if (unlock_err && err == NULL)
2153 free(remote_refname);
2157 static const struct got_error *
2158 delete_ref(struct got_repository *repo, struct got_reference *ref)
2160 const struct got_error *err = NULL;
2161 struct got_object_id *id = NULL;
2162 char *id_str = NULL;
2165 if (got_ref_is_symbolic(ref)) {
2166 target = got_ref_get_symref_target(ref);
2168 err = got_ref_resolve(&id, repo, ref);
2171 err = got_object_id_str(&id_str, id);
2177 err = got_ref_delete(ref, repo);
2181 printf("Deleted %s: %s\n", got_ref_get_name(ref), target);
2188 static const struct got_error *
2189 delete_refs_for_remote(struct got_repository *repo, const char *remote_name)
2191 const struct got_error *err = NULL;
2192 struct got_reflist_head refs;
2193 struct got_reflist_entry *re;
2198 if (asprintf(&prefix, "refs/remotes/%s", remote_name) == -1) {
2199 err = got_error_from_errno("asprintf");
2202 err = got_ref_list(&refs, repo, prefix, got_ref_cmp_by_name, NULL);
2206 TAILQ_FOREACH(re, &refs, entry)
2207 delete_ref(repo, re->ref);
2209 got_ref_list_free(&refs);
2213 static const struct got_error *
2214 cmd_fetch(int argc, char *argv[])
2216 const struct got_error *error = NULL, *unlock_err;
2217 char *cwd = NULL, *repo_path = NULL;
2218 const char *remote_name;
2219 char *proto = NULL, *host = NULL, *port = NULL;
2220 char *repo_name = NULL, *server_path = NULL;
2221 const struct got_remote_repo *remotes, *remote = NULL;
2223 char *id_str = NULL;
2224 struct got_repository *repo = NULL;
2225 struct got_worktree *worktree = NULL;
2226 const struct got_gotconfig *repo_conf = NULL, *worktree_conf = NULL;
2227 struct got_pathlist_head refs, symrefs, wanted_branches, wanted_refs;
2228 struct got_pathlist_entry *pe;
2229 struct got_object_id *pack_hash = NULL;
2230 int i, ch, fetchfd = -1, fetchstatus;
2231 pid_t fetchpid = -1;
2232 struct got_fetch_progress_arg fpa;
2233 int verbosity = 0, fetch_all_branches = 0, list_refs_only = 0;
2234 int delete_refs = 0, replace_tags = 0, delete_remote = 0;
2235 int *pack_fds = NULL;
2238 TAILQ_INIT(&symrefs);
2239 TAILQ_INIT(&wanted_branches);
2240 TAILQ_INIT(&wanted_refs);
2242 while ((ch = getopt(argc, argv, "ab:dlr:tvqR:X")) != -1) {
2245 fetch_all_branches = 1;
2248 error = got_pathlist_append(&wanted_branches,
2260 repo_path = realpath(optarg, NULL);
2261 if (repo_path == NULL)
2262 return got_error_from_errno2("realpath",
2264 got_path_strip_trailing_slashes(repo_path);
2272 else if (verbosity < 3)
2279 error = got_pathlist_append(&wanted_refs,
2295 if (fetch_all_branches && !TAILQ_EMPTY(&wanted_branches))
2296 option_conflict('a', 'b');
2297 if (list_refs_only) {
2298 if (!TAILQ_EMPTY(&wanted_branches))
2299 option_conflict('l', 'b');
2300 if (fetch_all_branches)
2301 option_conflict('l', 'a');
2303 option_conflict('l', 'd');
2305 option_conflict('l', 'X');
2307 if (delete_remote) {
2308 if (fetch_all_branches)
2309 option_conflict('X', 'a');
2310 if (!TAILQ_EMPTY(&wanted_branches))
2311 option_conflict('X', 'b');
2313 option_conflict('X', 'd');
2315 option_conflict('X', 't');
2316 if (!TAILQ_EMPTY(&wanted_refs))
2317 option_conflict('X', 'R');
2322 errx(1, "-X option requires a remote name");
2323 remote_name = GOT_FETCH_DEFAULT_REMOTE_NAME;
2324 } else if (argc == 1)
2325 remote_name = argv[0];
2329 cwd = getcwd(NULL, 0);
2331 error = got_error_from_errno("getcwd");
2335 error = got_repo_pack_fds_open(&pack_fds);
2339 if (repo_path == NULL) {
2340 error = got_worktree_open(&worktree, cwd);
2341 if (error && error->code != GOT_ERR_NOT_WORKTREE)
2347 strdup(got_worktree_get_repo_path(worktree));
2348 if (repo_path == NULL)
2349 error = got_error_from_errno("strdup");
2353 repo_path = strdup(cwd);
2354 if (repo_path == NULL) {
2355 error = got_error_from_errno("strdup");
2361 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
2365 if (delete_remote) {
2366 error = delete_refs_for_remote(repo, remote_name);
2367 goto done; /* nothing else to do */
2371 worktree_conf = got_worktree_get_gotconfig(worktree);
2372 if (worktree_conf) {
2373 got_gotconfig_get_remotes(&nremotes, &remotes,
2375 for (i = 0; i < nremotes; i++) {
2376 if (strcmp(remotes[i].name, remote_name) == 0) {
2377 remote = &remotes[i];
2383 if (remote == NULL) {
2384 repo_conf = got_repo_get_gotconfig(repo);
2386 got_gotconfig_get_remotes(&nremotes, &remotes,
2388 for (i = 0; i < nremotes; i++) {
2389 if (strcmp(remotes[i].name, remote_name) == 0) {
2390 remote = &remotes[i];
2396 if (remote == NULL) {
2397 got_repo_get_gitconfig_remotes(&nremotes, &remotes, repo);
2398 for (i = 0; i < nremotes; i++) {
2399 if (strcmp(remotes[i].name, remote_name) == 0) {
2400 remote = &remotes[i];
2405 if (remote == NULL) {
2406 error = got_error_path(remote_name, GOT_ERR_NO_REMOTE);
2410 if (TAILQ_EMPTY(&wanted_branches)) {
2411 if (!fetch_all_branches)
2412 fetch_all_branches = remote->fetch_all_branches;
2413 for (i = 0; i < remote->nfetch_branches; i++) {
2414 got_pathlist_append(&wanted_branches,
2415 remote->fetch_branches[i], NULL);
2418 if (TAILQ_EMPTY(&wanted_refs)) {
2419 for (i = 0; i < remote->nfetch_refs; i++) {
2420 got_pathlist_append(&wanted_refs,
2421 remote->fetch_refs[i], NULL);
2425 error = got_dial_parse_uri(&proto, &host, &port, &server_path,
2426 &repo_name, remote->fetch_url);
2430 if (strcmp(proto, "git") == 0) {
2432 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2433 "sendfd dns inet unveil", NULL) == -1)
2436 } else if (strcmp(proto, "git+ssh") == 0 ||
2437 strcmp(proto, "ssh") == 0) {
2439 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2440 "sendfd unveil", NULL) == -1)
2443 } else if (strcmp(proto, "http") == 0 ||
2444 strcmp(proto, "git+http") == 0) {
2445 error = got_error_path(proto, GOT_ERR_NOT_IMPL);
2448 error = got_error_path(proto, GOT_ERR_BAD_PROTO);
2452 error = got_dial_apply_unveil(proto);
2456 error = apply_unveil(got_repo_get_path(repo), 0, NULL);
2461 printf("Connecting to \"%s\" %s%s%s\n", remote->name, host,
2462 port ? ":" : "", port ? port : "");
2464 error = got_fetch_connect(&fetchpid, &fetchfd, proto, host, port,
2465 server_path, verbosity);
2469 fpa.last_scaled_size[0] = '\0';
2470 fpa.last_p_indexed = -1;
2471 fpa.last_p_resolved = -1;
2472 fpa.verbosity = verbosity;
2474 fpa.create_configs = 0;
2475 fpa.configs_created = 0;
2476 memset(&fpa.config_info, 0, sizeof(fpa.config_info));
2477 error = got_fetch_pack(&pack_hash, &refs, &symrefs, remote->name,
2478 remote->mirror_references, fetch_all_branches, &wanted_branches,
2479 &wanted_refs, list_refs_only, verbosity, fetchfd, repo,
2480 fetch_progress, &fpa);
2484 if (list_refs_only) {
2485 error = list_remote_refs(&symrefs, &refs);
2489 if (pack_hash == NULL) {
2491 printf("Already up-to-date\n");
2492 } else if (verbosity >= 0) {
2493 error = got_object_id_str(&id_str, pack_hash);
2496 printf("\nFetched %s.pack\n", id_str);
2501 /* Update references provided with the pack file. */
2502 TAILQ_FOREACH(pe, &refs, entry) {
2503 const char *refname = pe->path;
2504 struct got_object_id *id = pe->data;
2505 struct got_reference *ref;
2506 char *remote_refname;
2508 if (is_wanted_ref(&wanted_refs, refname) &&
2509 !remote->mirror_references) {
2510 error = update_wanted_ref(refname, id,
2511 remote->name, verbosity, repo);
2517 if (remote->mirror_references ||
2518 strncmp("refs/tags/", refname, 10) == 0) {
2519 error = got_ref_open(&ref, repo, refname, 1);
2521 if (error->code != GOT_ERR_NOT_REF)
2523 error = create_ref(refname, id, verbosity,
2528 error = update_ref(ref, id, replace_tags,
2530 unlock_err = got_ref_unlock(ref);
2531 if (unlock_err && error == NULL)
2537 } else if (strncmp("refs/heads/", refname, 11) == 0) {
2538 if (asprintf(&remote_refname, "refs/remotes/%s/%s",
2539 remote_name, refname + 11) == -1) {
2540 error = got_error_from_errno("asprintf");
2544 error = got_ref_open(&ref, repo, remote_refname, 1);
2546 if (error->code != GOT_ERR_NOT_REF)
2548 error = create_ref(remote_refname, id,
2553 error = update_ref(ref, id, replace_tags,
2555 unlock_err = got_ref_unlock(ref);
2556 if (unlock_err && error == NULL)
2563 /* Also create a local branch if none exists yet. */
2564 error = got_ref_open(&ref, repo, refname, 1);
2566 if (error->code != GOT_ERR_NOT_REF)
2568 error = create_ref(refname, id, verbosity,
2573 unlock_err = got_ref_unlock(ref);
2574 if (unlock_err && error == NULL)
2581 error = delete_missing_refs(&refs, &symrefs, remote,
2587 if (!remote->mirror_references) {
2588 /* Update remote HEAD reference if the server provided one. */
2589 TAILQ_FOREACH(pe, &symrefs, entry) {
2590 struct got_reference *target_ref;
2591 const char *refname = pe->path;
2592 const char *target = pe->data;
2593 char *remote_refname = NULL, *remote_target = NULL;
2595 if (strcmp(refname, GOT_REF_HEAD) != 0)
2598 if (strncmp("refs/heads/", target, 11) != 0)
2601 if (asprintf(&remote_refname, "refs/remotes/%s/%s",
2602 remote->name, refname) == -1) {
2603 error = got_error_from_errno("asprintf");
2606 if (asprintf(&remote_target, "refs/remotes/%s/%s",
2607 remote->name, target + 11) == -1) {
2608 error = got_error_from_errno("asprintf");
2609 free(remote_refname);
2613 error = got_ref_open(&target_ref, repo, remote_target,
2616 free(remote_refname);
2617 free(remote_target);
2618 if (error->code == GOT_ERR_NOT_REF) {
2624 error = update_symref(remote_refname, target_ref,
2626 free(remote_refname);
2627 free(remote_target);
2628 got_ref_close(target_ref);
2635 if (kill(fetchpid, SIGTERM) == -1)
2636 error = got_error_from_errno("kill");
2637 if (waitpid(fetchpid, &fetchstatus, 0) == -1 && error == NULL)
2638 error = got_error_from_errno("waitpid");
2640 if (fetchfd != -1 && close(fetchfd) == -1 && error == NULL)
2641 error = got_error_from_errno("close");
2643 const struct got_error *close_err = got_repo_close(repo);
2648 got_worktree_close(worktree);
2650 const struct got_error *pack_err =
2651 got_repo_pack_fds_close(pack_fds);
2655 TAILQ_FOREACH(pe, &refs, entry) {
2656 free((void *)pe->path);
2659 got_pathlist_free(&refs);
2660 TAILQ_FOREACH(pe, &symrefs, entry) {
2661 free((void *)pe->path);
2664 got_pathlist_free(&symrefs);
2665 got_pathlist_free(&wanted_branches);
2666 got_pathlist_free(&wanted_refs);
2681 usage_checkout(void)
2683 fprintf(stderr, "usage: %s checkout [-E] [-b branch] [-c commit] "
2684 "[-p prefix] [-q] repository-path [worktree-path]\n",
2690 show_worktree_base_ref_warning(void)
2692 fprintf(stderr, "%s: warning: could not create a reference "
2693 "to the work tree's base commit; the commit could be "
2694 "garbage-collected by Git or 'gotadmin cleanup'; making the "
2695 "repository writable and running 'got update' will prevent this\n",
2699 struct got_checkout_progress_arg {
2700 const char *worktree_path;
2701 int had_base_commit_ref_error;
2705 static const struct got_error *
2706 checkout_progress(void *arg, unsigned char status, const char *path)
2708 struct got_checkout_progress_arg *a = arg;
2710 /* Base commit bump happens silently. */
2711 if (status == GOT_STATUS_BUMP_BASE)
2714 if (status == GOT_STATUS_BASE_REF_ERR) {
2715 a->had_base_commit_ref_error = 1;
2719 while (path[0] == '/')
2722 if (a->verbosity >= 0)
2723 printf("%c %s/%s\n", status, a->worktree_path, path);
2728 static const struct got_error *
2729 check_cancelled(void *arg)
2731 if (sigint_received || sigpipe_received)
2732 return got_error(GOT_ERR_CANCELLED);
2736 static const struct got_error *
2737 check_linear_ancestry(struct got_object_id *commit_id,
2738 struct got_object_id *base_commit_id, int allow_forwards_in_time_only,
2739 struct got_repository *repo)
2741 const struct got_error *err = NULL;
2742 struct got_object_id *yca_id;
2744 err = got_commit_graph_find_youngest_common_ancestor(&yca_id,
2745 commit_id, base_commit_id, 1, repo, check_cancelled, NULL);
2750 return got_error(GOT_ERR_ANCESTRY);
2753 * Require a straight line of history between the target commit
2754 * and the work tree's base commit.
2756 * Non-linear situations such as this require a rebase:
2758 * (commit) D F (base_commit)
2766 * 'got update' only handles linear cases:
2767 * Update forwards in time: A (base/yca) - B - C - D (commit)
2768 * Update backwards in time: D (base) - C - B - A (commit/yca)
2770 if (allow_forwards_in_time_only) {
2771 if (got_object_id_cmp(base_commit_id, yca_id) != 0)
2772 return got_error(GOT_ERR_ANCESTRY);
2773 } else if (got_object_id_cmp(commit_id, yca_id) != 0 &&
2774 got_object_id_cmp(base_commit_id, yca_id) != 0)
2775 return got_error(GOT_ERR_ANCESTRY);
2781 static const struct got_error *
2782 check_same_branch(struct got_object_id *commit_id,
2783 struct got_reference *head_ref, struct got_object_id *yca_id,
2784 struct got_repository *repo)
2786 const struct got_error *err = NULL;
2787 struct got_commit_graph *graph = NULL;
2788 struct got_object_id *head_commit_id = NULL;
2789 int is_same_branch = 0;
2791 err = got_ref_resolve(&head_commit_id, repo, head_ref);
2795 if (got_object_id_cmp(head_commit_id, commit_id) == 0) {
2799 if (yca_id && got_object_id_cmp(commit_id, yca_id) == 0) {
2804 err = got_commit_graph_open(&graph, "/", 1);
2808 err = got_commit_graph_iter_start(graph, head_commit_id, repo,
2809 check_cancelled, NULL);
2814 struct got_object_id *id;
2815 err = got_commit_graph_iter_next(&id, graph, repo,
2816 check_cancelled, NULL);
2818 if (err->code == GOT_ERR_ITER_COMPLETED)
2824 if (yca_id && got_object_id_cmp(id, yca_id) == 0)
2826 if (got_object_id_cmp(id, commit_id) == 0) {
2834 got_commit_graph_close(graph);
2835 free(head_commit_id);
2836 if (!err && !is_same_branch)
2837 err = got_error(GOT_ERR_ANCESTRY);
2841 static const struct got_error *
2842 checkout_ancestry_error(struct got_reference *ref, const char *commit_id_str)
2844 static char msg[512];
2845 const char *branch_name;
2847 if (got_ref_is_symbolic(ref))
2848 branch_name = got_ref_get_symref_target(ref);
2850 branch_name = got_ref_get_name(ref);
2852 if (strncmp("refs/heads/", branch_name, 11) == 0)
2855 snprintf(msg, sizeof(msg),
2856 "target commit is not contained in branch '%s'; "
2857 "the branch to use must be specified with -b; "
2858 "if necessary a new branch can be created for "
2859 "this commit with 'got branch -c %s BRANCH_NAME'",
2860 branch_name, commit_id_str);
2862 return got_error_msg(GOT_ERR_ANCESTRY, msg);
2865 static const struct got_error *
2866 cmd_checkout(int argc, char *argv[])
2868 const struct got_error *error = NULL;
2869 struct got_repository *repo = NULL;
2870 struct got_reference *head_ref = NULL, *ref = NULL;
2871 struct got_worktree *worktree = NULL;
2872 char *repo_path = NULL;
2873 char *worktree_path = NULL;
2874 const char *path_prefix = "";
2875 const char *branch_name = GOT_REF_HEAD, *refname = NULL;
2876 char *commit_id_str = NULL;
2877 struct got_object_id *commit_id = NULL;
2879 int ch, same_path_prefix, allow_nonempty = 0, verbosity = 0;
2880 struct got_pathlist_head paths;
2881 struct got_checkout_progress_arg cpa;
2882 int *pack_fds = NULL;
2886 while ((ch = getopt(argc, argv, "b:c:Ep:q")) != -1) {
2889 branch_name = optarg;
2892 commit_id_str = strdup(optarg);
2893 if (commit_id_str == NULL)
2894 return got_error_from_errno("strdup");
2900 path_prefix = optarg;
2915 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
2916 "unveil", NULL) == -1)
2920 char *base, *dotgit;
2922 repo_path = realpath(argv[0], NULL);
2923 if (repo_path == NULL)
2924 return got_error_from_errno2("realpath", argv[0]);
2925 cwd = getcwd(NULL, 0);
2927 error = got_error_from_errno("getcwd");
2934 error = got_path_basename(&base, path);
2937 dotgit = strstr(base, ".git");
2940 if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
2941 error = got_error_from_errno("asprintf");
2946 } else if (argc == 2) {
2947 repo_path = realpath(argv[0], NULL);
2948 if (repo_path == NULL) {
2949 error = got_error_from_errno2("realpath", argv[0]);
2952 worktree_path = realpath(argv[1], NULL);
2953 if (worktree_path == NULL) {
2954 if (errno != ENOENT) {
2955 error = got_error_from_errno2("realpath",
2959 worktree_path = strdup(argv[1]);
2960 if (worktree_path == NULL) {
2961 error = got_error_from_errno("strdup");
2968 got_path_strip_trailing_slashes(repo_path);
2969 got_path_strip_trailing_slashes(worktree_path);
2971 error = got_repo_pack_fds_open(&pack_fds);
2975 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
2979 /* Pre-create work tree path for unveil(2) */
2980 error = got_path_mkdir(worktree_path);
2982 if (!(error->code == GOT_ERR_ERRNO && errno == EISDIR) &&
2983 !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
2985 if (!allow_nonempty &&
2986 !got_path_dir_is_empty(worktree_path)) {
2987 error = got_error_path(worktree_path,
2988 GOT_ERR_DIR_NOT_EMPTY);
2993 error = apply_unveil(got_repo_get_path(repo), 0, worktree_path);
2997 error = got_ref_open(&head_ref, repo, branch_name, 0);
3001 error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
3002 if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
3005 error = got_worktree_open(&worktree, worktree_path);
3009 error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
3013 if (!same_path_prefix) {
3014 error = got_error(GOT_ERR_PATH_PREFIX);
3018 if (commit_id_str) {
3019 struct got_reflist_head refs;
3021 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
3025 error = got_repo_match_object_id(&commit_id, NULL,
3026 commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
3027 got_ref_list_free(&refs);
3030 error = check_linear_ancestry(commit_id,
3031 got_worktree_get_base_commit_id(worktree), 0, repo);
3032 if (error != NULL) {
3033 if (error->code == GOT_ERR_ANCESTRY) {
3034 error = checkout_ancestry_error(
3035 head_ref, commit_id_str);
3039 error = check_same_branch(commit_id, head_ref, NULL, repo);
3041 if (error->code == GOT_ERR_ANCESTRY) {
3042 error = checkout_ancestry_error(
3043 head_ref, commit_id_str);
3047 error = got_worktree_set_base_commit_id(worktree, repo,
3051 /* Expand potentially abbreviated commit ID string. */
3052 free(commit_id_str);
3053 error = got_object_id_str(&commit_id_str, commit_id);
3057 commit_id = got_object_id_dup(
3058 got_worktree_get_base_commit_id(worktree));
3059 if (commit_id == NULL) {
3060 error = got_error_from_errno("got_object_id_dup");
3063 error = got_object_id_str(&commit_id_str, commit_id);
3068 error = got_pathlist_append(&paths, "", NULL);
3071 cpa.worktree_path = worktree_path;
3072 cpa.had_base_commit_ref_error = 0;
3073 cpa.verbosity = verbosity;
3074 error = got_worktree_checkout_files(worktree, &paths, repo,
3075 checkout_progress, &cpa, check_cancelled, NULL);
3079 if (got_ref_is_symbolic(head_ref)) {
3080 error = got_ref_resolve_symbolic(&ref, repo, head_ref);
3083 refname = got_ref_get_name(ref);
3085 refname = got_ref_get_name(head_ref);
3086 printf("Checked out %s: %s\n", refname, commit_id_str);
3087 printf("Now shut up and hack\n");
3088 if (cpa.had_base_commit_ref_error)
3089 show_worktree_base_ref_warning();
3092 const struct got_error *pack_err =
3093 got_repo_pack_fds_close(pack_fds);
3098 got_ref_close(head_ref);
3101 got_pathlist_free(&paths);
3102 free(commit_id_str);
3105 free(worktree_path);
3110 struct got_update_progress_arg {
3122 print_update_progress_stats(struct got_update_progress_arg *upa)
3124 if (!upa->did_something)
3127 if (upa->conflicts > 0)
3128 printf("Files with new merge conflicts: %d\n", upa->conflicts);
3129 if (upa->obstructed > 0)
3130 printf("File paths obstructed by a non-regular file: %d\n",
3132 if (upa->not_updated > 0)
3133 printf("Files not updated because of existing merge "
3134 "conflicts: %d\n", upa->not_updated);
3138 * The meaning of some status codes differs between merge-style operations and
3139 * update operations. For example, the ! status code means "file was missing"
3140 * if changes were merged into the work tree, and "missing file was restored"
3141 * if the work tree was updated. This function should be used by any operation
3142 * which merges changes into the work tree without updating the work tree.
3145 print_merge_progress_stats(struct got_update_progress_arg *upa)
3147 if (!upa->did_something)
3150 if (upa->conflicts > 0)
3151 printf("Files with new merge conflicts: %d\n", upa->conflicts);
3152 if (upa->obstructed > 0)
3153 printf("File paths obstructed by a non-regular file: %d\n",
3155 if (upa->missing > 0)
3156 printf("Files which had incoming changes but could not be "
3157 "found in the work tree: %d\n", upa->missing);
3158 if (upa->not_deleted > 0)
3159 printf("Files not deleted due to differences in deleted "
3160 "content: %d\n", upa->not_deleted);
3161 if (upa->unversioned > 0)
3162 printf("Files not merged because an unversioned file was "
3163 "found in the work tree: %d\n", upa->unversioned);
3169 fprintf(stderr, "usage: %s update [-b branch] [-c commit] [-q] "
3175 static const struct got_error *
3176 update_progress(void *arg, unsigned char status, const char *path)
3178 struct got_update_progress_arg *upa = arg;
3180 if (status == GOT_STATUS_EXISTS ||
3181 status == GOT_STATUS_BASE_REF_ERR)
3184 upa->did_something = 1;
3186 /* Base commit bump happens silently. */
3187 if (status == GOT_STATUS_BUMP_BASE)
3190 if (status == GOT_STATUS_CONFLICT)
3192 if (status == GOT_STATUS_OBSTRUCTED)
3194 if (status == GOT_STATUS_CANNOT_UPDATE)
3196 if (status == GOT_STATUS_MISSING)
3198 if (status == GOT_STATUS_CANNOT_DELETE)
3200 if (status == GOT_STATUS_UNVERSIONED)
3203 while (path[0] == '/')
3205 if (upa->verbosity >= 0)
3206 printf("%c %s\n", status, path);
3211 static const struct got_error *
3212 switch_head_ref(struct got_reference *head_ref,
3213 struct got_object_id *commit_id, struct got_worktree *worktree,
3214 struct got_repository *repo)
3216 const struct got_error *err = NULL;
3218 int ref_has_moved = 0;
3220 /* Trivial case: switching between two different references. */
3221 if (strcmp(got_ref_get_name(head_ref),
3222 got_worktree_get_head_ref_name(worktree)) != 0) {
3223 printf("Switching work tree from %s to %s\n",
3224 got_worktree_get_head_ref_name(worktree),
3225 got_ref_get_name(head_ref));
3226 return got_worktree_set_head_ref(worktree, head_ref);
3229 err = check_linear_ancestry(commit_id,
3230 got_worktree_get_base_commit_id(worktree), 0, repo);
3232 if (err->code != GOT_ERR_ANCESTRY)
3239 /* Switching to a rebased branch with the same reference name. */
3240 err = got_object_id_str(&base_id_str,
3241 got_worktree_get_base_commit_id(worktree));
3244 printf("Reference %s now points at a different branch\n",
3245 got_worktree_get_head_ref_name(worktree));
3246 printf("Switching work tree from %s to %s\n", base_id_str,
3247 got_worktree_get_head_ref_name(worktree));
3251 static const struct got_error *
3252 check_rebase_or_histedit_in_progress(struct got_worktree *worktree)
3254 const struct got_error *err;
3257 err = got_worktree_rebase_in_progress(&in_progress, worktree);
3261 return got_error(GOT_ERR_REBASING);
3263 err = got_worktree_histedit_in_progress(&in_progress, worktree);
3267 return got_error(GOT_ERR_HISTEDIT_BUSY);
3272 static const struct got_error *
3273 check_merge_in_progress(struct got_worktree *worktree,
3274 struct got_repository *repo)
3276 const struct got_error *err;
3279 err = got_worktree_merge_in_progress(&in_progress, worktree, repo);
3283 return got_error(GOT_ERR_MERGE_BUSY);
3288 static const struct got_error *
3289 get_worktree_paths_from_argv(struct got_pathlist_head *paths, int argc,
3290 char *argv[], struct got_worktree *worktree)
3292 const struct got_error *err = NULL;
3294 struct got_pathlist_entry *new;
3300 return got_error_from_errno("strdup");
3301 return got_pathlist_append(paths, path, NULL);
3304 for (i = 0; i < argc; i++) {
3305 err = got_worktree_resolve_path(&path, worktree, argv[i]);
3308 err = got_pathlist_insert(&new, paths, path, NULL);
3309 if (err || new == NULL /* duplicate */) {
3319 static const struct got_error *
3320 wrap_not_worktree_error(const struct got_error *orig_err,
3321 const char *cmdname, const char *path)
3323 const struct got_error *err;
3324 struct got_repository *repo;
3325 static char msg[512];
3326 int *pack_fds = NULL;
3328 err = got_repo_pack_fds_open(&pack_fds);
3332 err = got_repo_open(&repo, path, NULL, pack_fds);
3336 snprintf(msg, sizeof(msg),
3337 "'got %s' needs a work tree in addition to a git repository\n"
3338 "Work trees can be checked out from this Git repository with "
3340 "The got(1) manual page contains more information.", cmdname);
3341 err = got_error_msg(GOT_ERR_NOT_WORKTREE, msg);
3342 got_repo_close(repo);
3344 const struct got_error *pack_err =
3345 got_repo_pack_fds_close(pack_fds);
3352 static const struct got_error *
3353 cmd_update(int argc, char *argv[])
3355 const struct got_error *error = NULL;
3356 struct got_repository *repo = NULL;
3357 struct got_worktree *worktree = NULL;
3358 char *worktree_path = NULL;
3359 struct got_object_id *commit_id = NULL;
3360 char *commit_id_str = NULL;
3361 const char *branch_name = NULL;
3362 struct got_reference *head_ref = NULL;
3363 struct got_pathlist_head paths;
3364 struct got_pathlist_entry *pe;
3365 int ch, verbosity = 0;
3366 struct got_update_progress_arg upa;
3367 int *pack_fds = NULL;
3371 while ((ch = getopt(argc, argv, "b:c:q")) != -1) {
3374 branch_name = optarg;
3377 commit_id_str = strdup(optarg);
3378 if (commit_id_str == NULL)
3379 return got_error_from_errno("strdup");
3394 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
3395 "unveil", NULL) == -1)
3398 worktree_path = getcwd(NULL, 0);
3399 if (worktree_path == NULL) {
3400 error = got_error_from_errno("getcwd");
3404 error = got_repo_pack_fds_open(&pack_fds);
3408 error = got_worktree_open(&worktree, worktree_path);
3410 if (error->code == GOT_ERR_NOT_WORKTREE)
3411 error = wrap_not_worktree_error(error, "update",
3416 error = check_rebase_or_histedit_in_progress(worktree);
3420 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
3425 error = apply_unveil(got_repo_get_path(repo), 0,
3426 got_worktree_get_root_path(worktree));
3430 error = check_merge_in_progress(worktree, repo);
3434 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
3438 error = got_ref_open(&head_ref, repo, branch_name ? branch_name :
3439 got_worktree_get_head_ref_name(worktree), 0);
3442 if (commit_id_str == NULL) {
3443 error = got_ref_resolve(&commit_id, repo, head_ref);
3446 error = got_object_id_str(&commit_id_str, commit_id);
3450 struct got_reflist_head refs;
3452 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
3456 error = got_repo_match_object_id(&commit_id, NULL,
3457 commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
3458 got_ref_list_free(&refs);
3459 free(commit_id_str);
3460 commit_id_str = NULL;
3463 error = got_object_id_str(&commit_id_str, commit_id);
3469 struct got_object_id *head_commit_id;
3470 TAILQ_FOREACH(pe, &paths, entry) {
3471 if (pe->path_len == 0)
3473 error = got_error_msg(GOT_ERR_BAD_PATH,
3474 "switching between branches requires that "
3475 "the entire work tree gets updated");
3478 error = got_ref_resolve(&head_commit_id, repo, head_ref);
3481 error = check_linear_ancestry(commit_id, head_commit_id, 0,
3483 free(head_commit_id);
3486 error = check_same_branch(commit_id, head_ref, NULL, repo);
3489 error = switch_head_ref(head_ref, commit_id, worktree, repo);
3493 error = check_linear_ancestry(commit_id,
3494 got_worktree_get_base_commit_id(worktree), 0, repo);
3495 if (error != NULL) {
3496 if (error->code == GOT_ERR_ANCESTRY)
3497 error = got_error(GOT_ERR_BRANCH_MOVED);
3500 error = check_same_branch(commit_id, head_ref, NULL, repo);
3505 if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
3507 error = got_worktree_set_base_commit_id(worktree, repo,
3513 memset(&upa, 0, sizeof(upa));
3514 upa.verbosity = verbosity;
3515 error = got_worktree_checkout_files(worktree, &paths, repo,
3516 update_progress, &upa, check_cancelled, NULL);
3520 if (upa.did_something) {
3521 printf("Updated to %s: %s\n",
3522 got_worktree_get_head_ref_name(worktree), commit_id_str);
3524 printf("Already up-to-date\n");
3526 print_update_progress_stats(&upa);
3529 const struct got_error *pack_err =
3530 got_repo_pack_fds_close(pack_fds);
3534 free(worktree_path);
3535 TAILQ_FOREACH(pe, &paths, entry)
3536 free((char *)pe->path);
3537 got_pathlist_free(&paths);
3539 free(commit_id_str);
3543 static const struct got_error *
3544 diff_blobs(struct got_object_id *blob_id1, struct got_object_id *blob_id2,
3545 const char *path, int diff_context, int ignore_whitespace,
3546 int force_text_diff, struct got_repository *repo, FILE *outfile)
3548 const struct got_error *err = NULL;
3549 struct got_blob_object *blob1 = NULL, *blob2 = NULL;
3550 FILE *f1 = NULL, *f2 = NULL;
3551 int fd1 = -1, fd2 = -1;
3553 fd1 = got_opentempfd();
3555 return got_error_from_errno("got_opentempfd");
3556 fd2 = got_opentempfd();
3558 err = got_error_from_errno("got_opentempfd");
3563 err = got_object_open_as_blob(&blob1, repo, blob_id1, 8192,
3567 f1 = got_opentemp();
3569 err = got_error_from_errno("got_opentemp");
3574 err = got_object_open_as_blob(&blob2, repo, blob_id2, 8192, fd2);
3578 f2 = got_opentemp();
3580 err = got_error_from_errno("got_opentemp");
3584 while (path[0] == '/')
3586 err = got_diff_blob(NULL, NULL, blob1, blob2, f1, f2, path, path,
3587 diff_context, ignore_whitespace, force_text_diff, outfile);
3589 if (fd1 != -1 && close(fd1) == -1 && err == NULL)
3590 err = got_error_from_errno("close");
3592 got_object_blob_close(blob1);
3593 if (fd2 != -1 && close(fd2) == -1 && err == NULL)
3594 err = got_error_from_errno("close");
3595 got_object_blob_close(blob2);
3596 if (f1 && fclose(f1) == EOF && err == NULL)
3597 err = got_error_from_errno("fclose");
3598 if (f2 && fclose(f2) == EOF && err == NULL)
3599 err = got_error_from_errno("fclose");
3603 static const struct got_error *
3604 diff_trees(struct got_object_id *tree_id1, struct got_object_id *tree_id2,
3605 const char *path, int diff_context, int ignore_whitespace,
3606 int force_text_diff, struct got_repository *repo, FILE *outfile)
3608 const struct got_error *err = NULL;
3609 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3610 struct got_diff_blob_output_unidiff_arg arg;
3611 FILE *f1 = NULL, *f2 = NULL;
3612 int fd1 = -1, fd2 = -1;
3615 err = got_object_open_as_tree(&tree1, repo, tree_id1);
3618 f1 = got_opentemp();
3620 err = got_error_from_errno("got_opentemp");
3624 fd1 = got_opentempfd();
3626 err = got_error_from_errno("got_opentempfd");
3631 err = got_object_open_as_tree(&tree2, repo, tree_id2);
3635 f2 = got_opentemp();
3637 err = got_error_from_errno("got_opentemp");
3640 fd2 = got_opentempfd();
3642 err = got_error_from_errno("got_opentempfd");
3645 arg.diff_context = diff_context;
3646 arg.ignore_whitespace = ignore_whitespace;
3647 arg.force_text_diff = force_text_diff;
3648 arg.outfile = outfile;
3649 arg.line_offsets = NULL;
3651 while (path[0] == '/')
3653 err = got_diff_tree(tree1, tree2, f1, f2, fd1, fd2, path, path, repo,
3654 got_diff_blob_output_unidiff, &arg, 1);
3657 got_object_tree_close(tree1);
3659 got_object_tree_close(tree2);
3660 if (f1 && fclose(f1) == EOF && err == NULL)
3661 err = got_error_from_errno("fclose");
3662 if (f2 && fclose(f2) == EOF && err == NULL)
3663 err = got_error_from_errno("fclose");
3664 if (fd1 != -1 && close(fd1) == -1 && err == NULL)
3665 err = got_error_from_errno("close");
3666 if (fd2 != -1 && close(fd2) == -1 && err == NULL)
3667 err = got_error_from_errno("close");
3671 static const struct got_error *
3672 get_changed_paths(struct got_pathlist_head *paths,
3673 struct got_commit_object *commit, struct got_repository *repo)
3675 const struct got_error *err = NULL;
3676 struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
3677 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3678 struct got_object_qid *qid;
3680 qid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
3682 struct got_commit_object *pcommit;
3683 err = got_object_open_as_commit(&pcommit, repo,
3688 tree_id1 = got_object_id_dup(
3689 got_object_commit_get_tree_id(pcommit));
3690 if (tree_id1 == NULL) {
3691 got_object_commit_close(pcommit);
3692 return got_error_from_errno("got_object_id_dup");
3694 got_object_commit_close(pcommit);
3699 err = got_object_open_as_tree(&tree1, repo, tree_id1);
3704 tree_id2 = got_object_commit_get_tree_id(commit);
3705 err = got_object_open_as_tree(&tree2, repo, tree_id2);
3709 err = got_diff_tree(tree1, tree2, NULL, NULL, -1, -1, "", "", repo,
3710 got_diff_tree_collect_changed_paths, paths, 0);
3713 got_object_tree_close(tree1);
3715 got_object_tree_close(tree2);
3720 static const struct got_error *
3721 print_patch(struct got_commit_object *commit, struct got_object_id *id,
3722 const char *path, int diff_context, struct got_repository *repo,
3725 const struct got_error *err = NULL;
3726 struct got_commit_object *pcommit = NULL;
3727 char *id_str1 = NULL, *id_str2 = NULL;
3728 struct got_object_id *obj_id1 = NULL, *obj_id2 = NULL;
3729 struct got_object_qid *qid;
3731 qid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
3733 err = got_object_open_as_commit(&pcommit, repo,
3737 err = got_object_id_str(&id_str1, &qid->id);
3742 err = got_object_id_str(&id_str2, id);
3746 if (path && path[0] != '\0') {
3748 err = got_object_id_by_path(&obj_id2, repo, commit, path);
3752 err = got_object_id_by_path(&obj_id1, repo,
3755 if (err->code != GOT_ERR_NO_TREE_ENTRY) {
3761 err = got_object_get_type(&obj_type, repo, obj_id2);
3767 "diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
3768 fprintf(outfile, "commit - %s\n",
3769 id_str1 ? id_str1 : "/dev/null");
3770 fprintf(outfile, "commit + %s\n", id_str2);
3772 case GOT_OBJ_TYPE_BLOB:
3773 err = diff_blobs(obj_id1, obj_id2, path, diff_context,
3774 0, 0, repo, outfile);
3776 case GOT_OBJ_TYPE_TREE:
3777 err = diff_trees(obj_id1, obj_id2, path, diff_context,
3778 0, 0, repo, outfile);
3781 err = got_error(GOT_ERR_OBJ_TYPE);
3787 obj_id2 = got_object_commit_get_tree_id(commit);
3789 obj_id1 = got_object_commit_get_tree_id(pcommit);
3791 "diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
3792 fprintf(outfile, "commit - %s\n",
3793 id_str1 ? id_str1 : "/dev/null");
3794 fprintf(outfile, "commit + %s\n", id_str2);
3795 err = diff_trees(obj_id1, obj_id2, "", diff_context, 0, 0,
3802 got_object_commit_close(pcommit);
3807 get_datestr(time_t *time, char *datebuf)
3809 struct tm mytm, *tm;
3812 tm = gmtime_r(time, &mytm);
3815 s = asctime_r(tm, datebuf);
3818 p = strchr(s, '\n');
3824 static const struct got_error *
3825 match_commit(int *have_match, struct got_object_id *id,
3826 struct got_commit_object *commit, regex_t *regex)
3828 const struct got_error *err = NULL;
3829 regmatch_t regmatch;
3830 char *id_str = NULL, *logmsg = NULL;
3834 err = got_object_id_str(&id_str, id);
3838 err = got_object_commit_get_logmsg(&logmsg, commit);
3842 if (regexec(regex, got_object_commit_get_author(commit), 1,
3843 ®match, 0) == 0 ||
3844 regexec(regex, got_object_commit_get_committer(commit), 1,
3845 ®match, 0) == 0 ||
3846 regexec(regex, id_str, 1, ®match, 0) == 0 ||
3847 regexec(regex, logmsg, 1, ®match, 0) == 0)
3856 match_changed_paths(int *have_match, struct got_pathlist_head *changed_paths,
3859 regmatch_t regmatch;
3860 struct got_pathlist_entry *pe;
3864 TAILQ_FOREACH(pe, changed_paths, entry) {
3865 if (regexec(regex, pe->path, 1, ®match, 0) == 0) {
3872 static const struct got_error *
3873 match_patch(int *have_match, struct got_commit_object *commit,
3874 struct got_object_id *id, const char *path, int diff_context,
3875 struct got_repository *repo, regex_t *regex, FILE *f)
3877 const struct got_error *err = NULL;
3879 size_t linesize = 0;
3881 regmatch_t regmatch;
3885 err = got_opentemp_truncate(f);
3889 err = print_patch(commit, id, path, diff_context, repo, f);
3893 if (fseeko(f, 0L, SEEK_SET) == -1) {
3894 err = got_error_from_errno("fseeko");
3898 while ((linelen = getline(&line, &linesize, f)) != -1) {
3899 if (regexec(regex, line, 1, ®match, 0) == 0) {
3909 #define GOT_COMMIT_SEP_STR "-----------------------------------------------\n"
3911 static const struct got_error*
3912 build_refs_str(char **refs_str, struct got_reflist_head *refs,
3913 struct got_object_id *id, struct got_repository *repo,
3916 static const struct got_error *err = NULL;
3917 struct got_reflist_entry *re;
3923 TAILQ_FOREACH(re, refs, entry) {
3924 struct got_tag_object *tag = NULL;
3925 struct got_object_id *ref_id;
3928 name = got_ref_get_name(re->ref);
3929 if (strcmp(name, GOT_REF_HEAD) == 0)
3931 if (strncmp(name, "refs/", 5) == 0)
3933 if (strncmp(name, "got/", 4) == 0)
3935 if (strncmp(name, "heads/", 6) == 0)
3937 if (strncmp(name, "remotes/", 8) == 0) {
3941 s = strstr(name, "/" GOT_REF_HEAD);
3942 if (s != NULL && s[strlen(s)] == '\0')
3945 err = got_ref_resolve(&ref_id, repo, re->ref);
3948 if (strncmp(name, "tags/", 5) == 0) {
3949 err = got_object_open_as_tag(&tag, repo, ref_id);
3951 if (err->code != GOT_ERR_OBJ_TYPE) {
3955 /* Ref points at something other than a tag. */
3960 cmp = got_object_id_cmp(tag ?
3961 got_object_tag_get_object_id(tag) : ref_id, id);
3964 got_object_tag_close(tag);
3968 if (asprintf(refs_str, "%s%s%s", s ? s : "",
3969 s ? ", " : "", name) == -1) {
3970 err = got_error_from_errno("asprintf");
3981 static const struct got_error *
3982 print_commit_oneline(struct got_commit_object *commit, struct got_object_id *id,
3983 struct got_repository *repo, struct got_reflist_object_id_map *refs_idmap)
3985 const struct got_error *err = NULL;
3986 char *ref_str = NULL, *id_str = NULL, *logmsg0 = NULL;
3987 char *comma, *s, *nl;
3988 struct got_reflist_head *refs;
3989 char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
3991 time_t committer_time;
3993 refs = got_reflist_object_id_map_lookup(refs_idmap, id);
3995 err = build_refs_str(&ref_str, refs, id, repo, 1);
3999 /* Display the first matching ref only. */
4000 if (ref_str && (comma = strchr(ref_str, ',')) != NULL)
4004 if (ref_str == NULL) {
4005 err = got_object_id_str(&id_str, id);
4010 committer_time = got_object_commit_get_committer_time(commit);
4011 if (gmtime_r(&committer_time, &tm) == NULL) {
4012 err = got_error_from_errno("gmtime_r");
4015 if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm) == 0) {
4016 err = got_error(GOT_ERR_NO_SPACE);
4020 err = got_object_commit_get_logmsg(&logmsg0, commit);
4025 while (isspace((unsigned char)s[0]))
4028 nl = strchr(s, '\n');
4034 printf("%s%-7s %s\n", datebuf, ref_str, s);
4036 printf("%s%.7s %s\n", datebuf, id_str, s);
4038 if (fflush(stdout) != 0 && err == NULL)
4039 err = got_error_from_errno("fflush");
4047 static const struct got_error *
4048 print_commit(struct got_commit_object *commit, struct got_object_id *id,
4049 struct got_repository *repo, const char *path,
4050 struct got_pathlist_head *changed_paths, int show_patch,
4051 int diff_context, struct got_reflist_object_id_map *refs_idmap,
4052 const char *custom_refs_str)
4054 const struct got_error *err = NULL;
4055 char *id_str, *datestr, *logmsg0, *logmsg, *line;
4057 time_t committer_time;
4058 const char *author, *committer;
4059 char *refs_str = NULL;
4061 err = got_object_id_str(&id_str, id);
4065 if (custom_refs_str == NULL) {
4066 struct got_reflist_head *refs;
4067 refs = got_reflist_object_id_map_lookup(refs_idmap, id);
4069 err = build_refs_str(&refs_str, refs, id, repo, 0);
4075 printf(GOT_COMMIT_SEP_STR);
4076 if (custom_refs_str)
4077 printf("commit %s (%s)\n", id_str, custom_refs_str);
4079 printf("commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
4080 refs_str ? refs_str : "", refs_str ? ")" : "");
4085 printf("from: %s\n", got_object_commit_get_author(commit));
4086 committer_time = got_object_commit_get_committer_time(commit);
4087 datestr = get_datestr(&committer_time, datebuf);
4089 printf("date: %s UTC\n", datestr);
4090 author = got_object_commit_get_author(commit);
4091 committer = got_object_commit_get_committer(commit);
4092 if (strcmp(author, committer) != 0)
4093 printf("via: %s\n", committer);
4094 if (got_object_commit_get_nparents(commit) > 1) {
4095 const struct got_object_id_queue *parent_ids;
4096 struct got_object_qid *qid;
4098 parent_ids = got_object_commit_get_parent_ids(commit);
4099 STAILQ_FOREACH(qid, parent_ids, entry) {
4100 err = got_object_id_str(&id_str, &qid->id);
4103 printf("parent %d: %s\n", n++, id_str);
4109 err = got_object_commit_get_logmsg(&logmsg0, commit);
4115 line = strsep(&logmsg, "\n");
4117 printf(" %s\n", line);
4121 if (changed_paths) {
4122 struct got_pathlist_entry *pe;
4123 TAILQ_FOREACH(pe, changed_paths, entry) {
4124 struct got_diff_changed_path *cp = pe->data;
4125 printf(" %c %s\n", cp->status, pe->path);
4130 err = print_patch(commit, id, path, diff_context, repo, stdout);
4135 if (fflush(stdout) != 0 && err == NULL)
4136 err = got_error_from_errno("fflush");
4143 static const struct got_error *
4144 print_commits(struct got_object_id *root_id, struct got_object_id *end_id,
4145 struct got_repository *repo, const char *path, int show_changed_paths,
4146 int show_patch, const char *search_pattern, int diff_context, int limit,
4147 int log_branches, int reverse_display_order,
4148 struct got_reflist_object_id_map *refs_idmap, int one_line,
4151 const struct got_error *err;
4152 struct got_commit_graph *graph;
4155 struct got_object_id_queue reversed_commits;
4156 struct got_object_qid *qid;
4157 struct got_commit_object *commit;
4158 struct got_pathlist_head changed_paths;
4159 struct got_pathlist_entry *pe;
4161 STAILQ_INIT(&reversed_commits);
4162 TAILQ_INIT(&changed_paths);
4164 if (search_pattern && regcomp(®ex, search_pattern,
4165 REG_EXTENDED | REG_NOSUB | REG_NEWLINE))
4166 return got_error_msg(GOT_ERR_REGEX, search_pattern);
4168 err = got_commit_graph_open(&graph, path, !log_branches);
4171 err = got_commit_graph_iter_start(graph, root_id, repo,
4172 check_cancelled, NULL);
4176 struct got_object_id *id;
4178 if (sigint_received || sigpipe_received)
4181 err = got_commit_graph_iter_next(&id, graph, repo,
4182 check_cancelled, NULL);
4184 if (err->code == GOT_ERR_ITER_COMPLETED)
4191 err = got_object_open_as_commit(&commit, repo, id);
4195 if (show_changed_paths && !reverse_display_order) {
4196 err = get_changed_paths(&changed_paths, commit, repo);
4201 if (search_pattern) {
4202 err = match_commit(&have_match, id, commit, ®ex);
4204 got_object_commit_close(commit);
4207 if (have_match == 0 && show_changed_paths)
4208 match_changed_paths(&have_match,
4209 &changed_paths, ®ex);
4210 if (have_match == 0 && show_patch) {
4211 err = match_patch(&have_match, commit, id,
4212 path, diff_context, repo, ®ex,
4217 if (have_match == 0) {
4218 got_object_commit_close(commit);
4219 TAILQ_FOREACH(pe, &changed_paths, entry) {
4220 free((char *)pe->path);
4223 got_pathlist_free(&changed_paths);
4228 if (reverse_display_order) {
4229 err = got_object_qid_alloc(&qid, id);
4232 STAILQ_INSERT_HEAD(&reversed_commits, qid, entry);
4233 got_object_commit_close(commit);
4236 err = print_commit_oneline(commit, id,
4239 err = print_commit(commit, id, repo, path,
4240 show_changed_paths ? &changed_paths : NULL,
4241 show_patch, diff_context, refs_idmap, NULL);
4242 got_object_commit_close(commit);
4246 if ((limit && --limit == 0) ||
4247 (end_id && got_object_id_cmp(id, end_id) == 0))
4250 TAILQ_FOREACH(pe, &changed_paths, entry) {
4251 free((char *)pe->path);
4254 got_pathlist_free(&changed_paths);
4256 if (reverse_display_order) {
4257 STAILQ_FOREACH(qid, &reversed_commits, entry) {
4258 err = got_object_open_as_commit(&commit, repo,
4262 if (show_changed_paths) {
4263 err = get_changed_paths(&changed_paths,
4269 err = print_commit_oneline(commit, &qid->id,
4272 err = print_commit(commit, &qid->id, repo, path,
4273 show_changed_paths ? &changed_paths : NULL,
4274 show_patch, diff_context, refs_idmap, NULL);
4275 got_object_commit_close(commit);
4278 TAILQ_FOREACH(pe, &changed_paths, entry) {
4279 free((char *)pe->path);
4282 got_pathlist_free(&changed_paths);
4286 while (!STAILQ_EMPTY(&reversed_commits)) {
4287 qid = STAILQ_FIRST(&reversed_commits);
4288 STAILQ_REMOVE_HEAD(&reversed_commits, entry);
4289 got_object_qid_free(qid);
4291 TAILQ_FOREACH(pe, &changed_paths, entry) {
4292 free((char *)pe->path);
4295 got_pathlist_free(&changed_paths);
4298 got_commit_graph_close(graph);
4305 fprintf(stderr, "usage: %s log [-b] [-p] [-P] [-s] [-c commit] "
4306 "[-C number] [ -l N ] [-x commit] [-S search-pattern] "
4307 "[-r repository-path] [-R] [path]\n", getprogname());
4312 get_default_log_limit(void)
4314 const char *got_default_log_limit;
4318 got_default_log_limit = getenv("GOT_LOG_DEFAULT_LIMIT");
4319 if (got_default_log_limit == NULL)
4321 n = strtonum(got_default_log_limit, 0, INT_MAX, &errstr);
4327 static const struct got_error *
4328 cmd_log(int argc, char *argv[])
4330 const struct got_error *error;
4331 struct got_repository *repo = NULL;
4332 struct got_worktree *worktree = NULL;
4333 struct got_object_id *start_id = NULL, *end_id = NULL;
4334 char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
4335 const char *start_commit = NULL, *end_commit = NULL;
4336 const char *search_pattern = NULL;
4337 int diff_context = -1, ch;
4338 int show_changed_paths = 0, show_patch = 0, limit = 0, log_branches = 0;
4339 int reverse_display_order = 0, one_line = 0;
4341 struct got_reflist_head refs;
4342 struct got_reflist_object_id_map *refs_idmap = NULL;
4343 FILE *tmpfile = NULL;
4344 int *pack_fds = NULL;
4349 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
4355 limit = get_default_log_limit();
4357 while ((ch = getopt(argc, argv, "bpPc:C:l:r:RsS:x:")) != -1) {
4363 show_changed_paths = 1;
4366 start_commit = optarg;
4369 diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
4372 errx(1, "number of context lines is %s: %s",
4376 limit = strtonum(optarg, 0, INT_MAX, &errstr);
4378 errx(1, "number of commits is %s: %s",
4385 repo_path = realpath(optarg, NULL);
4386 if (repo_path == NULL)
4387 return got_error_from_errno2("realpath",
4389 got_path_strip_trailing_slashes(repo_path);
4392 reverse_display_order = 1;
4398 search_pattern = optarg;
4401 end_commit = optarg;
4412 if (diff_context == -1)
4414 else if (!show_patch)
4415 errx(1, "-C requires -p");
4417 if (one_line && (show_patch || show_changed_paths))
4418 errx(1, "cannot use -s with -p or -P");
4420 cwd = getcwd(NULL, 0);
4422 error = got_error_from_errno("getcwd");
4426 error = got_repo_pack_fds_open(&pack_fds);
4430 if (repo_path == NULL) {
4431 error = got_worktree_open(&worktree, cwd);
4432 if (error && error->code != GOT_ERR_NOT_WORKTREE)
4439 error = got_worktree_resolve_path(&path, worktree,
4444 path = strdup(argv[0]);
4446 error = got_error_from_errno("strdup");
4450 } else if (argc != 0)
4453 if (repo_path == NULL) {
4454 repo_path = worktree ?
4455 strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
4457 if (repo_path == NULL) {
4458 error = got_error_from_errno("strdup");
4462 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4466 error = apply_unveil(got_repo_get_path(repo), 1,
4467 worktree ? got_worktree_get_root_path(worktree) : NULL);
4471 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
4475 error = got_reflist_object_id_map_create(&refs_idmap, &refs, repo);
4479 if (start_commit == NULL) {
4480 struct got_reference *head_ref;
4481 struct got_commit_object *commit = NULL;
4482 error = got_ref_open(&head_ref, repo,
4483 worktree ? got_worktree_get_head_ref_name(worktree)
4487 error = got_ref_resolve(&start_id, repo, head_ref);
4488 got_ref_close(head_ref);
4491 error = got_object_open_as_commit(&commit, repo,
4495 got_object_commit_close(commit);
4497 error = got_repo_match_object_id(&start_id, NULL,
4498 start_commit, GOT_OBJ_TYPE_COMMIT, &refs, repo);
4502 if (end_commit != NULL) {
4503 error = got_repo_match_object_id(&end_id, NULL,
4504 end_commit, GOT_OBJ_TYPE_COMMIT, &refs, repo);
4511 * If a path was specified on the command line it was resolved
4512 * to a path in the work tree above. Prepend the work tree's
4513 * path prefix to obtain the corresponding in-repository path.
4517 prefix = got_worktree_get_path_prefix(worktree);
4518 if (asprintf(&in_repo_path, "%s%s%s", prefix,
4519 (path[0] != '\0') ? "/" : "", path) == -1) {
4520 error = got_error_from_errno("asprintf");
4525 error = got_repo_map_path(&in_repo_path, repo,
4531 path = in_repo_path;
4535 /* Release work tree lock. */
4536 got_worktree_close(worktree);
4540 if (search_pattern && show_patch) {
4541 tmpfile = got_opentemp();
4542 if (tmpfile == NULL) {
4543 error = got_error_from_errno("got_opentemp");
4548 error = print_commits(start_id, end_id, repo, path ? path : "",
4549 show_changed_paths, show_patch, search_pattern, diff_context,
4550 limit, log_branches, reverse_display_order, refs_idmap, one_line,
4557 got_worktree_close(worktree);
4559 const struct got_error *close_err = got_repo_close(repo);
4564 const struct got_error *pack_err =
4565 got_repo_pack_fds_close(pack_fds);
4570 got_reflist_object_id_map_free(refs_idmap);
4571 if (tmpfile && fclose(tmpfile) == EOF && error == NULL)
4572 error = got_error_from_errno("fclose");
4573 got_ref_list_free(&refs);
4580 fprintf(stderr, "usage: %s diff [-a] [-c commit] [-C number] "
4581 "[-r repository-path] [-s] [-w] [-P] "
4582 "[object1 object2 | path ...]\n", getprogname());
4586 struct print_diff_arg {
4587 struct got_repository *repo;
4588 struct got_worktree *worktree;
4593 int ignore_whitespace;
4594 int force_text_diff;
4598 * Create a file which contains the target path of a symlink so we can feed
4599 * it as content to the diff engine.
4601 static const struct got_error *
4602 get_symlink_target_file(int *fd, int dirfd, const char *de_name,
4603 const char *abspath)
4605 const struct got_error *err = NULL;
4606 char target_path[PATH_MAX];
4607 ssize_t target_len, outlen;
4612 target_len = readlinkat(dirfd, de_name, target_path, PATH_MAX);
4613 if (target_len == -1)
4614 return got_error_from_errno2("readlinkat", abspath);
4616 target_len = readlink(abspath, target_path, PATH_MAX);
4617 if (target_len == -1)
4618 return got_error_from_errno2("readlink", abspath);
4621 *fd = got_opentempfd();
4623 return got_error_from_errno("got_opentempfd");
4625 outlen = write(*fd, target_path, target_len);
4627 err = got_error_from_errno("got_opentempfd");
4631 if (lseek(*fd, 0, SEEK_SET) == -1) {
4632 err = got_error_from_errno2("lseek", abspath);
4643 static const struct got_error *
4644 print_diff(void *arg, unsigned char status, unsigned char staged_status,
4645 const char *path, struct got_object_id *blob_id,
4646 struct got_object_id *staged_blob_id, struct got_object_id *commit_id,
4647 int dirfd, const char *de_name)
4649 struct print_diff_arg *a = arg;
4650 const struct got_error *err = NULL;
4651 struct got_blob_object *blob1 = NULL;
4652 int fd = -1, fd1 = -1, fd2 = -1;
4653 FILE *f1 = NULL, *f2 = NULL;
4654 char *abspath = NULL, *label1 = NULL;
4658 if (a->diff_staged) {
4659 if (staged_status != GOT_STATUS_MODIFY &&
4660 staged_status != GOT_STATUS_ADD &&
4661 staged_status != GOT_STATUS_DELETE)
4664 if (staged_status == GOT_STATUS_DELETE)
4666 if (status == GOT_STATUS_NONEXISTENT)
4667 return got_error_set_errno(ENOENT, path);
4668 if (status != GOT_STATUS_MODIFY &&
4669 status != GOT_STATUS_ADD &&
4670 status != GOT_STATUS_DELETE &&
4671 status != GOT_STATUS_CONFLICT)
4675 if (!a->header_shown) {
4676 printf("diff %s%s\n", a->diff_staged ? "-s " : "",
4677 got_worktree_get_root_path(a->worktree));
4678 printf("commit - %s\n", a->id_str);
4679 printf("path + %s%s\n",
4680 got_worktree_get_root_path(a->worktree),
4681 a->diff_staged ? " (staged changes)" : "");
4682 a->header_shown = 1;
4685 if (a->diff_staged) {
4686 const char *label1 = NULL, *label2 = NULL;
4687 switch (staged_status) {
4688 case GOT_STATUS_MODIFY:
4692 case GOT_STATUS_ADD:
4695 case GOT_STATUS_DELETE:
4699 return got_error(GOT_ERR_FILE_STATUS);
4701 f1 = got_opentemp();
4703 err = got_error_from_errno("got_opentemp");
4706 f2 = got_opentemp();
4708 err = got_error_from_errno("got_opentemp");
4711 fd1 = got_opentempfd();
4713 err = got_error_from_errno("got_opentempfd");
4716 fd2 = got_opentempfd();
4718 err = got_error_from_errno("got_opentempfd");
4721 err = got_diff_objects_as_blobs(NULL, NULL, f1, f2, fd1, fd2,
4722 blob_id, staged_blob_id, label1, label2, a->diff_context,
4723 a->ignore_whitespace, a->force_text_diff, a->repo, stdout);
4727 fd1 = got_opentempfd();
4729 err = got_error_from_errno("got_opentempfd");
4733 if (staged_status == GOT_STATUS_ADD ||
4734 staged_status == GOT_STATUS_MODIFY) {
4736 err = got_object_open_as_blob(&blob1, a->repo, staged_blob_id,
4740 err = got_object_id_str(&id_str, staged_blob_id);
4743 if (asprintf(&label1, "%s (staged)", id_str) == -1) {
4744 err = got_error_from_errno("asprintf");
4749 } else if (status != GOT_STATUS_ADD) {
4750 err = got_object_open_as_blob(&blob1, a->repo, blob_id, 8192,
4756 if (status != GOT_STATUS_DELETE) {
4757 if (asprintf(&abspath, "%s/%s",
4758 got_worktree_get_root_path(a->worktree), path) == -1) {
4759 err = got_error_from_errno("asprintf");
4764 fd = openat(dirfd, de_name,
4765 O_RDONLY | O_NOFOLLOW | O_CLOEXEC);
4767 if (!got_err_open_nofollow_on_symlink()) {
4768 err = got_error_from_errno2("openat",
4772 err = get_symlink_target_file(&fd, dirfd,
4778 fd = open(abspath, O_RDONLY | O_NOFOLLOW | O_CLOEXEC);
4780 if (!got_err_open_nofollow_on_symlink()) {
4781 err = got_error_from_errno2("open",
4785 err = get_symlink_target_file(&fd, dirfd,
4791 if (fstat(fd, &sb) == -1) {
4792 err = got_error_from_errno2("fstat", abspath);
4795 f2 = fdopen(fd, "r");
4797 err = got_error_from_errno2("fdopen", abspath);
4805 f1 = got_opentemp();
4807 err = got_error_from_errno("got_opentemp");
4810 err = got_object_blob_dump_to_file(&size1, NULL, NULL, f1,
4816 err = got_diff_blob_file(blob1, f1, size1, label1, f2, sb.st_size,
4817 path, a->diff_context, a->ignore_whitespace, a->force_text_diff,
4820 if (fd1 != -1 && close(fd1) == -1 && err == NULL)
4821 err = got_error_from_errno("close");
4822 if (fd2 != -1 && close(fd2) == -1 && err == NULL)
4823 err = got_error_from_errno("close");
4825 got_object_blob_close(blob1);
4826 if (f1 && fclose(f1) == EOF && err == NULL)
4827 err = got_error_from_errno("fclose");
4828 if (f2 && fclose(f2) == EOF && err == NULL)
4829 err = got_error_from_errno("fclose");
4830 if (fd != -1 && close(fd) == -1 && err == NULL)
4831 err = got_error_from_errno("close");
4836 static const struct got_error *
4837 cmd_diff(int argc, char *argv[])
4839 const struct got_error *error;
4840 struct got_repository *repo = NULL;
4841 struct got_worktree *worktree = NULL;
4842 char *cwd = NULL, *repo_path = NULL;
4843 const char *commit_args[2] = { NULL, NULL };
4844 int ncommit_args = 0;
4845 struct got_object_id *ids[2] = { NULL, NULL };
4846 char *labels[2] = { NULL, NULL };
4847 int type1 = GOT_OBJ_TYPE_ANY, type2 = GOT_OBJ_TYPE_ANY;
4848 int diff_context = 3, diff_staged = 0, ignore_whitespace = 0, ch, i;
4849 int force_text_diff = 0, force_path = 0, rflag = 0;
4851 struct got_reflist_head refs;
4852 struct got_pathlist_head paths;
4853 struct got_pathlist_entry *pe;
4854 FILE *f1 = NULL, *f2 = NULL;
4855 int fd1 = -1, fd2 = -1;
4856 int *pack_fds = NULL;
4862 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
4867 while ((ch = getopt(argc, argv, "ac:C:r:swP")) != -1) {
4870 force_text_diff = 1;
4873 if (ncommit_args >= 2)
4874 errx(1, "too many -c options used");
4875 commit_args[ncommit_args++] = optarg;
4878 diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
4881 errx(1, "number of context lines is %s: %s",
4885 repo_path = realpath(optarg, NULL);
4886 if (repo_path == NULL)
4887 return got_error_from_errno2("realpath",
4889 got_path_strip_trailing_slashes(repo_path);
4896 ignore_whitespace = 1;
4910 cwd = getcwd(NULL, 0);
4912 error = got_error_from_errno("getcwd");
4916 error = got_repo_pack_fds_open(&pack_fds);
4920 if (repo_path == NULL) {
4921 error = got_worktree_open(&worktree, cwd);
4922 if (error && error->code != GOT_ERR_NOT_WORKTREE)
4928 strdup(got_worktree_get_repo_path(worktree));
4929 if (repo_path == NULL) {
4930 error = got_error_from_errno("strdup");
4934 repo_path = strdup(cwd);
4935 if (repo_path == NULL) {
4936 error = got_error_from_errno("strdup");
4942 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4947 if (rflag || worktree == NULL || ncommit_args > 0) {
4949 error = got_error_msg(GOT_ERR_NOT_IMPL,
4950 "-P option can only be used when diffing "
4955 error = got_error_msg(GOT_ERR_NOT_IMPL,
4956 "-s option can only be used when diffing "
4962 error = apply_unveil(got_repo_get_path(repo), 1,
4963 worktree ? got_worktree_get_root_path(worktree) : NULL);
4967 if ((!force_path && argc == 2) || ncommit_args > 0) {
4968 int obj_type = (ncommit_args > 0 ?
4969 GOT_OBJ_TYPE_COMMIT : GOT_OBJ_TYPE_ANY);
4970 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
4974 for (i = 0; i < (ncommit_args > 0 ? ncommit_args : argc); i++) {
4976 if (ncommit_args > 0)
4977 arg = commit_args[i];
4980 error = got_repo_match_object_id(&ids[i], &labels[i],
4981 arg, obj_type, &refs, repo);
4983 if (error->code != GOT_ERR_NOT_REF &&
4984 error->code != GOT_ERR_NO_OBJ)
4986 if (ncommit_args > 0)
4994 if (ncommit_args == 0 && (ids[0] == NULL || ids[1] == NULL)) {
4995 struct print_diff_arg arg;
4998 if (worktree == NULL) {
4999 if (argc == 2 && ids[0] == NULL) {
5000 error = got_error_path(argv[0], GOT_ERR_NO_OBJ);
5002 } else if (argc == 2 && ids[1] == NULL) {
5003 error = got_error_path(argv[1], GOT_ERR_NO_OBJ);
5005 } else if (argc > 0) {
5006 error = got_error_fmt(GOT_ERR_NOT_WORKTREE,
5007 "%s", "specified paths cannot be resolved");
5010 error = got_error(GOT_ERR_NOT_WORKTREE);
5015 error = get_worktree_paths_from_argv(&paths, argc, argv,
5020 error = got_object_id_str(&id_str,
5021 got_worktree_get_base_commit_id(worktree));
5025 arg.worktree = worktree;
5026 arg.diff_context = diff_context;
5027 arg.id_str = id_str;
5028 arg.header_shown = 0;
5029 arg.diff_staged = diff_staged;
5030 arg.ignore_whitespace = ignore_whitespace;
5031 arg.force_text_diff = force_text_diff;
5033 error = got_worktree_status(worktree, &paths, repo, 0,
5034 print_diff, &arg, check_cancelled, NULL);
5039 if (ncommit_args == 1) {
5040 struct got_commit_object *commit;
5041 error = got_object_open_as_commit(&commit, repo, ids[0]);
5045 labels[1] = labels[0];
5047 if (got_object_commit_get_nparents(commit) > 0) {
5048 const struct got_object_id_queue *pids;
5049 struct got_object_qid *pid;
5050 pids = got_object_commit_get_parent_ids(commit);
5051 pid = STAILQ_FIRST(pids);
5052 ids[0] = got_object_id_dup(&pid->id);
5053 if (ids[0] == NULL) {
5054 error = got_error_from_errno(
5055 "got_object_id_dup");
5056 got_object_commit_close(commit);
5059 error = got_object_id_str(&labels[0], ids[0]);
5061 got_object_commit_close(commit);
5066 labels[0] = strdup("/dev/null");
5067 if (labels[0] == NULL) {
5068 error = got_error_from_errno("strdup");
5069 got_object_commit_close(commit);
5074 got_object_commit_close(commit);
5077 if (ncommit_args == 0 && argc > 2) {
5078 error = got_error_msg(GOT_ERR_BAD_PATH,
5079 "path arguments cannot be used when diffing two objects");
5084 error = got_object_get_type(&type1, repo, ids[0]);
5089 error = got_object_get_type(&type2, repo, ids[1]);
5092 if (type1 != GOT_OBJ_TYPE_ANY && type1 != type2) {
5093 error = got_error(GOT_ERR_OBJ_TYPE);
5096 if (type1 == GOT_OBJ_TYPE_BLOB && argc > 0) {
5097 error = got_error_msg(GOT_ERR_OBJ_TYPE,
5098 "path arguments cannot be used when diffing blobs");
5102 for (i = 0; ncommit_args > 0 && i < argc; i++) {
5104 struct got_pathlist_entry *new;
5108 error = got_worktree_resolve_path(&p, worktree,
5112 prefix = got_worktree_get_path_prefix(worktree);
5113 while (prefix[0] == '/')
5115 if (asprintf(&in_repo_path, "%s%s%s", prefix,
5116 (p[0] != '\0' && prefix[0] != '\0') ? "/" : "",
5118 error = got_error_from_errno("asprintf");
5124 char *mapped_path, *s;
5125 error = got_repo_map_path(&mapped_path, repo, argv[i]);
5131 in_repo_path = strdup(s);
5132 if (in_repo_path == NULL) {
5133 error = got_error_from_errno("asprintf");
5140 error = got_pathlist_insert(&new, &paths, in_repo_path, NULL);
5141 if (error || new == NULL /* duplicate */)
5148 /* Release work tree lock. */
5149 got_worktree_close(worktree);
5153 f1 = got_opentemp();
5155 error = got_error_from_errno("got_opentemp");
5159 f2 = got_opentemp();
5161 error = got_error_from_errno("got_opentemp");
5165 fd1 = got_opentempfd();
5167 error = got_error_from_errno("got_opentempfd");
5171 fd2 = got_opentempfd();
5173 error = got_error_from_errno("got_opentempfd");
5177 switch (type1 == GOT_OBJ_TYPE_ANY ? type2 : type1) {
5178 case GOT_OBJ_TYPE_BLOB:
5179 error = got_diff_objects_as_blobs(NULL, NULL, f1, f2,
5180 fd1, fd2, ids[0], ids[1], NULL, NULL, diff_context,
5181 ignore_whitespace, force_text_diff, repo, stdout);
5183 case GOT_OBJ_TYPE_TREE:
5184 error = got_diff_objects_as_trees(NULL, NULL, f1, f2, fd1, fd2,
5185 ids[0], ids[1], &paths, "", "", diff_context,
5186 ignore_whitespace, force_text_diff, repo, stdout);
5188 case GOT_OBJ_TYPE_COMMIT:
5189 printf("diff %s %s\n", labels[0], labels[1]);
5190 error = got_diff_objects_as_commits(NULL, NULL, f1, f2,
5191 fd1, fd2, ids[0], ids[1], &paths, diff_context,
5192 ignore_whitespace, force_text_diff, repo, stdout);
5195 error = got_error(GOT_ERR_OBJ_TYPE);
5203 got_worktree_close(worktree);
5205 const struct got_error *close_err = got_repo_close(repo);
5210 const struct got_error *pack_err =
5211 got_repo_pack_fds_close(pack_fds);
5215 TAILQ_FOREACH(pe, &paths, entry)
5216 free((char *)pe->path);
5217 got_pathlist_free(&paths);
5218 got_ref_list_free(&refs);
5219 if (f1 && fclose(f1) == EOF && error == NULL)
5220 error = got_error_from_errno("fclose");
5221 if (f2 && fclose(f2) == EOF && error == NULL)
5222 error = got_error_from_errno("fclose");
5223 if (fd1 != -1 && close(fd1) == -1 && error == NULL)
5224 error = got_error_from_errno("close");
5225 if (fd2 != -1 && close(fd2) == -1 && error == NULL)
5226 error = got_error_from_errno("close");
5234 "usage: %s blame [-c commit] [-r repository-path] path\n",
5243 char datebuf[11]; /* YYYY-MM-DD + NUL */
5246 struct blame_cb_args {
5247 struct blame_line *lines;
5251 off_t *line_offsets;
5253 struct got_repository *repo;
5256 static const struct got_error *
5257 blame_cb(void *arg, int nlines, int lineno,
5258 struct got_commit_object *commit, struct got_object_id *id)
5260 const struct got_error *err = NULL;
5261 struct blame_cb_args *a = arg;
5262 struct blame_line *bline;
5264 size_t linesize = 0;
5267 time_t committer_time;
5269 if (nlines != a->nlines ||
5270 (lineno != -1 && lineno < 1) || lineno > a->nlines)
5271 return got_error(GOT_ERR_RANGE);
5273 if (sigint_received)
5274 return got_error(GOT_ERR_ITER_COMPLETED);
5277 return NULL; /* no change in this commit */
5279 /* Annotate this line. */
5280 bline = &a->lines[lineno - 1];
5281 if (bline->annotated)
5283 err = got_object_id_str(&bline->id_str, id);
5287 bline->committer = strdup(got_object_commit_get_committer(commit));
5288 if (bline->committer == NULL) {
5289 err = got_error_from_errno("strdup");
5293 committer_time = got_object_commit_get_committer_time(commit);
5294 if (gmtime_r(&committer_time, &tm) == NULL)
5295 return got_error_from_errno("gmtime_r");
5296 if (strftime(bline->datebuf, sizeof(bline->datebuf), "%G-%m-%d",
5298 err = got_error(GOT_ERR_NO_SPACE);
5301 bline->annotated = 1;
5303 /* Print lines annotated so far. */
5304 bline = &a->lines[a->lineno_cur - 1];
5305 if (!bline->annotated)
5308 offset = a->line_offsets[a->lineno_cur - 1];
5309 if (fseeko(a->f, offset, SEEK_SET) == -1) {
5310 err = got_error_from_errno("fseeko");
5314 while (bline->annotated) {
5315 char *smallerthan, *at, *nl, *committer;
5318 if (getline(&line, &linesize, a->f) == -1) {
5320 err = got_error_from_errno("getline");
5324 committer = bline->committer;
5325 smallerthan = strchr(committer, '<');
5326 if (smallerthan && smallerthan[1] != '\0')
5327 committer = smallerthan + 1;
5328 at = strchr(committer, '@');
5331 len = strlen(committer);
5333 committer[8] = '\0';
5335 nl = strchr(line, '\n');
5338 printf("%.*d) %.8s %s %-8s %s\n", a->nlines_prec, a->lineno_cur,
5339 bline->id_str, bline->datebuf, committer, line);
5342 bline = &a->lines[a->lineno_cur - 1];
5349 static const struct got_error *
5350 cmd_blame(int argc, char *argv[])
5352 const struct got_error *error;
5353 struct got_repository *repo = NULL;
5354 struct got_worktree *worktree = NULL;
5355 char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5356 char *link_target = NULL;
5357 struct got_object_id *obj_id = NULL;
5358 struct got_object_id *commit_id = NULL;
5359 struct got_commit_object *commit = NULL;
5360 struct got_blob_object *blob = NULL;
5361 char *commit_id_str = NULL;
5362 struct blame_cb_args bca;
5363 int ch, obj_type, i, fd1 = -1, fd2 = -1, fd3 = -1;
5365 int *pack_fds = NULL;
5366 FILE *f1 = NULL, *f2 = NULL;
5368 fd1 = got_opentempfd();
5370 return got_error_from_errno("got_opentempfd");
5372 memset(&bca, 0, sizeof(bca));
5375 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
5380 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5383 commit_id_str = optarg;
5386 repo_path = realpath(optarg, NULL);
5387 if (repo_path == NULL)
5388 return got_error_from_errno2("realpath",
5390 got_path_strip_trailing_slashes(repo_path);
5406 cwd = getcwd(NULL, 0);
5408 error = got_error_from_errno("getcwd");
5412 error = got_repo_pack_fds_open(&pack_fds);
5416 if (repo_path == NULL) {
5417 error = got_worktree_open(&worktree, cwd);
5418 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5424 strdup(got_worktree_get_repo_path(worktree));
5425 if (repo_path == NULL) {
5426 error = got_error_from_errno("strdup");
5431 repo_path = strdup(cwd);
5432 if (repo_path == NULL) {
5433 error = got_error_from_errno("strdup");
5439 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
5444 const char *prefix = got_worktree_get_path_prefix(worktree);
5447 error = got_worktree_resolve_path(&p, worktree, path);
5450 if (asprintf(&in_repo_path, "%s%s%s", prefix,
5451 (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
5453 error = got_error_from_errno("asprintf");
5458 error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5460 error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5463 error = got_repo_map_path(&in_repo_path, repo, path);
5468 if (commit_id_str == NULL) {
5469 struct got_reference *head_ref;
5470 error = got_ref_open(&head_ref, repo, worktree ?
5471 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
5474 error = got_ref_resolve(&commit_id, repo, head_ref);
5475 got_ref_close(head_ref);
5479 struct got_reflist_head refs;
5481 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
5485 error = got_repo_match_object_id(&commit_id, NULL,
5486 commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
5487 got_ref_list_free(&refs);
5493 /* Release work tree lock. */
5494 got_worktree_close(worktree);
5498 error = got_object_open_as_commit(&commit, repo, commit_id);
5502 error = got_object_resolve_symlinks(&link_target, in_repo_path,
5507 error = got_object_id_by_path(&obj_id, repo, commit,
5508 link_target ? link_target : in_repo_path);
5512 error = got_object_get_type(&obj_type, repo, obj_id);
5516 if (obj_type != GOT_OBJ_TYPE_BLOB) {
5517 error = got_error_path(link_target ? link_target : in_repo_path,
5522 error = got_object_open_as_blob(&blob, repo, obj_id, 8192, fd1);
5525 bca.f = got_opentemp();
5526 if (bca.f == NULL) {
5527 error = got_error_from_errno("got_opentemp");
5530 error = got_object_blob_dump_to_file(&filesize, &bca.nlines,
5531 &bca.line_offsets, bca.f, blob);
5532 if (error || bca.nlines == 0)
5535 /* Don't include \n at EOF in the blame line count. */
5536 if (bca.line_offsets[bca.nlines - 1] == filesize)
5539 bca.lines = calloc(bca.nlines, sizeof(*bca.lines));
5540 if (bca.lines == NULL) {
5541 error = got_error_from_errno("calloc");
5545 bca.nlines_prec = 0;
5553 fd2 = got_opentempfd();
5555 error = got_error_from_errno("got_opentempfd");
5558 fd3 = got_opentempfd();
5560 error = got_error_from_errno("got_opentempfd");
5563 f1 = got_opentemp();
5565 error = got_error_from_errno("got_opentemp");
5568 f2 = got_opentemp();
5570 error = got_error_from_errno("got_opentemp");
5573 error = got_blame(link_target ? link_target : in_repo_path, commit_id,
5574 repo, blame_cb, &bca, check_cancelled, NULL, fd2, fd3, f1, f2);
5583 got_object_commit_close(commit);
5585 if (fd1 != -1 && close(fd1) == -1 && error == NULL)
5586 error = got_error_from_errno("close");
5587 if (fd2 != -1 && close(fd2) == -1 && error == NULL)
5588 error = got_error_from_errno("close");
5589 if (fd3 != -1 && close(fd3) == -1 && error == NULL)
5590 error = got_error_from_errno("close");
5591 if (f1 && fclose(f1) == EOF && error == NULL)
5592 error = got_error_from_errno("fclose");
5593 if (f2 && fclose(f2) == EOF && error == NULL)
5594 error = got_error_from_errno("fclose");
5597 got_object_blob_close(blob);
5599 got_worktree_close(worktree);
5601 const struct got_error *close_err = got_repo_close(repo);
5606 const struct got_error *pack_err =
5607 got_repo_pack_fds_close(pack_fds);
5612 for (i = 0; i < bca.nlines; i++) {
5613 struct blame_line *bline = &bca.lines[i];
5614 free(bline->id_str);
5615 free(bline->committer);
5619 free(bca.line_offsets);
5620 if (bca.f && fclose(bca.f) == EOF && error == NULL)
5621 error = got_error_from_errno("fclose");
5629 "usage: %s tree [-c commit] [-r repository-path] [-iR] [path]\n",
5634 static const struct got_error *
5635 print_entry(struct got_tree_entry *te, const char *id, const char *path,
5636 const char *root_path, struct got_repository *repo)
5638 const struct got_error *err = NULL;
5639 int is_root_path = (strcmp(path, root_path) == 0);
5640 const char *modestr = "";
5641 mode_t mode = got_tree_entry_get_mode(te);
5642 char *link_target = NULL;
5644 path += strlen(root_path);
5645 while (path[0] == '/')
5648 if (got_object_tree_entry_is_submodule(te))
5650 else if (S_ISLNK(mode)) {
5653 err = got_tree_entry_get_symlink_target(&link_target, te, repo);
5656 for (i = 0; i < strlen(link_target); i++) {
5657 if (!isprint((unsigned char)link_target[i]))
5658 link_target[i] = '?';
5663 else if (S_ISDIR(mode))
5665 else if (mode & S_IXUSR)
5668 printf("%s%s%s%s%s%s%s\n", id ? id : "", path,
5669 is_root_path ? "" : "/", got_tree_entry_get_name(te), modestr,
5670 link_target ? " -> ": "", link_target ? link_target : "");
5676 static const struct got_error *
5677 print_tree(const char *path, struct got_commit_object *commit,
5678 int show_ids, int recurse, const char *root_path,
5679 struct got_repository *repo)
5681 const struct got_error *err = NULL;
5682 struct got_object_id *tree_id = NULL;
5683 struct got_tree_object *tree = NULL;
5686 err = got_object_id_by_path(&tree_id, repo, commit, path);
5690 err = got_object_open_as_tree(&tree, repo, tree_id);
5693 nentries = got_object_tree_get_nentries(tree);
5694 for (i = 0; i < nentries; i++) {
5695 struct got_tree_entry *te;
5698 if (sigint_received || sigpipe_received)
5701 te = got_object_tree_get_entry(tree, i);
5704 err = got_object_id_str(&id_str,
5705 got_tree_entry_get_id(te));
5708 if (asprintf(&id, "%s ", id_str) == -1) {
5709 err = got_error_from_errno("asprintf");
5715 err = print_entry(te, id, path, root_path, repo);
5720 if (recurse && S_ISDIR(got_tree_entry_get_mode(te))) {
5722 if (asprintf(&child_path, "%s%s%s", path,
5723 path[0] == '/' && path[1] == '\0' ? "" : "/",
5724 got_tree_entry_get_name(te)) == -1) {
5725 err = got_error_from_errno("asprintf");
5728 err = print_tree(child_path, commit, show_ids, 1,
5737 got_object_tree_close(tree);
5742 static const struct got_error *
5743 cmd_tree(int argc, char *argv[])
5745 const struct got_error *error;
5746 struct got_repository *repo = NULL;
5747 struct got_worktree *worktree = NULL;
5748 const char *path, *refname = NULL;
5749 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5750 struct got_object_id *commit_id = NULL;
5751 struct got_commit_object *commit = NULL;
5752 char *commit_id_str = NULL;
5753 int show_ids = 0, recurse = 0;
5755 int *pack_fds = NULL;
5758 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
5763 while ((ch = getopt(argc, argv, "c:r:iR")) != -1) {
5766 commit_id_str = optarg;
5769 repo_path = realpath(optarg, NULL);
5770 if (repo_path == NULL)
5771 return got_error_from_errno2("realpath",
5773 got_path_strip_trailing_slashes(repo_path);
5797 cwd = getcwd(NULL, 0);
5799 error = got_error_from_errno("getcwd");
5803 error = got_repo_pack_fds_open(&pack_fds);
5807 if (repo_path == NULL) {
5808 error = got_worktree_open(&worktree, cwd);
5809 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5815 strdup(got_worktree_get_repo_path(worktree));
5816 if (repo_path == NULL)
5817 error = got_error_from_errno("strdup");
5821 repo_path = strdup(cwd);
5822 if (repo_path == NULL) {
5823 error = got_error_from_errno("strdup");
5829 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
5834 const char *prefix = got_worktree_get_path_prefix(worktree);
5839 error = got_worktree_resolve_path(&p, worktree, path);
5842 if (asprintf(&in_repo_path, "%s%s%s", prefix,
5843 (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
5845 error = got_error_from_errno("asprintf");
5850 error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5854 error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5859 error = got_repo_map_path(&in_repo_path, repo, path);
5864 if (commit_id_str == NULL) {
5865 struct got_reference *head_ref;
5867 refname = got_worktree_get_head_ref_name(worktree);
5869 refname = GOT_REF_HEAD;
5870 error = got_ref_open(&head_ref, repo, refname, 0);
5873 error = got_ref_resolve(&commit_id, repo, head_ref);
5874 got_ref_close(head_ref);
5878 struct got_reflist_head refs;
5880 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
5884 error = got_repo_match_object_id(&commit_id, NULL,
5885 commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
5886 got_ref_list_free(&refs);
5892 /* Release work tree lock. */
5893 got_worktree_close(worktree);
5897 error = got_object_open_as_commit(&commit, repo, commit_id);
5901 error = print_tree(in_repo_path, commit, show_ids, recurse,
5902 in_repo_path, repo);
5909 got_object_commit_close(commit);
5911 got_worktree_close(worktree);
5913 const struct got_error *close_err = got_repo_close(repo);
5918 const struct got_error *pack_err =
5919 got_repo_pack_fds_close(pack_fds);
5929 fprintf(stderr, "usage: %s status [-I] [-s status-codes ] "
5930 "[-S status-codes] [path ...]\n", getprogname());
5934 struct got_status_arg {
5939 static const struct got_error *
5940 print_status(void *arg, unsigned char status, unsigned char staged_status,
5941 const char *path, struct got_object_id *blob_id,
5942 struct got_object_id *staged_blob_id, struct got_object_id *commit_id,
5943 int dirfd, const char *de_name)
5945 struct got_status_arg *st = arg;
5947 if (status == staged_status && (status == GOT_STATUS_DELETE))
5948 status = GOT_STATUS_NO_CHANGE;
5949 if (st != NULL && st->status_codes) {
5950 size_t ncodes = strlen(st->status_codes);
5953 for (i = 0; i < ncodes ; i++) {
5955 if (status == st->status_codes[i] ||
5956 staged_status == st->status_codes[i]) {
5961 if (status == st->status_codes[i] ||
5962 staged_status == st->status_codes[i])
5967 if (st->suppress && j == 0)
5974 printf("%c%c %s\n", status, staged_status, path);
5978 static const struct got_error *
5979 cmd_status(int argc, char *argv[])
5981 const struct got_error *error = NULL;
5982 struct got_repository *repo = NULL;
5983 struct got_worktree *worktree = NULL;
5984 struct got_status_arg st;
5986 struct got_pathlist_head paths;
5987 struct got_pathlist_entry *pe;
5988 int ch, i, no_ignores = 0;
5989 int *pack_fds = NULL;
5993 memset(&st, 0, sizeof(st));
5994 st.status_codes = NULL;
5997 while ((ch = getopt(argc, argv, "Is:S:")) != -1) {
6003 if (st.status_codes != NULL && st.suppress == 0)
6004 option_conflict('S', 's');
6008 for (i = 0; i < strlen(optarg); i++) {
6009 switch (optarg[i]) {
6010 case GOT_STATUS_MODIFY:
6011 case GOT_STATUS_ADD:
6012 case GOT_STATUS_DELETE:
6013 case GOT_STATUS_CONFLICT:
6014 case GOT_STATUS_MISSING:
6015 case GOT_STATUS_OBSTRUCTED:
6016 case GOT_STATUS_UNVERSIONED:
6017 case GOT_STATUS_MODE_CHANGE:
6018 case GOT_STATUS_NONEXISTENT:
6021 errx(1, "invalid status code '%c'",
6025 if (ch == 's' && st.suppress)
6026 option_conflict('s', 'S');
6027 st.status_codes = optarg;
6039 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
6043 cwd = getcwd(NULL, 0);
6045 error = got_error_from_errno("getcwd");
6049 error = got_repo_pack_fds_open(&pack_fds);
6053 error = got_worktree_open(&worktree, cwd);
6055 if (error->code == GOT_ERR_NOT_WORKTREE)
6056 error = wrap_not_worktree_error(error, "status", cwd);
6060 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
6065 error = apply_unveil(got_repo_get_path(repo), 1,
6066 got_worktree_get_root_path(worktree));
6070 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
6074 error = got_worktree_status(worktree, &paths, repo, no_ignores,
6075 print_status, &st, check_cancelled, NULL);
6078 const struct got_error *pack_err =
6079 got_repo_pack_fds_close(pack_fds);
6084 TAILQ_FOREACH(pe, &paths, entry)
6085 free((char *)pe->path);
6086 got_pathlist_free(&paths);
6095 "usage: %s ref [-r repository] [-l] [-t] [-c object] "
6096 "[-s reference] [-d] [name]\n",
6101 static const struct got_error *
6102 list_refs(struct got_repository *repo, const char *refname, int sort_by_time)
6104 static const struct got_error *err = NULL;
6105 struct got_reflist_head refs;
6106 struct got_reflist_entry *re;
6109 err = got_ref_list(&refs, repo, refname, sort_by_time ?
6110 got_ref_cmp_by_commit_timestamp_descending : got_ref_cmp_by_name,
6115 TAILQ_FOREACH(re, &refs, entry) {
6117 refstr = got_ref_to_str(re->ref);
6118 if (refstr == NULL) {
6119 err = got_error_from_errno("got_ref_to_str");
6122 printf("%s: %s\n", got_ref_get_name(re->ref), refstr);
6126 got_ref_list_free(&refs);
6130 static const struct got_error *
6131 delete_ref_by_name(struct got_repository *repo, const char *refname)
6133 const struct got_error *err;
6134 struct got_reference *ref;
6136 err = got_ref_open(&ref, repo, refname, 0);
6140 err = delete_ref(repo, ref);
6145 static const struct got_error *
6146 add_ref(struct got_repository *repo, const char *refname, const char *target)
6148 const struct got_error *err = NULL;
6149 struct got_object_id *id = NULL;
6150 struct got_reference *ref = NULL;
6151 struct got_reflist_head refs;
6154 * Don't let the user create a reference name with a leading '-'.
6155 * While technically a valid reference name, this case is usually
6156 * an unintended typo.
6158 if (refname[0] == '-')
6159 return got_error_path(refname, GOT_ERR_REF_NAME_MINUS);
6162 err = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
6165 err = got_repo_match_object_id(&id, NULL, target, GOT_OBJ_TYPE_ANY,
6167 got_ref_list_free(&refs);
6171 err = got_ref_alloc(&ref, refname, id);
6175 err = got_ref_write(ref, repo);
6183 static const struct got_error *
6184 add_symref(struct got_repository *repo, const char *refname, const char *target)
6186 const struct got_error *err = NULL;
6187 struct got_reference *ref = NULL;
6188 struct got_reference *target_ref = NULL;
6191 * Don't let the user create a reference name with a leading '-'.
6192 * While technically a valid reference name, this case is usually
6193 * an unintended typo.
6195 if (refname[0] == '-')
6196 return got_error_path(refname, GOT_ERR_REF_NAME_MINUS);
6198 err = got_ref_open(&target_ref, repo, target, 0);
6202 err = got_ref_alloc_symref(&ref, refname, target_ref);
6206 err = got_ref_write(ref, repo);
6209 got_ref_close(target_ref);
6215 static const struct got_error *
6216 cmd_ref(int argc, char *argv[])
6218 const struct got_error *error = NULL;
6219 struct got_repository *repo = NULL;
6220 struct got_worktree *worktree = NULL;
6221 char *cwd = NULL, *repo_path = NULL;
6222 int ch, do_list = 0, do_delete = 0, sort_by_time = 0;
6223 const char *obj_arg = NULL, *symref_target= NULL;
6224 char *refname = NULL;
6225 int *pack_fds = NULL;
6227 while ((ch = getopt(argc, argv, "c:dr:ls:t")) != -1) {
6236 repo_path = realpath(optarg, NULL);
6237 if (repo_path == NULL)
6238 return got_error_from_errno2("realpath",
6240 got_path_strip_trailing_slashes(repo_path);
6246 symref_target = optarg;
6257 if (obj_arg && do_list)
6258 option_conflict('c', 'l');
6259 if (obj_arg && do_delete)
6260 option_conflict('c', 'd');
6261 if (obj_arg && symref_target)
6262 option_conflict('c', 's');
6263 if (symref_target && do_delete)
6264 option_conflict('s', 'd');
6265 if (symref_target && do_list)
6266 option_conflict('s', 'l');
6267 if (do_delete && do_list)
6268 option_conflict('d', 'l');
6269 if (sort_by_time && !do_list)
6270 errx(1, "-t option requires -l option");
6276 if (argc != 0 && argc != 1)
6279 refname = strdup(argv[0]);
6280 if (refname == NULL) {
6281 error = got_error_from_errno("strdup");
6288 refname = strdup(argv[0]);
6289 if (refname == NULL) {
6290 error = got_error_from_errno("strdup");
6296 got_path_strip_trailing_slashes(refname);
6299 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
6300 "sendfd unveil", NULL) == -1)
6303 cwd = getcwd(NULL, 0);
6305 error = got_error_from_errno("getcwd");
6309 error = got_repo_pack_fds_open(&pack_fds);
6313 if (repo_path == NULL) {
6314 error = got_worktree_open(&worktree, cwd);
6315 if (error && error->code != GOT_ERR_NOT_WORKTREE)
6321 strdup(got_worktree_get_repo_path(worktree));
6322 if (repo_path == NULL)
6323 error = got_error_from_errno("strdup");
6327 repo_path = strdup(cwd);
6328 if (repo_path == NULL) {
6329 error = got_error_from_errno("strdup");
6335 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6341 /* Remove "cpath" promise. */
6342 if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
6348 error = apply_unveil(got_repo_get_path(repo), do_list,
6349 worktree ? got_worktree_get_root_path(worktree) : NULL);
6354 error = list_refs(repo, refname, sort_by_time);
6356 error = delete_ref_by_name(repo, refname);
6357 else if (symref_target)
6358 error = add_symref(repo, refname, symref_target);
6360 if (obj_arg == NULL)
6362 error = add_ref(repo, refname, obj_arg);
6367 const struct got_error *close_err = got_repo_close(repo);
6372 got_worktree_close(worktree);
6374 const struct got_error *pack_err =
6375 got_repo_pack_fds_close(pack_fds);
6388 "usage: %s branch [-c commit] [-d] [-r repository] [-l] [-t] "
6389 "[-n] [name]\n", getprogname());
6393 static const struct got_error *
6394 list_branch(struct got_repository *repo, struct got_worktree *worktree,
6395 struct got_reference *ref)
6397 const struct got_error *err = NULL;
6398 const char *refname, *marker = " ";
6401 refname = got_ref_get_name(ref);
6402 if (worktree && strcmp(refname,
6403 got_worktree_get_head_ref_name(worktree)) == 0) {
6404 struct got_object_id *id = NULL;
6406 err = got_ref_resolve(&id, repo, ref);
6409 if (got_object_id_cmp(id,
6410 got_worktree_get_base_commit_id(worktree)) == 0)
6417 if (strncmp(refname, "refs/heads/", 11) == 0)
6419 if (strncmp(refname, "refs/got/worktree/", 18) == 0)
6421 if (strncmp(refname, "refs/remotes/", 13) == 0)
6424 refstr = got_ref_to_str(ref);
6426 return got_error_from_errno("got_ref_to_str");
6428 printf("%s%s: %s\n", marker, refname, refstr);
6433 static const struct got_error *
6434 show_current_branch(struct got_repository *repo, struct got_worktree *worktree)
6436 const char *refname;
6438 if (worktree == NULL)
6439 return got_error(GOT_ERR_NOT_WORKTREE);
6441 refname = got_worktree_get_head_ref_name(worktree);
6443 if (strncmp(refname, "refs/heads/", 11) == 0)
6445 if (strncmp(refname, "refs/got/worktree/", 18) == 0)
6448 printf("%s\n", refname);
6453 static const struct got_error *
6454 list_branches(struct got_repository *repo, struct got_worktree *worktree,
6457 static const struct got_error *err = NULL;
6458 struct got_reflist_head refs;
6459 struct got_reflist_entry *re;
6460 struct got_reference *temp_ref = NULL;
6461 int rebase_in_progress, histedit_in_progress;
6466 err = got_worktree_rebase_in_progress(&rebase_in_progress,
6471 err = got_worktree_histedit_in_progress(&histedit_in_progress,
6476 if (rebase_in_progress || histedit_in_progress) {
6477 err = got_ref_open(&temp_ref, repo,
6478 got_worktree_get_head_ref_name(worktree), 0);
6481 list_branch(repo, worktree, temp_ref);
6482 got_ref_close(temp_ref);
6486 err = got_ref_list(&refs, repo, "refs/heads", sort_by_time ?
6487 got_ref_cmp_by_commit_timestamp_descending : got_ref_cmp_by_name,
6492 TAILQ_FOREACH(re, &refs, entry)
6493 list_branch(repo, worktree, re->ref);
6495 got_ref_list_free(&refs);
6497 err = got_ref_list(&refs, repo, "refs/remotes", sort_by_time ?
6498 got_ref_cmp_by_commit_timestamp_descending : got_ref_cmp_by_name,
6503 TAILQ_FOREACH(re, &refs, entry)
6504 list_branch(repo, worktree, re->ref);
6506 got_ref_list_free(&refs);
6511 static const struct got_error *
6512 delete_branch(struct got_repository *repo, struct got_worktree *worktree,
6513 const char *branch_name)
6515 const struct got_error *err = NULL;
6516 struct got_reference *ref = NULL;
6517 char *refname, *remote_refname = NULL;
6519 if (strncmp(branch_name, "refs/", 5) == 0)
6521 if (strncmp(branch_name, "heads/", 6) == 0)
6523 else if (strncmp(branch_name, "remotes/", 8) == 0)
6526 if (asprintf(&refname, "refs/heads/%s", branch_name) == -1)
6527 return got_error_from_errno("asprintf");
6529 if (asprintf(&remote_refname, "refs/remotes/%s",
6530 branch_name) == -1) {
6531 err = got_error_from_errno("asprintf");
6535 err = got_ref_open(&ref, repo, refname, 0);
6537 const struct got_error *err2;
6538 if (err->code != GOT_ERR_NOT_REF)
6541 * Keep 'err' intact such that if neither branch exists
6542 * we report "refs/heads" rather than "refs/remotes" in
6543 * our error message.
6545 err2 = got_ref_open(&ref, repo, remote_refname, 0);
6552 strcmp(got_worktree_get_head_ref_name(worktree),
6553 got_ref_get_name(ref)) == 0) {
6554 err = got_error_msg(GOT_ERR_SAME_BRANCH,
6555 "will not delete this work tree's current branch");
6559 err = delete_ref(repo, ref);
6564 free(remote_refname);
6568 static const struct got_error *
6569 add_branch(struct got_repository *repo, const char *branch_name,
6570 struct got_object_id *base_commit_id)
6572 const struct got_error *err = NULL;
6573 struct got_reference *ref = NULL;
6574 char *base_refname = NULL, *refname = NULL;
6577 * Don't let the user create a branch name with a leading '-'.
6578 * While technically a valid reference name, this case is usually
6579 * an unintended typo.
6581 if (branch_name[0] == '-')
6582 return got_error_path(branch_name, GOT_ERR_REF_NAME_MINUS);
6584 if (strncmp(branch_name, "refs/heads/", 11) == 0)
6587 if (asprintf(&refname, "refs/heads/%s", branch_name) == -1) {
6588 err = got_error_from_errno("asprintf");
6592 err = got_ref_open(&ref, repo, refname, 0);
6594 err = got_error(GOT_ERR_BRANCH_EXISTS);
6596 } else if (err->code != GOT_ERR_NOT_REF)
6599 err = got_ref_alloc(&ref, refname, base_commit_id);
6603 err = got_ref_write(ref, repo);
6612 static const struct got_error *
6613 cmd_branch(int argc, char *argv[])
6615 const struct got_error *error = NULL;
6616 struct got_repository *repo = NULL;
6617 struct got_worktree *worktree = NULL;
6618 char *cwd = NULL, *repo_path = NULL;
6619 int ch, do_list = 0, do_show = 0, do_update = 1, sort_by_time = 0;
6620 const char *delref = NULL, *commit_id_arg = NULL;
6621 struct got_reference *ref = NULL;
6622 struct got_pathlist_head paths;
6623 struct got_pathlist_entry *pe;
6624 struct got_object_id *commit_id = NULL;
6625 char *commit_id_str = NULL;
6626 int *pack_fds = NULL;
6630 while ((ch = getopt(argc, argv, "c:d:r:lnt")) != -1) {
6633 commit_id_arg = optarg;
6639 repo_path = realpath(optarg, NULL);
6640 if (repo_path == NULL)
6641 return got_error_from_errno2("realpath",
6643 got_path_strip_trailing_slashes(repo_path);
6660 if (do_list && delref)
6661 option_conflict('l', 'd');
6662 if (sort_by_time && !do_list)
6663 errx(1, "-t option requires -l option");
6668 if (!do_list && !delref && argc == 0)
6671 if ((do_list || delref || do_show) && commit_id_arg != NULL)
6672 errx(1, "-c option can only be used when creating a branch");
6674 if (do_list || delref) {
6677 } else if (!do_show && argc != 1)
6681 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
6682 "sendfd unveil", NULL) == -1)
6685 cwd = getcwd(NULL, 0);
6687 error = got_error_from_errno("getcwd");
6691 error = got_repo_pack_fds_open(&pack_fds);
6695 if (repo_path == NULL) {
6696 error = got_worktree_open(&worktree, cwd);
6697 if (error && error->code != GOT_ERR_NOT_WORKTREE)
6703 strdup(got_worktree_get_repo_path(worktree));
6704 if (repo_path == NULL)
6705 error = got_error_from_errno("strdup");
6709 repo_path = strdup(cwd);
6710 if (repo_path == NULL) {
6711 error = got_error_from_errno("strdup");
6717 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6722 if (do_list || do_show) {
6723 /* Remove "cpath" promise. */
6724 if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
6730 error = apply_unveil(got_repo_get_path(repo), do_list,
6731 worktree ? got_worktree_get_root_path(worktree) : NULL);
6736 error = show_current_branch(repo, worktree);
6738 error = list_branches(repo, worktree, sort_by_time);
6740 error = delete_branch(repo, worktree, delref);
6742 struct got_reflist_head refs;
6744 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
6748 if (commit_id_arg == NULL)
6749 commit_id_arg = worktree ?
6750 got_worktree_get_head_ref_name(worktree) :
6752 error = got_repo_match_object_id(&commit_id, NULL,
6753 commit_id_arg, GOT_OBJ_TYPE_COMMIT, &refs, repo);
6754 got_ref_list_free(&refs);
6757 error = add_branch(repo, argv[0], commit_id);
6760 if (worktree && do_update) {
6761 struct got_update_progress_arg upa;
6762 char *branch_refname = NULL;
6764 error = got_object_id_str(&commit_id_str, commit_id);
6767 error = get_worktree_paths_from_argv(&paths, 0, NULL,
6771 if (asprintf(&branch_refname, "refs/heads/%s", argv[0])
6773 error = got_error_from_errno("asprintf");
6776 error = got_ref_open(&ref, repo, branch_refname, 0);
6777 free(branch_refname);
6780 error = switch_head_ref(ref, commit_id, worktree,
6784 error = got_worktree_set_base_commit_id(worktree, repo,
6788 memset(&upa, 0, sizeof(upa));
6789 error = got_worktree_checkout_files(worktree, &paths,
6790 repo, update_progress, &upa, check_cancelled,
6794 if (upa.did_something) {
6795 printf("Updated to %s: %s\n",
6796 got_worktree_get_head_ref_name(worktree),
6799 print_update_progress_stats(&upa);
6806 const struct got_error *close_err = got_repo_close(repo);
6811 got_worktree_close(worktree);
6813 const struct got_error *pack_err =
6814 got_repo_pack_fds_close(pack_fds);
6821 free(commit_id_str);
6822 TAILQ_FOREACH(pe, &paths, entry)
6823 free((char *)pe->path);
6824 got_pathlist_free(&paths);
6833 "usage: %s tag [-c commit] [-r repository] [-l] "
6834 "[-m message] name\n", getprogname());
6839 static const struct got_error *
6840 sort_tags(struct got_reflist_head *sorted, struct got_reflist_head *tags)
6842 const struct got_error *err = NULL;
6843 struct got_reflist_entry *re, *se, *new;
6844 struct got_object_id *re_id, *se_id;
6845 struct got_tag_object *re_tag, *se_tag;
6846 time_t re_time, se_time;
6848 STAILQ_FOREACH(re, tags, entry) {
6849 se = STAILQ_FIRST(sorted);
6851 err = got_reflist_entry_dup(&new, re);
6854 STAILQ_INSERT_HEAD(sorted, new, entry);
6857 err = got_ref_resolve(&re_id, repo, re->ref);
6860 err = got_object_open_as_tag(&re_tag, repo, re_id);
6864 re_time = got_object_tag_get_tagger_time(re_tag);
6865 got_object_tag_close(re_tag);
6869 err = got_ref_resolve(&se_id, repo, re->ref);
6872 err = got_object_open_as_tag(&se_tag, repo, se_id);
6876 se_time = got_object_tag_get_tagger_time(se_tag);
6877 got_object_tag_close(se_tag);
6879 if (se_time > re_time) {
6880 err = got_reflist_entry_dup(&new, re);
6883 STAILQ_INSERT_AFTER(sorted, se, new, entry);
6886 se = STAILQ_NEXT(se, entry);
6895 static const struct got_error *
6896 get_tag_refname(char **refname, const char *tag_name)
6898 const struct got_error *err;
6900 if (strncmp("refs/tags/", tag_name, 10) == 0) {
6901 *refname = strdup(tag_name);
6902 if (*refname == NULL)
6903 return got_error_from_errno("strdup");
6904 } else if (asprintf(refname, "refs/tags/%s", tag_name) == -1) {
6905 err = got_error_from_errno("asprintf");
6913 static const struct got_error *
6914 list_tags(struct got_repository *repo, const char *tag_name)
6916 static const struct got_error *err = NULL;
6917 struct got_reflist_head refs;
6918 struct got_reflist_entry *re;
6919 char *wanted_refname = NULL;
6923 err = got_ref_list(&refs, repo, "refs/tags", got_ref_cmp_tags, repo);
6928 struct got_reference *ref;
6929 err = get_tag_refname(&wanted_refname, tag_name);
6932 /* Wanted tag reference should exist. */
6933 err = got_ref_open(&ref, repo, wanted_refname, 0);
6939 TAILQ_FOREACH(re, &refs, entry) {
6940 const char *refname;
6941 char *refstr, *tagmsg0, *tagmsg, *line, *id_str, *datestr;
6945 struct got_object_id *id;
6946 struct got_tag_object *tag;
6947 struct got_commit_object *commit = NULL;
6949 refname = got_ref_get_name(re->ref);
6950 if (strncmp(refname, "refs/tags/", 10) != 0 ||
6951 (wanted_refname && strcmp(refname, wanted_refname) != 0))
6954 refstr = got_ref_to_str(re->ref);
6955 if (refstr == NULL) {
6956 err = got_error_from_errno("got_ref_to_str");
6959 printf("%stag %s %s\n", GOT_COMMIT_SEP_STR, refname, refstr);
6962 err = got_ref_resolve(&id, repo, re->ref);
6965 err = got_object_open_as_tag(&tag, repo, id);
6967 if (err->code != GOT_ERR_OBJ_TYPE) {
6971 /* "lightweight" tag */
6972 err = got_object_open_as_commit(&commit, repo, id);
6977 tagger = got_object_commit_get_committer(commit);
6979 got_object_commit_get_committer_time(commit);
6980 err = got_object_id_str(&id_str, id);
6986 tagger = got_object_tag_get_tagger(tag);
6987 tagger_time = got_object_tag_get_tagger_time(tag);
6988 err = got_object_id_str(&id_str,
6989 got_object_tag_get_object_id(tag));
6993 printf("from: %s\n", tagger);
6994 datestr = get_datestr(&tagger_time, datebuf);
6996 printf("date: %s UTC\n", datestr);
6998 printf("object: %s %s\n", GOT_OBJ_LABEL_COMMIT, id_str);
7000 switch (got_object_tag_get_object_type(tag)) {
7001 case GOT_OBJ_TYPE_BLOB:
7002 printf("object: %s %s\n", GOT_OBJ_LABEL_BLOB,
7005 case GOT_OBJ_TYPE_TREE:
7006 printf("object: %s %s\n", GOT_OBJ_LABEL_TREE,
7009 case GOT_OBJ_TYPE_COMMIT:
7010 printf("object: %s %s\n", GOT_OBJ_LABEL_COMMIT,
7013 case GOT_OBJ_TYPE_TAG:
7014 printf("object: %s %s\n", GOT_OBJ_LABEL_TAG,
7023 err = got_object_commit_get_logmsg(&tagmsg0, commit);
7026 got_object_commit_close(commit);
7028 tagmsg0 = strdup(got_object_tag_get_message(tag));
7029 got_object_tag_close(tag);
7030 if (tagmsg0 == NULL) {
7031 err = got_error_from_errno("strdup");
7038 line = strsep(&tagmsg, "\n");
7040 printf(" %s\n", line);
7045 got_ref_list_free(&refs);
7046 free(wanted_refname);
7050 static const struct got_error *
7051 get_tag_message(char **tagmsg, char **tagmsg_path, const char *commit_id_str,
7052 const char *tag_name, const char *repo_path)
7054 const struct got_error *err = NULL;
7055 char *template = NULL, *initial_content = NULL;
7056 char *editor = NULL;
7057 int initial_content_len;
7060 if (asprintf(&template, GOT_TMPDIR_STR "/got-tagmsg") == -1) {
7061 err = got_error_from_errno("asprintf");
7065 initial_content_len = asprintf(&initial_content,
7066 "\n# tagging commit %s as %s\n",
7067 commit_id_str, tag_name);
7068 if (initial_content_len == -1) {
7069 err = got_error_from_errno("asprintf");
7073 err = got_opentemp_named_fd(tagmsg_path, &fd, template);
7077 if (write(fd, initial_content, initial_content_len) == -1) {
7078 err = got_error_from_errno2("write", *tagmsg_path);
7082 err = get_editor(&editor);
7085 err = edit_logmsg(tagmsg, editor, *tagmsg_path, initial_content,
7086 initial_content_len, 1);
7088 free(initial_content);
7092 if (fd != -1 && close(fd) == -1 && err == NULL)
7093 err = got_error_from_errno2("close", *tagmsg_path);
7095 /* Editor is done; we can now apply unveil(2) */
7097 err = apply_unveil(repo_path, 0, NULL);
7105 static const struct got_error *
7106 add_tag(struct got_repository *repo, const char *tagger,
7107 const char *tag_name, const char *commit_arg, const char *tagmsg_arg)
7109 const struct got_error *err = NULL;
7110 struct got_object_id *commit_id = NULL, *tag_id = NULL;
7111 char *label = NULL, *commit_id_str = NULL;
7112 struct got_reference *ref = NULL;
7113 char *refname = NULL, *tagmsg = NULL;
7114 char *tagmsg_path = NULL, *tag_id_str = NULL;
7115 int preserve_tagmsg = 0;
7116 struct got_reflist_head refs;
7121 * Don't let the user create a tag name with a leading '-'.
7122 * While technically a valid reference name, this case is usually
7123 * an unintended typo.
7125 if (tag_name[0] == '-')
7126 return got_error_path(tag_name, GOT_ERR_REF_NAME_MINUS);
7128 err = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
7132 err = got_repo_match_object_id(&commit_id, &label, commit_arg,
7133 GOT_OBJ_TYPE_COMMIT, &refs, repo);
7137 err = got_object_id_str(&commit_id_str, commit_id);
7141 err = get_tag_refname(&refname, tag_name);
7144 if (strncmp("refs/tags/", tag_name, 10) == 0)
7147 err = got_ref_open(&ref, repo, refname, 0);
7149 err = got_error(GOT_ERR_TAG_EXISTS);
7151 } else if (err->code != GOT_ERR_NOT_REF)
7154 if (tagmsg_arg == NULL) {
7155 err = get_tag_message(&tagmsg, &tagmsg_path, commit_id_str,
7156 tag_name, got_repo_get_path(repo));
7158 if (err->code != GOT_ERR_COMMIT_MSG_EMPTY &&
7159 tagmsg_path != NULL)
7160 preserve_tagmsg = 1;
7165 err = got_object_tag_create(&tag_id, tag_name, commit_id,
7166 tagger, time(NULL), tagmsg ? tagmsg : tagmsg_arg, repo);
7169 preserve_tagmsg = 1;
7173 err = got_ref_alloc(&ref, refname, tag_id);
7176 preserve_tagmsg = 1;
7180 err = got_ref_write(ref, repo);
7183 preserve_tagmsg = 1;
7187 err = got_object_id_str(&tag_id_str, tag_id);
7190 preserve_tagmsg = 1;
7193 printf("Created tag %s\n", tag_id_str);
7195 if (preserve_tagmsg) {
7196 fprintf(stderr, "%s: tag message preserved in %s\n",
7197 getprogname(), tagmsg_path);
7198 } else if (tagmsg_path && unlink(tagmsg_path) == -1 && err == NULL)
7199 err = got_error_from_errno2("unlink", tagmsg_path);
7204 free(commit_id_str);
7208 got_ref_list_free(&refs);
7212 static const struct got_error *
7213 cmd_tag(int argc, char *argv[])
7215 const struct got_error *error = NULL;
7216 struct got_repository *repo = NULL;
7217 struct got_worktree *worktree = NULL;
7218 char *cwd = NULL, *repo_path = NULL, *commit_id_str = NULL;
7219 char *gitconfig_path = NULL, *tagger = NULL;
7220 const char *tag_name = NULL, *commit_id_arg = NULL, *tagmsg = NULL;
7221 int ch, do_list = 0;
7222 int *pack_fds = NULL;
7224 while ((ch = getopt(argc, argv, "c:m:r:l")) != -1) {
7227 commit_id_arg = optarg;
7233 repo_path = realpath(optarg, NULL);
7234 if (repo_path == NULL)
7235 return got_error_from_errno2("realpath",
7237 got_path_strip_trailing_slashes(repo_path);
7252 if (commit_id_arg != NULL)
7254 "-c option can only be used when creating a tag");
7256 option_conflict('l', 'm');
7259 } else if (argc != 1)
7266 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
7267 "sendfd unveil", NULL) == -1)
7270 cwd = getcwd(NULL, 0);
7272 error = got_error_from_errno("getcwd");
7276 error = got_repo_pack_fds_open(&pack_fds);
7280 if (repo_path == NULL) {
7281 error = got_worktree_open(&worktree, cwd);
7282 if (error && error->code != GOT_ERR_NOT_WORKTREE)
7288 strdup(got_worktree_get_repo_path(worktree));
7289 if (repo_path == NULL)
7290 error = got_error_from_errno("strdup");
7294 repo_path = strdup(cwd);
7295 if (repo_path == NULL) {
7296 error = got_error_from_errno("strdup");
7304 /* Release work tree lock. */
7305 got_worktree_close(worktree);
7308 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
7313 /* Remove "cpath" promise. */
7314 if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
7318 error = apply_unveil(got_repo_get_path(repo), 1, NULL);
7321 error = list_tags(repo, tag_name);
7323 error = get_gitconfig_path(&gitconfig_path);
7326 error = got_repo_open(&repo, repo_path, gitconfig_path,
7331 error = get_author(&tagger, repo, worktree);
7335 /* Release work tree lock. */
7336 got_worktree_close(worktree);
7341 error = apply_unveil(got_repo_get_path(repo), 0, NULL);
7346 if (commit_id_arg == NULL) {
7347 struct got_reference *head_ref;
7348 struct got_object_id *commit_id;
7349 error = got_ref_open(&head_ref, repo,
7350 worktree ? got_worktree_get_head_ref_name(worktree)
7354 error = got_ref_resolve(&commit_id, repo, head_ref);
7355 got_ref_close(head_ref);
7358 error = got_object_id_str(&commit_id_str, commit_id);
7364 error = add_tag(repo, tagger, tag_name,
7365 commit_id_str ? commit_id_str : commit_id_arg, tagmsg);
7369 const struct got_error *close_err = got_repo_close(repo);
7374 got_worktree_close(worktree);
7376 const struct got_error *pack_err =
7377 got_repo_pack_fds_close(pack_fds);
7383 free(gitconfig_path);
7384 free(commit_id_str);
7392 fprintf(stderr, "usage: %s add [-R] [-I] path ...\n",
7397 static const struct got_error *
7398 add_progress(void *arg, unsigned char status, const char *path)
7400 while (path[0] == '/')
7402 printf("%c %s\n", status, path);
7406 static const struct got_error *
7407 cmd_add(int argc, char *argv[])
7409 const struct got_error *error = NULL;
7410 struct got_repository *repo = NULL;
7411 struct got_worktree *worktree = NULL;
7413 struct got_pathlist_head paths;
7414 struct got_pathlist_entry *pe;
7415 int ch, can_recurse = 0, no_ignores = 0;
7416 int *pack_fds = NULL;
7420 while ((ch = getopt(argc, argv, "IR")) != -1) {
7438 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
7445 cwd = getcwd(NULL, 0);
7447 error = got_error_from_errno("getcwd");
7451 error = got_repo_pack_fds_open(&pack_fds);
7455 error = got_worktree_open(&worktree, cwd);
7457 if (error->code == GOT_ERR_NOT_WORKTREE)
7458 error = wrap_not_worktree_error(error, "add", cwd);
7462 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
7467 error = apply_unveil(got_repo_get_path(repo), 1,
7468 got_worktree_get_root_path(worktree));
7472 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
7479 TAILQ_FOREACH(pe, &paths, entry) {
7480 if (asprintf(&ondisk_path, "%s/%s",
7481 got_worktree_get_root_path(worktree),
7483 error = got_error_from_errno("asprintf");
7486 if (lstat(ondisk_path, &sb) == -1) {
7487 if (errno == ENOENT) {
7491 error = got_error_from_errno2("lstat",
7497 if (S_ISDIR(sb.st_mode)) {
7498 error = got_error_msg(GOT_ERR_BAD_PATH,
7499 "adding directories requires -R option");
7505 error = got_worktree_schedule_add(worktree, &paths, add_progress,
7506 NULL, repo, no_ignores);
7509 const struct got_error *close_err = got_repo_close(repo);
7514 got_worktree_close(worktree);
7516 const struct got_error *pack_err =
7517 got_repo_pack_fds_close(pack_fds);
7521 TAILQ_FOREACH(pe, &paths, entry)
7522 free((char *)pe->path);
7523 got_pathlist_free(&paths);
7531 fprintf(stderr, "usage: %s remove [-f] [-k] [-R] [-s status-codes] "
7532 "path ...\n", getprogname());
7536 static const struct got_error *
7537 print_remove_status(void *arg, unsigned char status,
7538 unsigned char staged_status, const char *path)
7540 while (path[0] == '/')
7542 if (status == GOT_STATUS_NONEXISTENT)
7544 if (status == staged_status && (status == GOT_STATUS_DELETE))
7545 status = GOT_STATUS_NO_CHANGE;
7546 printf("%c%c %s\n", status, staged_status, path);
7550 static const struct got_error *
7551 cmd_remove(int argc, char *argv[])
7553 const struct got_error *error = NULL;
7554 struct got_worktree *worktree = NULL;
7555 struct got_repository *repo = NULL;
7556 const char *status_codes = NULL;
7558 struct got_pathlist_head paths;
7559 struct got_pathlist_entry *pe;
7560 int ch, delete_local_mods = 0, can_recurse = 0, keep_on_disk = 0, i;
7561 int ignore_missing_paths = 0;
7562 int *pack_fds = NULL;
7566 while ((ch = getopt(argc, argv, "fkRs:")) != -1) {
7569 delete_local_mods = 1;
7570 ignore_missing_paths = 1;
7579 for (i = 0; i < strlen(optarg); i++) {
7580 switch (optarg[i]) {
7581 case GOT_STATUS_MODIFY:
7582 delete_local_mods = 1;
7584 case GOT_STATUS_MISSING:
7585 ignore_missing_paths = 1;
7588 errx(1, "invalid status code '%c'",
7592 status_codes = optarg;
7604 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
7611 cwd = getcwd(NULL, 0);
7613 error = got_error_from_errno("getcwd");
7617 error = got_repo_pack_fds_open(&pack_fds);
7621 error = got_worktree_open(&worktree, cwd);
7623 if (error->code == GOT_ERR_NOT_WORKTREE)
7624 error = wrap_not_worktree_error(error, "remove", cwd);
7628 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
7633 error = apply_unveil(got_repo_get_path(repo), 1,
7634 got_worktree_get_root_path(worktree));
7638 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
7645 TAILQ_FOREACH(pe, &paths, entry) {
7646 if (asprintf(&ondisk_path, "%s/%s",
7647 got_worktree_get_root_path(worktree),
7649 error = got_error_from_errno("asprintf");
7652 if (lstat(ondisk_path, &sb) == -1) {
7653 if (errno == ENOENT) {
7657 error = got_error_from_errno2("lstat",
7663 if (S_ISDIR(sb.st_mode)) {
7664 error = got_error_msg(GOT_ERR_BAD_PATH,
7665 "removing directories requires -R option");
7671 error = got_worktree_schedule_delete(worktree, &paths,
7672 delete_local_mods, status_codes, print_remove_status, NULL,
7673 repo, keep_on_disk, ignore_missing_paths);
7676 const struct got_error *close_err = got_repo_close(repo);
7681 got_worktree_close(worktree);
7683 const struct got_error *pack_err =
7684 got_repo_pack_fds_close(pack_fds);
7688 TAILQ_FOREACH(pe, &paths, entry)
7689 free((char *)pe->path);
7690 got_pathlist_free(&paths);
7698 fprintf(stderr, "usage: %s patch [-n] [-p strip-count] "
7699 "[-R] [patchfile]\n", getprogname());
7703 static const struct got_error *
7704 patch_from_stdin(int *patchfd)
7706 const struct got_error *err = NULL;
7708 char *path, buf[BUFSIZ];
7709 sig_t sighup, sigint, sigquit;
7711 err = got_opentemp_named_fd(&path, patchfd,
7712 GOT_TMPDIR_STR "/got-patch");
7718 sighup = signal(SIGHUP, SIG_DFL);
7719 sigint = signal(SIGINT, SIG_DFL);
7720 sigquit = signal(SIGQUIT, SIG_DFL);
7723 r = read(0, buf, sizeof(buf));
7725 err = got_error_from_errno("read");
7730 if (write(*patchfd, buf, r) == -1) {
7731 err = got_error_from_errno("write");
7736 signal(SIGHUP, sighup);
7737 signal(SIGINT, sigint);
7738 signal(SIGQUIT, sigquit);
7740 if (err == NULL && lseek(*patchfd, 0, SEEK_SET) == -1)
7741 err = got_error_from_errno("lseek");
7751 static const struct got_error *
7752 patch_progress(void *arg, const char *old, const char *new,
7753 unsigned char status, const struct got_error *error, long old_from,
7754 long old_lines, long new_from, long new_lines, long offset,
7755 const struct got_error *hunk_err)
7757 const char *path = new == NULL ? old : new;
7759 while (*path == '/')
7763 printf("%c %s\n", status, path);
7766 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
7768 if (offset != 0 || hunk_err != NULL) {
7769 printf("@@ -%ld,%ld +%ld,%ld @@ ", old_from,
7770 old_lines, new_from, new_lines);
7771 if (hunk_err != NULL)
7772 printf("%s\n", hunk_err->msg);
7774 printf("applied with offset %ld\n", offset);
7780 static const struct got_error *
7781 cmd_patch(int argc, char *argv[])
7783 const struct got_error *error = NULL, *close_error = NULL;
7784 struct got_worktree *worktree = NULL;
7785 struct got_repository *repo = NULL;
7788 int ch, nop = 0, strip = -1, reverse = 0;
7790 int *pack_fds = NULL;
7792 while ((ch = getopt(argc, argv, "np:R")) != -1) {
7798 strip = strtonum(optarg, 0, INT_MAX, &errstr);
7800 errx(1, "pathname strip count is %s: %s",
7816 error = patch_from_stdin(&patchfd);
7819 } else if (argc == 1) {
7820 patchfd = open(argv[0], O_RDONLY);
7821 if (patchfd == -1) {
7822 error = got_error_from_errno2("open", argv[0]);
7828 if ((cwd = getcwd(NULL, 0)) == NULL) {
7829 error = got_error_from_errno("getcwd");
7833 error = got_repo_pack_fds_open(&pack_fds);
7837 error = got_worktree_open(&worktree, cwd);
7841 const char *repo_path = got_worktree_get_repo_path(worktree);
7842 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
7846 error = apply_unveil(got_repo_get_path(repo), 0,
7847 got_worktree_get_root_path(worktree));
7852 if (pledge("stdio rpath wpath cpath fattr proc exec sendfd flock",
7857 error = got_patch(patchfd, worktree, repo, nop, strip, reverse,
7858 &patch_progress, NULL, check_cancelled, NULL);
7862 close_error = got_repo_close(repo);
7864 error = close_error;
7866 if (worktree != NULL) {
7867 close_error = got_worktree_close(worktree);
7869 error = close_error;
7872 const struct got_error *pack_err =
7873 got_repo_pack_fds_close(pack_fds);
7884 fprintf(stderr, "usage: %s revert [-p] [-F response-script] [-R] "
7885 "path ...\n", getprogname());
7889 static const struct got_error *
7890 revert_progress(void *arg, unsigned char status, const char *path)
7892 if (status == GOT_STATUS_UNVERSIONED)
7895 while (path[0] == '/')
7897 printf("%c %s\n", status, path);
7901 struct choose_patch_arg {
7902 FILE *patch_script_file;
7906 static const struct got_error *
7907 show_change(unsigned char status, const char *path, FILE *patch_file, int n,
7908 int nchanges, const char *action)
7910 const struct got_error *err;
7912 size_t linesize = 0;
7916 case GOT_STATUS_ADD:
7917 printf("A %s\n%s this addition? [y/n] ", path, action);
7919 case GOT_STATUS_DELETE:
7920 printf("D %s\n%s this deletion? [y/n] ", path, action);
7922 case GOT_STATUS_MODIFY:
7923 if (fseek(patch_file, 0L, SEEK_SET) == -1)
7924 return got_error_from_errno("fseek");
7925 printf(GOT_COMMIT_SEP_STR);
7926 while ((linelen = getline(&line, &linesize, patch_file)) != -1)
7928 if (linelen == -1 && ferror(patch_file)) {
7929 err = got_error_from_errno("getline");
7934 printf(GOT_COMMIT_SEP_STR);
7935 printf("M %s (change %d of %d)\n%s this change? [y/n/q] ",
7936 path, n, nchanges, action);
7939 return got_error_path(path, GOT_ERR_FILE_STATUS);
7945 static const struct got_error *
7946 choose_patch(int *choice, void *arg, unsigned char status, const char *path,
7947 FILE *patch_file, int n, int nchanges)
7949 const struct got_error *err = NULL;
7951 size_t linesize = 0;
7954 struct choose_patch_arg *a = arg;
7956 *choice = GOT_PATCH_CHOICE_NONE;
7958 if (a->patch_script_file) {
7960 err = show_change(status, path, patch_file, n, nchanges,
7964 linelen = getline(&line, &linesize, a->patch_script_file);
7965 if (linelen == -1) {
7966 if (ferror(a->patch_script_file))
7967 return got_error_from_errno("getline");
7970 nl = strchr(line, '\n');
7973 if (strcmp(line, "y") == 0) {
7974 *choice = GOT_PATCH_CHOICE_YES;
7976 } else if (strcmp(line, "n") == 0) {
7977 *choice = GOT_PATCH_CHOICE_NO;
7979 } else if (strcmp(line, "q") == 0 &&
7980 status == GOT_STATUS_MODIFY) {
7981 *choice = GOT_PATCH_CHOICE_QUIT;
7984 printf("invalid response '%s'\n", line);
7989 while (resp != 'y' && resp != 'n' && resp != 'q') {
7990 err = show_change(status, path, patch_file, n, nchanges,
7997 if (status == GOT_STATUS_MODIFY) {
7998 if (resp != 'y' && resp != 'n' && resp != 'q') {
7999 printf("invalid response '%c'\n", resp);
8002 } else if (resp != 'y' && resp != 'n') {
8003 printf("invalid response '%c'\n", resp);
8009 *choice = GOT_PATCH_CHOICE_YES;
8010 else if (resp == 'n')
8011 *choice = GOT_PATCH_CHOICE_NO;
8012 else if (resp == 'q' && status == GOT_STATUS_MODIFY)
8013 *choice = GOT_PATCH_CHOICE_QUIT;
8018 static const struct got_error *
8019 cmd_revert(int argc, char *argv[])
8021 const struct got_error *error = NULL;
8022 struct got_worktree *worktree = NULL;
8023 struct got_repository *repo = NULL;
8024 char *cwd = NULL, *path = NULL;
8025 struct got_pathlist_head paths;
8026 struct got_pathlist_entry *pe;
8027 int ch, can_recurse = 0, pflag = 0;
8028 FILE *patch_script_file = NULL;
8029 const char *patch_script_path = NULL;
8030 struct choose_patch_arg cpa;
8031 int *pack_fds = NULL;
8035 while ((ch = getopt(argc, argv, "pF:R")) != -1) {
8041 patch_script_path = optarg;
8056 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
8057 "unveil", NULL) == -1)
8062 if (patch_script_path && !pflag)
8063 errx(1, "-F option can only be used together with -p option");
8065 cwd = getcwd(NULL, 0);
8067 error = got_error_from_errno("getcwd");
8071 error = got_repo_pack_fds_open(&pack_fds);
8075 error = got_worktree_open(&worktree, cwd);
8077 if (error->code == GOT_ERR_NOT_WORKTREE)
8078 error = wrap_not_worktree_error(error, "revert", cwd);
8082 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
8087 if (patch_script_path) {
8088 patch_script_file = fopen(patch_script_path, "re");
8089 if (patch_script_file == NULL) {
8090 error = got_error_from_errno2("fopen",
8095 error = apply_unveil(got_repo_get_path(repo), 1,
8096 got_worktree_get_root_path(worktree));
8100 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
8107 TAILQ_FOREACH(pe, &paths, entry) {
8108 if (asprintf(&ondisk_path, "%s/%s",
8109 got_worktree_get_root_path(worktree),
8111 error = got_error_from_errno("asprintf");
8114 if (lstat(ondisk_path, &sb) == -1) {
8115 if (errno == ENOENT) {
8119 error = got_error_from_errno2("lstat",
8125 if (S_ISDIR(sb.st_mode)) {
8126 error = got_error_msg(GOT_ERR_BAD_PATH,
8127 "reverting directories requires -R option");
8133 cpa.patch_script_file = patch_script_file;
8134 cpa.action = "revert";
8135 error = got_worktree_revert(worktree, &paths, revert_progress, NULL,
8136 pflag ? choose_patch : NULL, &cpa, repo);
8138 if (patch_script_file && fclose(patch_script_file) == EOF &&
8140 error = got_error_from_errno2("fclose", patch_script_path);
8142 const struct got_error *close_err = got_repo_close(repo);
8147 got_worktree_close(worktree);
8149 const struct got_error *pack_err =
8150 got_repo_pack_fds_close(pack_fds);
8162 fprintf(stderr, "usage: %s commit [-F path] [-m msg] [-N] [-S] "
8163 "[path ...]\n", getprogname());
8167 struct collect_commit_logmsg_arg {
8168 const char *cmdline_log;
8169 const char *prepared_log;
8170 int non_interactive;
8172 const char *worktree_path;
8173 const char *branch_name;
8174 const char *repo_path;
8179 static const struct got_error *
8180 read_prepared_logmsg(char **logmsg, const char *path)
8182 const struct got_error *err = NULL;
8188 memset(&sb, 0, sizeof(sb));
8190 f = fopen(path, "re");
8192 return got_error_from_errno2("fopen", path);
8194 if (fstat(fileno(f), &sb) == -1) {
8195 err = got_error_from_errno2("fstat", path);
8198 if (sb.st_size == 0) {
8199 err = got_error(GOT_ERR_COMMIT_MSG_EMPTY);
8203 *logmsg = malloc(sb.st_size + 1);
8204 if (*logmsg == NULL) {
8205 err = got_error_from_errno("malloc");
8209 r = fread(*logmsg, 1, sb.st_size, f);
8210 if (r != sb.st_size) {
8212 err = got_error_from_errno2("fread", path);
8214 err = got_error(GOT_ERR_IO);
8217 (*logmsg)[sb.st_size] = '\0';
8219 if (fclose(f) == EOF && err == NULL)
8220 err = got_error_from_errno2("fclose", path);
8229 static const struct got_error *
8230 collect_commit_logmsg(struct got_pathlist_head *commitable_paths, char **logmsg,
8233 char *initial_content = NULL;
8234 struct got_pathlist_entry *pe;
8235 const struct got_error *err = NULL;
8236 char *template = NULL;
8237 struct collect_commit_logmsg_arg *a = arg;
8238 int initial_content_len;
8242 /* if a message was specified on the command line, just use it */
8243 if (a->cmdline_log != NULL && strlen(a->cmdline_log) != 0) {
8244 len = strlen(a->cmdline_log) + 1;
8245 *logmsg = malloc(len + 1);
8246 if (*logmsg == NULL)
8247 return got_error_from_errno("malloc");
8248 strlcpy(*logmsg, a->cmdline_log, len);
8250 } else if (a->prepared_log != NULL && a->non_interactive)
8251 return read_prepared_logmsg(logmsg, a->prepared_log);
8253 if (asprintf(&template, "%s/logmsg", a->worktree_path) == -1)
8254 return got_error_from_errno("asprintf");
8256 err = got_opentemp_named_fd(&a->logmsg_path, &fd, template);
8260 if (a->prepared_log) {
8262 err = read_prepared_logmsg(&msg, a->prepared_log);
8265 if (write(fd, msg, strlen(msg)) == -1) {
8266 err = got_error_from_errno2("write", a->logmsg_path);
8273 initial_content_len = asprintf(&initial_content,
8274 "\n# changes to be committed on branch %s:\n",
8276 if (initial_content_len == -1) {
8277 err = got_error_from_errno("asprintf");
8281 if (write(fd, initial_content, initial_content_len) == -1) {
8282 err = got_error_from_errno2("write", a->logmsg_path);
8286 TAILQ_FOREACH(pe, commitable_paths, entry) {
8287 struct got_commitable *ct = pe->data;
8288 dprintf(fd, "# %c %s\n",
8289 got_commitable_get_status(ct),
8290 got_commitable_get_path(ct));
8293 err = edit_logmsg(logmsg, a->editor, a->logmsg_path, initial_content,
8294 initial_content_len, a->prepared_log ? 0 : 1);
8296 free(initial_content);
8299 if (fd != -1 && close(fd) == -1 && err == NULL)
8300 err = got_error_from_errno2("close", a->logmsg_path);
8302 /* Editor is done; we can now apply unveil(2) */
8304 err = apply_unveil(a->repo_path, 0, a->worktree_path);
8312 static const struct got_error *
8313 cmd_commit(int argc, char *argv[])
8315 const struct got_error *error = NULL;
8316 struct got_worktree *worktree = NULL;
8317 struct got_repository *repo = NULL;
8318 char *cwd = NULL, *id_str = NULL;
8319 struct got_object_id *id = NULL;
8320 const char *logmsg = NULL;
8321 char *prepared_logmsg = NULL;
8322 struct collect_commit_logmsg_arg cl_arg;
8323 char *gitconfig_path = NULL, *editor = NULL, *author = NULL;
8324 int ch, rebase_in_progress, histedit_in_progress, preserve_logmsg = 0;
8325 int allow_bad_symlinks = 0, non_interactive = 0, merge_in_progress = 0;
8326 struct got_pathlist_head paths;
8327 int *pack_fds = NULL;
8330 cl_arg.logmsg_path = NULL;
8332 while ((ch = getopt(argc, argv, "F:m:NS")) != -1) {
8336 option_conflict('F', 'm');
8337 prepared_logmsg = realpath(optarg, NULL);
8338 if (prepared_logmsg == NULL)
8339 return got_error_from_errno2("realpath",
8343 if (prepared_logmsg)
8344 option_conflict('m', 'F');
8348 non_interactive = 1;
8351 allow_bad_symlinks = 1;
8363 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
8364 "unveil", NULL) == -1)
8367 cwd = getcwd(NULL, 0);
8369 error = got_error_from_errno("getcwd");
8373 error = got_repo_pack_fds_open(&pack_fds);
8377 error = got_worktree_open(&worktree, cwd);
8379 if (error->code == GOT_ERR_NOT_WORKTREE)
8380 error = wrap_not_worktree_error(error, "commit", cwd);
8384 error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
8387 if (rebase_in_progress) {
8388 error = got_error(GOT_ERR_REBASING);
8392 error = got_worktree_histedit_in_progress(&histedit_in_progress,
8397 error = get_gitconfig_path(&gitconfig_path);
8400 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
8401 gitconfig_path, pack_fds);
8405 error = got_worktree_merge_in_progress(&merge_in_progress, worktree, repo);
8408 if (merge_in_progress) {
8409 error = got_error(GOT_ERR_MERGE_BUSY);
8413 error = get_author(&author, repo, worktree);
8418 * unveil(2) traverses exec(2); if an editor is used we have
8419 * to apply unveil after the log message has been written.
8421 if (logmsg == NULL || strlen(logmsg) == 0)
8422 error = get_editor(&editor);
8424 error = apply_unveil(got_repo_get_path(repo), 0,
8425 got_worktree_get_root_path(worktree));
8429 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
8433 cl_arg.editor = editor;
8434 cl_arg.cmdline_log = logmsg;
8435 cl_arg.prepared_log = prepared_logmsg;
8436 cl_arg.non_interactive = non_interactive;
8437 cl_arg.worktree_path = got_worktree_get_root_path(worktree);
8438 cl_arg.branch_name = got_worktree_get_head_ref_name(worktree);
8439 if (!histedit_in_progress) {
8440 if (strncmp(cl_arg.branch_name, "refs/heads/", 11) != 0) {
8441 error = got_error(GOT_ERR_COMMIT_BRANCH);
8444 cl_arg.branch_name += 11;
8446 cl_arg.repo_path = got_repo_get_path(repo);
8447 error = got_worktree_commit(&id, worktree, &paths, author, NULL,
8448 allow_bad_symlinks, collect_commit_logmsg, &cl_arg,
8449 print_status, NULL, repo);
8451 if (error->code != GOT_ERR_COMMIT_MSG_EMPTY &&
8452 cl_arg.logmsg_path != NULL)
8453 preserve_logmsg = 1;
8457 error = got_object_id_str(&id_str, id);
8460 printf("Created commit %s\n", id_str);
8462 if (preserve_logmsg) {
8463 fprintf(stderr, "%s: log message preserved in %s\n",
8464 getprogname(), cl_arg.logmsg_path);
8465 } else if (cl_arg.logmsg_path && unlink(cl_arg.logmsg_path) == -1 &&
8467 error = got_error_from_errno2("unlink", cl_arg.logmsg_path);
8468 free(cl_arg.logmsg_path);
8470 const struct got_error *close_err = got_repo_close(repo);
8475 got_worktree_close(worktree);
8477 const struct got_error *pack_err =
8478 got_repo_pack_fds_close(pack_fds);
8484 free(gitconfig_path);
8487 free(prepared_logmsg);
8494 fprintf(stderr, "usage: %s send [-a] [-b branch] [-d branch] [-f] "
8495 "[-r repository-path] [-t tag] [-T] [-q] [-v] "
8496 "[remote-repository]\n", getprogname());
8501 print_load_info(int print_colored, int print_found, int print_trees,
8502 int ncolored, int nfound, int ntrees)
8504 if (print_colored) {
8505 printf("%d commit%s colored", ncolored,
8506 ncolored == 1 ? "" : "s");
8509 printf("%s%d object%s found",
8510 ncolored > 0 ? "; " : "",
8511 nfound, nfound == 1 ? "" : "s");
8514 printf("; %d tree%s scanned", ntrees,
8515 ntrees == 1 ? "" : "s");
8519 struct got_send_progress_arg {
8520 char last_scaled_packsize[FMT_SCALED_STRSIZE];
8527 int last_nobj_total;
8531 int printed_something;
8533 struct got_pathlist_head *delete_branches;
8536 static const struct got_error *
8537 send_progress(void *arg, int ncolored, int nfound, int ntrees,
8538 off_t packfile_size, int ncommits, int nobj_total, int nobj_deltify,
8539 int nobj_written, off_t bytes_sent, const char *refname, int success)
8541 struct got_send_progress_arg *a = arg;
8542 char scaled_packsize[FMT_SCALED_STRSIZE];
8543 char scaled_sent[FMT_SCALED_STRSIZE];
8544 int p_deltify = 0, p_written = 0, p_sent = 0;
8545 int print_colored = 0, print_found = 0, print_trees = 0;
8546 int print_searching = 0, print_total = 0;
8547 int print_deltify = 0, print_written = 0, print_sent = 0;
8549 if (a->verbosity < 0)
8553 const char *status = success ? "accepted" : "rejected";
8556 struct got_pathlist_entry *pe;
8557 TAILQ_FOREACH(pe, a->delete_branches, entry) {
8558 const char *branchname = pe->path;
8559 if (got_path_cmp(branchname, refname,
8560 strlen(branchname), strlen(refname)) == 0) {
8562 a->sent_something = 1;
8568 if (a->printed_something)
8570 printf("Server has %s %s", status, refname);
8571 a->printed_something = 1;
8575 if (a->last_ncolored != ncolored) {
8577 a->last_ncolored = ncolored;
8580 if (a->last_nfound != nfound) {
8583 a->last_nfound = nfound;
8586 if (a->last_ntrees != ntrees) {
8590 a->last_ntrees = ntrees;
8593 if ((print_colored || print_found || print_trees) &&
8596 print_load_info(print_colored, print_found, print_trees,
8597 ncolored, nfound, ntrees);
8598 a->printed_something = 1;
8601 } else if (!a->loading_done) {
8603 print_load_info(1, 1, 1, ncolored, nfound, ntrees);
8605 a->loading_done = 1;
8608 if (fmt_scaled(packfile_size, scaled_packsize) == -1)
8609 return got_error_from_errno("fmt_scaled");
8610 if (fmt_scaled(bytes_sent, scaled_sent) == -1)
8611 return got_error_from_errno("fmt_scaled");
8613 if (a->last_ncommits != ncommits) {
8614 print_searching = 1;
8615 a->last_ncommits = ncommits;
8618 if (a->last_nobj_total != nobj_total) {
8619 print_searching = 1;
8621 a->last_nobj_total = nobj_total;
8624 if (packfile_size > 0 && (a->last_scaled_packsize[0] == '\0' ||
8625 strcmp(scaled_packsize, a->last_scaled_packsize)) != 0) {
8626 if (strlcpy(a->last_scaled_packsize, scaled_packsize,
8627 FMT_SCALED_STRSIZE) >= FMT_SCALED_STRSIZE)
8628 return got_error(GOT_ERR_NO_SPACE);
8631 if (nobj_deltify > 0 || nobj_written > 0) {
8632 if (nobj_deltify > 0) {
8633 p_deltify = (nobj_deltify * 100) / nobj_total;
8634 if (p_deltify != a->last_p_deltify) {
8635 a->last_p_deltify = p_deltify;
8636 print_searching = 1;
8641 if (nobj_written > 0) {
8642 p_written = (nobj_written * 100) / nobj_total;
8643 if (p_written != a->last_p_written) {
8644 a->last_p_written = p_written;
8645 print_searching = 1;
8653 if (bytes_sent > 0) {
8654 p_sent = (bytes_sent * 100) / packfile_size;
8655 if (p_sent != a->last_p_sent) {
8656 a->last_p_sent = p_sent;
8657 print_searching = 1;
8663 a->sent_something = 1;
8666 if (print_searching || print_total || print_deltify || print_written ||
8669 if (print_searching)
8670 printf("packing %d reference%s", ncommits,
8671 ncommits == 1 ? "" : "s");
8673 printf("; %d object%s", nobj_total,
8674 nobj_total == 1 ? "" : "s");
8676 printf("; deltify: %d%%", p_deltify);
8678 printf("; uploading pack: %*s %d%%", FMT_SCALED_STRSIZE - 2,
8679 scaled_packsize, p_sent);
8680 else if (print_written)
8681 printf("; writing pack: %*s %d%%", FMT_SCALED_STRSIZE - 2,
8682 scaled_packsize, p_written);
8683 if (print_searching || print_total || print_deltify ||
8684 print_written || print_sent) {
8685 a->printed_something = 1;
8691 static const struct got_error *
8692 cmd_send(int argc, char *argv[])
8694 const struct got_error *error = NULL;
8695 char *cwd = NULL, *repo_path = NULL;
8696 const char *remote_name;
8697 char *proto = NULL, *host = NULL, *port = NULL;
8698 char *repo_name = NULL, *server_path = NULL;
8699 const struct got_remote_repo *remotes, *remote = NULL;
8700 int nremotes, nbranches = 0, ntags = 0, ndelete_branches = 0;
8701 struct got_repository *repo = NULL;
8702 struct got_worktree *worktree = NULL;
8703 const struct got_gotconfig *repo_conf = NULL, *worktree_conf = NULL;
8704 struct got_pathlist_head branches;
8705 struct got_pathlist_head tags;
8706 struct got_reflist_head all_branches;
8707 struct got_reflist_head all_tags;
8708 struct got_pathlist_head delete_args;
8709 struct got_pathlist_head delete_branches;
8710 struct got_reflist_entry *re;
8711 struct got_pathlist_entry *pe;
8712 int i, ch, sendfd = -1, sendstatus;
8714 struct got_send_progress_arg spa;
8715 int verbosity = 0, overwrite_refs = 0;
8716 int send_all_branches = 0, send_all_tags = 0;
8717 struct got_reference *ref = NULL;
8718 int *pack_fds = NULL;
8720 TAILQ_INIT(&branches);
8722 TAILQ_INIT(&all_branches);
8723 TAILQ_INIT(&all_tags);
8724 TAILQ_INIT(&delete_args);
8725 TAILQ_INIT(&delete_branches);
8727 while ((ch = getopt(argc, argv, "ab:d:fr:t:Tvq")) != -1) {
8730 send_all_branches = 1;
8733 error = got_pathlist_append(&branches, optarg, NULL);
8739 error = got_pathlist_append(&delete_args, optarg, NULL);
8747 repo_path = realpath(optarg, NULL);
8748 if (repo_path == NULL)
8749 return got_error_from_errno2("realpath",
8751 got_path_strip_trailing_slashes(repo_path);
8754 error = got_pathlist_append(&tags, optarg, NULL);
8765 else if (verbosity < 3)
8779 if (send_all_branches && !TAILQ_EMPTY(&branches))
8780 option_conflict('a', 'b');
8781 if (send_all_tags && !TAILQ_EMPTY(&tags))
8782 option_conflict('T', 't');
8786 remote_name = GOT_SEND_DEFAULT_REMOTE_NAME;
8788 remote_name = argv[0];
8792 cwd = getcwd(NULL, 0);
8794 error = got_error_from_errno("getcwd");
8798 error = got_repo_pack_fds_open(&pack_fds);
8802 if (repo_path == NULL) {
8803 error = got_worktree_open(&worktree, cwd);
8804 if (error && error->code != GOT_ERR_NOT_WORKTREE)
8810 strdup(got_worktree_get_repo_path(worktree));
8811 if (repo_path == NULL)
8812 error = got_error_from_errno("strdup");
8816 repo_path = strdup(cwd);
8817 if (repo_path == NULL) {
8818 error = got_error_from_errno("strdup");
8824 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
8829 worktree_conf = got_worktree_get_gotconfig(worktree);
8830 if (worktree_conf) {
8831 got_gotconfig_get_remotes(&nremotes, &remotes,
8833 for (i = 0; i < nremotes; i++) {
8834 if (strcmp(remotes[i].name, remote_name) == 0) {
8835 remote = &remotes[i];
8841 if (remote == NULL) {
8842 repo_conf = got_repo_get_gotconfig(repo);
8844 got_gotconfig_get_remotes(&nremotes, &remotes,
8846 for (i = 0; i < nremotes; i++) {
8847 if (strcmp(remotes[i].name, remote_name) == 0) {
8848 remote = &remotes[i];
8854 if (remote == NULL) {
8855 got_repo_get_gitconfig_remotes(&nremotes, &remotes, repo);
8856 for (i = 0; i < nremotes; i++) {
8857 if (strcmp(remotes[i].name, remote_name) == 0) {
8858 remote = &remotes[i];
8863 if (remote == NULL) {
8864 error = got_error_path(remote_name, GOT_ERR_NO_REMOTE);
8868 error = got_dial_parse_uri(&proto, &host, &port, &server_path,
8869 &repo_name, remote->send_url);
8873 if (strcmp(proto, "git") == 0) {
8875 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
8876 "sendfd dns inet unveil", NULL) == -1)
8879 } else if (strcmp(proto, "git+ssh") == 0 ||
8880 strcmp(proto, "ssh") == 0) {
8882 if (pledge("stdio rpath wpath cpath fattr flock proc exec "
8883 "sendfd unveil", NULL) == -1)
8886 } else if (strcmp(proto, "http") == 0 ||
8887 strcmp(proto, "git+http") == 0) {
8888 error = got_error_path(proto, GOT_ERR_NOT_IMPL);
8891 error = got_error_path(proto, GOT_ERR_BAD_PROTO);
8895 error = got_dial_apply_unveil(proto);
8899 error = apply_unveil(got_repo_get_path(repo), 0, NULL);
8903 if (send_all_branches) {
8904 error = got_ref_list(&all_branches, repo, "refs/heads",
8905 got_ref_cmp_by_name, NULL);
8908 TAILQ_FOREACH(re, &all_branches, entry) {
8909 const char *branchname = got_ref_get_name(re->ref);
8910 error = got_pathlist_append(&branches,
8916 } else if (nbranches == 0) {
8917 for (i = 0; i < remote->nsend_branches; i++) {
8918 got_pathlist_append(&branches,
8919 remote->send_branches[i], NULL);
8923 if (send_all_tags) {
8924 error = got_ref_list(&all_tags, repo, "refs/tags",
8925 got_ref_cmp_by_name, NULL);
8928 TAILQ_FOREACH(re, &all_tags, entry) {
8929 const char *tagname = got_ref_get_name(re->ref);
8930 error = got_pathlist_append(&tags,
8939 * To prevent accidents only branches in refs/heads/ can be deleted
8940 * with 'got send -d'.
8941 * Deleting anything else requires local repository access or Git.
8943 TAILQ_FOREACH(pe, &delete_args, entry) {
8944 const char *branchname = pe->path;
8946 struct got_pathlist_entry *new;
8947 if (strncmp(branchname, "refs/heads/", 11) == 0) {
8948 s = strdup(branchname);
8950 error = got_error_from_errno("strdup");
8954 if (asprintf(&s, "refs/heads/%s", branchname) == -1) {
8955 error = got_error_from_errno("asprintf");
8959 error = got_pathlist_insert(&new, &delete_branches, s, NULL);
8960 if (error || new == NULL /* duplicate */)
8967 if (nbranches == 0 && ndelete_branches == 0) {
8968 struct got_reference *head_ref;
8970 error = got_ref_open(&head_ref, repo,
8971 got_worktree_get_head_ref_name(worktree), 0);
8973 error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
8976 if (got_ref_is_symbolic(head_ref)) {
8977 error = got_ref_resolve_symbolic(&ref, repo, head_ref);
8978 got_ref_close(head_ref);
8983 error = got_pathlist_append(&branches, got_ref_get_name(ref),
8991 printf("Connecting to \"%s\" %s%s%s\n", remote->name, host,
8992 port ? ":" : "", port ? port : "");
8994 error = got_send_connect(&sendpid, &sendfd, proto, host, port,
8995 server_path, verbosity);
8999 memset(&spa, 0, sizeof(spa));
9000 spa.last_scaled_packsize[0] = '\0';
9001 spa.last_p_deltify = -1;
9002 spa.last_p_written = -1;
9003 spa.verbosity = verbosity;
9004 spa.delete_branches = &delete_branches;
9005 error = got_send_pack(remote_name, &branches, &tags, &delete_branches,
9006 verbosity, overwrite_refs, sendfd, repo, send_progress, &spa,
9007 check_cancelled, NULL);
9008 if (spa.printed_something)
9012 if (!spa.sent_something && verbosity >= 0)
9013 printf("Already up-to-date\n");
9016 if (kill(sendpid, SIGTERM) == -1)
9017 error = got_error_from_errno("kill");
9018 if (waitpid(sendpid, &sendstatus, 0) == -1 && error == NULL)
9019 error = got_error_from_errno("waitpid");
9021 if (sendfd != -1 && close(sendfd) == -1 && error == NULL)
9022 error = got_error_from_errno("close");
9024 const struct got_error *close_err = got_repo_close(repo);
9029 got_worktree_close(worktree);
9031 const struct got_error *pack_err =
9032 got_repo_pack_fds_close(pack_fds);
9038 got_pathlist_free(&branches);
9039 got_pathlist_free(&tags);
9040 got_ref_list_free(&all_branches);
9041 got_ref_list_free(&all_tags);
9042 got_pathlist_free(&delete_args);
9043 TAILQ_FOREACH(pe, &delete_branches, entry)
9044 free((char *)pe->path);
9045 got_pathlist_free(&delete_branches);
9057 usage_cherrypick(void)
9059 fprintf(stderr, "usage: %s cherrypick commit-id\n", getprogname());
9063 static const struct got_error *
9064 cmd_cherrypick(int argc, char *argv[])
9066 const struct got_error *error = NULL;
9067 struct got_worktree *worktree = NULL;
9068 struct got_repository *repo = NULL;
9069 char *cwd = NULL, *commit_id_str = NULL;
9070 struct got_object_id *commit_id = NULL;
9071 struct got_commit_object *commit = NULL;
9072 struct got_object_qid *pid;
9074 struct got_update_progress_arg upa;
9075 int *pack_fds = NULL;
9077 while ((ch = getopt(argc, argv, "")) != -1) {
9089 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
9090 "unveil", NULL) == -1)
9096 cwd = getcwd(NULL, 0);
9098 error = got_error_from_errno("getcwd");
9102 error = got_repo_pack_fds_open(&pack_fds);
9106 error = got_worktree_open(&worktree, cwd);
9108 if (error->code == GOT_ERR_NOT_WORKTREE)
9109 error = wrap_not_worktree_error(error, "cherrypick",
9114 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
9119 error = apply_unveil(got_repo_get_path(repo), 0,
9120 got_worktree_get_root_path(worktree));
9124 error = got_repo_match_object_id(&commit_id, NULL, argv[0],
9125 GOT_OBJ_TYPE_COMMIT, NULL, repo);
9128 error = got_object_id_str(&commit_id_str, commit_id);
9132 error = got_object_open_as_commit(&commit, repo, commit_id);
9135 pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
9136 memset(&upa, 0, sizeof(upa));
9137 error = got_worktree_merge_files(worktree, pid ? &pid->id : NULL,
9138 commit_id, repo, update_progress, &upa, check_cancelled,
9143 if (upa.did_something)
9144 printf("Merged commit %s\n", commit_id_str);
9145 print_merge_progress_stats(&upa);
9148 got_object_commit_close(commit);
9149 free(commit_id_str);
9151 got_worktree_close(worktree);
9153 const struct got_error *close_err = got_repo_close(repo);
9158 const struct got_error *pack_err =
9159 got_repo_pack_fds_close(pack_fds);
9170 fprintf(stderr, "usage: %s backout commit-id\n", getprogname());
9174 static const struct got_error *
9175 cmd_backout(int argc, char *argv[])
9177 const struct got_error *error = NULL;
9178 struct got_worktree *worktree = NULL;
9179 struct got_repository *repo = NULL;
9180 char *cwd = NULL, *commit_id_str = NULL;
9181 struct got_object_id *commit_id = NULL;
9182 struct got_commit_object *commit = NULL;
9183 struct got_object_qid *pid;
9185 struct got_update_progress_arg upa;
9186 int *pack_fds = NULL;
9188 while ((ch = getopt(argc, argv, "")) != -1) {
9200 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
9201 "unveil", NULL) == -1)
9207 cwd = getcwd(NULL, 0);
9209 error = got_error_from_errno("getcwd");
9213 error = got_repo_pack_fds_open(&pack_fds);
9217 error = got_worktree_open(&worktree, cwd);
9219 if (error->code == GOT_ERR_NOT_WORKTREE)
9220 error = wrap_not_worktree_error(error, "backout", cwd);
9224 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
9229 error = apply_unveil(got_repo_get_path(repo), 0,
9230 got_worktree_get_root_path(worktree));
9234 error = got_repo_match_object_id(&commit_id, NULL, argv[0],
9235 GOT_OBJ_TYPE_COMMIT, NULL, repo);
9238 error = got_object_id_str(&commit_id_str, commit_id);
9242 error = got_object_open_as_commit(&commit, repo, commit_id);
9245 pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
9247 error = got_error(GOT_ERR_ROOT_COMMIT);
9251 memset(&upa, 0, sizeof(upa));
9252 error = got_worktree_merge_files(worktree, commit_id, &pid->id,
9253 repo, update_progress, &upa, check_cancelled, NULL);
9257 if (upa.did_something)
9258 printf("Backed out commit %s\n", commit_id_str);
9259 print_merge_progress_stats(&upa);
9262 got_object_commit_close(commit);
9263 free(commit_id_str);
9265 got_worktree_close(worktree);
9267 const struct got_error *close_err = got_repo_close(repo);
9272 const struct got_error *pack_err =
9273 got_repo_pack_fds_close(pack_fds);
9283 fprintf(stderr, "usage: %s rebase [-a] [-c] [-l] [-X] [branch]\n",
9289 trim_logmsg(char *logmsg, int limit)
9294 len = strlen(logmsg);
9298 nl = strchr(logmsg, '\n');
9303 static const struct got_error *
9304 get_short_logmsg(char **logmsg, int limit, struct got_commit_object *commit)
9306 const struct got_error *err;
9307 char *logmsg0 = NULL;
9310 err = got_object_commit_get_logmsg(&logmsg0, commit);
9315 while (isspace((unsigned char)s[0]))
9318 *logmsg = strdup(s);
9319 if (*logmsg == NULL) {
9320 err = got_error_from_errno("strdup");
9324 trim_logmsg(*logmsg, limit);
9330 static const struct got_error *
9331 show_rebase_merge_conflict(struct got_object_id *id,
9332 struct got_repository *repo)
9334 const struct got_error *err;
9335 struct got_commit_object *commit = NULL;
9336 char *id_str = NULL, *logmsg = NULL;
9338 err = got_object_open_as_commit(&commit, repo, id);
9342 err = got_object_id_str(&id_str, id);
9348 err = get_short_logmsg(&logmsg, 42, commit);
9352 printf("%s -> merge conflict: %s\n", id_str, logmsg);
9355 got_object_commit_close(commit);
9360 static const struct got_error *
9361 show_rebase_progress(struct got_commit_object *commit,
9362 struct got_object_id *old_id, struct got_object_id *new_id)
9364 const struct got_error *err;
9365 char *old_id_str = NULL, *new_id_str = NULL, *logmsg = NULL;
9367 err = got_object_id_str(&old_id_str, old_id);
9372 err = got_object_id_str(&new_id_str, new_id);
9377 old_id_str[12] = '\0';
9379 new_id_str[12] = '\0';
9381 err = get_short_logmsg(&logmsg, 42, commit);
9385 printf("%s -> %s: %s\n", old_id_str,
9386 new_id_str ? new_id_str : "no-op change", logmsg);
9394 static const struct got_error *
9395 rebase_complete(struct got_worktree *worktree, struct got_fileindex *fileindex,
9396 struct got_reference *branch, struct got_reference *new_base_branch,
9397 struct got_reference *tmp_branch, struct got_repository *repo,
9400 printf("Switching work tree to %s\n", got_ref_get_name(branch));
9401 return got_worktree_rebase_complete(worktree, fileindex,
9402 new_base_branch, tmp_branch, branch, repo, create_backup);
9405 static const struct got_error *
9406 rebase_commit(struct got_pathlist_head *merged_paths,
9407 struct got_worktree *worktree, struct got_fileindex *fileindex,
9408 struct got_reference *tmp_branch,
9409 struct got_object_id *commit_id, struct got_repository *repo)
9411 const struct got_error *error;
9412 struct got_commit_object *commit;
9413 struct got_object_id *new_commit_id;
9415 error = got_object_open_as_commit(&commit, repo, commit_id);
9419 error = got_worktree_rebase_commit(&new_commit_id, merged_paths,
9420 worktree, fileindex, tmp_branch, commit, commit_id, repo);
9422 if (error->code != GOT_ERR_COMMIT_NO_CHANGES)
9424 error = show_rebase_progress(commit, commit_id, NULL);
9426 error = show_rebase_progress(commit, commit_id, new_commit_id);
9427 free(new_commit_id);
9430 got_object_commit_close(commit);
9434 struct check_path_prefix_arg {
9435 const char *path_prefix;
9440 static const struct got_error *
9441 check_path_prefix_in_diff(void *arg, struct got_blob_object *blob1,
9442 struct got_blob_object *blob2, FILE *f1, FILE *f2,
9443 struct got_object_id *id1, struct got_object_id *id2,
9444 const char *path1, const char *path2,
9445 mode_t mode1, mode_t mode2, struct got_repository *repo)
9447 struct check_path_prefix_arg *a = arg;
9449 if ((path1 && !got_path_is_child(path1, a->path_prefix, a->len)) ||
9450 (path2 && !got_path_is_child(path2, a->path_prefix, a->len)))
9451 return got_error(a->errcode);
9456 static const struct got_error *
9457 check_path_prefix(struct got_object_id *parent_id,
9458 struct got_object_id *commit_id, const char *path_prefix,
9459 int errcode, struct got_repository *repo)
9461 const struct got_error *err;
9462 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
9463 struct got_commit_object *commit = NULL, *parent_commit = NULL;
9464 struct check_path_prefix_arg cpp_arg;
9466 if (got_path_is_root_dir(path_prefix))
9469 err = got_object_open_as_commit(&commit, repo, commit_id);
9473 err = got_object_open_as_commit(&parent_commit, repo, parent_id);
9477 err = got_object_open_as_tree(&tree1, repo,
9478 got_object_commit_get_tree_id(parent_commit));
9482 err = got_object_open_as_tree(&tree2, repo,
9483 got_object_commit_get_tree_id(commit));
9487 cpp_arg.path_prefix = path_prefix;
9488 while (cpp_arg.path_prefix[0] == '/')
9489 cpp_arg.path_prefix++;
9490 cpp_arg.len = strlen(cpp_arg.path_prefix);
9491 cpp_arg.errcode = errcode;
9492 err = got_diff_tree(tree1, tree2, NULL, NULL, -1, -1, "", "", repo,
9493 check_path_prefix_in_diff, &cpp_arg, 0);
9496 got_object_tree_close(tree1);
9498 got_object_tree_close(tree2);
9500 got_object_commit_close(commit);
9502 got_object_commit_close(parent_commit);
9506 static const struct got_error *
9507 collect_commits(struct got_object_id_queue *commits,
9508 struct got_object_id *initial_commit_id,
9509 struct got_object_id *iter_start_id, struct got_object_id *iter_stop_id,
9510 const char *path_prefix, int path_prefix_errcode,
9511 struct got_repository *repo)
9513 const struct got_error *err = NULL;
9514 struct got_commit_graph *graph = NULL;
9515 struct got_object_id *parent_id = NULL;
9516 struct got_object_qid *qid;
9517 struct got_object_id *commit_id = initial_commit_id;
9519 err = got_commit_graph_open(&graph, "/", 1);
9523 err = got_commit_graph_iter_start(graph, iter_start_id, repo,
9524 check_cancelled, NULL);
9527 while (got_object_id_cmp(commit_id, iter_stop_id) != 0) {
9528 err = got_commit_graph_iter_next(&parent_id, graph, repo,
9529 check_cancelled, NULL);
9531 if (err->code == GOT_ERR_ITER_COMPLETED) {
9532 err = got_error_msg(GOT_ERR_ANCESTRY,
9533 "ran out of commits to rebase before "
9534 "youngest common ancestor commit has "
9539 err = check_path_prefix(parent_id, commit_id,
9540 path_prefix, path_prefix_errcode, repo);
9544 err = got_object_qid_alloc(&qid, commit_id);
9547 STAILQ_INSERT_HEAD(commits, qid, entry);
9548 commit_id = parent_id;
9552 got_commit_graph_close(graph);
9556 static const struct got_error *
9557 get_commit_brief_str(char **brief_str, struct got_commit_object *commit)
9559 const struct got_error *err = NULL;
9560 time_t committer_time;
9562 char datebuf[11]; /* YYYY-MM-DD + NUL */
9563 char *author0 = NULL, *author, *smallerthan;
9564 char *logmsg0 = NULL, *logmsg, *newline;
9566 committer_time = got_object_commit_get_committer_time(commit);
9567 if (gmtime_r(&committer_time, &tm) == NULL)
9568 return got_error_from_errno("gmtime_r");
9569 if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d", &tm) == 0)
9570 return got_error(GOT_ERR_NO_SPACE);
9572 author0 = strdup(got_object_commit_get_author(commit));
9573 if (author0 == NULL)
9574 return got_error_from_errno("strdup");
9576 smallerthan = strchr(author, '<');
9577 if (smallerthan && smallerthan[1] != '\0')
9578 author = smallerthan + 1;
9579 author[strcspn(author, "@>")] = '\0';
9581 err = got_object_commit_get_logmsg(&logmsg0, commit);
9585 while (*logmsg == '\n')
9587 newline = strchr(logmsg, '\n');
9591 if (asprintf(brief_str, "%s %s %s",
9592 datebuf, author, logmsg) == -1)
9593 err = got_error_from_errno("asprintf");
9600 static const struct got_error *
9601 delete_backup_ref(struct got_reference *ref, struct got_object_id *id,
9602 struct got_repository *repo)
9604 const struct got_error *err;
9607 err = got_object_id_str(&id_str, id);
9611 err = got_ref_delete(ref, repo);
9615 printf("Deleted %s: %s\n", got_ref_get_name(ref), id_str);
9621 static const struct got_error *
9622 print_backup_ref(const char *branch_name, const char *new_id_str,
9623 struct got_object_id *old_commit_id, struct got_commit_object *old_commit,
9624 struct got_reflist_object_id_map *refs_idmap,
9625 struct got_repository *repo)
9627 const struct got_error *err = NULL;
9628 struct got_reflist_head *refs;
9629 char *refs_str = NULL;
9630 struct got_object_id *new_commit_id = NULL;
9631 struct got_commit_object *new_commit = NULL;
9632 char *new_commit_brief_str = NULL;
9633 struct got_object_id *yca_id = NULL;
9634 struct got_commit_object *yca_commit = NULL;
9635 char *yca_id_str = NULL, *yca_brief_str = NULL;
9636 char *custom_refs_str;
9638 if (asprintf(&custom_refs_str, "formerly %s", branch_name) == -1)
9639 return got_error_from_errno("asprintf");
9641 err = print_commit(old_commit, old_commit_id, repo, NULL, NULL,
9642 0, 0, refs_idmap, custom_refs_str);
9646 err = got_object_resolve_id_str(&new_commit_id, repo, new_id_str);
9650 refs = got_reflist_object_id_map_lookup(refs_idmap, new_commit_id);
9652 err = build_refs_str(&refs_str, refs, new_commit_id, repo, 0);
9657 err = got_object_open_as_commit(&new_commit, repo, new_commit_id);
9661 err = get_commit_brief_str(&new_commit_brief_str, new_commit);
9665 err = got_commit_graph_find_youngest_common_ancestor(&yca_id,
9666 old_commit_id, new_commit_id, 1, repo, check_cancelled, NULL);
9670 printf("has become commit %s%s%s%s\n %s\n", new_id_str,
9671 refs_str ? " (" : "", refs_str ? refs_str : "",
9672 refs_str ? ")" : "", new_commit_brief_str);
9673 if (yca_id && got_object_id_cmp(yca_id, new_commit_id) != 0 &&
9674 got_object_id_cmp(yca_id, old_commit_id) != 0) {
9678 err = got_object_open_as_commit(&yca_commit, repo, yca_id);
9682 err = get_commit_brief_str(&yca_brief_str, yca_commit);
9686 err = got_object_id_str(&yca_id_str, yca_id);
9690 refs = got_reflist_object_id_map_lookup(refs_idmap, yca_id);
9692 err = build_refs_str(&refs_str, refs, yca_id, repo, 0);
9696 printf("history forked at %s%s%s%s\n %s\n",
9698 refs_str ? " (" : "", refs_str ? refs_str : "",
9699 refs_str ? ")" : "", yca_brief_str);
9702 free(custom_refs_str);
9703 free(new_commit_id);
9707 free(yca_brief_str);
9709 got_object_commit_close(new_commit);
9711 got_object_commit_close(yca_commit);
9716 static const struct got_error *
9717 process_backup_refs(const char *backup_ref_prefix,
9718 const char *wanted_branch_name,
9719 int delete, struct got_repository *repo)
9721 const struct got_error *err;
9722 struct got_reflist_head refs, backup_refs;
9723 struct got_reflist_entry *re;
9724 const size_t backup_ref_prefix_len = strlen(backup_ref_prefix);
9725 struct got_object_id *old_commit_id = NULL;
9726 char *branch_name = NULL;
9727 struct got_commit_object *old_commit = NULL;
9728 struct got_reflist_object_id_map *refs_idmap = NULL;
9729 int wanted_branch_found = 0;
9732 TAILQ_INIT(&backup_refs);
9734 err = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
9738 err = got_reflist_object_id_map_create(&refs_idmap, &refs, repo);
9742 if (wanted_branch_name) {
9743 if (strncmp(wanted_branch_name, "refs/heads/", 11) == 0)
9744 wanted_branch_name += 11;
9747 err = got_ref_list(&backup_refs, repo, backup_ref_prefix,
9748 got_ref_cmp_by_commit_timestamp_descending, repo);
9752 TAILQ_FOREACH(re, &backup_refs, entry) {
9753 const char *refname = got_ref_get_name(re->ref);
9756 err = check_cancelled(NULL);
9760 err = got_ref_resolve(&old_commit_id, repo, re->ref);
9764 err = got_object_open_as_commit(&old_commit, repo,
9769 if (strncmp(backup_ref_prefix, refname,
9770 backup_ref_prefix_len) == 0)
9771 refname += backup_ref_prefix_len;
9773 while (refname[0] == '/')
9776 branch_name = strdup(refname);
9777 if (branch_name == NULL) {
9778 err = got_error_from_errno("strdup");
9781 slash = strrchr(branch_name, '/');
9784 refname += strlen(branch_name) + 1;
9787 if (wanted_branch_name == NULL ||
9788 strcmp(wanted_branch_name, branch_name) == 0) {
9789 wanted_branch_found = 1;
9791 err = delete_backup_ref(re->ref,
9792 old_commit_id, repo);
9794 err = print_backup_ref(branch_name, refname,
9795 old_commit_id, old_commit, refs_idmap,
9802 free(old_commit_id);
9803 old_commit_id = NULL;
9806 got_object_commit_close(old_commit);
9810 if (wanted_branch_name && !wanted_branch_found) {
9811 err = got_error_fmt(GOT_ERR_NOT_REF,
9812 "%s/%s/", backup_ref_prefix, wanted_branch_name);
9816 got_reflist_object_id_map_free(refs_idmap);
9817 got_ref_list_free(&refs);
9818 got_ref_list_free(&backup_refs);
9819 free(old_commit_id);
9822 got_object_commit_close(old_commit);
9826 static const struct got_error *
9827 abort_progress(void *arg, unsigned char status, const char *path)
9830 * Unversioned files should not clutter progress output when
9831 * an operation is aborted.
9833 if (status == GOT_STATUS_UNVERSIONED)
9836 return update_progress(arg, status, path);
9839 static const struct got_error *
9840 cmd_rebase(int argc, char *argv[])
9842 const struct got_error *error = NULL;
9843 struct got_worktree *worktree = NULL;
9844 struct got_repository *repo = NULL;
9845 struct got_fileindex *fileindex = NULL;
9847 struct got_reference *branch = NULL;
9848 struct got_reference *new_base_branch = NULL, *tmp_branch = NULL;
9849 struct got_object_id *commit_id = NULL, *parent_id = NULL;
9850 struct got_object_id *resume_commit_id = NULL;
9851 struct got_object_id *branch_head_commit_id = NULL, *yca_id = NULL;
9852 struct got_commit_object *commit = NULL;
9853 int ch, rebase_in_progress = 0, abort_rebase = 0, continue_rebase = 0;
9854 int histedit_in_progress = 0, merge_in_progress = 0;
9855 int create_backup = 1, list_backups = 0, delete_backups = 0;
9856 struct got_object_id_queue commits;
9857 struct got_pathlist_head merged_paths;
9858 const struct got_object_id_queue *parent_ids;
9859 struct got_object_qid *qid, *pid;
9860 struct got_update_progress_arg upa;
9861 int *pack_fds = NULL;
9863 STAILQ_INIT(&commits);
9864 TAILQ_INIT(&merged_paths);
9865 memset(&upa, 0, sizeof(upa));
9867 while ((ch = getopt(argc, argv, "aclX")) != -1) {
9873 continue_rebase = 1;
9891 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
9892 "unveil", NULL) == -1)
9897 option_conflict('l', 'a');
9898 if (continue_rebase)
9899 option_conflict('l', 'c');
9901 option_conflict('l', 'X');
9902 if (argc != 0 && argc != 1)
9904 } else if (delete_backups) {
9906 option_conflict('X', 'a');
9907 if (continue_rebase)
9908 option_conflict('X', 'c');
9910 option_conflict('l', 'X');
9911 if (argc != 0 && argc != 1)
9914 if (abort_rebase && continue_rebase)
9916 else if (abort_rebase || continue_rebase) {
9919 } else if (argc != 1)
9923 cwd = getcwd(NULL, 0);
9925 error = got_error_from_errno("getcwd");
9929 error = got_repo_pack_fds_open(&pack_fds);
9933 error = got_worktree_open(&worktree, cwd);
9935 if (list_backups || delete_backups) {
9936 if (error->code != GOT_ERR_NOT_WORKTREE)
9939 if (error->code == GOT_ERR_NOT_WORKTREE)
9940 error = wrap_not_worktree_error(error,
9946 error = got_repo_open(&repo,
9947 worktree ? got_worktree_get_repo_path(worktree) : cwd, NULL,
9952 error = apply_unveil(got_repo_get_path(repo), 0,
9953 worktree ? got_worktree_get_root_path(worktree) : NULL);
9957 if (list_backups || delete_backups) {
9958 error = process_backup_refs(
9959 GOT_WORKTREE_REBASE_BACKUP_REF_PREFIX,
9960 argc == 1 ? argv[0] : NULL, delete_backups, repo);
9961 goto done; /* nothing else to do */
9964 error = got_worktree_histedit_in_progress(&histedit_in_progress,
9968 if (histedit_in_progress) {
9969 error = got_error(GOT_ERR_HISTEDIT_BUSY);
9973 error = got_worktree_merge_in_progress(&merge_in_progress,
9977 if (merge_in_progress) {
9978 error = got_error(GOT_ERR_MERGE_BUSY);
9982 error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
9987 if (!rebase_in_progress) {
9988 error = got_error(GOT_ERR_NOT_REBASING);
9991 error = got_worktree_rebase_continue(&resume_commit_id,
9992 &new_base_branch, &tmp_branch, &branch, &fileindex,
9996 printf("Switching work tree to %s\n",
9997 got_ref_get_symref_target(new_base_branch));
9998 error = got_worktree_rebase_abort(worktree, fileindex, repo,
9999 new_base_branch, abort_progress, &upa);
10002 printf("Rebase of %s aborted\n", got_ref_get_name(branch));
10003 print_merge_progress_stats(&upa);
10004 goto done; /* nothing else to do */
10007 if (continue_rebase) {
10008 if (!rebase_in_progress) {
10009 error = got_error(GOT_ERR_NOT_REBASING);
10012 error = got_worktree_rebase_continue(&resume_commit_id,
10013 &new_base_branch, &tmp_branch, &branch, &fileindex,
10018 error = rebase_commit(NULL, worktree, fileindex, tmp_branch,
10019 resume_commit_id, repo);
10023 yca_id = got_object_id_dup(resume_commit_id);
10024 if (yca_id == NULL) {
10025 error = got_error_from_errno("got_object_id_dup");
10029 error = got_ref_open(&branch, repo, argv[0], 0);
10034 error = got_ref_resolve(&branch_head_commit_id, repo, branch);
10038 if (!continue_rebase) {
10039 struct got_object_id *base_commit_id;
10041 base_commit_id = got_worktree_get_base_commit_id(worktree);
10042 error = got_commit_graph_find_youngest_common_ancestor(&yca_id,
10043 base_commit_id, branch_head_commit_id, 1, repo,
10044 check_cancelled, NULL);
10047 if (yca_id == NULL) {
10048 error = got_error_msg(GOT_ERR_ANCESTRY,
10049 "specified branch shares no common ancestry "
10050 "with work tree's branch");
10054 error = check_same_branch(base_commit_id, branch, yca_id, repo);
10056 if (error->code != GOT_ERR_ANCESTRY)
10060 struct got_pathlist_head paths;
10061 printf("%s is already based on %s\n",
10062 got_ref_get_name(branch),
10063 got_worktree_get_head_ref_name(worktree));
10064 error = switch_head_ref(branch, branch_head_commit_id,
10068 error = got_worktree_set_base_commit_id(worktree, repo,
10069 branch_head_commit_id);
10072 TAILQ_INIT(&paths);
10073 error = got_pathlist_append(&paths, "", NULL);
10076 error = got_worktree_checkout_files(worktree,
10077 &paths, repo, update_progress, &upa,
10078 check_cancelled, NULL);
10079 got_pathlist_free(&paths);
10082 if (upa.did_something) {
10084 error = got_object_id_str(&id_str,
10085 branch_head_commit_id);
10088 printf("Updated to %s: %s\n",
10089 got_worktree_get_head_ref_name(worktree),
10093 printf("Already up-to-date\n");
10094 print_update_progress_stats(&upa);
10099 commit_id = branch_head_commit_id;
10100 error = got_object_open_as_commit(&commit, repo, commit_id);
10104 parent_ids = got_object_commit_get_parent_ids(commit);
10105 pid = STAILQ_FIRST(parent_ids);
10107 error = got_error(GOT_ERR_EMPTY_REBASE);
10110 error = collect_commits(&commits, commit_id, &pid->id,
10111 yca_id, got_worktree_get_path_prefix(worktree),
10112 GOT_ERR_REBASE_PATH, repo);
10113 got_object_commit_close(commit);
10118 if (!continue_rebase) {
10119 error = got_worktree_rebase_prepare(&new_base_branch,
10120 &tmp_branch, &fileindex, worktree, branch, repo);
10125 if (STAILQ_EMPTY(&commits)) {
10126 if (continue_rebase) {
10127 error = rebase_complete(worktree, fileindex,
10128 branch, new_base_branch, tmp_branch, repo,
10132 /* Fast-forward the reference of the branch. */
10133 struct got_object_id *new_head_commit_id;
10135 error = got_ref_resolve(&new_head_commit_id, repo,
10139 error = got_object_id_str(&id_str, new_head_commit_id);
10140 printf("Forwarding %s to commit %s\n",
10141 got_ref_get_name(branch), id_str);
10143 error = got_ref_change_ref(branch,
10144 new_head_commit_id);
10147 /* No backup needed since objects did not change. */
10153 STAILQ_FOREACH(qid, &commits, entry) {
10155 commit_id = &qid->id;
10156 parent_id = pid ? &pid->id : yca_id;
10159 memset(&upa, 0, sizeof(upa));
10160 error = got_worktree_rebase_merge_files(&merged_paths,
10161 worktree, fileindex, parent_id, commit_id, repo,
10162 update_progress, &upa, check_cancelled, NULL);
10166 print_merge_progress_stats(&upa);
10167 if (upa.conflicts > 0 || upa.missing > 0 ||
10168 upa.not_deleted > 0 || upa.unversioned > 0) {
10169 if (upa.conflicts > 0) {
10170 error = show_rebase_merge_conflict(&qid->id,
10175 got_worktree_rebase_pathlist_free(&merged_paths);
10179 error = rebase_commit(&merged_paths, worktree, fileindex,
10180 tmp_branch, commit_id, repo);
10181 got_worktree_rebase_pathlist_free(&merged_paths);
10186 if (upa.conflicts > 0 || upa.missing > 0 ||
10187 upa.not_deleted > 0 || upa.unversioned > 0) {
10188 error = got_worktree_rebase_postpone(worktree, fileindex);
10191 if (upa.conflicts > 0 && upa.missing == 0 &&
10192 upa.not_deleted == 0 && upa.unversioned == 0) {
10193 error = got_error_msg(GOT_ERR_CONFLICTS,
10194 "conflicts must be resolved before rebasing "
10196 } else if (upa.conflicts > 0) {
10197 error = got_error_msg(GOT_ERR_CONFLICTS,
10198 "conflicts must be resolved before rebasing "
10199 "can continue; changes destined for some "
10200 "files were not yet merged and should be "
10201 "merged manually if required before the "
10202 "rebase operation is continued");
10204 error = got_error_msg(GOT_ERR_CONFLICTS,
10205 "changes destined for some files were not "
10206 "yet merged and should be merged manually "
10207 "if required before the rebase operation "
10211 error = rebase_complete(worktree, fileindex, branch,
10212 new_base_branch, tmp_branch, repo, create_backup);
10214 got_object_id_queue_free(&commits);
10215 free(branch_head_commit_id);
10216 free(resume_commit_id);
10219 got_object_commit_close(commit);
10221 got_ref_close(branch);
10222 if (new_base_branch)
10223 got_ref_close(new_base_branch);
10225 got_ref_close(tmp_branch);
10227 got_worktree_close(worktree);
10229 const struct got_error *close_err = got_repo_close(repo);
10234 const struct got_error *pack_err =
10235 got_repo_pack_fds_close(pack_fds);
10243 usage_histedit(void)
10245 fprintf(stderr, "usage: %s histedit [-a] [-c] [-e] [-f] "
10246 "[-F histedit-script] [-m] [-l] [-X] [branch]\n",
10251 #define GOT_HISTEDIT_PICK 'p'
10252 #define GOT_HISTEDIT_EDIT 'e'
10253 #define GOT_HISTEDIT_FOLD 'f'
10254 #define GOT_HISTEDIT_DROP 'd'
10255 #define GOT_HISTEDIT_MESG 'm'
10257 static const struct got_histedit_cmd {
10258 unsigned char code;
10261 } got_histedit_cmds[] = {
10262 { GOT_HISTEDIT_PICK, "pick", "use commit" },
10263 { GOT_HISTEDIT_EDIT, "edit", "use commit but stop for amending" },
10264 { GOT_HISTEDIT_FOLD, "fold", "combine with next commit that will "
10266 { GOT_HISTEDIT_DROP, "drop", "remove commit from history" },
10267 { GOT_HISTEDIT_MESG, "mesg",
10268 "single-line log message for commit above (open editor if empty)" },
10271 struct got_histedit_list_entry {
10272 TAILQ_ENTRY(got_histedit_list_entry) entry;
10273 struct got_object_id *commit_id;
10274 const struct got_histedit_cmd *cmd;
10277 TAILQ_HEAD(got_histedit_list, got_histedit_list_entry);
10279 static const struct got_error *
10280 histedit_write_commit(struct got_object_id *commit_id, const char *cmdname,
10281 FILE *f, struct got_repository *repo)
10283 const struct got_error *err = NULL;
10284 char *logmsg = NULL, *id_str = NULL;
10285 struct got_commit_object *commit = NULL;
10288 err = got_object_open_as_commit(&commit, repo, commit_id);
10292 err = get_short_logmsg(&logmsg, 34, commit);
10296 err = got_object_id_str(&id_str, commit_id);
10300 n = fprintf(f, "%s %s %s\n", cmdname, id_str, logmsg);
10302 err = got_ferror(f, GOT_ERR_IO);
10305 got_object_commit_close(commit);
10311 static const struct got_error *
10312 histedit_write_commit_list(struct got_object_id_queue *commits,
10313 FILE *f, int edit_logmsg_only, int fold_only, int edit_only,
10314 struct got_repository *repo)
10316 const struct got_error *err = NULL;
10317 struct got_object_qid *qid;
10318 const char *histedit_cmd = NULL;
10320 if (STAILQ_EMPTY(commits))
10321 return got_error(GOT_ERR_EMPTY_HISTEDIT);
10323 STAILQ_FOREACH(qid, commits, entry) {
10324 histedit_cmd = got_histedit_cmds[0].name;
10326 histedit_cmd = "edit";
10327 else if (fold_only && STAILQ_NEXT(qid, entry) != NULL)
10328 histedit_cmd = "fold";
10329 err = histedit_write_commit(&qid->id, histedit_cmd, f, repo);
10332 if (edit_logmsg_only) {
10333 int n = fprintf(f, "%c\n", GOT_HISTEDIT_MESG);
10335 err = got_ferror(f, GOT_ERR_IO);
10344 static const struct got_error *
10345 write_cmd_list(FILE *f, const char *branch_name,
10346 struct got_object_id_queue *commits)
10348 const struct got_error *err = NULL;
10352 struct got_object_qid *qid;
10354 qid = STAILQ_FIRST(commits);
10355 err = got_object_id_str(&id_str, &qid->id);
10360 "# Editing the history of branch '%s' starting at\n"
10362 "# Commits will be processed in order from top to "
10363 "bottom of this file.\n", branch_name, id_str);
10365 err = got_ferror(f, GOT_ERR_IO);
10369 n = fprintf(f, "# Available histedit commands:\n");
10371 err = got_ferror(f, GOT_ERR_IO);
10375 for (i = 0; i < nitems(got_histedit_cmds); i++) {
10376 const struct got_histedit_cmd *cmd = &got_histedit_cmds[i];
10377 n = fprintf(f, "# %s (%c): %s\n", cmd->name, cmd->code,
10380 err = got_ferror(f, GOT_ERR_IO);
10389 static const struct got_error *
10390 histedit_syntax_error(int lineno)
10392 static char msg[42];
10395 ret = snprintf(msg, sizeof(msg), "histedit syntax error on line %d",
10397 if (ret == -1 || ret >= sizeof(msg))
10398 return got_error(GOT_ERR_HISTEDIT_SYNTAX);
10400 return got_error_msg(GOT_ERR_HISTEDIT_SYNTAX, msg);
10403 static const struct got_error *
10404 append_folded_commit_msg(char **new_msg, struct got_histedit_list_entry *hle,
10405 char *logmsg, struct got_repository *repo)
10407 const struct got_error *err;
10408 struct got_commit_object *folded_commit = NULL;
10409 char *id_str, *folded_logmsg = NULL;
10411 err = got_object_id_str(&id_str, hle->commit_id);
10415 err = got_object_open_as_commit(&folded_commit, repo, hle->commit_id);
10419 err = got_object_commit_get_logmsg(&folded_logmsg, folded_commit);
10422 if (asprintf(new_msg, "%s%s# log message of folded commit %s: %s",
10423 logmsg ? logmsg : "", logmsg ? "\n" : "", id_str,
10424 folded_logmsg) == -1) {
10425 err = got_error_from_errno("asprintf");
10429 got_object_commit_close(folded_commit);
10431 free(folded_logmsg);
10435 static struct got_histedit_list_entry *
10436 get_folded_commits(struct got_histedit_list_entry *hle)
10438 struct got_histedit_list_entry *prev, *folded = NULL;
10440 prev = TAILQ_PREV(hle, got_histedit_list, entry);
10441 while (prev && (prev->cmd->code == GOT_HISTEDIT_FOLD ||
10442 prev->cmd->code == GOT_HISTEDIT_DROP)) {
10443 if (prev->cmd->code == GOT_HISTEDIT_FOLD)
10445 prev = TAILQ_PREV(prev, got_histedit_list, entry);
10451 static const struct got_error *
10452 histedit_edit_logmsg(struct got_histedit_list_entry *hle,
10453 struct got_repository *repo)
10455 char *logmsg_path = NULL, *id_str = NULL, *orig_logmsg = NULL;
10456 char *logmsg = NULL, *new_msg = NULL, *editor = NULL;
10457 const struct got_error *err = NULL;
10458 struct got_commit_object *commit = NULL;
10461 struct got_histedit_list_entry *folded = NULL;
10463 err = got_object_open_as_commit(&commit, repo, hle->commit_id);
10467 folded = get_folded_commits(hle);
10469 while (folded != hle) {
10470 if (folded->cmd->code == GOT_HISTEDIT_DROP) {
10471 folded = TAILQ_NEXT(folded, entry);
10474 err = append_folded_commit_msg(&new_msg, folded,
10480 folded = TAILQ_NEXT(folded, entry);
10484 err = got_object_id_str(&id_str, hle->commit_id);
10487 err = got_object_commit_get_logmsg(&orig_logmsg, commit);
10490 logmsg_len = asprintf(&new_msg,
10491 "%s\n# original log message of commit %s: %s",
10492 logmsg ? logmsg : "", id_str, orig_logmsg);
10493 if (logmsg_len == -1) {
10494 err = got_error_from_errno("asprintf");
10500 err = got_object_id_str(&id_str, hle->commit_id);
10504 err = got_opentemp_named_fd(&logmsg_path, &fd,
10505 GOT_TMPDIR_STR "/got-logmsg");
10509 write(fd, logmsg, logmsg_len);
10512 err = get_editor(&editor);
10516 err = edit_logmsg(&hle->logmsg, editor, logmsg_path, logmsg,
10519 if (err->code != GOT_ERR_COMMIT_MSG_EMPTY)
10522 hle->logmsg = strdup(new_msg);
10523 if (hle->logmsg == NULL)
10524 err = got_error_from_errno("strdup");
10527 if (logmsg_path && unlink(logmsg_path) != 0 && err == NULL)
10528 err = got_error_from_errno2("unlink", logmsg_path);
10534 got_object_commit_close(commit);
10538 static const struct got_error *
10539 histedit_parse_list(struct got_histedit_list *histedit_cmds,
10540 FILE *f, struct got_repository *repo)
10542 const struct got_error *err = NULL;
10543 char *line = NULL, *p, *end;
10547 const struct got_histedit_cmd *cmd;
10548 struct got_object_id *commit_id = NULL;
10549 struct got_histedit_list_entry *hle = NULL;
10552 len = getline(&line, &size, f);
10554 const struct got_error *getline_err;
10557 getline_err = got_error_from_errno("getline");
10558 err = got_ferror(f, getline_err->code);
10563 while (isspace((unsigned char)p[0]))
10565 if (p[0] == '#' || p[0] == '\0') {
10571 for (i = 0; i < nitems(got_histedit_cmds); i++) {
10572 cmd = &got_histedit_cmds[i];
10573 if (strncmp(cmd->name, p, strlen(cmd->name)) == 0 &&
10574 isspace((unsigned char)p[strlen(cmd->name)])) {
10575 p += strlen(cmd->name);
10578 if (p[0] == cmd->code && isspace((unsigned char)p[1])) {
10583 if (i == nitems(got_histedit_cmds)) {
10584 err = histedit_syntax_error(lineno);
10587 while (isspace((unsigned char)p[0]))
10589 if (cmd->code == GOT_HISTEDIT_MESG) {
10590 if (hle == NULL || hle->logmsg != NULL) {
10591 err = got_error(GOT_ERR_HISTEDIT_CMD);
10594 if (p[0] == '\0') {
10595 err = histedit_edit_logmsg(hle, repo);
10599 hle->logmsg = strdup(p);
10600 if (hle->logmsg == NULL) {
10601 err = got_error_from_errno("strdup");
10610 while (end[0] && !isspace((unsigned char)end[0]))
10614 err = got_object_resolve_id_str(&commit_id, repo, p);
10616 /* override error code */
10617 err = histedit_syntax_error(lineno);
10621 hle = malloc(sizeof(*hle));
10623 err = got_error_from_errno("malloc");
10627 hle->commit_id = commit_id;
10628 hle->logmsg = NULL;
10632 TAILQ_INSERT_TAIL(histedit_cmds, hle, entry);
10640 static const struct got_error *
10641 histedit_check_script(struct got_histedit_list *histedit_cmds,
10642 struct got_object_id_queue *commits, struct got_repository *repo)
10644 const struct got_error *err = NULL;
10645 struct got_object_qid *qid;
10646 struct got_histedit_list_entry *hle;
10647 static char msg[92];
10650 if (TAILQ_EMPTY(histedit_cmds))
10651 return got_error_msg(GOT_ERR_EMPTY_HISTEDIT,
10652 "histedit script contains no commands");
10653 if (STAILQ_EMPTY(commits))
10654 return got_error(GOT_ERR_EMPTY_HISTEDIT);
10656 TAILQ_FOREACH(hle, histedit_cmds, entry) {
10657 struct got_histedit_list_entry *hle2;
10658 TAILQ_FOREACH(hle2, histedit_cmds, entry) {
10661 if (got_object_id_cmp(hle->commit_id,
10662 hle2->commit_id) != 0)
10664 err = got_object_id_str(&id_str, hle->commit_id);
10667 snprintf(msg, sizeof(msg), "commit %s is listed "
10668 "more than once in histedit script", id_str);
10670 return got_error_msg(GOT_ERR_HISTEDIT_CMD, msg);
10674 STAILQ_FOREACH(qid, commits, entry) {
10675 TAILQ_FOREACH(hle, histedit_cmds, entry) {
10676 if (got_object_id_cmp(&qid->id, hle->commit_id) == 0)
10680 err = got_object_id_str(&id_str, &qid->id);
10683 snprintf(msg, sizeof(msg),
10684 "commit %s missing from histedit script", id_str);
10686 return got_error_msg(GOT_ERR_HISTEDIT_CMD, msg);
10690 hle = TAILQ_LAST(histedit_cmds, got_histedit_list);
10691 if (hle && hle->cmd->code == GOT_HISTEDIT_FOLD)
10692 return got_error_msg(GOT_ERR_HISTEDIT_CMD,
10693 "last commit in histedit script cannot be folded");
10698 static const struct got_error *
10699 histedit_run_editor(struct got_histedit_list *histedit_cmds,
10700 const char *path, struct got_object_id_queue *commits,
10701 struct got_repository *repo)
10703 const struct got_error *err = NULL;
10707 err = get_editor(&editor);
10711 if (spawn_editor(editor, path) == -1) {
10712 err = got_error_from_errno("failed spawning editor");
10716 f = fopen(path, "re");
10718 err = got_error_from_errno("fopen");
10721 err = histedit_parse_list(histedit_cmds, f, repo);
10725 err = histedit_check_script(histedit_cmds, commits, repo);
10727 if (f && fclose(f) == EOF && err == NULL)
10728 err = got_error_from_errno("fclose");
10733 static const struct got_error *
10734 histedit_edit_list_retry(struct got_histedit_list *, const struct got_error *,
10735 struct got_object_id_queue *, const char *, const char *,
10736 struct got_repository *);
10738 static const struct got_error *
10739 histedit_edit_script(struct got_histedit_list *histedit_cmds,
10740 struct got_object_id_queue *commits, const char *branch_name,
10741 int edit_logmsg_only, int fold_only, int edit_only,
10742 struct got_repository *repo)
10744 const struct got_error *err;
10748 err = got_opentemp_named(&path, &f, "got-histedit");
10752 err = write_cmd_list(f, branch_name, commits);
10756 err = histedit_write_commit_list(commits, f, edit_logmsg_only,
10757 fold_only, edit_only, repo);
10761 if (edit_logmsg_only || fold_only || edit_only) {
10763 err = histedit_parse_list(histedit_cmds, f, repo);
10765 if (fclose(f) == EOF) {
10766 err = got_error_from_errno("fclose");
10770 err = histedit_run_editor(histedit_cmds, path, commits, repo);
10772 if (err->code != GOT_ERR_HISTEDIT_SYNTAX &&
10773 err->code != GOT_ERR_HISTEDIT_CMD)
10775 err = histedit_edit_list_retry(histedit_cmds, err,
10776 commits, path, branch_name, repo);
10780 if (f && fclose(f) == EOF && err == NULL)
10781 err = got_error_from_errno("fclose");
10782 if (path && unlink(path) != 0 && err == NULL)
10783 err = got_error_from_errno2("unlink", path);
10788 static const struct got_error *
10789 histedit_save_list(struct got_histedit_list *histedit_cmds,
10790 struct got_worktree *worktree, struct got_repository *repo)
10792 const struct got_error *err = NULL;
10795 struct got_histedit_list_entry *hle;
10796 struct got_commit_object *commit = NULL;
10798 err = got_worktree_get_histedit_script_path(&path, worktree);
10802 f = fopen(path, "we");
10804 err = got_error_from_errno2("fopen", path);
10807 TAILQ_FOREACH(hle, histedit_cmds, entry) {
10808 err = histedit_write_commit(hle->commit_id, hle->cmd->name, f,
10814 int n = fprintf(f, "%c %s\n",
10815 GOT_HISTEDIT_MESG, hle->logmsg);
10817 err = got_ferror(f, GOT_ERR_IO);
10823 if (f && fclose(f) == EOF && err == NULL)
10824 err = got_error_from_errno("fclose");
10827 got_object_commit_close(commit);
10832 histedit_free_list(struct got_histedit_list *histedit_cmds)
10834 struct got_histedit_list_entry *hle;
10836 while ((hle = TAILQ_FIRST(histedit_cmds))) {
10837 TAILQ_REMOVE(histedit_cmds, hle, entry);
10842 static const struct got_error *
10843 histedit_load_list(struct got_histedit_list *histedit_cmds,
10844 const char *path, struct got_repository *repo)
10846 const struct got_error *err = NULL;
10849 f = fopen(path, "re");
10851 err = got_error_from_errno2("fopen", path);
10855 err = histedit_parse_list(histedit_cmds, f, repo);
10857 if (f && fclose(f) == EOF && err == NULL)
10858 err = got_error_from_errno("fclose");
10862 static const struct got_error *
10863 histedit_edit_list_retry(struct got_histedit_list *histedit_cmds,
10864 const struct got_error *edit_err, struct got_object_id_queue *commits,
10865 const char *path, const char *branch_name, struct got_repository *repo)
10867 const struct got_error *err = NULL, *prev_err = edit_err;
10870 while (resp != 'c' && resp != 'r' && resp != 'a') {
10871 printf("%s: %s\n(c)ontinue editing, (r)estart editing, "
10872 "or (a)bort: ", getprogname(), prev_err->msg);
10877 histedit_free_list(histedit_cmds);
10878 err = histedit_run_editor(histedit_cmds, path, commits,
10881 if (err->code != GOT_ERR_HISTEDIT_SYNTAX &&
10882 err->code != GOT_ERR_HISTEDIT_CMD)
10889 } else if (resp == 'r') {
10890 histedit_free_list(histedit_cmds);
10891 err = histedit_edit_script(histedit_cmds,
10892 commits, branch_name, 0, 0, 0, repo);
10894 if (err->code != GOT_ERR_HISTEDIT_SYNTAX &&
10895 err->code != GOT_ERR_HISTEDIT_CMD)
10902 } else if (resp == 'a') {
10903 err = got_error(GOT_ERR_HISTEDIT_CANCEL);
10906 printf("invalid response '%c'\n", resp);
10912 static const struct got_error *
10913 histedit_complete(struct got_worktree *worktree,
10914 struct got_fileindex *fileindex, struct got_reference *tmp_branch,
10915 struct got_reference *branch, struct got_repository *repo)
10917 printf("Switching work tree to %s\n",
10918 got_ref_get_symref_target(branch));
10919 return got_worktree_histedit_complete(worktree, fileindex, tmp_branch,
10923 static const struct got_error *
10924 show_histedit_progress(struct got_commit_object *commit,
10925 struct got_histedit_list_entry *hle, struct got_object_id *new_id)
10927 const struct got_error *err;
10928 char *old_id_str = NULL, *new_id_str = NULL, *logmsg = NULL;
10930 err = got_object_id_str(&old_id_str, hle->commit_id);
10935 err = got_object_id_str(&new_id_str, new_id);
10940 old_id_str[12] = '\0';
10942 new_id_str[12] = '\0';
10945 logmsg = strdup(hle->logmsg);
10946 if (logmsg == NULL) {
10947 err = got_error_from_errno("strdup");
10950 trim_logmsg(logmsg, 42);
10952 err = get_short_logmsg(&logmsg, 42, commit);
10957 switch (hle->cmd->code) {
10958 case GOT_HISTEDIT_PICK:
10959 case GOT_HISTEDIT_EDIT:
10960 printf("%s -> %s: %s\n", old_id_str,
10961 new_id_str ? new_id_str : "no-op change", logmsg);
10963 case GOT_HISTEDIT_DROP:
10964 case GOT_HISTEDIT_FOLD:
10965 printf("%s -> %s commit: %s\n", old_id_str, hle->cmd->name,
10977 static const struct got_error *
10978 histedit_commit(struct got_pathlist_head *merged_paths,
10979 struct got_worktree *worktree, struct got_fileindex *fileindex,
10980 struct got_reference *tmp_branch, struct got_histedit_list_entry *hle,
10981 struct got_repository *repo)
10983 const struct got_error *err;
10984 struct got_commit_object *commit;
10985 struct got_object_id *new_commit_id;
10987 if ((hle->cmd->code == GOT_HISTEDIT_EDIT || get_folded_commits(hle))
10988 && hle->logmsg == NULL) {
10989 err = histedit_edit_logmsg(hle, repo);
10994 err = got_object_open_as_commit(&commit, repo, hle->commit_id);
10998 err = got_worktree_histedit_commit(&new_commit_id, merged_paths,
10999 worktree, fileindex, tmp_branch, commit, hle->commit_id,
11000 hle->logmsg, repo);
11002 if (err->code != GOT_ERR_COMMIT_NO_CHANGES)
11004 err = show_histedit_progress(commit, hle, NULL);
11006 err = show_histedit_progress(commit, hle, new_commit_id);
11007 free(new_commit_id);
11010 got_object_commit_close(commit);
11014 static const struct got_error *
11015 histedit_skip_commit(struct got_histedit_list_entry *hle,
11016 struct got_worktree *worktree, struct got_repository *repo)
11018 const struct got_error *error;
11019 struct got_commit_object *commit;
11021 error = got_worktree_histedit_skip_commit(worktree, hle->commit_id,
11026 error = got_object_open_as_commit(&commit, repo, hle->commit_id);
11030 error = show_histedit_progress(commit, hle, NULL);
11031 got_object_commit_close(commit);
11035 static const struct got_error *
11036 check_local_changes(void *arg, unsigned char status,
11037 unsigned char staged_status, const char *path,
11038 struct got_object_id *blob_id, struct got_object_id *staged_blob_id,
11039 struct got_object_id *commit_id, int dirfd, const char *de_name)
11041 int *have_local_changes = arg;
11044 case GOT_STATUS_ADD:
11045 case GOT_STATUS_DELETE:
11046 case GOT_STATUS_MODIFY:
11047 case GOT_STATUS_CONFLICT:
11048 *have_local_changes = 1;
11049 return got_error(GOT_ERR_CANCELLED);
11054 switch (staged_status) {
11055 case GOT_STATUS_ADD:
11056 case GOT_STATUS_DELETE:
11057 case GOT_STATUS_MODIFY:
11058 *have_local_changes = 1;
11059 return got_error(GOT_ERR_CANCELLED);
11067 static const struct got_error *
11068 cmd_histedit(int argc, char *argv[])
11070 const struct got_error *error = NULL;
11071 struct got_worktree *worktree = NULL;
11072 struct got_fileindex *fileindex = NULL;
11073 struct got_repository *repo = NULL;
11075 struct got_reference *branch = NULL;
11076 struct got_reference *tmp_branch = NULL;
11077 struct got_object_id *resume_commit_id = NULL;
11078 struct got_object_id *base_commit_id = NULL;
11079 struct got_object_id *head_commit_id = NULL;
11080 struct got_commit_object *commit = NULL;
11081 int ch, rebase_in_progress = 0, merge_in_progress = 0;
11082 struct got_update_progress_arg upa;
11083 int edit_in_progress = 0, abort_edit = 0, continue_edit = 0;
11084 int edit_logmsg_only = 0, fold_only = 0, edit_only = 0;
11085 int list_backups = 0, delete_backups = 0;
11086 const char *edit_script_path = NULL;
11087 struct got_object_id_queue commits;
11088 struct got_pathlist_head merged_paths;
11089 const struct got_object_id_queue *parent_ids;
11090 struct got_object_qid *pid;
11091 struct got_histedit_list histedit_cmds;
11092 struct got_histedit_list_entry *hle;
11093 int *pack_fds = NULL;
11095 STAILQ_INIT(&commits);
11096 TAILQ_INIT(&histedit_cmds);
11097 TAILQ_INIT(&merged_paths);
11098 memset(&upa, 0, sizeof(upa));
11100 while ((ch = getopt(argc, argv, "acefF:mlX")) != -1) {
11115 edit_script_path = optarg;
11118 edit_logmsg_only = 1;
11124 delete_backups = 1;
11136 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
11137 "unveil", NULL) == -1)
11140 if (abort_edit && continue_edit)
11141 option_conflict('a', 'c');
11142 if (edit_script_path && edit_logmsg_only)
11143 option_conflict('F', 'm');
11144 if (abort_edit && edit_logmsg_only)
11145 option_conflict('a', 'm');
11146 if (continue_edit && edit_logmsg_only)
11147 option_conflict('c', 'm');
11148 if (abort_edit && fold_only)
11149 option_conflict('a', 'f');
11150 if (continue_edit && fold_only)
11151 option_conflict('c', 'f');
11152 if (fold_only && edit_logmsg_only)
11153 option_conflict('f', 'm');
11154 if (edit_script_path && fold_only)
11155 option_conflict('F', 'f');
11156 if (abort_edit && edit_only)
11157 option_conflict('a', 'e');
11158 if (continue_edit && edit_only)
11159 option_conflict('c', 'e');
11160 if (edit_only && edit_logmsg_only)
11161 option_conflict('e', 'm');
11162 if (edit_script_path && edit_only)
11163 option_conflict('F', 'e');
11164 if (list_backups) {
11166 option_conflict('l', 'a');
11168 option_conflict('l', 'c');
11169 if (edit_script_path)
11170 option_conflict('l', 'F');
11171 if (edit_logmsg_only)
11172 option_conflict('l', 'm');
11174 option_conflict('l', 'f');
11176 option_conflict('l', 'e');
11177 if (delete_backups)
11178 option_conflict('l', 'X');
11179 if (argc != 0 && argc != 1)
11181 } else if (delete_backups) {
11183 option_conflict('X', 'a');
11185 option_conflict('X', 'c');
11186 if (edit_script_path)
11187 option_conflict('X', 'F');
11188 if (edit_logmsg_only)
11189 option_conflict('X', 'm');
11191 option_conflict('X', 'f');
11193 option_conflict('X', 'e');
11195 option_conflict('X', 'l');
11196 if (argc != 0 && argc != 1)
11198 } else if (argc != 0)
11202 * This command cannot apply unveil(2) in all cases because the
11203 * user may choose to run an editor to edit the histedit script
11204 * and to edit individual commit log messages.
11205 * unveil(2) traverses exec(2); if an editor is used we have to
11206 * apply unveil after edit script and log messages have been written.
11207 * XXX TODO: Make use of unveil(2) where possible.
11210 cwd = getcwd(NULL, 0);
11212 error = got_error_from_errno("getcwd");
11216 error = got_repo_pack_fds_open(&pack_fds);
11220 error = got_worktree_open(&worktree, cwd);
11222 if (list_backups || delete_backups) {
11223 if (error->code != GOT_ERR_NOT_WORKTREE)
11226 if (error->code == GOT_ERR_NOT_WORKTREE)
11227 error = wrap_not_worktree_error(error,
11233 if (list_backups || delete_backups) {
11234 error = got_repo_open(&repo,
11235 worktree ? got_worktree_get_repo_path(worktree) : cwd,
11239 error = apply_unveil(got_repo_get_path(repo), 0,
11240 worktree ? got_worktree_get_root_path(worktree) : NULL);
11243 error = process_backup_refs(
11244 GOT_WORKTREE_HISTEDIT_BACKUP_REF_PREFIX,
11245 argc == 1 ? argv[0] : NULL, delete_backups, repo);
11246 goto done; /* nothing else to do */
11249 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
11254 error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
11257 if (rebase_in_progress) {
11258 error = got_error(GOT_ERR_REBASING);
11262 error = got_worktree_merge_in_progress(&merge_in_progress, worktree,
11266 if (merge_in_progress) {
11267 error = got_error(GOT_ERR_MERGE_BUSY);
11271 error = got_worktree_histedit_in_progress(&edit_in_progress, worktree);
11275 if (edit_in_progress && edit_logmsg_only) {
11276 error = got_error_msg(GOT_ERR_HISTEDIT_BUSY,
11277 "histedit operation is in progress in this "
11278 "work tree and must be continued or aborted "
11279 "before the -m option can be used");
11282 if (edit_in_progress && fold_only) {
11283 error = got_error_msg(GOT_ERR_HISTEDIT_BUSY,
11284 "histedit operation is in progress in this "
11285 "work tree and must be continued or aborted "
11286 "before the -f option can be used");
11289 if (edit_in_progress && edit_only) {
11290 error = got_error_msg(GOT_ERR_HISTEDIT_BUSY,
11291 "histedit operation is in progress in this "
11292 "work tree and must be continued or aborted "
11293 "before the -e option can be used");
11297 if (edit_in_progress && abort_edit) {
11298 error = got_worktree_histedit_continue(&resume_commit_id,
11299 &tmp_branch, &branch, &base_commit_id, &fileindex,
11303 printf("Switching work tree to %s\n",
11304 got_ref_get_symref_target(branch));
11305 error = got_worktree_histedit_abort(worktree, fileindex, repo,
11306 branch, base_commit_id, abort_progress, &upa);
11309 printf("Histedit of %s aborted\n",
11310 got_ref_get_symref_target(branch));
11311 print_merge_progress_stats(&upa);
11312 goto done; /* nothing else to do */
11313 } else if (abort_edit) {
11314 error = got_error(GOT_ERR_NOT_HISTEDIT);
11318 if (continue_edit) {
11321 if (!edit_in_progress) {
11322 error = got_error(GOT_ERR_NOT_HISTEDIT);
11326 error = got_worktree_get_histedit_script_path(&path, worktree);
11330 error = histedit_load_list(&histedit_cmds, path, repo);
11335 error = got_worktree_histedit_continue(&resume_commit_id,
11336 &tmp_branch, &branch, &base_commit_id, &fileindex,
11341 error = got_ref_resolve(&head_commit_id, repo, branch);
11345 error = got_object_open_as_commit(&commit, repo,
11349 parent_ids = got_object_commit_get_parent_ids(commit);
11350 pid = STAILQ_FIRST(parent_ids);
11352 error = got_error(GOT_ERR_EMPTY_HISTEDIT);
11355 error = collect_commits(&commits, head_commit_id, &pid->id,
11356 base_commit_id, got_worktree_get_path_prefix(worktree),
11357 GOT_ERR_HISTEDIT_PATH, repo);
11358 got_object_commit_close(commit);
11363 if (edit_in_progress) {
11364 error = got_error(GOT_ERR_HISTEDIT_BUSY);
11368 error = got_ref_open(&branch, repo,
11369 got_worktree_get_head_ref_name(worktree), 0);
11373 if (strncmp(got_ref_get_name(branch), "refs/heads/", 11) != 0) {
11374 error = got_error_msg(GOT_ERR_COMMIT_BRANCH,
11375 "will not edit commit history of a branch outside "
11376 "the \"refs/heads/\" reference namespace");
11380 error = got_ref_resolve(&head_commit_id, repo, branch);
11381 got_ref_close(branch);
11386 error = got_object_open_as_commit(&commit, repo,
11390 parent_ids = got_object_commit_get_parent_ids(commit);
11391 pid = STAILQ_FIRST(parent_ids);
11393 error = got_error(GOT_ERR_EMPTY_HISTEDIT);
11396 error = collect_commits(&commits, head_commit_id, &pid->id,
11397 got_worktree_get_base_commit_id(worktree),
11398 got_worktree_get_path_prefix(worktree),
11399 GOT_ERR_HISTEDIT_PATH, repo);
11400 got_object_commit_close(commit);
11405 if (STAILQ_EMPTY(&commits)) {
11406 error = got_error(GOT_ERR_EMPTY_HISTEDIT);
11410 error = got_worktree_histedit_prepare(&tmp_branch, &branch,
11411 &base_commit_id, &fileindex, worktree, repo);
11415 if (edit_script_path) {
11416 error = histedit_load_list(&histedit_cmds,
11417 edit_script_path, repo);
11419 got_worktree_histedit_abort(worktree, fileindex,
11420 repo, branch, base_commit_id,
11421 abort_progress, &upa);
11422 print_merge_progress_stats(&upa);
11426 const char *branch_name;
11427 branch_name = got_ref_get_symref_target(branch);
11428 if (strncmp(branch_name, "refs/heads/", 11) == 0)
11430 error = histedit_edit_script(&histedit_cmds, &commits,
11431 branch_name, edit_logmsg_only, fold_only,
11434 got_worktree_histedit_abort(worktree, fileindex,
11435 repo, branch, base_commit_id,
11436 abort_progress, &upa);
11437 print_merge_progress_stats(&upa);
11443 error = histedit_save_list(&histedit_cmds, worktree,
11446 got_worktree_histedit_abort(worktree, fileindex,
11447 repo, branch, base_commit_id,
11448 abort_progress, &upa);
11449 print_merge_progress_stats(&upa);
11455 error = histedit_check_script(&histedit_cmds, &commits, repo);
11459 TAILQ_FOREACH(hle, &histedit_cmds, entry) {
11460 if (resume_commit_id) {
11461 if (got_object_id_cmp(hle->commit_id,
11462 resume_commit_id) != 0)
11465 resume_commit_id = NULL;
11466 if (hle->cmd->code == GOT_HISTEDIT_DROP ||
11467 hle->cmd->code == GOT_HISTEDIT_FOLD) {
11468 error = histedit_skip_commit(hle, worktree,
11473 struct got_pathlist_head paths;
11474 int have_changes = 0;
11476 TAILQ_INIT(&paths);
11477 error = got_pathlist_append(&paths, "", NULL);
11480 error = got_worktree_status(worktree, &paths,
11481 repo, 0, check_local_changes, &have_changes,
11482 check_cancelled, NULL);
11483 got_pathlist_free(&paths);
11485 if (error->code != GOT_ERR_CANCELLED)
11487 if (sigint_received || sigpipe_received)
11490 if (have_changes) {
11491 error = histedit_commit(NULL, worktree,
11492 fileindex, tmp_branch, hle, repo);
11496 error = got_object_open_as_commit(
11497 &commit, repo, hle->commit_id);
11500 error = show_histedit_progress(commit,
11502 got_object_commit_close(commit);
11511 if (hle->cmd->code == GOT_HISTEDIT_DROP) {
11512 error = histedit_skip_commit(hle, worktree, repo);
11518 error = got_object_open_as_commit(&commit, repo,
11522 parent_ids = got_object_commit_get_parent_ids(commit);
11523 pid = STAILQ_FIRST(parent_ids);
11525 error = got_worktree_histedit_merge_files(&merged_paths,
11526 worktree, fileindex, &pid->id, hle->commit_id, repo,
11527 update_progress, &upa, check_cancelled, NULL);
11530 got_object_commit_close(commit);
11533 print_merge_progress_stats(&upa);
11534 if (upa.conflicts > 0 || upa.missing > 0 ||
11535 upa.not_deleted > 0 || upa.unversioned > 0) {
11536 if (upa.conflicts > 0) {
11537 error = show_rebase_merge_conflict(
11538 hle->commit_id, repo);
11542 got_worktree_rebase_pathlist_free(&merged_paths);
11546 if (hle->cmd->code == GOT_HISTEDIT_EDIT) {
11548 error = got_object_id_str(&id_str, hle->commit_id);
11551 printf("Stopping histedit for amending commit %s\n",
11554 got_worktree_rebase_pathlist_free(&merged_paths);
11555 error = got_worktree_histedit_postpone(worktree,
11560 if (hle->cmd->code == GOT_HISTEDIT_FOLD) {
11561 error = histedit_skip_commit(hle, worktree, repo);
11567 error = histedit_commit(&merged_paths, worktree, fileindex,
11568 tmp_branch, hle, repo);
11569 got_worktree_rebase_pathlist_free(&merged_paths);
11574 if (upa.conflicts > 0 || upa.missing > 0 ||
11575 upa.not_deleted > 0 || upa.unversioned > 0) {
11576 error = got_worktree_histedit_postpone(worktree, fileindex);
11579 if (upa.conflicts > 0 && upa.missing == 0 &&
11580 upa.not_deleted == 0 && upa.unversioned == 0) {
11581 error = got_error_msg(GOT_ERR_CONFLICTS,
11582 "conflicts must be resolved before histedit "
11584 } else if (upa.conflicts > 0) {
11585 error = got_error_msg(GOT_ERR_CONFLICTS,
11586 "conflicts must be resolved before histedit "
11587 "can continue; changes destined for some "
11588 "files were not yet merged and should be "
11589 "merged manually if required before the "
11590 "histedit operation is continued");
11592 error = got_error_msg(GOT_ERR_CONFLICTS,
11593 "changes destined for some files were not "
11594 "yet merged and should be merged manually "
11595 "if required before the histedit operation "
11599 error = histedit_complete(worktree, fileindex, tmp_branch,
11602 got_object_id_queue_free(&commits);
11603 histedit_free_list(&histedit_cmds);
11604 free(head_commit_id);
11605 free(base_commit_id);
11606 free(resume_commit_id);
11608 got_object_commit_close(commit);
11610 got_ref_close(branch);
11612 got_ref_close(tmp_branch);
11614 got_worktree_close(worktree);
11616 const struct got_error *close_err = got_repo_close(repo);
11621 const struct got_error *pack_err =
11622 got_repo_pack_fds_close(pack_fds);
11630 usage_integrate(void)
11632 fprintf(stderr, "usage: %s integrate branch\n", getprogname());
11636 static const struct got_error *
11637 cmd_integrate(int argc, char *argv[])
11639 const struct got_error *error = NULL;
11640 struct got_repository *repo = NULL;
11641 struct got_worktree *worktree = NULL;
11642 char *cwd = NULL, *refname = NULL, *base_refname = NULL;
11643 const char *branch_arg = NULL;
11644 struct got_reference *branch_ref = NULL, *base_branch_ref = NULL;
11645 struct got_fileindex *fileindex = NULL;
11646 struct got_object_id *commit_id = NULL, *base_commit_id = NULL;
11648 struct got_update_progress_arg upa;
11649 int *pack_fds = NULL;
11651 while ((ch = getopt(argc, argv, "")) != -1) {
11664 branch_arg = argv[0];
11666 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
11667 "unveil", NULL) == -1)
11670 cwd = getcwd(NULL, 0);
11672 error = got_error_from_errno("getcwd");
11676 error = got_repo_pack_fds_open(&pack_fds);
11680 error = got_worktree_open(&worktree, cwd);
11682 if (error->code == GOT_ERR_NOT_WORKTREE)
11683 error = wrap_not_worktree_error(error, "integrate",
11688 error = check_rebase_or_histedit_in_progress(worktree);
11692 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
11697 error = apply_unveil(got_repo_get_path(repo), 0,
11698 got_worktree_get_root_path(worktree));
11702 error = check_merge_in_progress(worktree, repo);
11706 if (asprintf(&refname, "refs/heads/%s", branch_arg) == -1) {
11707 error = got_error_from_errno("asprintf");
11711 error = got_worktree_integrate_prepare(&fileindex, &branch_ref,
11712 &base_branch_ref, worktree, refname, repo);
11716 refname = strdup(got_ref_get_name(branch_ref));
11717 if (refname == NULL) {
11718 error = got_error_from_errno("strdup");
11719 got_worktree_integrate_abort(worktree, fileindex, repo,
11720 branch_ref, base_branch_ref);
11723 base_refname = strdup(got_ref_get_name(base_branch_ref));
11724 if (base_refname == NULL) {
11725 error = got_error_from_errno("strdup");
11726 got_worktree_integrate_abort(worktree, fileindex, repo,
11727 branch_ref, base_branch_ref);
11731 error = got_ref_resolve(&commit_id, repo, branch_ref);
11735 error = got_ref_resolve(&base_commit_id, repo, base_branch_ref);
11739 if (got_object_id_cmp(commit_id, base_commit_id) == 0) {
11740 error = got_error_msg(GOT_ERR_SAME_BRANCH,
11741 "specified branch has already been integrated");
11742 got_worktree_integrate_abort(worktree, fileindex, repo,
11743 branch_ref, base_branch_ref);
11747 error = check_linear_ancestry(commit_id, base_commit_id, 1, repo);
11749 if (error->code == GOT_ERR_ANCESTRY)
11750 error = got_error(GOT_ERR_REBASE_REQUIRED);
11751 got_worktree_integrate_abort(worktree, fileindex, repo,
11752 branch_ref, base_branch_ref);
11756 memset(&upa, 0, sizeof(upa));
11757 error = got_worktree_integrate_continue(worktree, fileindex, repo,
11758 branch_ref, base_branch_ref, update_progress, &upa,
11759 check_cancelled, NULL);
11763 printf("Integrated %s into %s\n", refname, base_refname);
11764 print_update_progress_stats(&upa);
11767 const struct got_error *close_err = got_repo_close(repo);
11772 got_worktree_close(worktree);
11774 const struct got_error *pack_err =
11775 got_repo_pack_fds_close(pack_fds);
11780 free(base_commit_id);
11783 free(base_refname);
11790 fprintf(stderr, "usage: %s merge [-a] [-c] [-n] [branch]\n",
11795 static const struct got_error *
11796 cmd_merge(int argc, char *argv[])
11798 const struct got_error *error = NULL;
11799 struct got_worktree *worktree = NULL;
11800 struct got_repository *repo = NULL;
11801 struct got_fileindex *fileindex = NULL;
11802 char *cwd = NULL, *id_str = NULL, *author = NULL;
11803 struct got_reference *branch = NULL, *wt_branch = NULL;
11804 struct got_object_id *branch_tip = NULL, *yca_id = NULL;
11805 struct got_object_id *wt_branch_tip = NULL;
11806 int ch, merge_in_progress = 0, abort_merge = 0, continue_merge = 0;
11807 int interrupt_merge = 0;
11808 struct got_update_progress_arg upa;
11809 struct got_object_id *merge_commit_id = NULL;
11810 char *branch_name = NULL;
11811 int *pack_fds = NULL;
11813 memset(&upa, 0, sizeof(upa));
11815 while ((ch = getopt(argc, argv, "acn")) != -1) {
11821 continue_merge = 1;
11824 interrupt_merge = 1;
11836 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
11837 "unveil", NULL) == -1)
11841 if (abort_merge && continue_merge)
11842 option_conflict('a', 'c');
11843 if (abort_merge || continue_merge) {
11846 } else if (argc != 1)
11849 cwd = getcwd(NULL, 0);
11851 error = got_error_from_errno("getcwd");
11855 error = got_repo_pack_fds_open(&pack_fds);
11859 error = got_worktree_open(&worktree, cwd);
11861 if (error->code == GOT_ERR_NOT_WORKTREE)
11862 error = wrap_not_worktree_error(error,
11867 error = got_repo_open(&repo,
11868 worktree ? got_worktree_get_repo_path(worktree) : cwd, NULL,
11873 error = apply_unveil(got_repo_get_path(repo), 0,
11874 worktree ? got_worktree_get_root_path(worktree) : NULL);
11878 error = check_rebase_or_histedit_in_progress(worktree);
11882 error = got_worktree_merge_in_progress(&merge_in_progress, worktree,
11888 if (!merge_in_progress) {
11889 error = got_error(GOT_ERR_NOT_MERGING);
11892 error = got_worktree_merge_continue(&branch_name,
11893 &branch_tip, &fileindex, worktree, repo);
11896 error = got_worktree_merge_abort(worktree, fileindex, repo,
11897 abort_progress, &upa);
11900 printf("Merge of %s aborted\n", branch_name);
11901 goto done; /* nothing else to do */
11904 error = get_author(&author, repo, worktree);
11908 if (continue_merge) {
11909 if (!merge_in_progress) {
11910 error = got_error(GOT_ERR_NOT_MERGING);
11913 error = got_worktree_merge_continue(&branch_name,
11914 &branch_tip, &fileindex, worktree, repo);
11918 error = got_ref_open(&branch, repo, argv[0], 0);
11921 branch_name = strdup(got_ref_get_name(branch));
11922 if (branch_name == NULL) {
11923 error = got_error_from_errno("strdup");
11926 error = got_ref_resolve(&branch_tip, repo, branch);
11931 error = got_ref_open(&wt_branch, repo,
11932 got_worktree_get_head_ref_name(worktree), 0);
11935 error = got_ref_resolve(&wt_branch_tip, repo, wt_branch);
11938 error = got_commit_graph_find_youngest_common_ancestor(&yca_id,
11939 wt_branch_tip, branch_tip, 0, repo,
11940 check_cancelled, NULL);
11941 if (error && error->code != GOT_ERR_ANCESTRY)
11944 if (!continue_merge) {
11945 error = check_path_prefix(wt_branch_tip, branch_tip,
11946 got_worktree_get_path_prefix(worktree),
11947 GOT_ERR_MERGE_PATH, repo);
11951 error = check_same_branch(wt_branch_tip, branch,
11954 if (error->code != GOT_ERR_ANCESTRY)
11958 static char msg[512];
11959 snprintf(msg, sizeof(msg),
11960 "cannot create a merge commit because "
11961 "%s is based on %s; %s can be integrated "
11962 "with 'got integrate' instead", branch_name,
11963 got_worktree_get_head_ref_name(worktree),
11965 error = got_error_msg(GOT_ERR_SAME_BRANCH, msg);
11969 error = got_worktree_merge_prepare(&fileindex, worktree,
11974 error = got_worktree_merge_branch(worktree, fileindex,
11975 yca_id, branch_tip, repo, update_progress, &upa,
11976 check_cancelled, NULL);
11979 print_merge_progress_stats(&upa);
11980 if (!upa.did_something) {
11981 error = got_worktree_merge_abort(worktree, fileindex,
11982 repo, abort_progress, &upa);
11985 printf("Already up-to-date\n");
11990 if (interrupt_merge) {
11991 error = got_worktree_merge_postpone(worktree, fileindex);
11994 printf("Merge of %s interrupted on request\n", branch_name);
11995 } else if (upa.conflicts > 0 || upa.missing > 0 ||
11996 upa.not_deleted > 0 || upa.unversioned > 0) {
11997 error = got_worktree_merge_postpone(worktree, fileindex);
12000 if (upa.conflicts > 0 && upa.missing == 0 &&
12001 upa.not_deleted == 0 && upa.unversioned == 0) {
12002 error = got_error_msg(GOT_ERR_CONFLICTS,
12003 "conflicts must be resolved before merging "
12005 } else if (upa.conflicts > 0) {
12006 error = got_error_msg(GOT_ERR_CONFLICTS,
12007 "conflicts must be resolved before merging "
12008 "can continue; changes destined for some "
12009 "files were not yet merged and "
12010 "should be merged manually if required before the "
12011 "merge operation is continued");
12013 error = got_error_msg(GOT_ERR_CONFLICTS,
12014 "changes destined for some "
12015 "files were not yet merged and should be "
12016 "merged manually if required before the "
12017 "merge operation is continued");
12021 error = got_worktree_merge_commit(&merge_commit_id, worktree,
12022 fileindex, author, NULL, 1, branch_tip, branch_name,
12023 repo, continue_merge ? print_status : NULL, NULL);
12026 error = got_worktree_merge_complete(worktree, fileindex, repo);
12029 error = got_object_id_str(&id_str, merge_commit_id);
12032 printf("Merged %s into %s: %s\n", branch_name,
12033 got_worktree_get_head_ref_name(worktree),
12039 free(merge_commit_id);
12045 got_ref_close(branch);
12047 got_ref_close(wt_branch);
12049 got_worktree_close(worktree);
12051 const struct got_error *close_err = got_repo_close(repo);
12056 const struct got_error *pack_err =
12057 got_repo_pack_fds_close(pack_fds);
12067 fprintf(stderr, "usage: %s stage [-l] | [-p] [-F response-script] "
12068 "[-S] [file-path ...]\n",
12073 static const struct got_error *
12074 print_stage(void *arg, unsigned char status, unsigned char staged_status,
12075 const char *path, struct got_object_id *blob_id,
12076 struct got_object_id *staged_blob_id, struct got_object_id *commit_id,
12077 int dirfd, const char *de_name)
12079 const struct got_error *err = NULL;
12080 char *id_str = NULL;
12082 if (staged_status != GOT_STATUS_ADD &&
12083 staged_status != GOT_STATUS_MODIFY &&
12084 staged_status != GOT_STATUS_DELETE)
12087 if (staged_status == GOT_STATUS_ADD ||
12088 staged_status == GOT_STATUS_MODIFY)
12089 err = got_object_id_str(&id_str, staged_blob_id);
12091 err = got_object_id_str(&id_str, blob_id);
12095 printf("%s %c %s\n", id_str, staged_status, path);
12100 static const struct got_error *
12101 cmd_stage(int argc, char *argv[])
12103 const struct got_error *error = NULL;
12104 struct got_repository *repo = NULL;
12105 struct got_worktree *worktree = NULL;
12107 struct got_pathlist_head paths;
12108 struct got_pathlist_entry *pe;
12109 int ch, list_stage = 0, pflag = 0, allow_bad_symlinks = 0;
12110 FILE *patch_script_file = NULL;
12111 const char *patch_script_path = NULL;
12112 struct choose_patch_arg cpa;
12113 int *pack_fds = NULL;
12115 TAILQ_INIT(&paths);
12117 while ((ch = getopt(argc, argv, "lpF:S")) != -1) {
12126 patch_script_path = optarg;
12129 allow_bad_symlinks = 1;
12141 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
12142 "unveil", NULL) == -1)
12145 if (list_stage && (pflag || patch_script_path))
12146 errx(1, "-l option cannot be used with other options");
12147 if (patch_script_path && !pflag)
12148 errx(1, "-F option can only be used together with -p option");
12150 cwd = getcwd(NULL, 0);
12152 error = got_error_from_errno("getcwd");
12156 error = got_repo_pack_fds_open(&pack_fds);
12160 error = got_worktree_open(&worktree, cwd);
12162 if (error->code == GOT_ERR_NOT_WORKTREE)
12163 error = wrap_not_worktree_error(error, "stage", cwd);
12167 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
12172 if (patch_script_path) {
12173 patch_script_file = fopen(patch_script_path, "re");
12174 if (patch_script_file == NULL) {
12175 error = got_error_from_errno2("fopen",
12176 patch_script_path);
12180 error = apply_unveil(got_repo_get_path(repo), 0,
12181 got_worktree_get_root_path(worktree));
12185 error = check_merge_in_progress(worktree, repo);
12189 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
12194 error = got_worktree_status(worktree, &paths, repo, 0,
12195 print_stage, NULL, check_cancelled, NULL);
12197 cpa.patch_script_file = patch_script_file;
12198 cpa.action = "stage";
12199 error = got_worktree_stage(worktree, &paths,
12200 pflag ? NULL : print_status, NULL,
12201 pflag ? choose_patch : NULL, &cpa,
12202 allow_bad_symlinks, repo);
12205 if (patch_script_file && fclose(patch_script_file) == EOF &&
12207 error = got_error_from_errno2("fclose", patch_script_path);
12209 const struct got_error *close_err = got_repo_close(repo);
12214 got_worktree_close(worktree);
12216 const struct got_error *pack_err =
12217 got_repo_pack_fds_close(pack_fds);
12221 TAILQ_FOREACH(pe, &paths, entry)
12222 free((char *)pe->path);
12223 got_pathlist_free(&paths);
12229 usage_unstage(void)
12231 fprintf(stderr, "usage: %s unstage [-p] [-F response-script] "
12232 "[file-path ...]\n",
12238 static const struct got_error *
12239 cmd_unstage(int argc, char *argv[])
12241 const struct got_error *error = NULL;
12242 struct got_repository *repo = NULL;
12243 struct got_worktree *worktree = NULL;
12245 struct got_pathlist_head paths;
12246 struct got_pathlist_entry *pe;
12248 struct got_update_progress_arg upa;
12249 FILE *patch_script_file = NULL;
12250 const char *patch_script_path = NULL;
12251 struct choose_patch_arg cpa;
12252 int *pack_fds = NULL;
12254 TAILQ_INIT(&paths);
12256 while ((ch = getopt(argc, argv, "pF:")) != -1) {
12262 patch_script_path = optarg;
12274 if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
12275 "unveil", NULL) == -1)
12278 if (patch_script_path && !pflag)
12279 errx(1, "-F option can only be used together with -p option");
12281 cwd = getcwd(NULL, 0);
12283 error = got_error_from_errno("getcwd");
12287 error = got_repo_pack_fds_open(&pack_fds);
12291 error = got_worktree_open(&worktree, cwd);
12293 if (error->code == GOT_ERR_NOT_WORKTREE)
12294 error = wrap_not_worktree_error(error, "unstage", cwd);
12298 error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
12303 if (patch_script_path) {
12304 patch_script_file = fopen(patch_script_path, "re");
12305 if (patch_script_file == NULL) {
12306 error = got_error_from_errno2("fopen",
12307 patch_script_path);
12312 error = apply_unveil(got_repo_get_path(repo), 0,
12313 got_worktree_get_root_path(worktree));
12317 error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
12321 cpa.patch_script_file = patch_script_file;
12322 cpa.action = "unstage";
12323 memset(&upa, 0, sizeof(upa));
12324 error = got_worktree_unstage(worktree, &paths, update_progress,
12325 &upa, pflag ? choose_patch : NULL, &cpa, repo);
12327 print_merge_progress_stats(&upa);
12329 if (patch_script_file && fclose(patch_script_file) == EOF &&
12331 error = got_error_from_errno2("fclose", patch_script_path);
12333 const struct got_error *close_err = got_repo_close(repo);
12338 got_worktree_close(worktree);
12340 const struct got_error *pack_err =
12341 got_repo_pack_fds_close(pack_fds);
12345 TAILQ_FOREACH(pe, &paths, entry)
12346 free((char *)pe->path);
12347 got_pathlist_free(&paths);
12355 fprintf(stderr, "usage: %s cat [-r repository ] [ -c commit ] [ -P ] "
12356 "arg1 [arg2 ...]\n", getprogname());
12360 static const struct got_error *
12361 cat_blob(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
12363 const struct got_error *err;
12364 struct got_blob_object *blob;
12367 fd = got_opentempfd();
12369 return got_error_from_errno("got_opentempfd");
12371 err = got_object_open_as_blob(&blob, repo, id, 8192, fd);
12375 err = got_object_blob_dump_to_file(NULL, NULL, NULL, outfile, blob);
12377 if (fd != -1 && close(fd) == -1 && err == NULL)
12378 err = got_error_from_errno("close");
12380 got_object_blob_close(blob);
12384 static const struct got_error *
12385 cat_tree(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
12387 const struct got_error *err;
12388 struct got_tree_object *tree;
12391 err = got_object_open_as_tree(&tree, repo, id);
12395 nentries = got_object_tree_get_nentries(tree);
12396 for (i = 0; i < nentries; i++) {
12397 struct got_tree_entry *te;
12399 if (sigint_received || sigpipe_received)
12401 te = got_object_tree_get_entry(tree, i);
12402 err = got_object_id_str(&id_str, got_tree_entry_get_id(te));
12405 fprintf(outfile, "%s %.7o %s\n", id_str,
12406 got_tree_entry_get_mode(te),
12407 got_tree_entry_get_name(te));
12411 got_object_tree_close(tree);
12416 format_gmtoff(char *buf, size_t sz, time_t gmtoff)
12426 h = (long long)gmtoff / 3600;
12427 m = ((long long)gmtoff - h*3600) / 60;
12428 snprintf(buf, sz, "%c%02lld%02lld", sign, h, m);
12431 static const struct got_error *
12432 cat_commit(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
12434 const struct got_error *err;
12435 struct got_commit_object *commit;
12436 const struct got_object_id_queue *parent_ids;
12437 struct got_object_qid *pid;
12438 char *id_str = NULL;
12439 const char *logmsg = NULL;
12442 err = got_object_open_as_commit(&commit, repo, id);
12446 err = got_object_id_str(&id_str, got_object_commit_get_tree_id(commit));
12450 fprintf(outfile, "%s%s\n", GOT_COMMIT_LABEL_TREE, id_str);
12451 parent_ids = got_object_commit_get_parent_ids(commit);
12452 fprintf(outfile, "numparents %d\n",
12453 got_object_commit_get_nparents(commit));
12454 STAILQ_FOREACH(pid, parent_ids, entry) {
12456 err = got_object_id_str(&pid_str, &pid->id);
12459 fprintf(outfile, "%s%s\n", GOT_COMMIT_LABEL_PARENT, pid_str);
12462 format_gmtoff(gmtoff, sizeof(gmtoff),
12463 got_object_commit_get_author_gmtoff(commit));
12464 fprintf(outfile, "%s%s %lld %s\n", GOT_COMMIT_LABEL_AUTHOR,
12465 got_object_commit_get_author(commit),
12466 (long long)got_object_commit_get_author_time(commit),
12469 format_gmtoff(gmtoff, sizeof(gmtoff),
12470 got_object_commit_get_committer_gmtoff(commit));
12471 fprintf(outfile, "%s%s %lld %s\n", GOT_COMMIT_LABEL_COMMITTER,
12472 got_object_commit_get_author(commit),
12473 (long long)got_object_commit_get_committer_time(commit),
12476 logmsg = got_object_commit_get_logmsg_raw(commit);
12477 fprintf(outfile, "messagelen %zd\n", strlen(logmsg));
12478 fprintf(outfile, "%s", logmsg);
12481 got_object_commit_close(commit);
12485 static const struct got_error *
12486 cat_tag(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
12488 const struct got_error *err;
12489 struct got_tag_object *tag;
12490 char *id_str = NULL;
12491 const char *tagmsg = NULL;
12494 err = got_object_open_as_tag(&tag, repo, id);
12498 err = got_object_id_str(&id_str, got_object_tag_get_object_id(tag));
12502 fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_OBJECT, id_str);
12504 switch (got_object_tag_get_object_type(tag)) {
12505 case GOT_OBJ_TYPE_BLOB:
12506 fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
12507 GOT_OBJ_LABEL_BLOB);
12509 case GOT_OBJ_TYPE_TREE:
12510 fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
12511 GOT_OBJ_LABEL_TREE);
12513 case GOT_OBJ_TYPE_COMMIT:
12514 fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
12515 GOT_OBJ_LABEL_COMMIT);
12517 case GOT_OBJ_TYPE_TAG:
12518 fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
12519 GOT_OBJ_LABEL_TAG);
12525 fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TAG,
12526 got_object_tag_get_name(tag));
12528 format_gmtoff(gmtoff, sizeof(gmtoff),
12529 got_object_tag_get_tagger_gmtoff(tag));
12530 fprintf(outfile, "%s%s %lld %s\n", GOT_TAG_LABEL_TAGGER,
12531 got_object_tag_get_tagger(tag),
12532 (long long)got_object_tag_get_tagger_time(tag),
12535 tagmsg = got_object_tag_get_message(tag);
12536 fprintf(outfile, "messagelen %zd\n", strlen(tagmsg));
12537 fprintf(outfile, "%s", tagmsg);
12540 got_object_tag_close(tag);
12544 static const struct got_error *
12545 cmd_cat(int argc, char *argv[])
12547 const struct got_error *error;
12548 struct got_repository *repo = NULL;
12549 struct got_worktree *worktree = NULL;
12550 char *cwd = NULL, *repo_path = NULL, *label = NULL;
12551 const char *commit_id_str = NULL;
12552 struct got_object_id *id = NULL, *commit_id = NULL;
12553 struct got_commit_object *commit = NULL;
12554 int ch, obj_type, i, force_path = 0;
12555 struct got_reflist_head refs;
12556 int *pack_fds = NULL;
12561 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
12566 while ((ch = getopt(argc, argv, "c:r:P")) != -1) {
12569 commit_id_str = optarg;
12572 repo_path = realpath(optarg, NULL);
12573 if (repo_path == NULL)
12574 return got_error_from_errno2("realpath",
12576 got_path_strip_trailing_slashes(repo_path);
12590 cwd = getcwd(NULL, 0);
12592 error = got_error_from_errno("getcwd");
12596 error = got_repo_pack_fds_open(&pack_fds);
12600 if (repo_path == NULL) {
12601 error = got_worktree_open(&worktree, cwd);
12602 if (error && error->code != GOT_ERR_NOT_WORKTREE)
12605 repo_path = strdup(
12606 got_worktree_get_repo_path(worktree));
12607 if (repo_path == NULL) {
12608 error = got_error_from_errno("strdup");
12612 /* Release work tree lock. */
12613 got_worktree_close(worktree);
12618 if (repo_path == NULL) {
12619 repo_path = strdup(cwd);
12620 if (repo_path == NULL)
12621 return got_error_from_errno("strdup");
12624 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
12629 error = apply_unveil(got_repo_get_path(repo), 1, NULL);
12633 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
12637 if (commit_id_str == NULL)
12638 commit_id_str = GOT_REF_HEAD;
12639 error = got_repo_match_object_id(&commit_id, NULL,
12640 commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
12644 error = got_object_open_as_commit(&commit, repo, commit_id);
12648 for (i = 0; i < argc; i++) {
12650 error = got_object_id_by_path(&id, repo, commit,
12655 error = got_repo_match_object_id(&id, &label, argv[i],
12656 GOT_OBJ_TYPE_ANY, NULL /* do not resolve tags */,
12659 if (error->code != GOT_ERR_BAD_OBJ_ID_STR &&
12660 error->code != GOT_ERR_NOT_REF)
12662 error = got_object_id_by_path(&id, repo,
12669 error = got_object_get_type(&obj_type, repo, id);
12673 switch (obj_type) {
12674 case GOT_OBJ_TYPE_BLOB:
12675 error = cat_blob(id, repo, stdout);
12677 case GOT_OBJ_TYPE_TREE:
12678 error = cat_tree(id, repo, stdout);
12680 case GOT_OBJ_TYPE_COMMIT:
12681 error = cat_commit(id, repo, stdout);
12683 case GOT_OBJ_TYPE_TAG:
12684 error = cat_tag(id, repo, stdout);
12687 error = got_error(GOT_ERR_OBJ_TYPE);
12702 got_object_commit_close(commit);
12704 got_worktree_close(worktree);
12706 const struct got_error *close_err = got_repo_close(repo);
12711 const struct got_error *pack_err =
12712 got_repo_pack_fds_close(pack_fds);
12717 got_ref_list_free(&refs);
12724 fprintf(stderr, "usage: %s info [path ...]\n",
12729 static const struct got_error *
12730 print_path_info(void *arg, const char *path, mode_t mode, time_t mtime,
12731 struct got_object_id *blob_id, struct got_object_id *staged_blob_id,
12732 struct got_object_id *commit_id)
12734 const struct got_error *err = NULL;
12735 char *id_str = NULL;
12737 struct tm mytm, *tm;
12738 struct got_pathlist_head *paths = arg;
12739 struct got_pathlist_entry *pe;
12742 * Clear error indication from any of the path arguments which
12743 * would cause this file index entry to be displayed.
12745 TAILQ_FOREACH(pe, paths, entry) {
12746 if (got_path_cmp(path, pe->path, strlen(path),
12747 pe->path_len) == 0 ||
12748 got_path_is_child(path, pe->path, pe->path_len))
12749 pe->data = NULL; /* no error */
12752 printf(GOT_COMMIT_SEP_STR);
12754 printf("symlink: %s\n", path);
12755 else if (S_ISREG(mode)) {
12756 printf("file: %s\n", path);
12757 printf("mode: %o\n", mode & (S_IRWXU | S_IRWXG | S_IRWXO));
12758 } else if (S_ISDIR(mode))
12759 printf("directory: %s\n", path);
12761 printf("something: %s\n", path);
12763 tm = localtime_r(&mtime, &mytm);
12766 if (strftime(datebuf, sizeof(datebuf), "%c %Z", tm) == 0)
12767 return got_error(GOT_ERR_NO_SPACE);
12768 printf("timestamp: %s\n", datebuf);
12771 err = got_object_id_str(&id_str, blob_id);
12774 printf("based on blob: %s\n", id_str);
12778 if (staged_blob_id) {
12779 err = got_object_id_str(&id_str, staged_blob_id);
12782 printf("based on staged blob: %s\n", id_str);
12787 err = got_object_id_str(&id_str, commit_id);
12790 printf("based on commit: %s\n", id_str);
12797 static const struct got_error *
12798 cmd_info(int argc, char *argv[])
12800 const struct got_error *error = NULL;
12801 struct got_worktree *worktree = NULL;
12802 char *cwd = NULL, *id_str = NULL;
12803 struct got_pathlist_head paths;
12804 struct got_pathlist_entry *pe;
12805 char *uuidstr = NULL;
12806 int ch, show_files = 0;
12807 int *pack_fds = NULL;
12809 TAILQ_INIT(&paths);
12811 while ((ch = getopt(argc, argv, "")) != -1) {
12823 if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
12827 cwd = getcwd(NULL, 0);
12829 error = got_error_from_errno("getcwd");
12833 error = got_repo_pack_fds_open(&pack_fds);
12837 error = got_worktree_open(&worktree, cwd);
12839 if (error->code == GOT_ERR_NOT_WORKTREE)
12840 error = wrap_not_worktree_error(error, "info", cwd);
12845 /* Remove "cpath" promise. */
12846 if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
12850 error = apply_unveil(NULL, 0, got_worktree_get_root_path(worktree));
12855 error = get_worktree_paths_from_argv(&paths, argc, argv,
12862 error = got_object_id_str(&id_str,
12863 got_worktree_get_base_commit_id(worktree));
12867 error = got_worktree_get_uuid(&uuidstr, worktree);
12871 printf("work tree: %s\n", got_worktree_get_root_path(worktree));
12872 printf("work tree base commit: %s\n", id_str);
12873 printf("work tree path prefix: %s\n",
12874 got_worktree_get_path_prefix(worktree));
12875 printf("work tree branch reference: %s\n",
12876 got_worktree_get_head_ref_name(worktree));
12877 printf("work tree UUID: %s\n", uuidstr);
12878 printf("repository: %s\n", got_worktree_get_repo_path(worktree));
12881 struct got_pathlist_entry *pe;
12882 TAILQ_FOREACH(pe, &paths, entry) {
12883 if (pe->path_len == 0)
12886 * Assume this path will fail. This will be corrected
12887 * in print_path_info() in case the path does suceeed.
12889 pe->data = (void *)got_error_path(pe->path,
12892 error = got_worktree_path_info(worktree, &paths,
12893 print_path_info, &paths, check_cancelled, NULL);
12896 TAILQ_FOREACH(pe, &paths, entry) {
12897 if (pe->data != NULL) {
12898 error = pe->data; /* bad path */
12905 const struct got_error *pack_err =
12906 got_repo_pack_fds_close(pack_fds);
12910 TAILQ_FOREACH(pe, &paths, entry)
12911 free((char *)pe->path);
12912 got_pathlist_free(&paths);