1 5c860e29 2018-03-12 stsp /*
   2 f42b1b34 2018-03-12 stsp  * Copyright (c) 2017 Martin Pieuchot <mpi@openbsd.org>
   3 f42b1b34 2018-03-12 stsp  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
   4 5c860e29 2018-03-12 stsp  *
   5 5c860e29 2018-03-12 stsp  * Permission to use, copy, modify, and distribute this software for any
   6 5c860e29 2018-03-12 stsp  * purpose with or without fee is hereby granted, provided that the above
   7 5c860e29 2018-03-12 stsp  * copyright notice and this permission notice appear in all copies.
   8 5c860e29 2018-03-12 stsp  *
   9 5c860e29 2018-03-12 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10 5c860e29 2018-03-12 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11 5c860e29 2018-03-12 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12 5c860e29 2018-03-12 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13 5c860e29 2018-03-12 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14 5c860e29 2018-03-12 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15 5c860e29 2018-03-12 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16 5c860e29 2018-03-12 stsp  */
  17 5c860e29 2018-03-12 stsp 
  18 f42b1b34 2018-03-12 stsp #include <sys/queue.h>
  19 64a96a6d 2018-04-01 stsp #include <sys/limits.h>
  20 c0768b0f 2018-06-10 stsp #include <sys/types.h>
  21 5de5890b 2018-10-18 stsp #include <sys/stat.h>
  22 f42b1b34 2018-03-12 stsp 
  23 5c860e29 2018-03-12 stsp #include <err.h>
  24 5c860e29 2018-03-12 stsp #include <errno.h>
  25 5c860e29 2018-03-12 stsp #include <locale.h>
  26 99437157 2018-11-11 stsp #include <signal.h>
  27 5c860e29 2018-03-12 stsp #include <stdio.h>
  28 5c860e29 2018-03-12 stsp #include <stdlib.h>
  29 5c860e29 2018-03-12 stsp #include <string.h>
  30 5c860e29 2018-03-12 stsp #include <unistd.h>
  31 c09a553d 2018-03-12 stsp #include <libgen.h>
  32 c0768b0f 2018-06-10 stsp #include <time.h>
  33 5c860e29 2018-03-12 stsp 
  34 f42b1b34 2018-03-12 stsp #include "got_error.h"
  35 f42b1b34 2018-03-12 stsp #include "got_object.h"
  36 5261c201 2018-04-01 stsp #include "got_reference.h"
  37 f42b1b34 2018-03-12 stsp #include "got_repository.h"
  38 c09a553d 2018-03-12 stsp #include "got_worktree.h"
  39 79109fed 2018-03-27 stsp #include "got_diff.h"
  40 372ccdbb 2018-06-10 stsp #include "got_commit_graph.h"
  41 404c43c4 2018-06-21 stsp #include "got_blame.h"
  42 63219cd2 2019-01-04 stsp #include "got_privsep.h"
  43 5c860e29 2018-03-12 stsp 
  44 5c860e29 2018-03-12 stsp #ifndef nitems
  45 5c860e29 2018-03-12 stsp #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
  46 5c860e29 2018-03-12 stsp #endif
  47 99437157 2018-11-11 stsp 
  48 99437157 2018-11-11 stsp static volatile sig_atomic_t sigint_received;
  49 99437157 2018-11-11 stsp static volatile sig_atomic_t sigpipe_received;
  50 99437157 2018-11-11 stsp 
  51 99437157 2018-11-11 stsp static void
  52 99437157 2018-11-11 stsp catch_sigint(int signo)
  53 99437157 2018-11-11 stsp {
  54 99437157 2018-11-11 stsp 	sigint_received = 1;
  55 99437157 2018-11-11 stsp }
  56 99437157 2018-11-11 stsp 
  57 99437157 2018-11-11 stsp static void
  58 99437157 2018-11-11 stsp catch_sigpipe(int signo)
  59 99437157 2018-11-11 stsp {
  60 99437157 2018-11-11 stsp 	sigpipe_received = 1;
  61 99437157 2018-11-11 stsp }
  62 5c860e29 2018-03-12 stsp 
  63 99437157 2018-11-11 stsp 
  64 5c860e29 2018-03-12 stsp struct cmd {
  65 5c860e29 2018-03-12 stsp 	const char	 *cmd_name;
  66 d7d4f210 2018-03-12 stsp 	const struct got_error *(*cmd_main)(int, char *[]);
  67 1b6b95a8 2018-03-12 stsp 	void		(*cmd_usage)(void);
  68 46a0db7d 2018-03-12 stsp 	const char	 *cmd_descr;
  69 5c860e29 2018-03-12 stsp };
  70 5c860e29 2018-03-12 stsp 
  71 4ed7e80c 2018-05-20 stsp __dead static void	usage(void);
  72 4ed7e80c 2018-05-20 stsp __dead static void	usage_checkout(void);
  73 507dc3bb 2018-12-29 stsp __dead static void	usage_update(void);
  74 4ed7e80c 2018-05-20 stsp __dead static void	usage_log(void);
  75 4ed7e80c 2018-05-20 stsp __dead static void	usage_diff(void);
  76 404c43c4 2018-06-21 stsp __dead static void	usage_blame(void);
  77 5de5890b 2018-10-18 stsp __dead static void	usage_tree(void);
  78 5c860e29 2018-03-12 stsp 
  79 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_checkout(int, char *[]);
  80 507dc3bb 2018-12-29 stsp static const struct got_error*		cmd_update(int, char *[]);
  81 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_log(int, char *[]);
  82 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_diff(int, char *[]);
  83 404c43c4 2018-06-21 stsp static const struct got_error*		cmd_blame(int, char *[]);
  84 5de5890b 2018-10-18 stsp static const struct got_error*		cmd_tree(int, char *[]);
  85 4ed7e80c 2018-05-20 stsp #ifdef notyet
  86 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_status(int, char *[]);
  87 4ed7e80c 2018-05-20 stsp #endif
  88 5c860e29 2018-03-12 stsp 
  89 4ed7e80c 2018-05-20 stsp static struct cmd got_commands[] = {
  90 c09a553d 2018-03-12 stsp 	{ "checkout",	cmd_checkout,	usage_checkout,
  91 0bb8a95e 2018-03-12 stsp 	    "check out a new work tree from a repository" },
  92 507dc3bb 2018-12-29 stsp 	{ "update",	cmd_update,	usage_update,
  93 507dc3bb 2018-12-29 stsp 	    "update a work tree to a different commit" },
  94 1b6b95a8 2018-03-12 stsp 	{ "log",	cmd_log,	usage_log,
  95 1b6b95a8 2018-03-12 stsp 	    "show repository history" },
  96 b00d56cd 2018-04-01 stsp 	{ "diff",	cmd_diff,	usage_diff,
  97 b00d56cd 2018-04-01 stsp 	    "compare files and directories" },
  98 404c43c4 2018-06-21 stsp 	{ "blame",	cmd_blame,	usage_blame,
  99 404c43c4 2018-06-21 stsp 	    " show when lines in a file were changed" },
 100 5de5890b 2018-10-18 stsp 	{ "tree",	cmd_tree,	usage_tree,
 101 5de5890b 2018-10-18 stsp 	    " list files and directories in repository" },
 102 f42b1b34 2018-03-12 stsp #ifdef notyet
 103 1b6b95a8 2018-03-12 stsp 	{ "status",	cmd_status,	usage_status,
 104 1b6b95a8 2018-03-12 stsp 	    "show modification status of files" },
 105 f42b1b34 2018-03-12 stsp #endif
 106 5c860e29 2018-03-12 stsp };
 107 5c860e29 2018-03-12 stsp 
 108 5c860e29 2018-03-12 stsp int
 109 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
 110 5c860e29 2018-03-12 stsp {
 111 5c860e29 2018-03-12 stsp 	struct cmd *cmd;
 112 5c860e29 2018-03-12 stsp 	unsigned int i;
 113 5c860e29 2018-03-12 stsp 	int ch;
 114 1b6b95a8 2018-03-12 stsp 	int hflag = 0;
 115 5c860e29 2018-03-12 stsp 
 116 5c860e29 2018-03-12 stsp 	setlocale(LC_ALL, "");
 117 5c860e29 2018-03-12 stsp 
 118 1b6b95a8 2018-03-12 stsp 	while ((ch = getopt(argc, argv, "h")) != -1) {
 119 5c860e29 2018-03-12 stsp 		switch (ch) {
 120 1b6b95a8 2018-03-12 stsp 		case 'h':
 121 1b6b95a8 2018-03-12 stsp 			hflag = 1;
 122 1b6b95a8 2018-03-12 stsp 			break;
 123 5c860e29 2018-03-12 stsp 		default:
 124 5c860e29 2018-03-12 stsp 			usage();
 125 5c860e29 2018-03-12 stsp 			/* NOTREACHED */
 126 5c860e29 2018-03-12 stsp 		}
 127 5c860e29 2018-03-12 stsp 	}
 128 5c860e29 2018-03-12 stsp 
 129 5c860e29 2018-03-12 stsp 	argc -= optind;
 130 5c860e29 2018-03-12 stsp 	argv += optind;
 131 1e70621d 2018-03-27 stsp 	optind = 0;
 132 5c860e29 2018-03-12 stsp 
 133 5c860e29 2018-03-12 stsp 	if (argc <= 0)
 134 5c860e29 2018-03-12 stsp 		usage();
 135 5c860e29 2018-03-12 stsp 
 136 99437157 2018-11-11 stsp 	signal(SIGINT, catch_sigint);
 137 99437157 2018-11-11 stsp 	signal(SIGPIPE, catch_sigpipe);
 138 99437157 2018-11-11 stsp 
 139 5c860e29 2018-03-12 stsp 	for (i = 0; i < nitems(got_commands); i++) {
 140 d7d4f210 2018-03-12 stsp 		const struct got_error *error;
 141 d7d4f210 2018-03-12 stsp 
 142 5c860e29 2018-03-12 stsp 		cmd = &got_commands[i];
 143 5c860e29 2018-03-12 stsp 
 144 5c860e29 2018-03-12 stsp 		if (strncmp(cmd->cmd_name, argv[0], strlen(argv[0])))
 145 5c860e29 2018-03-12 stsp 			continue;
 146 5c860e29 2018-03-12 stsp 
 147 1b6b95a8 2018-03-12 stsp 		if (hflag)
 148 1b6b95a8 2018-03-12 stsp 			got_commands[i].cmd_usage();
 149 1b6b95a8 2018-03-12 stsp 
 150 d7d4f210 2018-03-12 stsp 		error = got_commands[i].cmd_main(argc, argv);
 151 80d5f134 2018-11-11 stsp 		if (error && !(sigint_received || sigpipe_received)) {
 152 d7d4f210 2018-03-12 stsp 			fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
 153 d7d4f210 2018-03-12 stsp 			return 1;
 154 d7d4f210 2018-03-12 stsp 		}
 155 d7d4f210 2018-03-12 stsp 
 156 d7d4f210 2018-03-12 stsp 		return 0;
 157 5c860e29 2018-03-12 stsp 	}
 158 5c860e29 2018-03-12 stsp 
 159 20ecf764 2018-03-12 stsp 	fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
 160 5c860e29 2018-03-12 stsp 	return 1;
 161 5c860e29 2018-03-12 stsp }
 162 5c860e29 2018-03-12 stsp 
 163 4ed7e80c 2018-05-20 stsp __dead static void
 164 5c860e29 2018-03-12 stsp usage(void)
 165 5c860e29 2018-03-12 stsp {
 166 46a0db7d 2018-03-12 stsp 	int i;
 167 46a0db7d 2018-03-12 stsp 
 168 987e94ba 2018-03-12 stsp 	fprintf(stderr, "usage: %s [-h] command [arg ...]\n\n"
 169 987e94ba 2018-03-12 stsp 	    "Available commands:\n", getprogname());
 170 46a0db7d 2018-03-12 stsp 	for (i = 0; i < nitems(got_commands); i++) {
 171 46a0db7d 2018-03-12 stsp 		struct cmd *cmd = &got_commands[i];
 172 46a0db7d 2018-03-12 stsp 		fprintf(stderr, "    %s: %s\n", cmd->cmd_name, cmd->cmd_descr);
 173 46a0db7d 2018-03-12 stsp 	}
 174 5c860e29 2018-03-12 stsp 	exit(1);
 175 5c860e29 2018-03-12 stsp }
 176 5c860e29 2018-03-12 stsp 
 177 0266afb7 2019-01-04 stsp static const struct got_error *
 178 0266afb7 2019-01-04 stsp apply_unveil(const char *repo_path, const char *worktree_path)
 179 0266afb7 2019-01-04 stsp {
 180 0266afb7 2019-01-04 stsp 	const struct got_error *error;
 181 0266afb7 2019-01-04 stsp 
 182 0266afb7 2019-01-04 stsp 	if (repo_path && unveil(repo_path, "r") != 0)
 183 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 184 0266afb7 2019-01-04 stsp 
 185 0266afb7 2019-01-04 stsp 	if (worktree_path && unveil(worktree_path, "rwc") != 0)
 186 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 187 0266afb7 2019-01-04 stsp 
 188 f12d0dbe 2019-01-04 stsp 	if (unveil("/tmp", "rwc") != 0)
 189 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 190 0266afb7 2019-01-04 stsp 
 191 0266afb7 2019-01-04 stsp 	error = got_privsep_unveil_exec_helpers();
 192 0266afb7 2019-01-04 stsp 	if (error != NULL)
 193 0266afb7 2019-01-04 stsp 		return error;
 194 0266afb7 2019-01-04 stsp 
 195 0266afb7 2019-01-04 stsp 	if (unveil(NULL, NULL) != 0)
 196 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 197 0266afb7 2019-01-04 stsp 
 198 0266afb7 2019-01-04 stsp 	return NULL;
 199 0266afb7 2019-01-04 stsp }
 200 0266afb7 2019-01-04 stsp 
 201 4ed7e80c 2018-05-20 stsp __dead static void
 202 c09a553d 2018-03-12 stsp usage_checkout(void)
 203 c09a553d 2018-03-12 stsp {
 204 0bb8a95e 2018-03-12 stsp 	fprintf(stderr, "usage: %s checkout [-p prefix] repository-path "
 205 0bb8a95e 2018-03-12 stsp 	    "[worktree-path]\n", getprogname());
 206 c09a553d 2018-03-12 stsp 	exit(1);
 207 92a684f4 2018-03-12 stsp }
 208 92a684f4 2018-03-12 stsp 
 209 92a684f4 2018-03-12 stsp static void
 210 a0eb853d 2018-12-29 stsp checkout_progress(void *arg, unsigned char status, const char *path)
 211 92a684f4 2018-03-12 stsp {
 212 92a684f4 2018-03-12 stsp 	char *worktree_path = arg;
 213 92a684f4 2018-03-12 stsp 
 214 92a684f4 2018-03-12 stsp 	while (path[0] == '/')
 215 92a684f4 2018-03-12 stsp 		path++;
 216 92a684f4 2018-03-12 stsp 
 217 d7b62c98 2018-12-27 stsp 	printf("%c  %s/%s\n", status, worktree_path, path);
 218 99437157 2018-11-11 stsp }
 219 99437157 2018-11-11 stsp 
 220 99437157 2018-11-11 stsp static const struct got_error *
 221 99437157 2018-11-11 stsp checkout_cancel(void *arg)
 222 99437157 2018-11-11 stsp {
 223 99437157 2018-11-11 stsp 	if (sigint_received || sigpipe_received)
 224 99437157 2018-11-11 stsp 		return got_error(GOT_ERR_CANCELLED);
 225 99437157 2018-11-11 stsp 	return NULL;
 226 c09a553d 2018-03-12 stsp }
 227 c09a553d 2018-03-12 stsp 
 228 4ed7e80c 2018-05-20 stsp static const struct got_error *
 229 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
 230 c09a553d 2018-03-12 stsp {
 231 c09a553d 2018-03-12 stsp 	const struct got_error *error = NULL;
 232 c09a553d 2018-03-12 stsp 	struct got_repository *repo = NULL;
 233 c09a553d 2018-03-12 stsp 	struct got_reference *head_ref = NULL;
 234 c09a553d 2018-03-12 stsp 	struct got_worktree *worktree = NULL;
 235 c09a553d 2018-03-12 stsp 	char *repo_path = NULL;
 236 c09a553d 2018-03-12 stsp 	char *worktree_path = NULL;
 237 0bb8a95e 2018-03-12 stsp 	const char *path_prefix = "";
 238 e5dc7198 2018-12-29 stsp 	int ch, same_path_prefix;
 239 c09a553d 2018-03-12 stsp 
 240 0bb8a95e 2018-03-12 stsp 	while ((ch = getopt(argc, argv, "p:")) != -1) {
 241 0bb8a95e 2018-03-12 stsp 		switch (ch) {
 242 0bb8a95e 2018-03-12 stsp 		case 'p':
 243 0bb8a95e 2018-03-12 stsp 			path_prefix = optarg;
 244 0bb8a95e 2018-03-12 stsp 			break;
 245 0bb8a95e 2018-03-12 stsp 		default:
 246 0bb8a95e 2018-03-12 stsp 			usage();
 247 0bb8a95e 2018-03-12 stsp 			/* NOTREACHED */
 248 0bb8a95e 2018-03-12 stsp 		}
 249 0bb8a95e 2018-03-12 stsp 	}
 250 0bb8a95e 2018-03-12 stsp 
 251 0bb8a95e 2018-03-12 stsp 	argc -= optind;
 252 0bb8a95e 2018-03-12 stsp 	argv += optind;
 253 0bb8a95e 2018-03-12 stsp 
 254 6715a751 2018-03-16 stsp #ifndef PROFILE
 255 63219cd2 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
 256 63219cd2 2019-01-04 stsp 	    NULL) == -1)
 257 c09a553d 2018-03-12 stsp 		err(1, "pledge");
 258 6715a751 2018-03-16 stsp #endif
 259 0bb8a95e 2018-03-12 stsp 	if (argc == 1) {
 260 c09a553d 2018-03-12 stsp 		char *cwd, *base, *dotgit;
 261 76089277 2018-04-01 stsp 		repo_path = realpath(argv[0], NULL);
 262 76089277 2018-04-01 stsp 		if (repo_path == NULL)
 263 76089277 2018-04-01 stsp 			return got_error_from_errno();
 264 c09a553d 2018-03-12 stsp 		cwd = getcwd(NULL, 0);
 265 76089277 2018-04-01 stsp 		if (cwd == NULL) {
 266 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 267 76089277 2018-04-01 stsp 			goto done;
 268 76089277 2018-04-01 stsp 		}
 269 5d7c1dab 2018-04-01 stsp 		if (path_prefix[0])
 270 5d7c1dab 2018-04-01 stsp 			base = basename(path_prefix);
 271 5d7c1dab 2018-04-01 stsp 		else
 272 5d7c1dab 2018-04-01 stsp 			base = basename(repo_path);
 273 76089277 2018-04-01 stsp 		if (base == NULL) {
 274 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 275 76089277 2018-04-01 stsp 			goto done;
 276 76089277 2018-04-01 stsp 		}
 277 c09a553d 2018-03-12 stsp 		dotgit = strstr(base, ".git");
 278 c09a553d 2018-03-12 stsp 		if (dotgit)
 279 c09a553d 2018-03-12 stsp 			*dotgit = '\0';
 280 c09a553d 2018-03-12 stsp 		if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
 281 0a585a0d 2018-03-17 stsp 			error = got_error_from_errno();
 282 c09a553d 2018-03-12 stsp 			free(cwd);
 283 76089277 2018-04-01 stsp 			goto done;
 284 c09a553d 2018-03-12 stsp 		}
 285 c09a553d 2018-03-12 stsp 		free(cwd);
 286 0bb8a95e 2018-03-12 stsp 	} else if (argc == 2) {
 287 76089277 2018-04-01 stsp 		repo_path = realpath(argv[0], NULL);
 288 76089277 2018-04-01 stsp 		if (repo_path == NULL) {
 289 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 290 76089277 2018-04-01 stsp 			goto done;
 291 76089277 2018-04-01 stsp 		}
 292 f7b38925 2018-04-01 stsp 		worktree_path = realpath(argv[1], NULL);
 293 76089277 2018-04-01 stsp 		if (worktree_path == NULL) {
 294 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 295 76089277 2018-04-01 stsp 			goto done;
 296 76089277 2018-04-01 stsp 		}
 297 c09a553d 2018-03-12 stsp 	} else
 298 c09a553d 2018-03-12 stsp 		usage_checkout();
 299 63219cd2 2019-01-04 stsp 
 300 0266afb7 2019-01-04 stsp 	error = apply_unveil(repo_path, worktree_path);
 301 0266afb7 2019-01-04 stsp 	if (error)
 302 63219cd2 2019-01-04 stsp 		goto done;
 303 c09a553d 2018-03-12 stsp 
 304 c09a553d 2018-03-12 stsp 	error = got_repo_open(&repo, repo_path);
 305 c09a553d 2018-03-12 stsp 	if (error != NULL)
 306 c09a553d 2018-03-12 stsp 		goto done;
 307 c09a553d 2018-03-12 stsp 	error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
 308 c09a553d 2018-03-12 stsp 	if (error != NULL)
 309 c09a553d 2018-03-12 stsp 		goto done;
 310 c09a553d 2018-03-12 stsp 
 311 0bb8a95e 2018-03-12 stsp 	error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
 312 d70b8e30 2018-12-27 stsp 	if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
 313 c09a553d 2018-03-12 stsp 		goto done;
 314 c09a553d 2018-03-12 stsp 
 315 c09a553d 2018-03-12 stsp 	error = got_worktree_open(&worktree, worktree_path);
 316 c09a553d 2018-03-12 stsp 	if (error != NULL)
 317 c09a553d 2018-03-12 stsp 		goto done;
 318 c09a553d 2018-03-12 stsp 
 319 e5dc7198 2018-12-29 stsp 	error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
 320 e5dc7198 2018-12-29 stsp 	    path_prefix);
 321 e5dc7198 2018-12-29 stsp 	if (error != NULL)
 322 e5dc7198 2018-12-29 stsp 		goto done;
 323 e5dc7198 2018-12-29 stsp 	if (!same_path_prefix) {
 324 49520a32 2018-12-29 stsp 		error = got_error(GOT_ERR_PATH_PREFIX);
 325 49520a32 2018-12-29 stsp 		goto done;
 326 49520a32 2018-12-29 stsp 	}
 327 49520a32 2018-12-29 stsp 
 328 93a30277 2018-12-24 stsp 	error = got_worktree_checkout_files(worktree, repo,
 329 99437157 2018-11-11 stsp 	    checkout_progress, worktree_path, checkout_cancel, NULL);
 330 c09a553d 2018-03-12 stsp 	if (error != NULL)
 331 c09a553d 2018-03-12 stsp 		goto done;
 332 c09a553d 2018-03-12 stsp 
 333 b65ae19a 2018-04-24 stsp 	printf("Now shut up and hack\n");
 334 c09a553d 2018-03-12 stsp 
 335 c09a553d 2018-03-12 stsp done:
 336 76089277 2018-04-01 stsp 	free(repo_path);
 337 507dc3bb 2018-12-29 stsp 	free(worktree_path);
 338 507dc3bb 2018-12-29 stsp 	return error;
 339 507dc3bb 2018-12-29 stsp }
 340 507dc3bb 2018-12-29 stsp 
 341 507dc3bb 2018-12-29 stsp __dead static void
 342 507dc3bb 2018-12-29 stsp usage_update(void)
 343 507dc3bb 2018-12-29 stsp {
 344 507dc3bb 2018-12-29 stsp 	fprintf(stderr, "usage: %s update [-c commit] [worktree-path]\n",
 345 507dc3bb 2018-12-29 stsp 	    getprogname());
 346 507dc3bb 2018-12-29 stsp 	exit(1);
 347 507dc3bb 2018-12-29 stsp }
 348 507dc3bb 2018-12-29 stsp 
 349 507dc3bb 2018-12-29 stsp static void
 350 507dc3bb 2018-12-29 stsp update_progress(void *arg, unsigned char status, const char *path)
 351 507dc3bb 2018-12-29 stsp {
 352 507dc3bb 2018-12-29 stsp 	if (status == GOT_STATUS_EXISTS)
 353 507dc3bb 2018-12-29 stsp 		return;
 354 507dc3bb 2018-12-29 stsp 
 355 507dc3bb 2018-12-29 stsp 	while (path[0] == '/')
 356 507dc3bb 2018-12-29 stsp 		path++;
 357 507dc3bb 2018-12-29 stsp 	printf("%c  %s\n", status, path);
 358 507dc3bb 2018-12-29 stsp }
 359 507dc3bb 2018-12-29 stsp 
 360 507dc3bb 2018-12-29 stsp static const struct got_error *
 361 be7061eb 2018-12-30 stsp check_ancestry(struct got_worktree *worktree, struct got_object_id *commit_id,
 362 be7061eb 2018-12-30 stsp     struct got_repository *repo)
 363 be7061eb 2018-12-30 stsp {
 364 be7061eb 2018-12-30 stsp 	const struct got_error *err;
 365 be7061eb 2018-12-30 stsp 	struct got_reference *head_ref = NULL;
 366 be7061eb 2018-12-30 stsp 	struct got_object_id *head_commit_id = NULL;
 367 be7061eb 2018-12-30 stsp 	struct got_commit_graph *graph = NULL;
 368 be7061eb 2018-12-30 stsp 
 369 be7061eb 2018-12-30 stsp 	head_ref = got_worktree_get_head_ref(worktree);
 370 be7061eb 2018-12-30 stsp 	if (head_ref == NULL)
 371 be7061eb 2018-12-30 stsp 		return got_error_from_errno();
 372 be7061eb 2018-12-30 stsp 
 373 2944241d 2018-12-30 stsp 	/* TODO: Check the reflog. The head ref may have been rebased. */
 374 be7061eb 2018-12-30 stsp 	err = got_ref_resolve(&head_commit_id, repo, head_ref);
 375 be7061eb 2018-12-30 stsp 	if (err)
 376 be7061eb 2018-12-30 stsp 		goto done;
 377 be7061eb 2018-12-30 stsp 
 378 be7061eb 2018-12-30 stsp 	err = got_commit_graph_open(&graph, head_commit_id, "/", 1, repo);
 379 be7061eb 2018-12-30 stsp 	if (err)
 380 be7061eb 2018-12-30 stsp 		goto done;
 381 be7061eb 2018-12-30 stsp 
 382 be7061eb 2018-12-30 stsp 	err = got_commit_graph_iter_start(graph, head_commit_id, repo);
 383 be7061eb 2018-12-30 stsp 	if (err)
 384 be7061eb 2018-12-30 stsp 		goto done;
 385 be7061eb 2018-12-30 stsp 	while (1) {
 386 be7061eb 2018-12-30 stsp 		struct got_object_id *id;
 387 be7061eb 2018-12-30 stsp 
 388 be7061eb 2018-12-30 stsp 		if (sigint_received || sigpipe_received)
 389 be7061eb 2018-12-30 stsp 			break;
 390 be7061eb 2018-12-30 stsp 
 391 be7061eb 2018-12-30 stsp 		err = got_commit_graph_iter_next(&id, graph);
 392 be7061eb 2018-12-30 stsp 		if (err) {
 393 be7061eb 2018-12-30 stsp 			if (err->code == GOT_ERR_ITER_COMPLETED) {
 394 be7061eb 2018-12-30 stsp 				err = got_error(GOT_ERR_ANCESTRY);
 395 be7061eb 2018-12-30 stsp 				break;
 396 be7061eb 2018-12-30 stsp 			}
 397 be7061eb 2018-12-30 stsp 			if (err->code != GOT_ERR_ITER_NEED_MORE)
 398 be7061eb 2018-12-30 stsp 				break;
 399 be7061eb 2018-12-30 stsp 			err = got_commit_graph_fetch_commits(graph, 1, repo);
 400 be7061eb 2018-12-30 stsp 			if (err)
 401 be7061eb 2018-12-30 stsp 				break;
 402 be7061eb 2018-12-30 stsp 			else
 403 be7061eb 2018-12-30 stsp 				continue;
 404 be7061eb 2018-12-30 stsp 		}
 405 be7061eb 2018-12-30 stsp 		if (id == NULL)
 406 be7061eb 2018-12-30 stsp 			break;
 407 be7061eb 2018-12-30 stsp 		if (got_object_id_cmp(id, commit_id) == 0)
 408 be7061eb 2018-12-30 stsp 			break;
 409 be7061eb 2018-12-30 stsp 	}
 410 be7061eb 2018-12-30 stsp done:
 411 be7061eb 2018-12-30 stsp 	if (head_ref)
 412 be7061eb 2018-12-30 stsp 		got_ref_close(head_ref);
 413 be7061eb 2018-12-30 stsp 	if (graph)
 414 be7061eb 2018-12-30 stsp 		got_commit_graph_close(graph);
 415 be7061eb 2018-12-30 stsp 	return err;
 416 be7061eb 2018-12-30 stsp }
 417 be7061eb 2018-12-30 stsp 
 418 be7061eb 2018-12-30 stsp static const struct got_error *
 419 507dc3bb 2018-12-29 stsp cmd_update(int argc, char *argv[])
 420 507dc3bb 2018-12-29 stsp {
 421 507dc3bb 2018-12-29 stsp 	const struct got_error *error = NULL;
 422 507dc3bb 2018-12-29 stsp 	struct got_repository *repo = NULL;
 423 507dc3bb 2018-12-29 stsp 	struct got_worktree *worktree = NULL;
 424 507dc3bb 2018-12-29 stsp 	char *worktree_path = NULL;
 425 507dc3bb 2018-12-29 stsp 	struct got_object_id *commit_id = NULL;
 426 9c4b8182 2019-01-02 stsp 	char *commit_id_str = NULL;
 427 507dc3bb 2018-12-29 stsp 	int ch;
 428 507dc3bb 2018-12-29 stsp 
 429 507dc3bb 2018-12-29 stsp 	while ((ch = getopt(argc, argv, "c:")) != -1) {
 430 507dc3bb 2018-12-29 stsp 		switch (ch) {
 431 507dc3bb 2018-12-29 stsp 		case 'c':
 432 9c4b8182 2019-01-02 stsp 			commit_id_str = strdup(optarg);
 433 9c4b8182 2019-01-02 stsp 			if (commit_id_str == NULL)
 434 9c4b8182 2019-01-02 stsp 				return got_error_from_errno();
 435 507dc3bb 2018-12-29 stsp 			break;
 436 507dc3bb 2018-12-29 stsp 		default:
 437 507dc3bb 2018-12-29 stsp 			usage();
 438 507dc3bb 2018-12-29 stsp 			/* NOTREACHED */
 439 507dc3bb 2018-12-29 stsp 		}
 440 507dc3bb 2018-12-29 stsp 	}
 441 507dc3bb 2018-12-29 stsp 
 442 507dc3bb 2018-12-29 stsp 	argc -= optind;
 443 507dc3bb 2018-12-29 stsp 	argv += optind;
 444 507dc3bb 2018-12-29 stsp 
 445 507dc3bb 2018-12-29 stsp #ifndef PROFILE
 446 0266afb7 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
 447 0266afb7 2019-01-04 stsp 	    NULL) == -1)
 448 507dc3bb 2018-12-29 stsp 		err(1, "pledge");
 449 507dc3bb 2018-12-29 stsp #endif
 450 507dc3bb 2018-12-29 stsp 	if (argc == 0) {
 451 507dc3bb 2018-12-29 stsp 		worktree_path = getcwd(NULL, 0);
 452 507dc3bb 2018-12-29 stsp 		if (worktree_path == NULL) {
 453 507dc3bb 2018-12-29 stsp 			error = got_error_from_errno();
 454 507dc3bb 2018-12-29 stsp 			goto done;
 455 507dc3bb 2018-12-29 stsp 		}
 456 507dc3bb 2018-12-29 stsp 	} else if (argc == 1) {
 457 507dc3bb 2018-12-29 stsp 		worktree_path = realpath(argv[0], NULL);
 458 507dc3bb 2018-12-29 stsp 		if (worktree_path == NULL) {
 459 507dc3bb 2018-12-29 stsp 			error = got_error_from_errno();
 460 507dc3bb 2018-12-29 stsp 			goto done;
 461 507dc3bb 2018-12-29 stsp 		}
 462 507dc3bb 2018-12-29 stsp 	} else
 463 507dc3bb 2018-12-29 stsp 		usage_update();
 464 507dc3bb 2018-12-29 stsp 
 465 507dc3bb 2018-12-29 stsp 	error = got_worktree_open(&worktree, worktree_path);
 466 507dc3bb 2018-12-29 stsp 	if (error != NULL)
 467 507dc3bb 2018-12-29 stsp 		goto done;
 468 507dc3bb 2018-12-29 stsp 
 469 507dc3bb 2018-12-29 stsp 	error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
 470 507dc3bb 2018-12-29 stsp 	if (error != NULL)
 471 507dc3bb 2018-12-29 stsp 		goto done;
 472 507dc3bb 2018-12-29 stsp 
 473 7839bc15 2019-01-06 stsp 	error = apply_unveil(got_repo_get_path(repo), worktree_path);
 474 0266afb7 2019-01-04 stsp 	if (error)
 475 0266afb7 2019-01-04 stsp 		goto done;
 476 0266afb7 2019-01-04 stsp 
 477 507dc3bb 2018-12-29 stsp 	if (commit_id_str == NULL) {
 478 507dc3bb 2018-12-29 stsp 		struct got_reference *head_ref;
 479 507dc3bb 2018-12-29 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
 480 507dc3bb 2018-12-29 stsp 		if (error != NULL)
 481 507dc3bb 2018-12-29 stsp 			goto done;
 482 507dc3bb 2018-12-29 stsp 		error = got_ref_resolve(&commit_id, repo, head_ref);
 483 9c4b8182 2019-01-02 stsp 		if (error != NULL)
 484 9c4b8182 2019-01-02 stsp 			goto done;
 485 9c4b8182 2019-01-02 stsp 		error = got_object_id_str(&commit_id_str, commit_id);
 486 507dc3bb 2018-12-29 stsp 		if (error != NULL)
 487 507dc3bb 2018-12-29 stsp 			goto done;
 488 507dc3bb 2018-12-29 stsp 	} else {
 489 507dc3bb 2018-12-29 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
 490 507dc3bb 2018-12-29 stsp 		    commit_id_str);
 491 507dc3bb 2018-12-29 stsp 		if (error != NULL)
 492 507dc3bb 2018-12-29 stsp 			goto done;
 493 507dc3bb 2018-12-29 stsp 	}
 494 35c965b2 2018-12-29 stsp 
 495 be7061eb 2018-12-30 stsp 	error = check_ancestry(worktree, commit_id, repo);
 496 be7061eb 2018-12-30 stsp 	if (error != NULL)
 497 be7061eb 2018-12-30 stsp 		goto done;
 498 507dc3bb 2018-12-29 stsp 
 499 507dc3bb 2018-12-29 stsp 	if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
 500 507dc3bb 2018-12-29 stsp 	    commit_id) != 0) {
 501 507dc3bb 2018-12-29 stsp 		error = got_worktree_set_base_commit_id(worktree, repo,
 502 507dc3bb 2018-12-29 stsp 		    commit_id);
 503 507dc3bb 2018-12-29 stsp 		if (error)
 504 507dc3bb 2018-12-29 stsp 			goto done;
 505 507dc3bb 2018-12-29 stsp 	}
 506 507dc3bb 2018-12-29 stsp 
 507 507dc3bb 2018-12-29 stsp 	error = got_worktree_checkout_files(worktree, repo,
 508 507dc3bb 2018-12-29 stsp 	    update_progress, NULL, checkout_cancel, NULL);
 509 507dc3bb 2018-12-29 stsp 	if (error != NULL)
 510 507dc3bb 2018-12-29 stsp 		goto done;
 511 9c4b8182 2019-01-02 stsp 
 512 9c4b8182 2019-01-02 stsp 	printf("Updated to commit %s\n", commit_id_str);
 513 507dc3bb 2018-12-29 stsp done:
 514 c09a553d 2018-03-12 stsp 	free(worktree_path);
 515 507dc3bb 2018-12-29 stsp 	free(commit_id);
 516 9c4b8182 2019-01-02 stsp 	free(commit_id_str);
 517 c09a553d 2018-03-12 stsp 	return error;
 518 c09a553d 2018-03-12 stsp }
 519 c09a553d 2018-03-12 stsp 
 520 f42b1b34 2018-03-12 stsp static const struct got_error *
 521 79109fed 2018-03-27 stsp print_patch(struct got_commit_object *commit, struct got_object_id *id,
 522 c0cc5c62 2018-10-18 stsp     int diff_context, struct got_repository *repo)
 523 5c860e29 2018-03-12 stsp {
 524 f42b1b34 2018-03-12 stsp 	const struct got_error *err = NULL;
 525 79109fed 2018-03-27 stsp 	struct got_tree_object *tree1 = NULL, *tree2;
 526 79f35eb3 2018-06-11 stsp 	struct got_object_qid *qid;
 527 0f2b3dca 2018-12-22 stsp 	char *id_str1 = NULL, *id_str2;
 528 79109fed 2018-03-27 stsp 
 529 45d799e2 2018-12-23 stsp 	err = got_object_open_as_tree(&tree2, repo,
 530 45d799e2 2018-12-23 stsp 	    got_object_commit_get_tree_id(commit));
 531 79109fed 2018-03-27 stsp 	if (err)
 532 79109fed 2018-03-27 stsp 		return err;
 533 79109fed 2018-03-27 stsp 
 534 45d799e2 2018-12-23 stsp 	qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
 535 79f35eb3 2018-06-11 stsp 	if (qid != NULL) {
 536 79109fed 2018-03-27 stsp 		struct got_commit_object *pcommit;
 537 79109fed 2018-03-27 stsp 
 538 117e771c 2018-07-23 stsp 		err = got_object_open_as_commit(&pcommit, repo, qid->id);
 539 79109fed 2018-03-27 stsp 		if (err)
 540 79109fed 2018-03-27 stsp 			return err;
 541 79109fed 2018-03-27 stsp 
 542 45d799e2 2018-12-23 stsp 		err = got_object_open_as_tree(&tree1, repo,
 543 45d799e2 2018-12-23 stsp 		    got_object_commit_get_tree_id(pcommit));
 544 79109fed 2018-03-27 stsp 		got_object_commit_close(pcommit);
 545 0f2b3dca 2018-12-22 stsp 		if (err)
 546 0f2b3dca 2018-12-22 stsp 			return err;
 547 0f2b3dca 2018-12-22 stsp 
 548 0f2b3dca 2018-12-22 stsp 		err = got_object_id_str(&id_str1, qid->id);
 549 79109fed 2018-03-27 stsp 		if (err)
 550 79109fed 2018-03-27 stsp 			return err;
 551 79109fed 2018-03-27 stsp 	}
 552 79109fed 2018-03-27 stsp 
 553 0f2b3dca 2018-12-22 stsp 	err = got_object_id_str(&id_str2, id);
 554 0f2b3dca 2018-12-22 stsp 	if (err)
 555 0f2b3dca 2018-12-22 stsp 		goto done;
 556 0f2b3dca 2018-12-22 stsp 
 557 56765ebb 2018-12-23 stsp 	printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
 558 54156555 2018-12-24 stsp 	err = got_diff_tree(tree1, tree2, "", "", diff_context, repo, stdout);
 559 0f2b3dca 2018-12-22 stsp done:
 560 79109fed 2018-03-27 stsp 	if (tree1)
 561 79109fed 2018-03-27 stsp 		got_object_tree_close(tree1);
 562 79109fed 2018-03-27 stsp 	got_object_tree_close(tree2);
 563 0f2b3dca 2018-12-22 stsp 	free(id_str1);
 564 0f2b3dca 2018-12-22 stsp 	free(id_str2);
 565 79109fed 2018-03-27 stsp 	return err;
 566 79109fed 2018-03-27 stsp }
 567 79109fed 2018-03-27 stsp 
 568 4bb494d5 2018-06-16 stsp static char *
 569 6c281f94 2018-06-11 stsp get_datestr(time_t *time, char *datebuf)
 570 6c281f94 2018-06-11 stsp {
 571 6c281f94 2018-06-11 stsp 	char *p, *s = ctime_r(time, datebuf);
 572 6c281f94 2018-06-11 stsp 	p = strchr(s, '\n');
 573 6c281f94 2018-06-11 stsp 	if (p)
 574 6c281f94 2018-06-11 stsp 		*p = '\0';
 575 6c281f94 2018-06-11 stsp 	return s;
 576 6c281f94 2018-06-11 stsp }
 577 6c281f94 2018-06-11 stsp 
 578 79109fed 2018-03-27 stsp static const struct got_error *
 579 79109fed 2018-03-27 stsp print_commit(struct got_commit_object *commit, struct got_object_id *id,
 580 c0cc5c62 2018-10-18 stsp     struct got_repository *repo, int show_patch, int diff_context)
 581 79109fed 2018-03-27 stsp {
 582 79109fed 2018-03-27 stsp 	const struct got_error *err = NULL;
 583 621015ac 2018-07-23 stsp 	char *id_str, *datestr, *logmsg0, *logmsg, *line;
 584 6c281f94 2018-06-11 stsp 	char datebuf[26];
 585 45d799e2 2018-12-23 stsp 	time_t committer_time;
 586 45d799e2 2018-12-23 stsp 	const char *author, *committer;
 587 5c860e29 2018-03-12 stsp 
 588 832c249c 2018-06-10 stsp 	err = got_object_id_str(&id_str, id);
 589 8bf5b3c9 2018-03-17 stsp 	if (err)
 590 8bf5b3c9 2018-03-17 stsp 		return err;
 591 788c352e 2018-06-16 stsp 
 592 33d869be 2018-12-22 stsp 	printf("-----------------------------------------------\n");
 593 832c249c 2018-06-10 stsp 	printf("commit %s\n", id_str);
 594 832c249c 2018-06-10 stsp 	free(id_str);
 595 45d799e2 2018-12-23 stsp 	printf("from: %s\n", got_object_commit_get_author(commit));
 596 45d799e2 2018-12-23 stsp 	committer_time = got_object_commit_get_committer_time(commit);
 597 45d799e2 2018-12-23 stsp 	datestr = get_datestr(&committer_time, datebuf);
 598 dab5fe87 2018-09-14 stsp 	printf("date: %s UTC\n", datestr);
 599 45d799e2 2018-12-23 stsp 	author = got_object_commit_get_author(commit);
 600 45d799e2 2018-12-23 stsp 	committer = got_object_commit_get_committer(commit);
 601 45d799e2 2018-12-23 stsp 	if (strcmp(author, committer) != 0)
 602 45d799e2 2018-12-23 stsp 		printf("via: %s\n", committer);
 603 45d799e2 2018-12-23 stsp 	if (got_object_commit_get_nparents(commit) > 1) {
 604 45d799e2 2018-12-23 stsp 		const struct got_object_id_queue *parent_ids;
 605 79f35eb3 2018-06-11 stsp 		struct got_object_qid *qid;
 606 3fe1abad 2018-06-10 stsp 		int n = 1;
 607 45d799e2 2018-12-23 stsp 		parent_ids = got_object_commit_get_parent_ids(commit);
 608 45d799e2 2018-12-23 stsp 		SIMPLEQ_FOREACH(qid, parent_ids, entry) {
 609 79f35eb3 2018-06-11 stsp 			err = got_object_id_str(&id_str, qid->id);
 610 a0603db2 2018-06-10 stsp 			if (err)
 611 a0603db2 2018-06-10 stsp 				return err;
 612 3fe1abad 2018-06-10 stsp 			printf("parent %d: %s\n", n++, id_str);
 613 a0603db2 2018-06-10 stsp 			free(id_str);
 614 a0603db2 2018-06-10 stsp 		}
 615 a0603db2 2018-06-10 stsp 	}
 616 832c249c 2018-06-10 stsp 
 617 45d799e2 2018-12-23 stsp 	logmsg0 = strdup(got_object_commit_get_logmsg(commit));
 618 621015ac 2018-07-23 stsp 	if (logmsg0 == NULL)
 619 832c249c 2018-06-10 stsp 		return got_error_from_errno();
 620 8bf5b3c9 2018-03-17 stsp 
 621 621015ac 2018-07-23 stsp 	logmsg = logmsg0;
 622 832c249c 2018-06-10 stsp 	do {
 623 832c249c 2018-06-10 stsp 		line = strsep(&logmsg, "\n");
 624 832c249c 2018-06-10 stsp 		if (line)
 625 832c249c 2018-06-10 stsp 			printf(" %s\n", line);
 626 832c249c 2018-06-10 stsp 	} while (line);
 627 621015ac 2018-07-23 stsp 	free(logmsg0);
 628 832c249c 2018-06-10 stsp 
 629 971751ac 2018-03-27 stsp 	if (show_patch) {
 630 c0cc5c62 2018-10-18 stsp 		err = print_patch(commit, id, diff_context, repo);
 631 971751ac 2018-03-27 stsp 		if (err == 0)
 632 971751ac 2018-03-27 stsp 			printf("\n");
 633 971751ac 2018-03-27 stsp 	}
 634 07862c20 2018-09-15 stsp 
 635 d82de447 2018-09-15 stsp 	fflush(stdout);
 636 07862c20 2018-09-15 stsp 	return err;
 637 f42b1b34 2018-03-12 stsp }
 638 5c860e29 2018-03-12 stsp 
 639 f42b1b34 2018-03-12 stsp static const struct got_error *
 640 15a94983 2018-12-23 stsp print_commits(struct got_object_id *root_id, struct got_repository *repo,
 641 15a94983 2018-12-23 stsp     char *path, int show_patch, int diff_context, int limit,
 642 15a94983 2018-12-23 stsp     int first_parent_traversal)
 643 f42b1b34 2018-03-12 stsp {
 644 f42b1b34 2018-03-12 stsp 	const struct got_error *err;
 645 372ccdbb 2018-06-10 stsp 	struct got_commit_graph *graph;
 646 372ccdbb 2018-06-10 stsp 
 647 31cedeaf 2018-09-15 stsp 	err = got_commit_graph_open(&graph, root_id, path,
 648 31cedeaf 2018-09-15 stsp 	    first_parent_traversal, repo);
 649 f42b1b34 2018-03-12 stsp 	if (err)
 650 f42b1b34 2018-03-12 stsp 		return err;
 651 31cedeaf 2018-09-15 stsp 	err = got_commit_graph_iter_start(graph, root_id, repo);
 652 372ccdbb 2018-06-10 stsp 	if (err)
 653 fcc85cad 2018-07-22 stsp 		goto done;
 654 31cedeaf 2018-09-15 stsp 	while (1) {
 655 372ccdbb 2018-06-10 stsp 		struct got_commit_object *commit;
 656 372ccdbb 2018-06-10 stsp 		struct got_object_id *id;
 657 8bf5b3c9 2018-03-17 stsp 
 658 84453469 2018-11-11 stsp 		if (sigint_received || sigpipe_received)
 659 84453469 2018-11-11 stsp 			break;
 660 84453469 2018-11-11 stsp 
 661 b43fbaa0 2018-06-11 stsp 		err = got_commit_graph_iter_next(&id, graph);
 662 372ccdbb 2018-06-10 stsp 		if (err) {
 663 9ba79e04 2018-06-11 stsp 			if (err->code == GOT_ERR_ITER_COMPLETED) {
 664 9ba79e04 2018-06-11 stsp 				err = NULL;
 665 9ba79e04 2018-06-11 stsp 				break;
 666 9ba79e04 2018-06-11 stsp 			}
 667 372ccdbb 2018-06-10 stsp 			if (err->code != GOT_ERR_ITER_NEED_MORE)
 668 372ccdbb 2018-06-10 stsp 				break;
 669 31cedeaf 2018-09-15 stsp 			err = got_commit_graph_fetch_commits(graph, 1, repo);
 670 372ccdbb 2018-06-10 stsp 			if (err)
 671 372ccdbb 2018-06-10 stsp 				break;
 672 372ccdbb 2018-06-10 stsp 			else
 673 372ccdbb 2018-06-10 stsp 				continue;
 674 7e665116 2018-04-02 stsp 		}
 675 b43fbaa0 2018-06-11 stsp 		if (id == NULL)
 676 7e665116 2018-04-02 stsp 			break;
 677 b43fbaa0 2018-06-11 stsp 
 678 f8e900f3 2018-06-11 stsp 		err = got_object_open_as_commit(&commit, repo, id);
 679 b43fbaa0 2018-06-11 stsp 		if (err)
 680 fcc85cad 2018-07-22 stsp 			break;
 681 c0cc5c62 2018-10-18 stsp 		err = print_commit(commit, id, repo, show_patch, diff_context);
 682 b43fbaa0 2018-06-11 stsp 		got_object_commit_close(commit);
 683 372ccdbb 2018-06-10 stsp 		if (err || (limit && --limit == 0))
 684 7e665116 2018-04-02 stsp 			break;
 685 31cedeaf 2018-09-15 stsp 	}
 686 fcc85cad 2018-07-22 stsp done:
 687 372ccdbb 2018-06-10 stsp 	got_commit_graph_close(graph);
 688 f42b1b34 2018-03-12 stsp 	return err;
 689 f42b1b34 2018-03-12 stsp }
 690 5c860e29 2018-03-12 stsp 
 691 4ed7e80c 2018-05-20 stsp __dead static void
 692 6f3d1eb0 2018-03-12 stsp usage_log(void)
 693 6f3d1eb0 2018-03-12 stsp {
 694 c0cc5c62 2018-10-18 stsp 	fprintf(stderr, "usage: %s log [-c commit] [-C number] [-f] [ -l N ] [-p] "
 695 04ca23f4 2018-07-16 stsp 	    "[-r repository-path] [path]\n", getprogname());
 696 6f3d1eb0 2018-03-12 stsp 	exit(1);
 697 6f3d1eb0 2018-03-12 stsp }
 698 6f3d1eb0 2018-03-12 stsp 
 699 4ed7e80c 2018-05-20 stsp static const struct got_error *
 700 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
 701 f42b1b34 2018-03-12 stsp {
 702 f42b1b34 2018-03-12 stsp 	const struct got_error *error;
 703 04ca23f4 2018-07-16 stsp 	struct got_repository *repo = NULL;
 704 15a94983 2018-12-23 stsp 	struct got_commit_object *commit = NULL;
 705 3235492e 2018-04-01 stsp 	struct got_object_id *id = NULL;
 706 04ca23f4 2018-07-16 stsp 	char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
 707 d142fc45 2018-04-01 stsp 	char *start_commit = NULL;
 708 c0cc5c62 2018-10-18 stsp 	int diff_context = 3, ch;
 709 1fd6d7ea 2018-06-13 stsp 	int show_patch = 0, limit = 0, first_parent_traversal = 0;
 710 64a96a6d 2018-04-01 stsp 	const char *errstr;
 711 5c860e29 2018-03-12 stsp 
 712 6715a751 2018-03-16 stsp #ifndef PROFILE
 713 6098196c 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
 714 6098196c 2019-01-04 stsp 	    NULL)
 715 ad242220 2018-09-08 stsp 	    == -1)
 716 f42b1b34 2018-03-12 stsp 		err(1, "pledge");
 717 6715a751 2018-03-16 stsp #endif
 718 79109fed 2018-03-27 stsp 
 719 c0cc5c62 2018-10-18 stsp 	while ((ch = getopt(argc, argv, "pc:C:l:fr:")) != -1) {
 720 79109fed 2018-03-27 stsp 		switch (ch) {
 721 79109fed 2018-03-27 stsp 		case 'p':
 722 79109fed 2018-03-27 stsp 			show_patch = 1;
 723 d142fc45 2018-04-01 stsp 			break;
 724 d142fc45 2018-04-01 stsp 		case 'c':
 725 d142fc45 2018-04-01 stsp 			start_commit = optarg;
 726 64a96a6d 2018-04-01 stsp 			break;
 727 c0cc5c62 2018-10-18 stsp 		case 'C':
 728 4a8520aa 2018-10-18 stsp 			diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
 729 4a8520aa 2018-10-18 stsp 			    &errstr);
 730 c0cc5c62 2018-10-18 stsp 			if (errstr != NULL)
 731 c0cc5c62 2018-10-18 stsp 				err(1, "-C option %s", errstr);
 732 c0cc5c62 2018-10-18 stsp 			break;
 733 64a96a6d 2018-04-01 stsp 		case 'l':
 734 64a96a6d 2018-04-01 stsp 			limit = strtonum(optarg, 1, INT_MAX, &errstr);
 735 64a96a6d 2018-04-01 stsp 			if (errstr != NULL)
 736 64a96a6d 2018-04-01 stsp 				err(1, "-l option %s", errstr);
 737 79109fed 2018-03-27 stsp 			break;
 738 0ed6ed4c 2018-06-13 stsp 		case 'f':
 739 0ed6ed4c 2018-06-13 stsp 			first_parent_traversal = 1;
 740 a0603db2 2018-06-10 stsp 			break;
 741 04ca23f4 2018-07-16 stsp 		case 'r':
 742 04ca23f4 2018-07-16 stsp 			repo_path = realpath(optarg, NULL);
 743 04ca23f4 2018-07-16 stsp 			if (repo_path == NULL)
 744 04ca23f4 2018-07-16 stsp 				err(1, "-r option");
 745 04ca23f4 2018-07-16 stsp 			break;
 746 79109fed 2018-03-27 stsp 		default:
 747 79109fed 2018-03-27 stsp 			usage();
 748 79109fed 2018-03-27 stsp 			/* NOTREACHED */
 749 79109fed 2018-03-27 stsp 		}
 750 79109fed 2018-03-27 stsp 	}
 751 79109fed 2018-03-27 stsp 
 752 79109fed 2018-03-27 stsp 	argc -= optind;
 753 79109fed 2018-03-27 stsp 	argv += optind;
 754 79109fed 2018-03-27 stsp 
 755 04ca23f4 2018-07-16 stsp 	if (argc == 0)
 756 04ca23f4 2018-07-16 stsp 		path = strdup("");
 757 04ca23f4 2018-07-16 stsp 	else if (argc == 1)
 758 04ca23f4 2018-07-16 stsp 		path = strdup(argv[0]);
 759 04ca23f4 2018-07-16 stsp 	else
 760 6f3d1eb0 2018-03-12 stsp 		usage_log();
 761 04ca23f4 2018-07-16 stsp 	if (path == NULL)
 762 04ca23f4 2018-07-16 stsp 		return got_error_from_errno();
 763 f42b1b34 2018-03-12 stsp 
 764 04ca23f4 2018-07-16 stsp 	cwd = getcwd(NULL, 0);
 765 04ca23f4 2018-07-16 stsp 	if (cwd == NULL) {
 766 04ca23f4 2018-07-16 stsp 		error = got_error_from_errno();
 767 04ca23f4 2018-07-16 stsp 		goto done;
 768 04ca23f4 2018-07-16 stsp 	}
 769 04ca23f4 2018-07-16 stsp 	if (repo_path == NULL) {
 770 04ca23f4 2018-07-16 stsp 		repo_path = strdup(cwd);
 771 04ca23f4 2018-07-16 stsp 		if (repo_path == NULL) {
 772 04ca23f4 2018-07-16 stsp 			error = got_error_from_errno();
 773 04ca23f4 2018-07-16 stsp 			goto done;
 774 04ca23f4 2018-07-16 stsp 		}
 775 04ca23f4 2018-07-16 stsp 	}
 776 04ca23f4 2018-07-16 stsp 
 777 6098196c 2019-01-04 stsp 	error = apply_unveil(repo_path, NULL);
 778 6098196c 2019-01-04 stsp 	if (error)
 779 6098196c 2019-01-04 stsp 		goto done;
 780 6098196c 2019-01-04 stsp 
 781 f42b1b34 2018-03-12 stsp 	error = got_repo_open(&repo, repo_path);
 782 d7d4f210 2018-03-12 stsp 	if (error != NULL)
 783 04ca23f4 2018-07-16 stsp 		goto done;
 784 f42b1b34 2018-03-12 stsp 
 785 d142fc45 2018-04-01 stsp 	if (start_commit == NULL) {
 786 3235492e 2018-04-01 stsp 		struct got_reference *head_ref;
 787 3235492e 2018-04-01 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
 788 3235492e 2018-04-01 stsp 		if (error != NULL)
 789 3235492e 2018-04-01 stsp 			return error;
 790 3235492e 2018-04-01 stsp 		error = got_ref_resolve(&id, repo, head_ref);
 791 3235492e 2018-04-01 stsp 		got_ref_close(head_ref);
 792 3235492e 2018-04-01 stsp 		if (error != NULL)
 793 3235492e 2018-04-01 stsp 			return error;
 794 15a94983 2018-12-23 stsp 		error = got_object_open_as_commit(&commit, repo, id);
 795 3235492e 2018-04-01 stsp 	} else {
 796 d1f2edc9 2018-06-13 stsp 		struct got_reference *ref;
 797 d1f2edc9 2018-06-13 stsp 		error = got_ref_open(&ref, repo, start_commit);
 798 3235492e 2018-04-01 stsp 		if (error == NULL) {
 799 d1f2edc9 2018-06-13 stsp 			error = got_ref_resolve(&id, repo, ref);
 800 d1f2edc9 2018-06-13 stsp 			got_ref_close(ref);
 801 d1f2edc9 2018-06-13 stsp 			if (error != NULL)
 802 d1f2edc9 2018-06-13 stsp 				return error;
 803 15a94983 2018-12-23 stsp 			error = got_object_open_as_commit(&commit, repo, id);
 804 d1f2edc9 2018-06-13 stsp 			if (error != NULL)
 805 d1f2edc9 2018-06-13 stsp 				return error;
 806 d1f2edc9 2018-06-13 stsp 		}
 807 15a94983 2018-12-23 stsp 		if (commit == NULL) {
 808 15a94983 2018-12-23 stsp 			error = got_object_resolve_id_str(&id, repo,
 809 d1f2edc9 2018-06-13 stsp 			    start_commit);
 810 d1f2edc9 2018-06-13 stsp 			if (error != NULL)
 811 d1f2edc9 2018-06-13 stsp 				return error;
 812 3235492e 2018-04-01 stsp 		}
 813 3235492e 2018-04-01 stsp 	}
 814 d7d4f210 2018-03-12 stsp 	if (error != NULL)
 815 04ca23f4 2018-07-16 stsp 		goto done;
 816 04ca23f4 2018-07-16 stsp 
 817 23721109 2018-10-22 stsp 	error = got_repo_map_path(&in_repo_path, repo, path, 1);
 818 04ca23f4 2018-07-16 stsp 	if (error != NULL)
 819 04ca23f4 2018-07-16 stsp 		goto done;
 820 04ca23f4 2018-07-16 stsp 	if (in_repo_path) {
 821 04ca23f4 2018-07-16 stsp 		free(path);
 822 04ca23f4 2018-07-16 stsp 		path = in_repo_path;
 823 04ca23f4 2018-07-16 stsp 	}
 824 04ca23f4 2018-07-16 stsp 
 825 15a94983 2018-12-23 stsp 	error = print_commits(id, repo, path, show_patch,
 826 c0cc5c62 2018-10-18 stsp 	    diff_context, limit, first_parent_traversal);
 827 04ca23f4 2018-07-16 stsp done:
 828 04ca23f4 2018-07-16 stsp 	free(path);
 829 04ca23f4 2018-07-16 stsp 	free(repo_path);
 830 04ca23f4 2018-07-16 stsp 	free(cwd);
 831 f42b1b34 2018-03-12 stsp 	free(id);
 832 ad242220 2018-09-08 stsp 	if (repo) {
 833 ad242220 2018-09-08 stsp 		const struct got_error *repo_error;
 834 ad242220 2018-09-08 stsp 		repo_error = got_repo_close(repo);
 835 ad242220 2018-09-08 stsp 		if (error == NULL)
 836 ad242220 2018-09-08 stsp 			error = repo_error;
 837 ad242220 2018-09-08 stsp 	}
 838 8bf5b3c9 2018-03-17 stsp 	return error;
 839 5c860e29 2018-03-12 stsp }
 840 5c860e29 2018-03-12 stsp 
 841 4ed7e80c 2018-05-20 stsp __dead static void
 842 3f8b7d6a 2018-04-01 stsp usage_diff(void)
 843 3f8b7d6a 2018-04-01 stsp {
 844 c0cc5c62 2018-10-18 stsp 	fprintf(stderr, "usage: %s diff [-C number] [repository-path] "
 845 c0cc5c62 2018-10-18 stsp 	    "object1 object2\n", getprogname());
 846 3f8b7d6a 2018-04-01 stsp 	exit(1);
 847 b00d56cd 2018-04-01 stsp }
 848 b00d56cd 2018-04-01 stsp 
 849 4ed7e80c 2018-05-20 stsp static const struct got_error *
 850 b00d56cd 2018-04-01 stsp cmd_diff(int argc, char *argv[])
 851 b00d56cd 2018-04-01 stsp {
 852 b00d56cd 2018-04-01 stsp 	const struct got_error *error;
 853 b00d56cd 2018-04-01 stsp 	struct got_repository *repo = NULL;
 854 b00d56cd 2018-04-01 stsp 	char *repo_path = NULL;
 855 15a94983 2018-12-23 stsp 	struct got_object_id *id1 = NULL, *id2 = NULL;
 856 15a94983 2018-12-23 stsp 	char *id_str1 = NULL, *id_str2 = NULL;
 857 15a94983 2018-12-23 stsp 	int type1, type2;
 858 c0cc5c62 2018-10-18 stsp 	int diff_context = 3, ch;
 859 c0cc5c62 2018-10-18 stsp 	const char *errstr;
 860 b00d56cd 2018-04-01 stsp 
 861 b00d56cd 2018-04-01 stsp #ifndef PROFILE
 862 25eccc22 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
 863 25eccc22 2019-01-04 stsp 	    NULL) == -1)
 864 b00d56cd 2018-04-01 stsp 		err(1, "pledge");
 865 b00d56cd 2018-04-01 stsp #endif
 866 b00d56cd 2018-04-01 stsp 
 867 c0cc5c62 2018-10-18 stsp 	while ((ch = getopt(argc, argv, "C:")) != -1) {
 868 b00d56cd 2018-04-01 stsp 		switch (ch) {
 869 c0cc5c62 2018-10-18 stsp 		case 'C':
 870 c0cc5c62 2018-10-18 stsp 			diff_context = strtonum(optarg, 1, INT_MAX, &errstr);
 871 c0cc5c62 2018-10-18 stsp 			if (errstr != NULL)
 872 c0cc5c62 2018-10-18 stsp 				err(1, "-C option %s", errstr);
 873 c0cc5c62 2018-10-18 stsp 			break;
 874 b00d56cd 2018-04-01 stsp 		default:
 875 b00d56cd 2018-04-01 stsp 			usage();
 876 b00d56cd 2018-04-01 stsp 			/* NOTREACHED */
 877 b00d56cd 2018-04-01 stsp 		}
 878 b00d56cd 2018-04-01 stsp 	}
 879 b00d56cd 2018-04-01 stsp 
 880 b00d56cd 2018-04-01 stsp 	argc -= optind;
 881 b00d56cd 2018-04-01 stsp 	argv += optind;
 882 b00d56cd 2018-04-01 stsp 
 883 b00d56cd 2018-04-01 stsp 	if (argc == 0) {
 884 b00d56cd 2018-04-01 stsp 		usage_diff(); /* TODO show local worktree changes */
 885 3f8b7d6a 2018-04-01 stsp 	} else if (argc == 2) {
 886 3f8b7d6a 2018-04-01 stsp 		repo_path = getcwd(NULL, 0);
 887 3f8b7d6a 2018-04-01 stsp 		if (repo_path == NULL)
 888 e1e3f570 2018-04-01 stsp 			return got_error_from_errno();
 889 15a94983 2018-12-23 stsp 		id_str1 = argv[0];
 890 15a94983 2018-12-23 stsp 		id_str2 = argv[1];
 891 b00d56cd 2018-04-01 stsp 	} else if (argc == 3) {
 892 76089277 2018-04-01 stsp 		repo_path = realpath(argv[0], NULL);
 893 76089277 2018-04-01 stsp 		if (repo_path == NULL)
 894 76089277 2018-04-01 stsp 			return got_error_from_errno();
 895 15a94983 2018-12-23 stsp 		id_str1 = argv[1];
 896 15a94983 2018-12-23 stsp 		id_str2 = argv[2];
 897 b00d56cd 2018-04-01 stsp 	} else
 898 b00d56cd 2018-04-01 stsp 		usage_diff();
 899 25eccc22 2019-01-04 stsp 
 900 25eccc22 2019-01-04 stsp 	error = apply_unveil(repo_path, NULL);
 901 25eccc22 2019-01-04 stsp 	if (error)
 902 25eccc22 2019-01-04 stsp 		goto done;
 903 b00d56cd 2018-04-01 stsp 
 904 b00d56cd 2018-04-01 stsp 	error = got_repo_open(&repo, repo_path);
 905 76089277 2018-04-01 stsp 	free(repo_path);
 906 b00d56cd 2018-04-01 stsp 	if (error != NULL)
 907 b00d56cd 2018-04-01 stsp 		goto done;
 908 b00d56cd 2018-04-01 stsp 
 909 15a94983 2018-12-23 stsp 	error = got_object_resolve_id_str(&id1, repo, id_str1);
 910 6402fb3c 2018-09-15 stsp 	if (error)
 911 b00d56cd 2018-04-01 stsp 		goto done;
 912 b00d56cd 2018-04-01 stsp 
 913 15a94983 2018-12-23 stsp 	error = got_object_resolve_id_str(&id2, repo, id_str2);
 914 6402fb3c 2018-09-15 stsp 	if (error)
 915 b00d56cd 2018-04-01 stsp 		goto done;
 916 b00d56cd 2018-04-01 stsp 
 917 15a94983 2018-12-23 stsp 	error = got_object_get_type(&type1, repo, id1);
 918 15a94983 2018-12-23 stsp 	if (error)
 919 15a94983 2018-12-23 stsp 		goto done;
 920 15a94983 2018-12-23 stsp 
 921 15a94983 2018-12-23 stsp 	error = got_object_get_type(&type2, repo, id2);
 922 15a94983 2018-12-23 stsp 	if (error)
 923 15a94983 2018-12-23 stsp 		goto done;
 924 15a94983 2018-12-23 stsp 
 925 15a94983 2018-12-23 stsp 	if (type1 != type2) {
 926 b00d56cd 2018-04-01 stsp 		error = got_error(GOT_ERR_OBJ_TYPE);
 927 b00d56cd 2018-04-01 stsp 		goto done;
 928 b00d56cd 2018-04-01 stsp 	}
 929 b00d56cd 2018-04-01 stsp 
 930 15a94983 2018-12-23 stsp 	switch (type1) {
 931 b00d56cd 2018-04-01 stsp 	case GOT_OBJ_TYPE_BLOB:
 932 15a94983 2018-12-23 stsp 		error = got_diff_objects_as_blobs(id1, id2, NULL, NULL,
 933 54156555 2018-12-24 stsp 		    diff_context, repo, stdout);
 934 b00d56cd 2018-04-01 stsp 		break;
 935 b00d56cd 2018-04-01 stsp 	case GOT_OBJ_TYPE_TREE:
 936 15a94983 2018-12-23 stsp 		error = got_diff_objects_as_trees(id1, id2, "", "",
 937 54156555 2018-12-24 stsp 		    diff_context, repo, stdout);
 938 b00d56cd 2018-04-01 stsp 		break;
 939 b00d56cd 2018-04-01 stsp 	case GOT_OBJ_TYPE_COMMIT:
 940 15a94983 2018-12-23 stsp 		printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null",
 941 15a94983 2018-12-23 stsp 		    id_str2);
 942 15a94983 2018-12-23 stsp 		error = got_diff_objects_as_commits(id1, id2, diff_context,
 943 c0cc5c62 2018-10-18 stsp 		    repo, stdout);
 944 b00d56cd 2018-04-01 stsp 		break;
 945 b00d56cd 2018-04-01 stsp 	default:
 946 b00d56cd 2018-04-01 stsp 		error = got_error(GOT_ERR_OBJ_TYPE);
 947 b00d56cd 2018-04-01 stsp 	}
 948 b00d56cd 2018-04-01 stsp 
 949 b00d56cd 2018-04-01 stsp done:
 950 15a94983 2018-12-23 stsp 	free(id1);
 951 15a94983 2018-12-23 stsp 	free(id2);
 952 ad242220 2018-09-08 stsp 	if (repo) {
 953 ad242220 2018-09-08 stsp 		const struct got_error *repo_error;
 954 ad242220 2018-09-08 stsp 		repo_error = got_repo_close(repo);
 955 ad242220 2018-09-08 stsp 		if (error == NULL)
 956 ad242220 2018-09-08 stsp 			error = repo_error;
 957 ad242220 2018-09-08 stsp 	}
 958 b00d56cd 2018-04-01 stsp 	return error;
 959 404c43c4 2018-06-21 stsp }
 960 404c43c4 2018-06-21 stsp 
 961 404c43c4 2018-06-21 stsp __dead static void
 962 404c43c4 2018-06-21 stsp usage_blame(void)
 963 404c43c4 2018-06-21 stsp {
 964 66bea077 2018-08-02 stsp 	fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
 965 404c43c4 2018-06-21 stsp 	    getprogname());
 966 404c43c4 2018-06-21 stsp 	exit(1);
 967 b00d56cd 2018-04-01 stsp }
 968 b00d56cd 2018-04-01 stsp 
 969 404c43c4 2018-06-21 stsp static const struct got_error *
 970 404c43c4 2018-06-21 stsp cmd_blame(int argc, char *argv[])
 971 404c43c4 2018-06-21 stsp {
 972 404c43c4 2018-06-21 stsp 	const struct got_error *error;
 973 404c43c4 2018-06-21 stsp 	struct got_repository *repo = NULL;
 974 66bea077 2018-08-02 stsp 	char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
 975 404c43c4 2018-06-21 stsp 	struct got_object_id *commit_id = NULL;
 976 404c43c4 2018-06-21 stsp 	char *commit_id_str = NULL;
 977 404c43c4 2018-06-21 stsp 	int ch;
 978 404c43c4 2018-06-21 stsp 
 979 404c43c4 2018-06-21 stsp #ifndef PROFILE
 980 36e2fb66 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
 981 36e2fb66 2019-01-04 stsp 	    NULL) == -1)
 982 404c43c4 2018-06-21 stsp 		err(1, "pledge");
 983 404c43c4 2018-06-21 stsp #endif
 984 404c43c4 2018-06-21 stsp 
 985 66bea077 2018-08-02 stsp 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
 986 404c43c4 2018-06-21 stsp 		switch (ch) {
 987 404c43c4 2018-06-21 stsp 		case 'c':
 988 404c43c4 2018-06-21 stsp 			commit_id_str = optarg;
 989 404c43c4 2018-06-21 stsp 			break;
 990 66bea077 2018-08-02 stsp 		case 'r':
 991 66bea077 2018-08-02 stsp 			repo_path = realpath(optarg, NULL);
 992 66bea077 2018-08-02 stsp 			if (repo_path == NULL)
 993 66bea077 2018-08-02 stsp 				err(1, "-r option");
 994 66bea077 2018-08-02 stsp 			break;
 995 404c43c4 2018-06-21 stsp 		default:
 996 404c43c4 2018-06-21 stsp 			usage();
 997 404c43c4 2018-06-21 stsp 			/* NOTREACHED */
 998 404c43c4 2018-06-21 stsp 		}
 999 404c43c4 2018-06-21 stsp 	}
