feat: introduce tmp path for stdin files

Signed-off-by: Brian McGee <brian@bmcgee.ie>

Author: brianmcgee

cmd/root_test.go

@@ -90,7 +90,11 @@ func (f *Formatter) Apply(ctx context.Context, files []*walk.File) error {
 
 	// append paths to the args
 	for _, file := range files {
-		args = append(args, file.RelPath)
+		if file.TmpPath != "" {
+			args = append(args, file.TmpPath)
+		} else {
+			args = append(args, file.RelPath)
+		}
 	}
 
 	// execute the command

format/formatter.go

@@ -17,6 +17,7 @@ with pkgs; [
   shfmt
   statix
   deadnix
+  just
   opentofu
   dos2unix
   yamlfmt

nix/packages/treefmt/formatters.nix

@@ -0,0 +1,7 @@
+# print this message
+help:
+        just --list --list-submodules --unsorted
+
+# interactive recipes launcher
+choose:
+    just --choose --unsorted

test/examples/just/justfile

@@ -89,3 +89,8 @@ includes = ["*.yaml"]
 
 [formatter.foo-fmt]
 command = "foo-fmt"
+
+[formatter.just]
+command = "just"
+options = ["--fmt", "--unstable", "--justfile"]
+includes = ["**/justfile"]

test/examples/treefmt.toml

@@ -106,7 +106,7 @@ func (c *CachedReader) Read(ctx context.Context, files []*File) (n int, err erro
 
 			// set a release function which inserts this file into the update channel
 			file.AddReleaseFunc(func(ctx context.Context) error {
-				if !GetNoCache(ctx) {
+				if !GetNoCache(ctx) && file.TmpPath == "" {
 					c.updateCh <- file
 				}
 

walk/cached.go

@@ -33,6 +33,7 @@ var examplesPaths = []string{
 	"html/index.html",
 	"html/scripts/.gitkeep",
 	"javascript/source/hello.js",
+	"just/justfile",
 	"nix/sources.nix",
 	"nixpkgs.toml",
 	"python/main.py",
@@ -78,8 +79,8 @@ func TestFilesystemReader(t *testing.T) {
 		}
 	}
 
-	as.Equal(32, count)
-	as.Equal(32, statz.Value(stats.Traversed))
+	as.Equal(33, count)
+	as.Equal(33, statz.Value(stats.Traversed))
 	as.Equal(0, statz.Value(stats.Matched))
 	as.Equal(0, statz.Value(stats.Formatted))
 	as.Equal(0, statz.Value(stats.Changed))

walk/filesystem_test.go

@@ -29,12 +29,12 @@ func TestGitReader(t *testing.T) {
 	reader, err := walk.NewGitReader(tempDir, "", &statz)
 	as.NoError(err)
 
-	files := make([]*walk.File, 33)
+	files := make([]*walk.File, 34)
 	ctx, cancel := context.WithTimeout(t.Context(), 100*time.Millisecond)
 	n, err := reader.Read(ctx, files)
 
 	cancel()
-	as.Equal(32, n)
+	as.Equal(33, n)
 	as.ErrorIs(err, io.EOF)
 
 	// add everything to the git index
@@ -63,8 +63,8 @@ func TestGitReader(t *testing.T) {
 		}
 	}
 
-	as.Equal(32, count)
-	as.Equal(32, statz.Value(stats.Traversed))
+	as.Equal(33, count)
+	as.Equal(33, statz.Value(stats.Traversed))
 	as.Equal(0, statz.Value(stats.Matched))
 	as.Equal(0, statz.Value(stats.Formatted))
 	as.Equal(0, statz.Value(stats.Changed))

walk/git_test.go

@@ -67,8 +67,8 @@ func TestJujutsuReader(t *testing.T) {
 		}
 	}
 
-	as.Equal(32, count)
-	as.Equal(32, statz.Value(stats.Traversed))
+	as.Equal(33, count)
+	as.Equal(33, statz.Value(stats.Traversed))
 	as.Equal(0, statz.Value(stats.Matched))
 	as.Equal(0, statz.Value(stats.Formatted))
 	as.Equal(0, statz.Value(stats.Changed))

walk/jujutsu_test.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"path"
 	"path/filepath"
 
 	"github.com/numtide/treefmt/v2/stats"
@@ -25,10 +26,11 @@ func (s StdinReader) Read(_ context.Context, files []*File) (n int, err error) {
 		return 0, io.EOF
 	}
 
-	// read stdin into a temporary file with the same file extension
-	pattern := "*" + filepath.Ext(s.path)
+	// we attempt to preserve any file extensions as some formatters will not behave correctly without it
+	pattern := "treefmt-stdin-*" + filepath.Ext(s.path)
 
-	file, err := os.CreateTemp(s.root, pattern)
+	// create a temporary file to dump stdin into
+	file, err := os.CreateTemp("", pattern)
 	if err != nil {
 		return 0, fmt.Errorf("failed to create a temporary file for processing stdin: %w", err)
 	}
@@ -43,14 +45,10 @@ func (s StdinReader) Read(_ context.Context, files []*File) (n int, err error) {
 		return 0, fmt.Errorf("failed to get file info for temporary file: %w", err)
 	}
 
-	relPath, err := filepath.Rel(s.root, file.Name())
-	if err != nil {
-		return 0, fmt.Errorf("failed to get relative path for temporary file: %w", err)
-	}
-
 	files[0] = &File{
-		Path:    file.Name(),
-		RelPath: relPath,
+		Path:    path.Join(s.root, s.path),
+		RelPath: s.path,
+		TmpPath: file.Name(),
 		Info:    info,
 	}