This makes git checkout, git reset, etc use smudgeToFile.

Includes test cases.

(There's a call to convert_to_working_tree in merge-recursive.c
that could also be made to use smudgeToFile as well.)

Signed-off-by: Joey Hess <jo...@joeyh.name>
---
 entry.c               | 40 ++++++++++++++++++++++++++++++++--------
 t/t0021-conversion.sh | 34 ++++++++++++++++++++++++++++++++--
 2 files changed, 64 insertions(+), 10 deletions(-)

diff --git a/entry.c b/entry.c
index 519e042..81d12a1 100644
--- a/entry.c
+++ b/entry.c
@@ -146,6 +146,7 @@ static int write_entry(struct cache_entry *ce,
        unsigned long size;
        size_t wrote, newsize = 0;
        struct stat st;
+       int regular_file, smudge_to_file;
 
        if (ce_mode_s_ifmt == S_IFREG) {
                struct stream_filter *filter = get_stream_filter(ce->name, 
ce->sha1);
@@ -175,8 +176,13 @@ static int write_entry(struct cache_entry *ce,
 
                /*
                 * Convert from git internal format to working tree format
+                * unless the smudgeToFile filter can write to the
+                * file directly.
                 */
-               if (ce_mode_s_ifmt == S_IFREG &&
+               regular_file = ce_mode_s_ifmt == S_IFREG;
+               smudge_to_file = regular_file
+                       && can_smudge_to_file(ce->name);
+               if (regular_file && !smudge_to_file &&
                    convert_to_working_tree(ce->name, new, size, &buf)) {
                        free(new);
                        new = strbuf_detach(&buf, &newsize);
@@ -189,13 +195,31 @@ static int write_entry(struct cache_entry *ce,
                        return error_errno("unable to create file %s", path);
                }
 
-               wrote = write_in_full(fd, new, size);
-               if (!to_tempfile)
-                       fstat_done = fstat_output(fd, state, &st);
-               close(fd);
-               free(new);
-               if (wrote != size)
-                       return error("unable to write file %s", path);
+               if (!smudge_to_file) {
+                       wrote = write_in_full(fd, new, size);
+                       if (!to_tempfile)
+                               fstat_done = fstat_output(fd, state, &st);
+                       close(fd);
+                       free(new);
+                       if (wrote != size)
+                               return error("unable to write file %s", path);
+               }
+               else {
+                       close(fd);
+                       convert_to_working_tree_filter_to_file(ce->name, path, 
new, size);
+                       free(new);
+                       /*
+                        * The smudgeToFile filter may have replaced the
+                        * file; open 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);
+               }
                break;
        case S_IFGITLINK:
                if (to_tempfile)
diff --git a/t/t0021-conversion.sh b/t/t0021-conversion.sh
index bd84b80..ea18b17 100755
--- a/t/t0021-conversion.sh
+++ b/t/t0021-conversion.sh
@@ -14,12 +14,20 @@ chmod +x rot13.sh
 
 cat <<EOF >rot13-from-file.sh
 #!$SHELL_PATH
-fsfile="\$1"
+srcfile="\$1"
 touch rot13-from-file.ran
-cat "\$fsfile" | ./rot13.sh
+cat "\$srcfile" | ./rot13.sh
 EOF
 chmod +x rot13-from-file.sh
 
+cat <<EOF >rot13-to-file.sh
+#!$SHELL_PATH
+destfile="\$1"
+touch rot13-to-file.ran
+./rot13.sh >"\$destfile"
+EOF
+chmod +x rot13-to-file.sh
+
 test_expect_success setup '
        git config filter.rot13.smudge ./rot13.sh &&
        git config filter.rot13.clean ./rot13.sh &&
@@ -291,6 +299,17 @@ test_expect_success 'cleanFromFile filter is used when 
adding a file' '
        test_cmp test fstest.t
 '
 
+test_expect_success 'smudgeToFile filter is used when checking out a file' '
+       test_config filter.rot13.smudgeToFile ./rot13-to-file.sh &&
+
+       rm -f fstest.t &&
+       git checkout -- fstest.t &&
+       test_cmp test fstest.t &&
+
+       test -e rot13-to-file.ran &&
+       rm -f rot13-to-file.ran
+'
+
 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 &&
@@ -304,4 +323,15 @@ test_expect_success 'cleanFromFile filter is not used when 
clean filter is not c
        test_cmp test actual
 '
 
+test_expect_success 'smudgeToFile filter is not used when smudge filter is not 
configured' '
+       test_config filter.nosmudge.clean ./rot13.sh &&
+       test_config filter.nosmudge.smudgeToFile ./rot13-to-file.sh &&
+
+       echo "*.no filter=nosmudge" >.gitattributes &&
+
+       rm -f fstest.t &&
+       git checkout -- fstest.t &&
+       test ! -e rot13-to-file.ran
+'
+
 test_done
-- 
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