summaryrefslogtreecommitdiff
path: root/t/helper/test-oidmap.c
blob: 0acf99931ee176982d61078dc0e2d882377c05fd (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
#include "test-tool.h"
#include "cache.h"
#include "oidmap.h"
#include "strbuf.h"
 
/* key is an oid and value is a name (could be a refname for example) */
struct test_entry {
	struct oidmap_entry entry;
	char name[FLEX_ARRAY];
};
 
#define DELIM " \t\r\n"
 
/*
 * Read stdin line by line and print result of commands to stdout:
 *
 * hash oidkey -> sha1hash(oidkey)
 * put oidkey namevalue -> NULL / old namevalue
 * get oidkey -> NULL / namevalue
 * remove oidkey -> NULL / old namevalue
 * iterate -> oidkey1 namevalue1\noidkey2 namevalue2\n...
 *
 */
int cmd__oidmap(int argc, const char **argv)
{
	struct strbuf line = STRBUF_INIT;
	struct oidmap map = OIDMAP_INIT;
 
	setup_git_directory();
 
	/* init oidmap */
	oidmap_init(&map, 0);
 
	/* process commands from stdin */
	while (strbuf_getline(&line, stdin) != EOF) {
		char *cmd, *p1 = NULL, *p2 = NULL;
		struct test_entry *entry;
		struct object_id oid;
 
		/* break line into command and up to two parameters */
		cmd = strtok(line.buf, DELIM);
		/* ignore empty lines */
		if (!cmd || *cmd == '#')
			continue;
 
		p1 = strtok(NULL, DELIM);
		if (p1)
			p2 = strtok(NULL, DELIM);
 
		if (!strcmp("put", cmd) && p1 && p2) {
 
			if (get_oid(p1, &oid)) {
				printf("Unknown oid: %s\n", p1);
				continue;
			}
 
			/* create entry with oid_key = p1, name_value = p2 */
			FLEX_ALLOC_STR(entry, name, p2);
			oidcpy(&entry->entry.oid, &oid);
 
			/* add / replace entry */
			entry = oidmap_put(&map, entry);
 
			/* print and free replaced entry, if any */
			puts(entry ? entry->name : "NULL");
			free(entry);
 
		} else if (!strcmp("get", cmd) && p1) {
 
			if (get_oid(p1, &oid)) {
				printf("Unknown oid: %s\n", p1);
				continue;
			}
 
			/* lookup entry in oidmap */
			entry = oidmap_get(&map, &oid);
 
			/* print result */
			puts(entry ? entry->name : "NULL");
 
		} else if (!strcmp("remove", cmd) && p1) {
 
			if (get_oid(p1, &oid)) {
				printf("Unknown oid: %s\n", p1);
				continue;
			}
 
			/* remove entry from oidmap */
			entry = oidmap_remove(&map, &oid);
 
			/* print result and free entry*/
			puts(entry ? entry->name : "NULL");
			free(entry);
 
		} else if (!strcmp("iterate", cmd)) {
 
			struct oidmap_iter iter;
			oidmap_iter_init(&map, &iter);
			while ((entry = oidmap_iter_next(&iter)))
				printf("%s %s\n", oid_to_hex(&entry->entry.oid), entry->name);
 
		} else {
 
			printf("Unknown command %s\n", cmd);
 
		}
	}
 
	strbuf_release(&line);
	oidmap_free(&map, 1);
	return 0;
}