[thirdparty/git.git] / archive.c

#include "cache.h"
#include "commit.h"
#include "tree-walk.h"
#include "attr.h"
#include "archive.h"
#include "parse-options.h"
#include "unpack-trees.h"

static char const * const archive_usage[] = {
	"git archive [options] <tree-ish> [<path>...]",
	"git archive --list",
	"git archive --remote <repo> [--exec <cmd>] [options] <tree-ish> [<path>...]",
	"git archive --remote <repo> [--exec <cmd>] --list",
	NULL
};

#define USES_ZLIB_COMPRESSION 1

static const struct archiver {
	const char *name;
	write_archive_fn_t write_archive;
	unsigned int flags;
} archivers[] = {
	{ "tar", write_tar_archive },
	{ "zip", write_zip_archive, USES_ZLIB_COMPRESSION },
};

static void format_subst(const struct commit *commit,
                         const char *src, size_t len,
                         struct strbuf *buf)
{
	char *to_free = NULL;
	struct strbuf fmt = STRBUF_INIT;
	struct pretty_print_context ctx = {0};
	ctx.date_mode = DATE_NORMAL;
	ctx.abbrev = DEFAULT_ABBREV;

	if (src == buf->buf)
		to_free = strbuf_detach(buf, NULL);
	for (;;) {
		const char *b, *c;

		b = memmem(src, len, "$Format:", 8);
		if (!b)
			break;
		c = memchr(b + 8, '$', (src + len) - b - 8);
		if (!c)
			break;

		strbuf_reset(&fmt);
		strbuf_add(&fmt, b + 8, c - b - 8);

		strbuf_add(buf, src, b - src);
		format_commit_message(commit, fmt.buf, buf, &ctx);
		len -= c + 1 - src;
		src  = c + 1;
	}
	strbuf_add(buf, src, len);
	strbuf_release(&fmt);
	free(to_free);
}

static void *sha1_file_to_archive(const char *path, const unsigned char *sha1,
		unsigned int mode, enum object_type *type,
		unsigned long *sizep, const struct commit *commit)
{
	void *buffer;

	buffer = read_sha1_file(sha1, type, sizep);
	if (buffer && S_ISREG(mode)) {
		struct strbuf buf = STRBUF_INIT;
		size_t size = 0;

		strbuf_attach(&buf, buffer, *sizep, *sizep + 1);
		convert_to_working_tree(path, buf.buf, buf.len, &buf);
		if (commit)
			format_subst(commit, buf.buf, buf.len, &buf);
		buffer = strbuf_detach(&buf, &size);
		*sizep = size;
	}

	return buffer;
}

static void setup_archive_check(struct git_attr_check *check)
{
	static struct git_attr *attr_export_ignore;
	static struct git_attr *attr_export_subst;

	if (!attr_export_ignore) {
		attr_export_ignore = git_attr("export-ignore");
		attr_export_subst = git_attr("export-subst");
	}
	check[0].attr = attr_export_ignore;
	check[1].attr = attr_export_subst;
}

struct archiver_context {
	struct archiver_args *args;
	write_archive_entry_fn_t write_entry;
};

static int write_archive_entry(const unsigned char *sha1, const char *base,
		int baselen, const char *filename, unsigned mode, int stage,
		void *context)
{
	static struct strbuf path = STRBUF_INIT;
	struct archiver_context *c = context;
	struct archiver_args *args = c->args;
	write_archive_entry_fn_t write_entry = c->write_entry;
	struct git_attr_check check[2];
	const char *path_without_prefix;
	int convert = 0;
	int err;
	enum object_type type;
	unsigned long size;
	void *buffer;

	strbuf_reset(&path);
	strbuf_grow(&path, PATH_MAX);
	strbuf_add(&path, args->base, args->baselen);
	strbuf_add(&path, base, baselen);
	strbuf_addstr(&path, filename);
	path_without_prefix = path.buf + args->baselen;

	setup_archive_check(check);
	if (!git_checkattr(path_without_prefix, ARRAY_SIZE(check), check)) {
		if (ATTR_TRUE(check[0].value))
			return 0;
		convert = ATTR_TRUE(check[1].value);
	}

	if (S_ISDIR(mode) || S_ISGITLINK(mode)) {
		strbuf_addch(&path, '/');
		if (args->verbose)
			fprintf(stderr, "%.*s\n", (int)path.len, path.buf);
		err = write_entry(args, sha1, path.buf, path.len, mode, NULL, 0);
		if (err)
			return err;
		return (S_ISDIR(mode) ? READ_TREE_RECURSIVE : 0);
	}

	buffer = sha1_file_to_archive(path_without_prefix, sha1, mode,
			&type, &size, convert ? args->commit : NULL);
	if (!buffer)
		return error("cannot read %s", sha1_to_hex(sha1));
	if (args->verbose)
		fprintf(stderr, "%.*s\n", (int)path.len, path.buf);
	err = write_entry(args, sha1, path.buf, path.len, mode, buffer, size);
	free(buffer);
	return err;
}

