|
9 | 9 | #include "gettext.h"
|
10 | 10 | #include "parse-options.h"
|
11 | 11 | #include "string-list.h"
|
| 12 | +#include "tempfile.h" |
12 | 13 | #include "trailer.h"
|
13 | 14 | #include "config.h"
|
14 | 15 |
|
@@ -91,6 +92,102 @@ static int parse_opt_parse(const struct option *opt, const char *arg,
|
91 | 92 | return 0;
|
92 | 93 | }
|
93 | 94 |
|
| 95 | +static struct tempfile *trailers_tempfile; |
| 96 | + |
| 97 | +static FILE *create_in_place_tempfile(const char *file) |
| 98 | +{ |
| 99 | + struct stat st; |
| 100 | + struct strbuf filename_template = STRBUF_INIT; |
| 101 | + const char *tail; |
| 102 | + FILE *outfile; |
| 103 | + |
| 104 | + if (stat(file, &st)) |
| 105 | + die_errno(_("could not stat %s"), file); |
| 106 | + if (!S_ISREG(st.st_mode)) |
| 107 | + die(_("file %s is not a regular file"), file); |
| 108 | + if (!(st.st_mode & S_IWUSR)) |
| 109 | + die(_("file %s is not writable by user"), file); |
| 110 | + |
| 111 | + /* Create temporary file in the same directory as the original */ |
| 112 | + tail = strrchr(file, '/'); |
| 113 | + if (tail) |
| 114 | + strbuf_add(&filename_template, file, tail - file + 1); |
| 115 | + strbuf_addstr(&filename_template, "git-interpret-trailers-XXXXXX"); |
| 116 | + |
| 117 | + trailers_tempfile = xmks_tempfile_m(filename_template.buf, st.st_mode); |
| 118 | + strbuf_release(&filename_template); |
| 119 | + outfile = fdopen_tempfile(trailers_tempfile, "w"); |
| 120 | + if (!outfile) |
| 121 | + die_errno(_("could not open temporary file")); |
| 122 | + |
| 123 | + return outfile; |
| 124 | +} |
| 125 | + |
| 126 | +static void read_input_file(struct strbuf *sb, const char *file) |
| 127 | +{ |
| 128 | + if (file) { |
| 129 | + if (strbuf_read_file(sb, file, 0) < 0) |
| 130 | + die_errno(_("could not read input file '%s'"), file); |
| 131 | + } else { |
| 132 | + if (strbuf_read(sb, fileno(stdin), 0) < 0) |
| 133 | + die_errno(_("could not read from stdin")); |
| 134 | + } |
| 135 | +} |
| 136 | + |
| 137 | +static void interpret_trailers(const struct process_trailer_options *opts, |
| 138 | + struct list_head *new_trailer_head, |
| 139 | + const char *file) |
| 140 | +{ |
| 141 | + LIST_HEAD(head); |
| 142 | + struct strbuf sb = STRBUF_INIT; |
| 143 | + struct strbuf trailer_block = STRBUF_INIT; |
| 144 | + struct trailer_info info; |
| 145 | + FILE *outfile = stdout; |
| 146 | + |
| 147 | + trailer_config_init(); |
| 148 | + |
| 149 | + read_input_file(&sb, file); |
| 150 | + |
| 151 | + if (opts->in_place) |
| 152 | + outfile = create_in_place_tempfile(file); |
| 153 | + |
| 154 | + parse_trailers(opts, &info, sb.buf, &head); |
| 155 | + |
| 156 | + /* Print the lines before the trailers */ |
| 157 | + if (!opts->only_trailers) |
| 158 | + fwrite(sb.buf, 1, info.trailer_block_start, outfile); |
| 159 | + |
| 160 | + if (!opts->only_trailers && !info.blank_line_before_trailer) |
| 161 | + fprintf(outfile, "\n"); |
| 162 | + |
| 163 | + |
| 164 | + if (!opts->only_input) { |
| 165 | + LIST_HEAD(config_head); |
| 166 | + LIST_HEAD(arg_head); |
| 167 | + parse_trailers_from_config(&config_head); |
| 168 | + parse_trailers_from_command_line_args(&arg_head, new_trailer_head); |
| 169 | + list_splice(&config_head, &arg_head); |
| 170 | + process_trailers_lists(&head, &arg_head); |
| 171 | + } |
| 172 | + |
| 173 | + /* Print trailer block. */ |
| 174 | + format_trailers(opts, &head, &trailer_block); |
| 175 | + free_trailers(&head); |
| 176 | + fwrite(trailer_block.buf, 1, trailer_block.len, outfile); |
| 177 | + strbuf_release(&trailer_block); |
| 178 | + |
| 179 | + /* Print the lines after the trailers as is */ |
| 180 | + if (!opts->only_trailers) |
| 181 | + fwrite(sb.buf + info.trailer_block_end, 1, sb.len - info.trailer_block_end, outfile); |
| 182 | + trailer_info_release(&info); |
| 183 | + |
| 184 | + if (opts->in_place) |
| 185 | + if (rename_tempfile(&trailers_tempfile, file)) |
| 186 | + die_errno(_("could not rename temporary file to %s"), file); |
| 187 | + |
| 188 | + strbuf_release(&sb); |
| 189 | +} |
| 190 | + |
94 | 191 | int cmd_interpret_trailers(int argc, const char **argv, const char *prefix)
|
95 | 192 | {
|
96 | 193 | struct process_trailer_options opts = PROCESS_TRAILER_OPTIONS_INIT;
|
@@ -132,11 +229,11 @@ int cmd_interpret_trailers(int argc, const char **argv, const char *prefix)
|
132 | 229 | if (argc) {
|
133 | 230 | int i;
|
134 | 231 | for (i = 0; i < argc; i++)
|
135 |
| - process_trailers(argv[i], &opts, &trailers); |
| 232 | + interpret_trailers(&opts, &trailers, argv[i]); |
136 | 233 | } else {
|
137 | 234 | if (opts.in_place)
|
138 | 235 | die(_("no input file given for in-place editing"));
|
139 |
| - process_trailers(NULL, &opts, &trailers); |
| 236 | + interpret_trailers(&opts, &trailers, NULL); |
140 | 237 | }
|
141 | 238 |
|
142 | 239 | new_trailers_clear(&trailers);
|
|
0 commit comments