summaryrefslogtreecommitdiff
path: root/src/cli/cmd_hash_object.c
blob: 5cfe9146a6bdcbfe73f56ed940142c80fb7f92ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
/*
 * Copyright (C) the libgit2 contributors. All rights reserved.
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */

#include <git2.h>
#include "cli.h"
#include "cmd.h"

#include "futils.h"

#define COMMAND_NAME "hash-object"

static int show_help;
static char *type_name;
static int write_object, read_stdin, literally;
static char **filenames;

static const cli_opt_spec opts[] = {
	{ CLI_OPT_TYPE_SWITCH,   "help",      0, &show_help,    1,
	  CLI_OPT_USAGE_HIDDEN | CLI_OPT_USAGE_STOP_PARSING, NULL,
	  "display help about the " COMMAND_NAME " command" },

	{ CLI_OPT_TYPE_VALUE,     NULL,      't', &type_name,    0,
	  CLI_OPT_USAGE_DEFAULT, "type",     "the type of object to hash (default: \"blob\")" },
	{ CLI_OPT_TYPE_SWITCH,    NULL,      'w', &write_object, 1,
	  CLI_OPT_USAGE_DEFAULT,  NULL,      "write the object to the object database" },
	{ CLI_OPT_TYPE_SWITCH,   "literally", 0, &literally,    1,
	  CLI_OPT_USAGE_DEFAULT,  NULL,      "do not validate the object contents" },
	{ CLI_OPT_TYPE_SWITCH,   "stdin",     0, &read_stdin,   1,
	  CLI_OPT_USAGE_REQUIRED, NULL,      "read content from stdin" },
	{ CLI_OPT_TYPE_ARGS,     "file",      0, &filenames,    0,
	  CLI_OPT_USAGE_CHOICE,  "file",     "the file (or files) to read and hash" },
	{ 0 },
};

static void print_help(void)
{
	cli_opt_usage_fprint(stdout, PROGRAM_NAME, COMMAND_NAME, opts);
	printf("\n");

	printf("Compute the object ID for a given file and optionally write that file\nto the object database.\n");
	printf("\n");

	printf("Options:\n");

	cli_opt_help_fprint(stdout, opts);
}

static int hash_buf(git_odb *odb, git_str *buf, git_object_t type)
{
	git_oid oid;

	if (!literally) {
		int valid = 0;

		if (git_object_rawcontent_is_valid(&valid, buf->ptr, buf->size, type) < 0 || !valid)
			return cli_error_git();
	}

	if (write_object) {
		if (git_odb_write(&oid, odb, buf->ptr, buf->size, type) < 0)
			return cli_error_git();
	} else {
		if (git_odb_hash(&oid, buf->ptr, buf->size, type) < 0)
			return cli_error_git();
	}

	if (printf("%s\n", git_oid_tostr_s(&oid)) < 0)
		return cli_error_os();

	return 0;
}

int cmd_hash_object(int argc, char **argv)
{
	git_repository *repo = NULL;
	git_odb *odb = NULL;
	git_str buf = GIT_STR_INIT;
	cli_opt invalid_opt;
	git_object_t type = GIT_OBJECT_BLOB;
	char **filename;
	int ret = 0;

	if (cli_opt_parse(&invalid_opt, opts, argv + 1, argc - 1, CLI_OPT_PARSE_GNU))
		return cli_opt_usage_error(COMMAND_NAME, opts, &invalid_opt);

	if (show_help) {
		print_help();
		return 0;
	}

	if (type_name && (type = git_object_string2type(type_name)) == GIT_OBJECT_INVALID)
		return cli_error_usage("invalid object type '%s'", type_name);

	if (write_object &&
	    (git_repository_open_ext(&repo, ".", GIT_REPOSITORY_OPEN_FROM_ENV, NULL) < 0 ||
	     git_repository_odb(&odb, repo) < 0)) {
		ret = cli_error_git();
		goto done;
	}

	/*
	 * TODO: we're reading blobs, we shouldn't pull them all into main
	 * memory, we should just stream them into the odb instead.
	 * (Or create a `git_odb_writefile` API.)
	 */
	if (read_stdin) {
		if (git_futils_readbuffer_fd_full(&buf, fileno(stdin)) < 0) {
			ret = cli_error_git();
			goto done;
		}

		if ((ret = hash_buf(odb, &buf, type)) != 0)
			goto done;
	} else {
		for (filename = filenames; *filename; filename++) {
			if (git_futils_readbuffer(&buf, *filename) < 0) {
				ret = cli_error_git();
				goto done;
			}

			if ((ret = hash_buf(odb, &buf, type)) != 0)
				goto done;
		}
	}

done:
	git_str_dispose(&buf);
	git_odb_free(odb);
	git_repository_free(repo);
	return ret;
}