int write_archive_entries(struct archiver_args *args,
		write_archive_entry_fn_t write_entry)
{
	struct archiver_context context;
	struct unpack_trees_options opts;
	struct tree_desc t;
	struct pathspec pathspec;
	int err;

	if (args->baselen > 0 && args->base[args->baselen - 1] == '/') {
		size_t len = args->baselen;

		while (len > 1 && args->base[len - 2] == '/')
			len--;
		if (args->verbose)
			fprintf(stderr, "%.*s\n", (int)len, args->base);
		err = write_entry(args, args->tree->object.sha1, args->base,
				len, 040777, NULL, 0);
		if (err)
			return err;
	}

	context.args = args;
	context.write_entry = write_entry;

	/*
	 * Setup index and instruct attr to read index only
	 */
	if (!args->worktree_attributes) {
		memset(&opts, 0, sizeof(opts));
		opts.index_only = 1;
		opts.head_idx = -1;
		opts.src_index = &the_index;
		opts.dst_index = &the_index;
		opts.fn = oneway_merge;
		init_tree_desc(&t, args->tree->buffer, args->tree->size);
		if (unpack_trees(1, &t, &opts))
			return -1;
		git_attr_set_direction(GIT_ATTR_INDEX, &the_index);
	}

	init_pathspec(&pathspec, args->pathspec);
	err = read_tree_recursive(args->tree, "", 0, 0, &pathspec,
				  write_archive_entry, &context);
	free_pathspec(&pathspec);
	if (err == READ_TREE_RECURSIVE)
		err = 0;
	return err;
}

static const struct archiver *lookup_archiver(const char *name)
{
	int i;

	if (!name)
		return NULL;

	for (i = 0; i < ARRAY_SIZE(archivers); i++) {
		if (!strcmp(name, archivers[i].name))
			return &archivers[i];
	}
	return NULL;
}

static int reject_entry(const unsigned char *sha1, const char *base,
			int baselen, const char *filename, unsigned mode,
			int stage, void *context)
{
	return -1;
}

static int path_exists(struct tree *tree, const char *path)
{
	const char *paths[] = { path, NULL };
	struct pathspec pathspec;
	int ret;

	init_pathspec(&pathspec, paths);
	ret = read_tree_recursive(tree, "", 0, 0, &pathspec, reject_entry, NULL);
	free_pathspec(&pathspec);
	return ret != 0;
}

static void parse_pathspec_arg(const char **pathspec,
		struct archiver_args *ar_args)
{
	ar_args->pathspec = pathspec = get_pathspec("", pathspec);
	if (pathspec) {
		while (*pathspec) {
			if (!path_exists(ar_args->tree, *pathspec))
				die("path not found: %s", *pathspec);
			pathspec++;
		}
	}
}

static void parse_treeish_arg(const char **argv,
		struct archiver_args *ar_args, const char *prefix)
{
	const char *name = argv[0];
	const unsigned char *commit_sha1;
	time_t archive_time;
	struct tree *tree;
	const struct commit *commit;
	unsigned char sha1[20];

	if (get_sha1(name, sha1))
		die("Not a valid object name");

	commit = lookup_commit_reference_gently(sha1, 1);
	if (commit) {
		commit_sha1 = commit->object.sha1;
		archive_time = commit->date;
	} else {
		commit_sha1 = NULL;
		archive_time = time(NULL);
	}

	tree = parse_tree_indirect(sha1);
	if (tree == NULL)
		die("not a tree object");

	if (prefix) {
		unsigned char tree_sha1[20];
		unsigned int mode;
		int err;

		err = get_tree_entry(tree->object.sha1, prefix,
				     tree_sha1, &mode);
		if (err || !S_ISDIR(mode))
			die("current working directory is untracked");

		tree = parse_tree_indirect(tree_sha1);
	}
	ar_args->tree = tree;
	ar_args->commit_sha1 = commit_sha1;
	ar_args->commit = commit;
	ar_args->time = archive_time;
}

