git am updates the work tree and so should use the smudgeToFile filter.

This includes some refactoring into convert_to_working_tree_filter_to_file
to make it check the file after running the smudgeToFile command, and clean
up from a failing command.

Signed-off-by: Joey Hess <jo...@joeyh.name>
---
 apply.c               | 16 ++++++++++++++++
 convert.c             | 25 +++++++++++++++++++++++--
 entry.c               | 21 ++++-----------------
 t/t0021-conversion.sh | 13 +++++++++++++
 4 files changed, 56 insertions(+), 19 deletions(-)

diff --git a/apply.c b/apply.c
index 4a6b2db..7db8344 100644
--- a/apply.c
+++ b/apply.c
@@ -4322,6 +4322,22 @@ static int try_create_file(const char *path, unsigned 
int mode, const char *buf,
        if (fd < 0)
                return 1;
 
+       if (can_smudge_to_file(path)) {
+               close(fd);
+               fd = convert_to_working_tree_filter_to_file(path, path, buf, 
size);
+               if (fd < 0) {
+                       /* smudgeToFile filter failed; continue
+                        * with regular file creation instead. */
+                       fd = open(path, O_CREAT | O_EXCL | O_WRONLY, (mode & 
0100) ? 0777 : 0666);
+                       if (fd < 0)
+                               return -1;
+               }
+               else {
+                       close(fd);
+                       return 0;
+               }
+       }
+
        if (convert_to_working_tree(path, buf, size, &nbuf)) {
                size = nbuf.len;
                buf  = nbuf.buf;
diff --git a/convert.c b/convert.c
index e1b0b44..3746ad5 100644
--- a/convert.c
+++ b/convert.c
@@ -1030,13 +1030,34 @@ int convert_to_working_tree(const char *path, const 
char *src, size_t len, struc
        return convert_to_working_tree_internal(path, NULL, src, len, dst, 0);
 }
 
+/*
+ * Returns fd open to read the worktree file on success.
+ * On failure, the worktree file will not exist.
+ */
 int convert_to_working_tree_filter_to_file(const char *path, const char 
*destpath, const char *src, size_t len)
 {
        struct strbuf output = STRBUF_INIT;
-       int ret = convert_to_working_tree_internal(path, destpath, src, len, 
&output, 0);
+       int ok = convert_to_working_tree_internal(path, destpath, src, len, 
&output, 0);
        /* The smudgeToFile filter stdout is not used. */
        strbuf_release(&output);
-       return ret;
+       if (ok) {
+               /*
+                * Open the file to make sure that it's present
+                * (and readable) after the command populated it.
+                */
+               int fd = open(path, O_RDONLY);
+               if (fd < 0)
+                       unlink(path);
+               return fd;
+       }
+       else {
+               /*
+                * The command could have created the file before failing,
+                * so delete it.
+                */
+               unlink(path);
+               return -1;
+       }
 }
 
 int renormalize_buffer(const char *path, const char *src, size_t len, struct 
strbuf *dst)
diff --git a/entry.c b/entry.c
index 7811e31..40662eb 100644
--- a/entry.c
+++ b/entry.c
@@ -191,34 +191,21 @@ static int write_entry(struct cache_entry *ce,
 
                if (smudge_to_file) {
                        close(fd);
-                       if (convert_to_working_tree_filter_to_file(ce->name, 
path, new, size)) {
+                       fd = convert_to_working_tree_filter_to_file(ce->name, 
path, new, size);
+                       if (fd >= 0) {
                                free(new);
-                               /*
-                                * The smudgeToFile filter may have replaced
-                                * or deleted the file; reopen it to make
-                                * sure that the file exists.
-                                */
-                               fd = open(path, O_RDONLY);
-                               if (fd < 0)
-                                       return error_errno("unable to create 
file %s", path);
                                if (!to_tempfile)
                                        fstat_done = fstat_output(fd, state, 
&st);
                                close(fd);
                        }
                        else {
-                               /*
-                                * The failing smudgeToFile filter may have
-                                * deleted or replaced the file; delete
-                                * the file and re-open for recovery write.
-                                */
-                               unlink(path);
+                               /* Fall through to normal write below. */
+                               smudge_to_file = 0;
                                fd = open_output_fd(path, ce, to_tempfile);
                                if (fd < 0) {
                                        free(new);
                                        return error_errno("unable to create 
file %s", path);
                                }
-                               /* Fall through to normal write below. */
-                               smudge_to_file = 0;
                        }
                }
 
diff --git a/t/t0021-conversion.sh b/t/t0021-conversion.sh
index 0efad9b..42b28aa 100755
--- a/t/t0021-conversion.sh
+++ b/t/t0021-conversion.sh
@@ -334,6 +334,19 @@ test_expect_success 'recovery from failure of smudgeToFile 
filter that deletes t
        test_cmp test fstest.t
 '
 
+test_expect_success 'smudgeToFile filter is used by git am' '
+       test_config filter.rot13.smudgeToFile ./rot13-to-file.sh &&
+
+       git commit fstest.t -m "added fstest.t" &&
+       git format-patch HEAD^ --stdout >fstest.patch &&
+       git reset --hard HEAD^ &&
+       git am fstest.patch &&
+
+       test -e rot13-to-file.ran &&
+       rm -f rot13-to-file.ran &&
+       test_cmp test fstest.t
+'
+
 test_expect_success 'cleanFromFile filter is not used when clean filter is not 
configured' '
        test_config filter.noclean.smudge ./rot13.sh &&
        test_config filter.noclean.cleanFromFile ./rot13-from-file.sh &&
-- 
2.8.1

--
To unsubscribe from this list: send the line "unsubscribe git" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to