summaryrefslogtreecommitdiff
path: root/merge-cache.c
blob: 3aa391be0a236ea1c432dadcd14824c2fe2b44ea (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
#include <sys/types.h>
#include <sys/wait.h>
 
#include "cache.h"
 
static const char *pgm = NULL;
static const char *arguments[5];
 
static void run_program(void)
{
	int pid = fork(), status;
 
	if (pid < 0)
		die("unable to fork");
	if (!pid) {
		execlp(pgm, arguments[0],
			    arguments[1],
			    arguments[2],
			    arguments[3],
			    arguments[4],
			    NULL);
		die("unable to execute '%s'", pgm);
	}
	if (waitpid(pid, &status, 0) < 0 || !WIFEXITED(status) || WEXITSTATUS(status))
		die("merge program failed");
}
 
static char *create_temp_file(int stage, unsigned char *sha1)
{
	static char template[4][50];
	char *path = template[stage];
	void *buf;
	char type[100];
	unsigned long size;
	int fd;
 
	buf = read_sha1_file(sha1, type, &size);
	if (!buf || strcmp(type, "blob"))
		die("unable to read blob object %s", sha1_to_hex(sha1));
 
	strcpy(path, ".merge_file_XXXXXX");
	fd = mkstemp(path);
	if (fd < 0)
		die("unable to create temp-file");
	if (write(fd, buf, size) != size)
		die("unable to write temp-file");
	close(fd);
	return path;
}
 
static int merge_entry(int pos, const char *path)
{
	int found;
 
	if (pos >= active_nr)
		die("merge-cache: %s not in the cache", path);
	arguments[0] = pgm;
	arguments[1] = "";
	arguments[2] = "";
	arguments[3] = "";
	arguments[4] = path;
	found = 0;
	do {
		struct cache_entry *ce = active_cache[pos];
		int stage = ce_stage(ce);
 
		if (strcmp(ce->name, path))
			break;
		found++;
		arguments[stage] = create_temp_file(stage, ce->sha1);
	} while (++pos < active_nr);
	if (!found)
		die("merge-cache: %s not in the cache", path);
	run_program();
	return found;
}
 
static void merge_file(const char *path)
{
	int pos = cache_name_pos(path, strlen(path));
 
	/*
	 * If it already exists in the cache as stage0, it's
	 * already merged and there is nothing to do.
	 */
	if (pos < 0)
		merge_entry(-pos-1, path);
}
 
static void merge_all(void)
{
	int i;
	for (i = 0; i < active_nr; i++) {
		struct cache_entry *ce = active_cache[i];
		if (!ce_stage(ce))
			continue;
		i += merge_entry(i, ce->name)-1;
	}
}
 
int main(int argc, char **argv)
{
	int i, force_file = 0;
 
	if (argc < 3)
		usage("merge-cache <merge-program> (-a | <filename>*)");
 
	read_cache();
 
	pgm = argv[1];
	for (i = 2; i < argc; i++) {
		char *arg = argv[i];
		if (!force_file && *arg == '-') {
			if (!strcmp(arg, "--")) {
				force_file = 1;
				continue;
			}
			if (!strcmp(arg, "-a")) {
				merge_all();
				continue;
			}
			die("merge-cache: unknown option %s", arg);
		}
		merge_file(arg);
	}
	return 0;
}