#define OPT__COMPR(s, v, h, p) \
	{ OPTION_SET_INT, (s), NULL, (v), NULL, (h), \
	  PARSE_OPT_NOARG | PARSE_OPT_NONEG, NULL, (p) }
#define OPT__COMPR_HIDDEN(s, v, p) \
	{ OPTION_SET_INT, (s), NULL, (v), NULL, "", \
	  PARSE_OPT_NOARG | PARSE_OPT_NONEG | PARSE_OPT_HIDDEN, NULL, (p) }

static int parse_archive_args(int argc, const char **argv,
		const struct archiver **ar, struct archiver_args *args)
{
	const char *format = "tar";
	const char *base = NULL;
	const char *remote = NULL;
	const char *exec = NULL;
	const char *output = NULL;
	int compression_level = -1;
	int verbose = 0;
	int i;
	int list = 0;
	int worktree_attributes = 0;
	struct option opts[] = {
		OPT_GROUP(""),
		OPT_STRING(0, "format", &format, "fmt", "archive format"),
		OPT_STRING(0, "prefix", &base, "prefix",
			"prepend prefix to each pathname in the archive"),
		OPT_STRING('o', "output", &output, "file",
			"write the archive to this file"),
		OPT_BOOLEAN(0, "worktree-attributes", &worktree_attributes,
			"read .gitattributes in working directory"),
		OPT__VERBOSE(&verbose, "report archived files on stderr"),
		OPT__COMPR('0', &compression_level, "store only", 0),
		OPT__COMPR('1', &compression_level, "compress faster", 1),
		OPT__COMPR_HIDDEN('2', &compression_level, 2),
		OPT__COMPR_HIDDEN('3', &compression_level, 3),
		OPT__COMPR_HIDDEN('4', &compression_level, 4),
		OPT__COMPR_HIDDEN('5', &compression_level, 5),
		OPT__COMPR_HIDDEN('6', &compression_level, 6),
		OPT__COMPR_HIDDEN('7', &compression_level, 7),
		OPT__COMPR_HIDDEN('8', &compression_level, 8),
		OPT__COMPR('9', &compression_level, "compress better", 9),
		OPT_GROUP(""),
		OPT_BOOLEAN('l', "list", &list,
			"list supported archive formats"),
		OPT_GROUP(""),
		OPT_STRING(0, "remote", &remote, "repo",
			"retrieve the archive from remote repository <repo>"),
		OPT_STRING(0, "exec", &exec, "cmd",
			"path to the remote git-upload-archive command"),
		OPT_END()
	};

	argc = parse_options(argc, argv, NULL, opts, archive_usage, 0);

	if (remote)
		die("Unexpected option --remote");
	if (exec)
		die("Option --exec can only be used together with --remote");
	if (output)
		die("Unexpected option --output");

	if (!base)
		base = "";

	if (list) {
		for (i = 0; i < ARRAY_SIZE(archivers); i++)
			printf("%s\n", archivers[i].name);
		exit(0);
	}

	/* We need at least one parameter -- tree-ish */
	if (argc < 1)
		usage_with_options(archive_usage, opts);
	*ar = lookup_archiver(format);
	if (!*ar)
		die("Unknown archive format '%s'", format);

	args->compression_level = Z_DEFAULT_COMPRESSION;
	if (compression_level != -1) {
		if ((*ar)->flags & USES_ZLIB_COMPRESSION)
			args->compression_level = compression_level;
		else {
			die("Argument not supported for format '%s': -%d",
					format, compression_level);
		}
	}
	args->verbose = verbose;
	args->base = base;
	args->baselen = strlen(base);
	args->worktree_attributes = worktree_attributes;

	return argc;
}

int write_archive(int argc, const char **argv, const char *prefix,
		int setup_prefix)
{
	const struct archiver *ar = NULL;
	struct archiver_args args;

	argc = parse_archive_args(argc, argv, &ar, &args);
	if (setup_prefix && prefix == NULL)
		prefix = setup_git_directory();

	parse_treeish_arg(argv, &args, prefix);
	parse_pathspec_arg(argv + 1, &args);

	git_config(git_default_config, NULL);

	return ar->write_archive(&args);
}
Commit	Line	Data
18125644 LH	1	#include "cache.h"
18125644 LH	2	#include "commit.h"
c0885435	3	#include "tree-walk.h"
18125644	4	#include "attr.h"
562e25ab	5	#include "archive.h"
4fac1d3a	6	#include "parse-options.h"
ba053ea9	7	#include "unpack-trees.h"
4fac1d3a RS	8
4fac1d3a RS	9	static char const * const archive_usage[] = {
62b4698e	10	"git archive [options] <tree-ish> [<path>...]",
4fac1d3a	11	"git archive --list",
62b4698e	12	"git archive --remote <repo> [--exec <cmd>] [options] <tree-ish> [<path>...]",
4fac1d3a RS	13	"git archive --remote <repo> [--exec <cmd>] --list",
	14	NULL
	15	};
c0885435 RS	16
	17	#define USES_ZLIB_COMPRESSION 1
	18
120a385a	19	static const struct archiver {
f15f736d RS	20	const char *name;
	21	write_archive_fn_t write_archive;
	22	unsigned int flags;
	23	} archivers[] = {
c0885435 RS	24	{ "tar", write_tar_archive },
	25	{ "zip", write_zip_archive, USES_ZLIB_COMPRESSION },
	26	};
	27
18125644 LH	28	static void format_subst(const struct commit *commit,
	29	const char *src, size_t len,
	30	struct strbuf *buf)
	31	{
	32	char *to_free = NULL;
f285a2d7	33	struct strbuf fmt = STRBUF_INIT;
dd2e794a TR	34	struct pretty_print_context ctx = {0};
dd2e794a TR	35	ctx.date_mode = DATE_NORMAL;
35039ced	36	ctx.abbrev = DEFAULT_ABBREV;
18125644 LH	37
	38	if (src == buf->buf)
	39	to_free = strbuf_detach(buf, NULL);
18125644 LH	40	for (;;) {
	41	const char b, c;
	42
	43	b = memmem(src, len, "$Format:", 8);
75b7dfbd	44	if (!b)
18125644	45	break;
75b7dfbd	46	c = memchr(b + 8, '$', (src + len) - b - 8);
18125644 LH	47	if (!c)
	48	break;
	49
	50	strbuf_reset(&fmt);
	51	strbuf_add(&fmt, b + 8, c - b - 8);
	52
	53	strbuf_add(buf, src, b - src);
dd2e794a	54	format_commit_message(commit, fmt.buf, buf, &ctx);
18125644 LH	55	len -= c + 1 - src;
	56	src = c + 1;
	57	}
	58	strbuf_add(buf, src, len);
	59	strbuf_release(&fmt);
	60	free(to_free);
	61	}
	62
1d11d5bb RS	63	static void sha1_file_to_archive(const char path, const unsigned char *sha1,
	64	unsigned int mode, enum object_type *type,
	65	unsigned long sizep, const struct commit commit)
18125644 LH	66	{
	67	void *buffer;
	68
	69	buffer = read_sha1_file(sha1, type, sizep);
	70	if (buffer && S_ISREG(mode)) {
f285a2d7	71	struct strbuf buf = STRBUF_INIT;
18125644 LH	72	size_t size = 0;
18125644 LH	73
18125644 LH	74	strbuf_attach(&buf, buffer, sizep, sizep + 1);
18125644 LH	75	convert_to_working_tree(path, buf.buf, buf.len, &buf);
1d11d5bb RS	76	if (commit)
1d11d5bb RS	77	format_subst(commit, buf.buf, buf.len, &buf);
18125644 LH	78	buffer = strbuf_detach(&buf, &size);
	79	*sizep = size;
	80	}
	81
	82	return buffer;
	83	}
	84
