diff --git a/go.mod b/go.mod index f291838eba7bca3a9c25b061f5974a4d7885b6ce..1d1403b4511c79f64657fa3d9d053cc7921e44ac 100644 --- a/go.mod +++ b/go.mod @@ -54,6 +54,7 @@ require ( github.com/stretchr/testify v1.11.1 github.com/tidwall/sjson v1.2.5 github.com/zeebo/xxh3 v1.0.2 + golang.org/x/mod v0.30.0 golang.org/x/sync v0.18.0 golang.org/x/text v0.31.0 gopkg.in/natefinch/lumberjack.v2 v2.2.1 diff --git a/go.sum b/go.sum index aaeba2379bc8d66cdbbbed1cfe491372abf9e76c..79efd285e06d69d320f089fe6d6af3f46bbd1106 100644 --- a/go.sum +++ b/go.sum @@ -378,6 +378,8 @@ golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= golang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= golang.org/x/mod v0.15.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c= golang.org/x/mod v0.17.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c= +golang.org/x/mod v0.30.0 h1:fDEXFVZ/fmCKProc/yAXXUijritrDzahmwwefnjoPFk= +golang.org/x/mod v0.30.0/go.mod h1:lAsf5O2EvJeSFMiBxXDki7sCgAxEUcZHXoXMKT4GJKc= golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= diff --git a/internal/stringext/string.go b/internal/stringext/string.go index 03456db93bc148f7c77e52da3c493c94fa79624f..9383ce1d78b8f0a776fc533526ee961d0123d734 100644 --- a/internal/stringext/string.go +++ b/internal/stringext/string.go @@ -1,6 +1,8 @@ package stringext import ( + "strings" + "golang.org/x/text/cases" "golang.org/x/text/language" ) @@ -8,3 +10,12 @@ import ( func Capitalize(text string) string { return cases.Title(language.English, cases.Compact).String(text) } + +func ContainsAny(str string, args ...string) bool { + for _, arg := range args { + if strings.Contains(str, arg) { + return true + } + } + return false +} diff --git a/internal/tui/tui.go b/internal/tui/tui.go index 2fbd9f612934511d18e5d8c0a35f042ebbb8d82a..45f0ae5ec410e85b3d30d620b4db5c499cff09c3 100644 --- a/internal/tui/tui.go +++ b/internal/tui/tui.go @@ -4,6 +4,7 @@ import ( "context" "fmt" "math/rand" + "regexp" "slices" "strings" "time" @@ -17,6 +18,7 @@ import ( "github.com/charmbracelet/crush/internal/event" "github.com/charmbracelet/crush/internal/permission" "github.com/charmbracelet/crush/internal/pubsub" + "github.com/charmbracelet/crush/internal/stringext" cmpChat "github.com/charmbracelet/crush/internal/tui/components/chat" "github.com/charmbracelet/crush/internal/tui/components/chat/splash" "github.com/charmbracelet/crush/internal/tui/components/completions" @@ -34,6 +36,7 @@ import ( "github.com/charmbracelet/crush/internal/tui/page/chat" "github.com/charmbracelet/crush/internal/tui/styles" "github.com/charmbracelet/crush/internal/tui/util" + "golang.org/x/mod/semver" "golang.org/x/text/cases" "golang.org/x/text/language" ) @@ -120,10 +123,19 @@ func (a *appModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) { a.sendProgressBar = slices.Contains(msg, "WT_SESSION") } case tea.TerminalVersionMsg: + if a.sendProgressBar { + return a, nil + } termVersion := strings.ToLower(msg.Name) - // Only enable progress bar for the following terminals. - if !a.sendProgressBar { - a.sendProgressBar = strings.Contains(termVersion, "ghostty") + switch { + case stringext.ContainsAny(termVersion, "ghostty", "rio"): + a.sendProgressBar = true + case strings.Contains(termVersion, "iterm2"): + // iTerm2 supports progress bars from version v3.6.6 + matches := regexp.MustCompile(`^iterm2 (\d+\.\d+\.\d+)$`).FindStringSubmatch(termVersion) + if len(matches) == 2 && semver.Compare("v"+matches[1], "v3.6.6") >= 0 { + a.sendProgressBar = true + } } return a, nil case tea.KeyboardEnhancementsMsg: