diff-tree: support list if input trees on stdin
[git] / commit-tree.c
1 /*
2  * GIT - The information manager from hell
3  *
4  * Copyright (C) Linus Torvalds, 2005
5  */
6 #include "cache.h"
7
8 #include <pwd.h>
9 #include <time.h>
10 #include <string.h>
11 #include <ctype.h>
12 #include <time.h>
13
14 #define BLOCKING (1ul << 14)
15
16 /*
17  * FIXME! Share the code with "write-tree.c"
18  */
19 static void init_buffer(char **bufp, unsigned int *sizep)
20 {
21         char *buf = xmalloc(BLOCKING);
22         *sizep = 0;
23         *bufp = buf;
24 }
25
26 static void add_buffer(char **bufp, unsigned int *sizep, const char *fmt, ...)
27 {
28         char one_line[2048];
29         va_list args;
30         int len;
31         unsigned long alloc, size, newsize;
32         char *buf;
33
34         va_start(args, fmt);
35         len = vsnprintf(one_line, sizeof(one_line), fmt, args);
36         va_end(args);
37         size = *sizep;
38         newsize = size + len;
39         alloc = (size + 32767) & ~32767;
40         buf = *bufp;
41         if (newsize > alloc) {
42                 alloc = (newsize + 32767) & ~32767;
43                 buf = xrealloc(buf, alloc);
44                 *bufp = buf;
45         }
46         *sizep = newsize;
47         memcpy(buf + size, one_line, len);
48 }
49
50 static void remove_special(char *p)
51 {
52         char c;
53         char *dst = p, *src = p;
54
55         for (;;) {
56                 c = *src;
57                 src++;
58                 switch(c) {
59                 case '\n': case '<': case '>':
60                         continue;
61                 }
62                 *dst++ = c;
63                 if (!c)
64                         break;
65         }
66
67         /*
68          * Go back, and remove crud from the end: some people
69          * have commas etc in their gecos field
70          */
71         dst--;
72         while (--dst >= p) {
73                 unsigned char c = *dst;
74                 switch (c) {
75                 case ',': case ';': case '.':
76                         *dst = 0;
77                         continue;
78                 }
79                 break;
80         }
81 }
82
83 static void check_valid(unsigned char *sha1, const char *expect)
84 {
85         void *buf;
86         char type[20];
87         unsigned long size;
88
89         buf = read_sha1_file(sha1, type, &size);
90         if (!buf || strcmp(type, expect))
91                 die("%s is not a valid '%s' object", sha1_to_hex(sha1), expect);
92         free(buf);
93 }
94
95 /*
96  * Having more than two parents is not strange at all, and this is
97  * how multi-way merges are represented.
98  */
99 #define MAXPARENT (16)
100
101 static char *commit_tree_usage = "commit-tree <sha1> [-p <sha1>]* < changelog";
102
103 int main(int argc, char **argv)
104 {
105         int i, len;
106         int parents = 0;
107         unsigned char tree_sha1[20];
108         unsigned char parent_sha1[MAXPARENT][20];
109         unsigned char commit_sha1[20];
110         char *gecos, *realgecos, *commitgecos;
111         char *email, *commitemail, realemail[1000];
112         char date[20], realdate[20];
113         char *audate;
114         char comment[1000];
115         struct passwd *pw;
116         char *buffer;
117         unsigned int size;
118
119         if (argc < 2 || get_sha1_hex(argv[1], tree_sha1) < 0)
120                 usage(commit_tree_usage);
121
122         check_valid(tree_sha1, "tree");
123         for (i = 2; i < argc; i += 2) {
124                 char *a, *b;
125                 a = argv[i]; b = argv[i+1];
126                 if (!b || strcmp(a, "-p") || get_sha1(b, parent_sha1[parents]))
127                         usage(commit_tree_usage);
128                 check_valid(parent_sha1[parents], "commit");
129                 parents++;
130         }
131         if (!parents)
132                 fprintf(stderr, "Committing initial tree %s\n", argv[1]);
133         pw = getpwuid(getuid());
134         if (!pw)
135                 die("You don't exist. Go away!");
136         realgecos = pw->pw_gecos;
137         len = strlen(pw->pw_name);
138         memcpy(realemail, pw->pw_name, len);
139         realemail[len] = '@';
140         gethostname(realemail+len+1, sizeof(realemail)-len-1);
141         if (!strchr(realemail+len+1, '.')) {
142                 strcat(realemail, ".");
143                 getdomainname(realemail+strlen(realemail), sizeof(realemail)-strlen(realemail)-1);
144         }
145
146         datestamp(realdate, sizeof(realdate));
147         strcpy(date, realdate);
148
149         commitgecos = getenv("COMMIT_AUTHOR_NAME") ? : realgecos;
150         commitemail = getenv("COMMIT_AUTHOR_EMAIL") ? : realemail;
151         gecos = getenv("AUTHOR_NAME") ? : realgecos;
152         email = getenv("AUTHOR_EMAIL") ? : realemail;
153         audate = getenv("AUTHOR_DATE");
154         if (audate)
155                 parse_date(audate, date, sizeof(date));
156
157         remove_special(gecos); remove_special(realgecos); remove_special(commitgecos);
158         remove_special(email); remove_special(realemail); remove_special(commitemail);
159
160         init_buffer(&buffer, &size);
161         add_buffer(&buffer, &size, "tree %s\n", sha1_to_hex(tree_sha1));
162
163         /*
164          * NOTE! This ordering means that the same exact tree merged with a
165          * different order of parents will be a _different_ changeset even
166          * if everything else stays the same.
167          */
168         for (i = 0; i < parents; i++)
169                 add_buffer(&buffer, &size, "parent %s\n", sha1_to_hex(parent_sha1[i]));
170
171         /* Person/date information */
172         add_buffer(&buffer, &size, "author %s <%s> %s\n", gecos, email, date);
173         add_buffer(&buffer, &size, "committer %s <%s> %s\n\n", commitgecos, commitemail, realdate);
174
175         /* And add the comment */
176         while (fgets(comment, sizeof(comment), stdin) != NULL)
177                 add_buffer(&buffer, &size, "%s", comment);
178
179         write_sha1_file(buffer, size, "commit", commit_sha1);
180         printf("%s\n", sha1_to_hex(commit_sha1));
181         return 0;
182 }