git.iko.soy/crush
on main help sign in
feat: read stdin (#101)
b0ad508Carlos Alexandro Becker
cmd/root.go+26
compare with main
unifiedsplit
@@ -3,6 +3,7 @@ package cmd
3import (
3import (
4 "context"
4 "context"
5 "fmt"
5 "fmt"
6+ "io"
6 "os"
7 "os"
7 "sync"
8 "sync"
8 "time"
9 "time"
@@ -18,6 +19,7 @@ import (
18 "github.com/charmbracelet/crush/internal/tui"
19 "github.com/charmbracelet/crush/internal/tui"
19 "github.com/charmbracelet/crush/internal/version"
20 "github.com/charmbracelet/crush/internal/version"
20 "github.com/charmbracelet/fang"
21 "github.com/charmbracelet/fang"
22+ "github.com/charmbracelet/x/term"
21 "github.com/spf13/cobra"
23 "github.com/spf13/cobra"
22)
24)
23
25
@@ -99,6 +101,12 @@ to assist developers in writing, debugging, and understanding code directly from
99 // Initialize MCP tools early for both modes
101 // Initialize MCP tools early for both modes
100 initMCPTools(ctx, app)
102 initMCPTools(ctx, app)
101
103
104+ prompt, err = maybePrependStdin(prompt)
105+ if err != nil {
106+ logging.Error("Failed to read stdin: %v", err)
107+ return err
108+ }
109+
102 // Non-interactive mode
110 // Non-interactive mode
103 if prompt != "" {
111 if prompt != "" {
104 // Run non-interactive flow using the App method
112 // Run non-interactive flow using the App method
@@ -301,3 +309,21 @@ func init() {
301 return format.SupportedFormats, cobra.ShellCompDirectiveNoFileComp
309 return format.SupportedFormats, cobra.ShellCompDirectiveNoFileComp
302 })
310 })
303}
311}
312+
313+func maybePrependStdin(prompt string) (string, error) {
314+ if term.IsTerminal(os.Stdin.Fd()) {
315+ return prompt, nil
316+ }
317+ fi, err := os.Stdin.Stat()
318+ if err != nil {
319+ return prompt, err
320+ }
321+ if fi.Mode()&os.ModeNamedPipe == 0 {
322+ return prompt, nil
323+ }
324+ bts, err := io.ReadAll(os.Stdin)
325+ if err != nil {
326+ return prompt, err
327+ }
328+ return string(bts) + "\n\n" + prompt, nil
329+}