1d11d5bb	85	static void setup_archive_check(struct git_attr_check *check)
008d896d RS	86	{
008d896d RS	87	static struct git_attr *attr_export_ignore;
1d11d5bb	88	static struct git_attr *attr_export_subst;
008d896d	89
1d11d5bb	90	if (!attr_export_ignore) {
7fb0eaa2 JH	91	attr_export_ignore = git_attr("export-ignore");
7fb0eaa2 JH	92	attr_export_subst = git_attr("export-subst");
1d11d5bb	93	}
008d896d	94	check[0].attr = attr_export_ignore;
1d11d5bb	95	check[1].attr = attr_export_subst;
008d896d	96	}
562e25ab RS	97
	98	struct archiver_context {
	99	struct archiver_args *args;
	100	write_archive_entry_fn_t write_entry;
	101	};
	102
	103	static int write_archive_entry(const unsigned char sha1, const char base,
	104	int baselen, const char *filename, unsigned mode, int stage,
	105	void *context)
	106	{
	107	static struct strbuf path = STRBUF_INIT;
	108	struct archiver_context *c = context;
	109	struct archiver_args *args = c->args;
	110	write_archive_entry_fn_t write_entry = c->write_entry;
1d11d5bb RS	111	struct git_attr_check check[2];
	112	const char *path_without_prefix;
	113	int convert = 0;
562e25ab RS	114	int err;
	115	enum object_type type;
	116	unsigned long size;
	117	void *buffer;
	118
	119	strbuf_reset(&path);
	120	strbuf_grow(&path, PATH_MAX);
ebfbdb34	121	strbuf_add(&path, args->base, args->baselen);
562e25ab RS	122	strbuf_add(&path, base, baselen);
562e25ab RS	123	strbuf_addstr(&path, filename);
1d11d5bb	124	path_without_prefix = path.buf + args->baselen;
562e25ab	125
1d11d5bb RS	126	setup_archive_check(check);
	127	if (!git_checkattr(path_without_prefix, ARRAY_SIZE(check), check)) {
	128	if (ATTR_TRUE(check[0].value))
	129	return 0;
	130	convert = ATTR_TRUE(check[1].value);
	131	}
562e25ab RS	132
	133	if (S_ISDIR(mode) \|\| S_ISGITLINK(mode)) {
	134	strbuf_addch(&path, '/');
	135	if (args->verbose)
	136	fprintf(stderr, "%.*s\n", (int)path.len, path.buf);
	137	err = write_entry(args, sha1, path.buf, path.len, mode, NULL, 0);
	138	if (err)
	139	return err;
d3bee161	140	return (S_ISDIR(mode) ? READ_TREE_RECURSIVE : 0);
562e25ab RS	141	}
562e25ab RS	142
1d11d5bb RS	143	buffer = sha1_file_to_archive(path_without_prefix, sha1, mode,
1d11d5bb RS	144	&type, &size, convert ? args->commit : NULL);
562e25ab RS	145	if (!buffer)
	146	return error("cannot read %s", sha1_to_hex(sha1));
	147	if (args->verbose)
	148	fprintf(stderr, "%.*s\n", (int)path.len, path.buf);
	149	err = write_entry(args, sha1, path.buf, path.len, mode, buffer, size);
	150	free(buffer);
	151	return err;
	152	}
	153
	154	int write_archive_entries(struct archiver_args *args,
	155	write_archive_entry_fn_t write_entry)
	156	{
	157	struct archiver_context context;
ba053ea9 NTND	158	struct unpack_trees_options opts;
ba053ea9 NTND	159	struct tree_desc t;
f0096c06	160	struct pathspec pathspec;
562e25ab RS	161	int err;
	162
	163	if (args->baselen > 0 && args->base[args->baselen - 1] == '/') {
	164	size_t len = args->baselen;
	165
	166	while (len > 1 && args->base[len - 2] == '/')
	167	len--;
	168	if (args->verbose)
	169	fprintf(stderr, "%.*s\n", (int)len, args->base);
	170	err = write_entry(args, args->tree->object.sha1, args->base,
	171	len, 040777, NULL, 0);
	172	if (err)
	173	return err;
	174	}
	175
	176	context.args = args;
	177	context.write_entry = write_entry;
	178
ba053ea9 NTND	179	/*
	180	* Setup index and instruct attr to read index only
	181	*/
	182	if (!args->worktree_attributes) {
	183	memset(&opts, 0, sizeof(opts));
	184	opts.index_only = 1;
	185	opts.head_idx = -1;
	186	opts.src_index = &the_index;
	187	opts.dst_index = &the_index;
	188	opts.fn = oneway_merge;
	189	init_tree_desc(&t, args->tree->buffer, args->tree->size);
	190	if (unpack_trees(1, &t, &opts))
	191	return -1;
	192	git_attr_set_direction(GIT_ATTR_INDEX, &the_index);
	193	}
	194
f0096c06 NTND	195	init_pathspec(&pathspec, args->pathspec);
f0096c06 NTND	196	err = read_tree_recursive(args->tree, "", 0, 0, &pathspec,
ebfbdb34	197	write_archive_entry, &context);
f0096c06	198	free_pathspec(&pathspec);
562e25ab RS	199	if (err == READ_TREE_RECURSIVE)
	200	err = 0;
	201	return err;
	202	}
