feat: introduce tmp path for stdin files

Signed-off-by: Brian McGee <[email protected]>

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

@@ -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,
 	}
 

walk/stdin.go

@@ -36,6 +36,7 @@ type ReleaseFunc func(ctx context.Context) error
 type File struct {
 	Path    string
 	RelPath string
+	TmpPath string
 	Info    fs.FileInfo
 
 	// FormattedInfo is the result of os.stat after formatting the file.
@@ -111,8 +112,13 @@ func (f *File) AddReleaseFunc(fn ReleaseFunc) {
 // Stat checks if the file has changed by comparing its current state (size, mod time) to when it was first read.
 // It returns a boolean indicating if the file has changed, the current file info, and an error if any.
 func (f *File) Stat() (changed bool, info fs.FileInfo, err error) {
+	p := f.Path
+	if f.TmpPath != "" {
+		p = f.TmpPath
+	}
+
 	// Get the file's current state
-	current, err := os.Stat(f.Path)
+	current, err := os.Stat(p)
 	if err != nil {
 		return false, nil, fmt.Errorf("failed to stat %s: %w", f.Path, err)
 	}