archive: unify file attribute handling
[git] / archive.c
1 #include "cache.h"
2 #include "commit.h"
3 #include "attr.h"
4 #include "archive.h"
5
6 static void format_subst(const struct commit *commit,
7                          const char *src, size_t len,
8                          struct strbuf *buf)
9 {
10         char *to_free = NULL;
11         struct strbuf fmt;
12
13         if (src == buf->buf)
14                 to_free = strbuf_detach(buf, NULL);
15         strbuf_init(&fmt, 0);
16         for (;;) {
17                 const char *b, *c;
18
19                 b = memmem(src, len, "$Format:", 8);
20                 if (!b)
21                         break;
22                 c = memchr(b + 8, '$', (src + len) - b - 8);
23                 if (!c)
24                         break;
25
26                 strbuf_reset(&fmt);
27                 strbuf_add(&fmt, b + 8, c - b - 8);
28
29                 strbuf_add(buf, src, b - src);
30                 format_commit_message(commit, fmt.buf, buf);
31                 len -= c + 1 - src;
32                 src  = c + 1;
33         }
34         strbuf_add(buf, src, len);
35         strbuf_release(&fmt);
36         free(to_free);
37 }
38
39 static void *sha1_file_to_archive(const char *path, const unsigned char *sha1,
40                 unsigned int mode, enum object_type *type,
41                 unsigned long *sizep, const struct commit *commit)
42 {
43         void *buffer;
44
45         buffer = read_sha1_file(sha1, type, sizep);
46         if (buffer && S_ISREG(mode)) {
47                 struct strbuf buf;
48                 size_t size = 0;
49
50                 strbuf_init(&buf, 0);
51                 strbuf_attach(&buf, buffer, *sizep, *sizep + 1);
52                 convert_to_working_tree(path, buf.buf, buf.len, &buf);
53                 if (commit)
54                         format_subst(commit, buf.buf, buf.len, &buf);
55                 buffer = strbuf_detach(&buf, &size);
56                 *sizep = size;
57         }
58
59         return buffer;
60 }
61
62 static void setup_archive_check(struct git_attr_check *check)
63 {
64         static struct git_attr *attr_export_ignore;
65         static struct git_attr *attr_export_subst;
66
67         if (!attr_export_ignore) {
68                 attr_export_ignore = git_attr("export-ignore", 13);
69                 attr_export_subst = git_attr("export-subst", 12);
70         }
71         check[0].attr = attr_export_ignore;
72         check[1].attr = attr_export_subst;
73 }
74
75 struct archiver_context {
76         struct archiver_args *args;
77         write_archive_entry_fn_t write_entry;
78 };
79
80 static int write_archive_entry(const unsigned char *sha1, const char *base,
81                 int baselen, const char *filename, unsigned mode, int stage,
82                 void *context)
83 {
84         static struct strbuf path = STRBUF_INIT;
85         struct archiver_context *c = context;
86         struct archiver_args *args = c->args;
87         write_archive_entry_fn_t write_entry = c->write_entry;
88         struct git_attr_check check[2];
89         const char *path_without_prefix;
90         int convert = 0;
91         int err;
92         enum object_type type;
93         unsigned long size;
94         void *buffer;
95
96         strbuf_reset(&path);
97         strbuf_grow(&path, PATH_MAX);
98         strbuf_add(&path, base, baselen);
99         strbuf_addstr(&path, filename);
100         path_without_prefix = path.buf + args->baselen;
101
102         setup_archive_check(check);
103         if (!git_checkattr(path_without_prefix, ARRAY_SIZE(check), check)) {
104                 if (ATTR_TRUE(check[0].value))
105                         return 0;
106                 convert = ATTR_TRUE(check[1].value);
107         }
108
109         if (S_ISDIR(mode) || S_ISGITLINK(mode)) {
110                 strbuf_addch(&path, '/');
111                 if (args->verbose)
112                         fprintf(stderr, "%.*s\n", (int)path.len, path.buf);
113                 err = write_entry(args, sha1, path.buf, path.len, mode, NULL, 0);
114                 if (err)
115                         return err;
116                 return READ_TREE_RECURSIVE;
117         }
118
119         buffer = sha1_file_to_archive(path_without_prefix, sha1, mode,
120                         &type, &size, convert ? args->commit : NULL);
121         if (!buffer)
122                 return error("cannot read %s", sha1_to_hex(sha1));
123         if (args->verbose)
124                 fprintf(stderr, "%.*s\n", (int)path.len, path.buf);
125         err = write_entry(args, sha1, path.buf, path.len, mode, buffer, size);
126         free(buffer);
127         return err;
128 }
129
130 int write_archive_entries(struct archiver_args *args,
131                 write_archive_entry_fn_t write_entry)
132 {
133         struct archiver_context context;
134         int err;
135
136         if (args->baselen > 0 && args->base[args->baselen - 1] == '/') {
137                 size_t len = args->baselen;
138
139                 while (len > 1 && args->base[len - 2] == '/')
140                         len--;
141                 if (args->verbose)
142                         fprintf(stderr, "%.*s\n", (int)len, args->base);
143                 err = write_entry(args, args->tree->object.sha1, args->base,
144                                 len, 040777, NULL, 0);
145                 if (err)
146                         return err;
147         }
148
149         context.args = args;
150         context.write_entry = write_entry;
151
152         err =  read_tree_recursive(args->tree, args->base, args->baselen, 0,
153                         args->pathspec, write_archive_entry, &context);
154         if (err == READ_TREE_RECURSIVE)
155                 err = 0;
156         return err;
157 }