1000 404c43c4 2018-06-21 stsp 
1001 404c43c4 2018-06-21 stsp 	argc -= optind;
1002 404c43c4 2018-06-21 stsp 	argv += optind;
1003 404c43c4 2018-06-21 stsp 
1004 a39318fd 2018-08-02 stsp 	if (argc == 1)
1005 404c43c4 2018-06-21 stsp 		path = argv[0];
1006 a39318fd 2018-08-02 stsp 	else
1007 404c43c4 2018-06-21 stsp 		usage_blame();
1008 404c43c4 2018-06-21 stsp 
1009 66bea077 2018-08-02 stsp 	cwd = getcwd(NULL, 0);
1010 66bea077 2018-08-02 stsp 	if (cwd == NULL) {
1011 66bea077 2018-08-02 stsp 		error = got_error_from_errno();
1012 66bea077 2018-08-02 stsp 		goto done;
1013 66bea077 2018-08-02 stsp 	}
1014 66bea077 2018-08-02 stsp 	if (repo_path == NULL) {
1015 66bea077 2018-08-02 stsp 		repo_path = strdup(cwd);
1016 66bea077 2018-08-02 stsp 		if (repo_path == NULL) {
1017 66bea077 2018-08-02 stsp 			error = got_error_from_errno();
1018 66bea077 2018-08-02 stsp 			goto done;
1019 66bea077 2018-08-02 stsp 		}
1020 66bea077 2018-08-02 stsp 	}
1021 36e2fb66 2019-01-04 stsp 
1022 36e2fb66 2019-01-04 stsp 	error = apply_unveil(repo_path, NULL);
1023 36e2fb66 2019-01-04 stsp 	if (error)
1024 36e2fb66 2019-01-04 stsp 		goto done;
1025 66bea077 2018-08-02 stsp 
1026 404c43c4 2018-06-21 stsp 	error = got_repo_open(&repo, repo_path);
1027 404c43c4 2018-06-21 stsp 	if (error != NULL)
1028 404c43c4 2018-06-21 stsp 		goto done;
1029 404c43c4 2018-06-21 stsp 
1030 23721109 2018-10-22 stsp 	error = got_repo_map_path(&in_repo_path, repo, path, 1);
1031 66bea077 2018-08-02 stsp 	if (error != NULL)
1032 66bea077 2018-08-02 stsp 		goto done;
1033 66bea077 2018-08-02 stsp 
1034 404c43c4 2018-06-21 stsp 	if (commit_id_str == NULL) {
1035 404c43c4 2018-06-21 stsp 		struct got_reference *head_ref;
1036 404c43c4 2018-06-21 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1037 404c43c4 2018-06-21 stsp 		if (error != NULL)
1038 66bea077 2018-08-02 stsp 			goto done;
1039 404c43c4 2018-06-21 stsp 		error = got_ref_resolve(&commit_id, repo, head_ref);
1040 404c43c4 2018-06-21 stsp 		got_ref_close(head_ref);
1041 404c43c4 2018-06-21 stsp 		if (error != NULL)
1042 66bea077 2018-08-02 stsp 			goto done;
1043 404c43c4 2018-06-21 stsp 	} else {
1044 15a94983 2018-12-23 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
1045 15a94983 2018-12-23 stsp 		    commit_id_str);
1046 404c43c4 2018-06-21 stsp 		if (error != NULL)
1047 66bea077 2018-08-02 stsp 			goto done;
1048 404c43c4 2018-06-21 stsp 	}
1049 404c43c4 2018-06-21 stsp 
1050 66bea077 2018-08-02 stsp 	error = got_blame(in_repo_path, commit_id, repo, stdout);
1051 404c43c4 2018-06-21 stsp done:
1052 66bea077 2018-08-02 stsp 	free(in_repo_path);
1053 66bea077 2018-08-02 stsp 	free(repo_path);
1054 66bea077 2018-08-02 stsp 	free(cwd);
1055 404c43c4 2018-06-21 stsp 	free(commit_id);
1056 ad242220 2018-09-08 stsp 	if (repo) {
1057 ad242220 2018-09-08 stsp 		const struct got_error *repo_error;
1058 ad242220 2018-09-08 stsp 		repo_error = got_repo_close(repo);
1059 ad242220 2018-09-08 stsp 		if (error == NULL)
1060 ad242220 2018-09-08 stsp 			error = repo_error;
1061 ad242220 2018-09-08 stsp 	}
1062 404c43c4 2018-06-21 stsp 	return error;
1063 5de5890b 2018-10-18 stsp }
1064 5de5890b 2018-10-18 stsp 
1065 5de5890b 2018-10-18 stsp __dead static void
1066 5de5890b 2018-10-18 stsp usage_tree(void)
1067 5de5890b 2018-10-18 stsp {
1068 5de5890b 2018-10-18 stsp 	fprintf(stderr, "usage: %s tree [-c commit] [-r repository-path] [-i] path\n",
1069 5de5890b 2018-10-18 stsp 	    getprogname());
1070 5de5890b 2018-10-18 stsp 	exit(1);
1071 5de5890b 2018-10-18 stsp }
1072 5de5890b 2018-10-18 stsp 
1073 5de5890b 2018-10-18 stsp 
1074 5de5890b 2018-10-18 stsp static const struct got_error *
1075 5de5890b 2018-10-18 stsp print_tree(const char *path, struct got_object_id *commit_id,
1076 5de5890b 2018-10-18 stsp     int show_ids, struct got_repository *repo)
1077 5de5890b 2018-10-18 stsp {
1078 5de5890b 2018-10-18 stsp 	const struct got_error *err = NULL;
1079 5de5890b 2018-10-18 stsp 	struct got_object_id *tree_id = NULL;
1080 5de5890b 2018-10-18 stsp 	struct got_tree_object *tree = NULL;
1081 5de5890b 2018-10-18 stsp 	const struct got_tree_entries *entries;
1082 5de5890b 2018-10-18 stsp 	struct got_tree_entry *te;
1083 5de5890b 2018-10-18 stsp 
1084 5de5890b 2018-10-18 stsp 	err = got_object_id_by_path(&tree_id, repo, commit_id, path);
1085 5de5890b 2018-10-18 stsp 	if (err)
1086 5de5890b 2018-10-18 stsp 		goto done;
1087 5de5890b 2018-10-18 stsp 
1088 5de5890b 2018-10-18 stsp 	err = got_object_open_as_tree(&tree, repo, tree_id);
1089 5de5890b 2018-10-18 stsp 	if (err)
1090 5de5890b 2018-10-18 stsp 		goto done;
1091 5de5890b 2018-10-18 stsp 	entries = got_object_tree_get_entries(tree);
1092 5de5890b 2018-10-18 stsp 	te = SIMPLEQ_FIRST(&entries->head);
1093 5de5890b 2018-10-18 stsp 	while (te) {
1094 5de5890b 2018-10-18 stsp 		char *id = NULL;
1095 84453469 2018-11-11 stsp 
1096 84453469 2018-11-11 stsp 		if (sigint_received || sigpipe_received)
1097 84453469 2018-11-11 stsp 			break;
1098 84453469 2018-11-11 stsp 
1099 5de5890b 2018-10-18 stsp 		if (show_ids) {
1100 5de5890b 2018-10-18 stsp 			char *id_str;
1101 5de5890b 2018-10-18 stsp 			err = got_object_id_str(&id_str, te->id);
1102 5de5890b 2018-10-18 stsp 			if (err)
1103 5de5890b 2018-10-18 stsp 				goto done;
1104 5de5890b 2018-10-18 stsp 			if (asprintf(&id, "%s ", id_str) == -1) {
1105 5de5890b 2018-10-18 stsp 				err = got_error_from_errno();
1106 5de5890b 2018-10-18 stsp 				free(id_str);
1107 5de5890b 2018-10-18 stsp 				goto done;
1108 5de5890b 2018-10-18 stsp 			}
1109 5de5890b 2018-10-18 stsp 			free(id_str);
1110 5de5890b 2018-10-18 stsp 		}
1111 5de5890b 2018-10-18 stsp 		printf("%s%s%s\n", id ? id : "",
1112 5de5890b 2018-10-18 stsp 		    te->name, S_ISDIR(te->mode) ? "/" : "");
1113 5de5890b 2018-10-18 stsp 		te = SIMPLEQ_NEXT(te, entry);
1114 5de5890b 2018-10-18 stsp 		free(id);
1115 5de5890b 2018-10-18 stsp 	}
1116 5de5890b 2018-10-18 stsp done:
1117 5de5890b 2018-10-18 stsp 	if (tree)
1118 5de5890b 2018-10-18 stsp 		got_object_tree_close(tree);
1119 5de5890b 2018-10-18 stsp 	free(tree_id);
1120 5de5890b 2018-10-18 stsp 	return err;
1121 404c43c4 2018-06-21 stsp }
1122 404c43c4 2018-06-21 stsp 
1123 5de5890b 2018-10-18 stsp static const struct got_error *
1124 5de5890b 2018-10-18 stsp cmd_tree(int argc, char *argv[])
1125 5de5890b 2018-10-18 stsp {
1126 5de5890b 2018-10-18 stsp 	const struct got_error *error;
1127 5de5890b 2018-10-18 stsp 	struct got_repository *repo = NULL;
1128 5de5890b 2018-10-18 stsp 	char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1129 5de5890b 2018-10-18 stsp 	struct got_object_id *commit_id = NULL;
1130 5de5890b 2018-10-18 stsp 	char *commit_id_str = NULL;
1131 5de5890b 2018-10-18 stsp 	int show_ids = 0;
1132 5de5890b 2018-10-18 stsp 	int ch;
1133 5de5890b 2018-10-18 stsp 
1134 5de5890b 2018-10-18 stsp #ifndef PROFILE
1135 0f8d269b 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1136 0f8d269b 2019-01-04 stsp 	    NULL) == -1)
1137 5de5890b 2018-10-18 stsp 		err(1, "pledge");
1138 5de5890b 2018-10-18 stsp #endif
1139 5de5890b 2018-10-18 stsp 
1140 5de5890b 2018-10-18 stsp 	while ((ch = getopt(argc, argv, "c:r:i")) != -1) {
1141 5de5890b 2018-10-18 stsp 		switch (ch) {
1142 5de5890b 2018-10-18 stsp 		case 'c':
1143 5de5890b 2018-10-18 stsp 			commit_id_str = optarg;
1144 5de5890b 2018-10-18 stsp 			break;
1145 5de5890b 2018-10-18 stsp 		case 'r':
1146 5de5890b 2018-10-18 stsp 			repo_path = realpath(optarg, NULL);
1147 5de5890b 2018-10-18 stsp 			if (repo_path == NULL)
1148 5de5890b 2018-10-18 stsp 				err(1, "-r option");
1149 5de5890b 2018-10-18 stsp 			break;
1150 5de5890b 2018-10-18 stsp 		case 'i':
1151 5de5890b 2018-10-18 stsp 			show_ids = 1;
1152 5de5890b 2018-10-18 stsp 			break;
1153 5de5890b 2018-10-18 stsp 		default:
1154 5de5890b 2018-10-18 stsp 			usage();
1155 5de5890b 2018-10-18 stsp 			/* NOTREACHED */
1156 5de5890b 2018-10-18 stsp 		}
1157 5de5890b 2018-10-18 stsp 	}
1158 5de5890b 2018-10-18 stsp 
1159 5de5890b 2018-10-18 stsp 	argc -= optind;
1160 5de5890b 2018-10-18 stsp 	argv += optind;
1161 5de5890b 2018-10-18 stsp 
1162 5de5890b 2018-10-18 stsp 	if (argc == 1)
1163 5de5890b 2018-10-18 stsp 		path = argv[0];
1164 5de5890b 2018-10-18 stsp 	else if (argc > 1)
1165 5de5890b 2018-10-18 stsp 		usage_tree();
1166 5de5890b 2018-10-18 stsp 	else
1167 5de5890b 2018-10-18 stsp 		path = "/";
1168 5de5890b 2018-10-18 stsp 
1169 5de5890b 2018-10-18 stsp 	cwd = getcwd(NULL, 0);
1170 5de5890b 2018-10-18 stsp 	if (cwd == NULL) {
1171 5de5890b 2018-10-18 stsp 		error = got_error_from_errno();
1172 5de5890b 2018-10-18 stsp 		goto done;
1173 5de5890b 2018-10-18 stsp 	}
1174 5de5890b 2018-10-18 stsp 	if (repo_path == NULL) {
1175 5de5890b 2018-10-18 stsp 		repo_path = strdup(cwd);
1176 5de5890b 2018-10-18 stsp 		if (repo_path == NULL) {
1177 5de5890b 2018-10-18 stsp 			error = got_error_from_errno();
1178 5de5890b 2018-10-18 stsp 			goto done;
1179 5de5890b 2018-10-18 stsp 		}
1180 5de5890b 2018-10-18 stsp 	}
1181 5de5890b 2018-10-18 stsp 
1182 0f8d269b 2019-01-04 stsp 	error = apply_unveil(repo_path, NULL);
1183 0f8d269b 2019-01-04 stsp 	if (error)
1184 0f8d269b 2019-01-04 stsp 		goto done;
1185 0f8d269b 2019-01-04 stsp 
1186 5de5890b 2018-10-18 stsp 	error = got_repo_open(&repo, repo_path);
1187 5de5890b 2018-10-18 stsp 	if (error != NULL)
1188 5de5890b 2018-10-18 stsp 		goto done;
1189 5de5890b 2018-10-18 stsp 
1190 23721109 2018-10-22 stsp 	error = got_repo_map_path(&in_repo_path, repo, path, 1);
1191 5de5890b 2018-10-18 stsp 	if (error != NULL)
1192 5de5890b 2018-10-18 stsp 		goto done;
1193 5de5890b 2018-10-18 stsp 
1194 5de5890b 2018-10-18 stsp 	if (commit_id_str == NULL) {
1195 5de5890b 2018-10-18 stsp 		struct got_reference *head_ref;
1196 5de5890b 2018-10-18 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1197 5de5890b 2018-10-18 stsp 		if (error != NULL)
1198 5de5890b 2018-10-18 stsp 			goto done;
1199 5de5890b 2018-10-18 stsp 		error = got_ref_resolve(&commit_id, repo, head_ref);
1200 5de5890b 2018-10-18 stsp 		got_ref_close(head_ref);
1201 5de5890b 2018-10-18 stsp 		if (error != NULL)
1202 5de5890b 2018-10-18 stsp 			goto done;
1203 5de5890b 2018-10-18 stsp 	} else {
1204 15a94983 2018-12-23 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
1205 15a94983 2018-12-23 stsp 		    commit_id_str);
1206 5de5890b 2018-10-18 stsp 		if (error != NULL)
1207 5de5890b 2018-10-18 stsp 			goto done;
1208 5de5890b 2018-10-18 stsp 	}
1209 5de5890b 2018-10-18 stsp 
1210 5de5890b 2018-10-18 stsp 	error = print_tree(in_repo_path, commit_id, show_ids, repo);
1211 5de5890b 2018-10-18 stsp done:
1212 5de5890b 2018-10-18 stsp 	free(in_repo_path);
1213 5de5890b 2018-10-18 stsp 	free(repo_path);
1214 5de5890b 2018-10-18 stsp 	free(cwd);
1215 5de5890b 2018-10-18 stsp 	free(commit_id);
1216 5de5890b 2018-10-18 stsp 	if (repo) {
1217 5de5890b 2018-10-18 stsp 		const struct got_error *repo_error;
1218 5de5890b 2018-10-18 stsp 		repo_error = got_repo_close(repo);
1219 5de5890b 2018-10-18 stsp 		if (error == NULL)
1220 5de5890b 2018-10-18 stsp 			error = repo_error;
1221 5de5890b 2018-10-18 stsp 	}
1222 5de5890b 2018-10-18 stsp 	return error;
1223 5de5890b 2018-10-18 stsp }
1224 5de5890b 2018-10-18 stsp 
1225 f42b1b34 2018-03-12 stsp #ifdef notyet
1226 4ed7e80c 2018-05-20 stsp static const struct got_error *
1227 5c860e29 2018-03-12 stsp cmd_status(int argc __unused, char *argv[] __unused)
1228 5c860e29 2018-03-12 stsp {
1229 5c860e29 2018-03-12 stsp 	git_repository *repo = NULL;
1230 5c860e29 2018-03-12 stsp 	git_status_list *status;
1231 5c860e29 2018-03-12 stsp 	git_status_options statusopts;
1232 5c860e29 2018-03-12 stsp 	size_t i;
1233 5c860e29 2018-03-12 stsp 
1234 5c860e29 2018-03-12 stsp 	git_libgit2_init();
1235 5c860e29 2018-03-12 stsp 
1236 5c860e29 2018-03-12 stsp 	if (git_repository_open_ext(&repo, ".", 0, NULL))
1237 5c860e29 2018-03-12 stsp 		errx(1, "git_repository_open: %s", giterr_last()->message);
1238 5c860e29 2018-03-12 stsp 
1239 5c860e29 2018-03-12 stsp 	if (git_repository_is_bare(repo))
1240 5c860e29 2018-03-12 stsp 		errx(1, "bar repository");
1241 5c860e29 2018-03-12 stsp 
1242 5c860e29 2018-03-12 stsp 	if (git_status_init_options(&statusopts, GIT_STATUS_OPTIONS_VERSION))
1243 5c860e29 2018-03-12 stsp 		errx(1, "git_status_init_options: %s", giterr_last()->message);
1244 5c860e29 2018-03-12 stsp 
1245 5c860e29 2018-03-12 stsp 	statusopts.show  = GIT_STATUS_SHOW_INDEX_AND_WORKDIR;
1246 5c860e29 2018-03-12 stsp 	statusopts.flags = GIT_STATUS_OPT_INCLUDE_UNTRACKED |
1247 5c860e29 2018-03-12 stsp 	    GIT_STATUS_OPT_RENAMES_HEAD_TO_INDEX |
1248 5c860e29 2018-03-12 stsp 	    GIT_STATUS_OPT_SORT_CASE_SENSITIVELY;
1249 5c860e29 2018-03-12 stsp 
1250 5c860e29 2018-03-12 stsp 	if (git_status_list_new(&status, repo, &statusopts))
1251 5c860e29 2018-03-12 stsp 		errx(1, "git_status_list_new: %s", giterr_last()->message);
1252 5c860e29 2018-03-12 stsp 
1253 5c860e29 2018-03-12 stsp 	for (i = 0; i < git_status_list_entrycount(status); i++) {
1254 5c860e29 2018-03-12 stsp 		const git_status_entry *se;
1255 5c860e29 2018-03-12 stsp 
1256 5c860e29 2018-03-12 stsp 		se = git_status_byindex(status, i);
1257 5c860e29 2018-03-12 stsp 		switch (se->status) {
1258 5c860e29 2018-03-12 stsp 		case GIT_STATUS_WT_NEW:
1259 5c860e29 2018-03-12 stsp 			printf("? %s\n", se->index_to_workdir->new_file.path);
1260 5c860e29 2018-03-12 stsp 			break;
1261 5c860e29 2018-03-12 stsp 		case GIT_STATUS_WT_MODIFIED:
1262 5c860e29 2018-03-12 stsp 			printf("M %s\n", se->index_to_workdir->new_file.path);
1263 5c860e29 2018-03-12 stsp 			break;
1264 5c860e29 2018-03-12 stsp 		case GIT_STATUS_WT_DELETED:
1265 5c860e29 2018-03-12 stsp 			printf("R %s\n", se->index_to_workdir->new_file.path);
1266 5c860e29 2018-03-12 stsp 			break;
1267 5c860e29 2018-03-12 stsp 		case GIT_STATUS_WT_RENAMED:
1268 5c860e29 2018-03-12 stsp 			printf("m %s -> %s\n",
1269 5c860e29 2018-03-12 stsp 			    se->index_to_workdir->old_file.path,
1270 5c860e29 2018-03-12 stsp 			    se->index_to_workdir->new_file.path);
1271 5c860e29 2018-03-12 stsp 			break;
1272 5c860e29 2018-03-12 stsp 		case GIT_STATUS_WT_TYPECHANGE:
1273 5c860e29 2018-03-12 stsp 			printf("t %s\n", se->index_to_workdir->new_file.path);
1274 5c860e29 2018-03-12 stsp 			break;
1275 5c860e29 2018-03-12 stsp 		case GIT_STATUS_INDEX_NEW:
1276 5c860e29 2018-03-12 stsp 			printf("A %s\n", se->head_to_index->new_file.path);
1277 5c860e29 2018-03-12 stsp 			break;
1278 5c860e29 2018-03-12 stsp 		case GIT_STATUS_INDEX_MODIFIED:
1279 5c860e29 2018-03-12 stsp 			printf("M %s\n", se->head_to_index->old_file.path);
1280 5c860e29 2018-03-12 stsp 			break;
1281 5c860e29 2018-03-12 stsp 		case GIT_STATUS_INDEX_DELETED:
1282 5c860e29 2018-03-12 stsp 			printf("R %s\n", se->head_to_index->old_file.path);
1283 5c860e29 2018-03-12 stsp 			break;
1284 5c860e29 2018-03-12 stsp 		case GIT_STATUS_INDEX_RENAMED:
1285 5c860e29 2018-03-12 stsp 			printf("m %s -> %s\n",
1286 5c860e29 2018-03-12 stsp 			    se->head_to_index->old_file.path,
1287 5c860e29 2018-03-12 stsp 			    se->head_to_index->new_file.path);
1288 5c860e29 2018-03-12 stsp 			break;
1289 5c860e29 2018-03-12 stsp 		case GIT_STATUS_INDEX_TYPECHANGE:
1290 5c860e29 2018-03-12 stsp 			printf("t %s\n", se->head_to_index->old_file.path);
1291 5c860e29 2018-03-12 stsp 			break;
1292 5c860e29 2018-03-12 stsp 		case GIT_STATUS_CURRENT:
1293 5c860e29 2018-03-12 stsp 		default:
1294 5c860e29 2018-03-12 stsp 			break;
1295 5c860e29 2018-03-12 stsp 		}
1296 5c860e29 2018-03-12 stsp 	}
1297 5c860e29 2018-03-12 stsp 
1298 5c860e29 2018-03-12 stsp 	git_status_list_free(status);
1299 5c860e29 2018-03-12 stsp 	git_repository_free(repo);
1300 5c860e29 2018-03-12 stsp 	git_libgit2_shutdown();
1301 5c860e29 2018-03-12 stsp 
1302 5c860e29 2018-03-12 stsp 	return 0;
1303 5c860e29 2018-03-12 stsp }
1304 f42b1b34 2018-03-12 stsp #endif