6e94e683	203
c0885435 RS	204	static const struct archiver lookup_archiver(const char name)
	205	{
	206	int i;
	207
4fac1d3a RS	208	if (!name)
	209	return NULL;
	210
c0885435 RS	211	for (i = 0; i < ARRAY_SIZE(archivers); i++) {
	212	if (!strcmp(name, archivers[i].name))
	213	return &archivers[i];
	214	}
	215	return NULL;
	216	}
	217
d5f53d6d RS	218	static int reject_entry(const unsigned char sha1, const char base,
	219	int baselen, const char *filename, unsigned mode,
	220	int stage, void *context)
	221	{
	222	return -1;
	223	}
	224
	225	static int path_exists(struct tree tree, const char path)
	226	{
f0096c06 NTND	227	const char *paths[] = { path, NULL };
	228	struct pathspec pathspec;
	229	int ret;
	230
	231	init_pathspec(&pathspec, paths);
	232	ret = read_tree_recursive(tree, "", 0, 0, &pathspec, reject_entry, NULL);
	233	free_pathspec(&pathspec);
	234	return ret != 0;
d5f53d6d RS	235	}
d5f53d6d RS	236
c0885435 RS	237	static void parse_pathspec_arg(const char **pathspec,
	238	struct archiver_args *ar_args)
	239	{
d5f53d6d RS	240	ar_args->pathspec = pathspec = get_pathspec("", pathspec);
	241	if (pathspec) {
	242	while (*pathspec) {
	243	if (!path_exists(ar_args->tree, *pathspec))
	244	die("path not found: %s", *pathspec);
	245	pathspec++;
	246	}
	247	}
c0885435 RS	248	}
	249
	250	static void parse_treeish_arg(const char **argv,
	251	struct archiver_args ar_args, const char prefix)
	252	{
	253	const char *name = argv[0];
	254	const unsigned char *commit_sha1;
	255	time_t archive_time;
	256	struct tree *tree;
	257	const struct commit *commit;
	258	unsigned char sha1[20];
	259
	260	if (get_sha1(name, sha1))
	261	die("Not a valid object name");
	262
	263	commit = lookup_commit_reference_gently(sha1, 1);
	264	if (commit) {
	265	commit_sha1 = commit->object.sha1;
	266	archive_time = commit->date;
	267	} else {
	268	commit_sha1 = NULL;
	269	archive_time = time(NULL);
	270	}
	271
	272	tree = parse_tree_indirect(sha1);
	273	if (tree == NULL)
	274	die("not a tree object");
	275
	276	if (prefix) {
	277	unsigned char tree_sha1[20];
	278	unsigned int mode;
	279	int err;
	280
	281	err = get_tree_entry(tree->object.sha1, prefix,
	282	tree_sha1, &mode);
	283	if (err \|\| !S_ISDIR(mode))
	284	die("current working directory is untracked");
	285
	286	tree = parse_tree_indirect(tree_sha1);
	287	}
	288	ar_args->tree = tree;
	289	ar_args->commit_sha1 = commit_sha1;
	290	ar_args->commit = commit;
	291	ar_args->time = archive_time;
	292	}
	293
4fac1d3a RS	294	#define OPT__COMPR(s, v, h, p) \
	295	{ OPTION_SET_INT, (s), NULL, (v), NULL, (h), \
	296	PARSE_OPT_NOARG \| PARSE_OPT_NONEG, NULL, (p) }
	297	#define OPT__COMPR_HIDDEN(s, v, p) \
	298	{ OPTION_SET_INT, (s), NULL, (v), NULL, "", \
	299	PARSE_OPT_NOARG \| PARSE_OPT_NONEG \| PARSE_OPT_HIDDEN, NULL, (p) }
	300
c0885435 RS	301	static int parse_archive_args(int argc, const char **argv,
	302	const struct archiver *ar, struct archiver_args args)
	303	{
	304	const char *format = "tar";
4fac1d3a RS	305	const char *base = NULL;
	306	const char *remote = NULL;
	307	const char *exec = NULL;
aec0c1bb	308	const char *output = NULL;
c0885435 RS	309	int compression_level = -1;
	310	int verbose = 0;
	311	int i;
4fac1d3a	312	int list = 0;
ba053ea9	313	int worktree_attributes = 0;
4fac1d3a RS	314	struct option opts[] = {
	315	OPT_GROUP(""),
	316	OPT_STRING(0, "format", &format, "fmt", "archive format"),
	317	OPT_STRING(0, "prefix", &base, "prefix",
	318	"prepend prefix to each pathname in the archive"),
05d3951e	319	OPT_STRING('o', "output", &output, "file",
aec0c1bb	320	"write the archive to this file"),
ba053ea9 NTND	321	OPT_BOOLEAN(0, "worktree-attributes", &worktree_attributes,
ba053ea9 NTND	322	"read .gitattributes in working directory"),
4ad5c804	323	OPT__VERBOSE(&verbose, "report archived files on stderr"),
4fac1d3a RS	324	OPT__COMPR('0', &compression_level, "store only", 0),
	325	OPT__COMPR('1', &compression_level, "compress faster", 1),
	326	OPT__COMPR_HIDDEN('2', &compression_level, 2),
	327	OPT__COMPR_HIDDEN('3', &compression_level, 3),
	328	OPT__COMPR_HIDDEN('4', &compression_level, 4),
	329	OPT__COMPR_HIDDEN('5', &compression_level, 5),
	330	OPT__COMPR_HIDDEN('6', &compression_level, 6),
	331	OPT__COMPR_HIDDEN('7', &compression_level, 7),
	332	OPT__COMPR_HIDDEN('8', &compression_level, 8),
	333	OPT__COMPR('9', &compression_level, "compress better", 9),
	334	OPT_GROUP(""),
	335	OPT_BOOLEAN('l', "list", &list,
	336	"list supported archive formats"),
	337	OPT_GROUP(""),
	338	OPT_STRING(0, "remote", &remote, "repo",
	339	"retrieve the archive from remote repository <repo>"),
	340	OPT_STRING(0, "exec", &exec, "cmd",
	341	"path to the remote git-upload-archive command"),
	342	OPT_END()
	343	};
	344
37782920	345	argc = parse_options(argc, argv, NULL, opts, archive_usage, 0);
4fac1d3a RS	346
	347	if (remote)
	348	die("Unexpected option --remote");
	349	if (exec)
	350	die("Option --exec can only be used together with --remote");
52e77876 RS	351	if (output)
52e77876 RS	352	die("Unexpected option --output");
4fac1d3a RS	353
	354	if (!base)
	355	base = "";
	356
	357	if (list) {
	358	for (i = 0; i < ARRAY_SIZE(archivers); i++)
	359	printf("%s\n", archivers[i].name);
	360	exit(0);
c0885435 RS	361	}
	362
	363	/* We need at least one parameter -- tree-ish */
4fac1d3a RS	364	if (argc < 1)
4fac1d3a RS	365	usage_with_options(archive_usage, opts);
c0885435 RS	366	*ar = lookup_archiver(format);
	367	if (!*ar)
	368	die("Unknown archive format '%s'", format);
	369
	370	args->compression_level = Z_DEFAULT_COMPRESSION;
	371	if (compression_level != -1) {
	372	if ((*ar)->flags & USES_ZLIB_COMPRESSION)
	373	args->compression_level = compression_level;
	374	else {
	375	die("Argument not supported for format '%s': -%d",
	376	format, compression_level);
	377	}
	378	}
	379	args->verbose = verbose;
	380	args->base = base;
	381	args->baselen = strlen(base);
ba053ea9	382	args->worktree_attributes = worktree_attributes;
c0885435	383
4fac1d3a	384	return argc;
c0885435 RS	385	}
c0885435 RS	386
6e94e683 RS	387	int write_archive(int argc, const char *argv, const char prefix,
	388	int setup_prefix)
	389	{
	390	const struct archiver *ar = NULL;
	391	struct archiver_args args;
6e94e683	392
4fac1d3a	393	argc = parse_archive_args(argc, argv, &ar, &args);
6e94e683 RS	394	if (setup_prefix && prefix == NULL)
	395	prefix = setup_git_directory();
	396
6e94e683 RS	397	parse_treeish_arg(argv, &args, prefix);
	398	parse_pathspec_arg(argv + 1, &args);
	399
ddff8563 CB	400	git_config(git_default_config, NULL);
ddff8563 CB	401
6e94e683 RS	402	return ar->write_archive(&args);
6e94e683 RS	403	}