feat!: v2

Signed-off-by: Carlos Alexandro Becker <[email protected]>

accesscontrol/accesscontrol.go

@@ -5,7 +5,7 @@ import (
 	"fmt"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
+	"github.com/charmbracelet/wish/v2"
 )
 
 // Middleware will exit 1 connections trying to execute commands that are not allowed.

accesscontrol/accesscontrol_test.go

@@ -5,8 +5,8 @@ import (
 	"testing"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish/accesscontrol"
-	"github.com/charmbracelet/wish/testsession"
+	"github.com/charmbracelet/wish/v2/accesscontrol"
+	"github.com/charmbracelet/wish/v2/testsession"
 	gossh "golang.org/x/crypto/ssh"
 )
 

activeterm/activeterm.go

@@ -3,7 +3,7 @@ package activeterm
 
 import (
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
+	"github.com/charmbracelet/wish/v2"
 )
 
 // Middleware will exit 1 connections trying with no active terminals.

activeterm/activeterm_test.go

@@ -4,8 +4,8 @@ import (
 	"testing"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish/activeterm"
-	"github.com/charmbracelet/wish/testsession"
+	"github.com/charmbracelet/wish/v2/activeterm"
+	"github.com/charmbracelet/wish/v2/testsession"
 	gossh "golang.org/x/crypto/ssh"
 )
 

bubbletea/query.go

@@ -59,7 +59,7 @@ func queryTerminal(
 	filter QueryTerminalFilter,
 	query string,
 ) error {
-	rd, err := input.NewDriver(in, "", 0)
+	rd, err := input.NewReader(in, "", 0)
 	if err != nil {
 		return err
 	}

bubbletea/tea.go

@@ -3,14 +3,11 @@ package bubbletea
 
 import (
 	"context"
-	"strings"
 
-	tea "github.com/charmbracelet/bubbletea"
-	"github.com/charmbracelet/lipgloss"
-	"github.com/charmbracelet/log"
+	tea "github.com/charmbracelet/bubbletea/v2"
+	"github.com/charmbracelet/log/v2"
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
-	"github.com/muesli/termenv"
+	"github.com/charmbracelet/wish/v2"
 )
 
 // BubbleTeaHandler is the function Bubble Tea apps implement to hook into the
@@ -40,35 +37,21 @@ type ProgramHandler func(sess ssh.Session) *tea.Program
 // It also captures window resize events and sends them to the tea.Program
 // as tea.WindowSizeMsgs.
 func Middleware(handler Handler) wish.Middleware {
-	return MiddlewareWithProgramHandler(newDefaultProgramHandler(handler), termenv.Ascii)
-}
-
-// MiddlewareWithColorProfile allows you to specify the minimum number of colors
-// this program needs to work properly.
-//
-// If the client's color profile has less colors than p, p will be forced.
-// Use with caution.
-func MiddlewareWithColorProfile(handler Handler, profile termenv.Profile) wish.Middleware {
-	return MiddlewareWithProgramHandler(newDefaultProgramHandler(handler), profile)
+	return MiddlewareWithProgramHandler(newDefaultProgramHandler(handler))
 }
 
 // MiddlewareWithProgramHandler allows you to specify the ProgramHandler to be
-// able to access the underlying tea.Program, and the minimum supported color
-// profile.
+// able to access the underlying tea.Program.
 //
 // This is useful for creating custom middlewares that need access to
 // tea.Program for instance to use p.Send() to send messages to tea.Program.
 //
 // Make sure to set the tea.WithInput and tea.WithOutput to the ssh.Session
 // otherwise the program will not function properly. The recommended way
 // of doing so is by using MakeOptions.
-//
-// If the client's color profile has less colors than p, p will be forced.
-// Use with caution.
-func MiddlewareWithProgramHandler(handler ProgramHandler, profile termenv.Profile) wish.Middleware {
+func MiddlewareWithProgramHandler(handler ProgramHandler) wish.Middleware {
 	return func(next ssh.Handler) ssh.Handler {
 		return func(sess ssh.Session) {
-			sess.Context().SetValue(minColorProfileKey, profile)
 			program := handler(sess)
 			if program == nil {
 				next(sess)
@@ -104,50 +87,12 @@ func MiddlewareWithProgramHandler(handler ProgramHandler, profile termenv.Profil
 	}
 }
 
-var minColorProfileKey struct{}
-
-var profileNames = [4]string{"TrueColor", "ANSI256", "ANSI", "Ascii"}
-
-// MakeRenderer returns a lipgloss renderer for the current session.
-// This function handle PTYs as well, and should be used to style your application.
-func MakeRenderer(sess ssh.Session) *lipgloss.Renderer {
-	cp, ok := sess.Context().Value(minColorProfileKey).(termenv.Profile)
-	if !ok {
-		cp = termenv.Ascii
-	}
-	r := newRenderer(sess)
-	if r.ColorProfile() > cp {
-		wish.Printf(sess, "Warning: Client's terminal is %q, forcing %q\r\n", profileNames[r.ColorProfile()], profileNames[cp])
-		r.SetColorProfile(cp)
-	}
-	return r
-}
-
 // MakeOptions returns the tea.WithInput and tea.WithOutput program options
 // taking into account possible Emulated or Allocated PTYs.
 func MakeOptions(sess ssh.Session) []tea.ProgramOption {
 	return makeOpts(sess)
 }
 
-type sshEnviron []string
-
-var _ termenv.Environ = sshEnviron(nil)
-
-// Environ implements termenv.Environ.
-func (e sshEnviron) Environ() []string {
-	return e
-}
-
-// Getenv implements termenv.Environ.
-func (e sshEnviron) Getenv(k string) string {
-	for _, v := range e {
-		if strings.HasPrefix(v, k+"=") {
-			return v[len(k)+1:]
-		}
-	}
-	return ""
-}
-
 func newDefaultProgramHandler(handler Handler) ProgramHandler {
 	return func(s ssh.Session) *tea.Program {
 		m, opts := handler(s)

bubbletea/tea_other.go

@@ -4,10 +4,8 @@
 package bubbletea
 
 import (
-	tea "github.com/charmbracelet/bubbletea"
-	"github.com/charmbracelet/lipgloss"
+	tea "github.com/charmbracelet/bubbletea/v2"
 	"github.com/charmbracelet/ssh"
-	"github.com/muesli/termenv"
 )
 
 func makeOpts(s ssh.Session) []tea.ProgramOption {
@@ -16,9 +14,3 @@ func makeOpts(s ssh.Session) []tea.ProgramOption {
 		tea.WithOutput(s),
 	}
 }
-
-func newRenderer(s ssh.Session) *lipgloss.Renderer {
-	pty, _, _ := s.Pty()
-	env := sshEnviron(append(s.Environ(), "TERM="+pty.Term))
-	return lipgloss.NewRenderer(s, termenv.WithEnvironment(env), termenv.WithUnsafe(), termenv.WithColorCache(true))
-}

bubbletea/tea_unix.go

@@ -4,85 +4,26 @@
 package bubbletea
 
 import (
-	"image/color"
-	"time"
-
-	tea "github.com/charmbracelet/bubbletea"
-	"github.com/charmbracelet/lipgloss"
+	tea "github.com/charmbracelet/bubbletea/v2"
+	"github.com/charmbracelet/colorprofile"
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/x/ansi"
-	"github.com/charmbracelet/x/input"
-	"github.com/charmbracelet/x/term"
-	"github.com/lucasb-eyer/go-colorful"
-	"github.com/muesli/termenv"
 )
 
 func makeOpts(s ssh.Session) []tea.ProgramOption {
 	pty, _, ok := s.Pty()
 	if !ok || s.EmulatedPty() {
+		envs := append(s.Environ(), "TERM="+pty.Term)
 		return []tea.ProgramOption{
 			tea.WithInput(s),
 			tea.WithOutput(s),
+			tea.WithEnvironment(envs),
+			tea.WithColorProfile(colorprofile.Env(envs)),
 		}
 	}
 
 	return []tea.ProgramOption{
 		tea.WithInput(pty.Slave),
 		tea.WithOutput(pty.Slave),
+		tea.WithEnvironment(s.Environ()),
 	}
 }
-
-func newRenderer(s ssh.Session) *lipgloss.Renderer {
-	pty, _, ok := s.Pty()
-	if !ok || pty.Term == "" || pty.Term == "dumb" {
-		return lipgloss.NewRenderer(s, termenv.WithProfile(termenv.Ascii))
-	}
-	env := sshEnviron(append(s.Environ(), "TERM="+pty.Term))
-	var r *lipgloss.Renderer
-	var bg color.Color
-	if ok && pty.Slave != nil {
-		r = lipgloss.NewRenderer(
-			pty.Slave,
-			termenv.WithEnvironment(env),
-			termenv.WithColorCache(true),
-		)
-		state, err := term.MakeRaw(pty.Slave.Fd())
-		if err == nil {
-			bg, _ = queryBackgroundColor(pty.Slave, pty.Slave)
-			term.Restore(pty.Slave.Fd(), state)
-		}
-	} else {
-		r = lipgloss.NewRenderer(
-			s,
-			termenv.WithEnvironment(env),
-			termenv.WithUnsafe(),
-			termenv.WithColorCache(true),
-		)
-		bg = querySessionBackgroundColor(s)
-	}
-	if bg != nil {
-		c, ok := colorful.MakeColor(bg)
-		if ok {
-			_, _, l := c.Hsl()
-			r.SetHasDarkBackground(l < 0.5)
-		}
-	}
-	return r
-}
-
-// copied from x/[email protected].
-func querySessionBackgroundColor(s ssh.Session) (bg color.Color) {
-	_ = queryTerminal(s, s, time.Second, func(events []input.Event) bool {
-		for _, e := range events {
-			switch e := e.(type) {
-			case input.BackgroundColorEvent:
-				bg = e.Color
-				continue // we need to consume the next DA1 event
-			case input.PrimaryDeviceAttributesEvent:
-				return false
-			}
-		}
-		return true
-	}, ansi.RequestBackgroundColor+ansi.RequestPrimaryDeviceAttributes)
-	return
-}

cmd.go

@@ -5,7 +5,7 @@ import (
 	"io"
 	"os/exec"
 
-	tea "github.com/charmbracelet/bubbletea"
+	tea "github.com/charmbracelet/bubbletea/v2"
 	"github.com/charmbracelet/ssh"
 )
 

cmd_test.go

@@ -8,7 +8,7 @@ import (
 	"time"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish/testsession"
+	"github.com/charmbracelet/wish/v2/testsession"
 )
 
 func TestCommandNoPty(t *testing.T) {

comment/comment.go

@@ -2,7 +2,7 @@ package comment
 
 import (
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
+	"github.com/charmbracelet/wish/v2"
 )
 
 // Middleware prints a comment at the end of the session.

comment/comment_test.go

@@ -4,7 +4,7 @@ import (
 	"testing"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish/testsession"
+	"github.com/charmbracelet/wish/v2/testsession"
 	gossh "golang.org/x/crypto/ssh"
 )
 

elapsed/elapsed.go

@@ -4,7 +4,7 @@ import (
 	"time"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
+	"github.com/charmbracelet/wish/v2"
 )
 
 // MiddlewareWithFormat returns a middleware that logs the elapsed time of the

elapsed/elapsed_test.go

@@ -5,7 +5,7 @@ import (
 	"time"
 
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish/testsession"
+	"github.com/charmbracelet/wish/v2/testsession"
 	gossh "golang.org/x/crypto/ssh"
 )
 

examples/banner/main.go

@@ -12,11 +12,11 @@ import (
 
 	_ "embed"
 
-	"github.com/charmbracelet/log"
+	"github.com/charmbracelet/log/v2"
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
-	"github.com/charmbracelet/wish/elapsed"
-	"github.com/charmbracelet/wish/logging"
+	"github.com/charmbracelet/wish/v2"
+	"github.com/charmbracelet/wish/v2/elapsed"
+	"github.com/charmbracelet/wish/v2/logging"
 )
 
 const (

examples/bubbletea-exec/main.go

@@ -10,14 +10,14 @@ import (
 	"syscall"
 	"time"
 
-	tea "github.com/charmbracelet/bubbletea"
-	"github.com/charmbracelet/lipgloss"
-	"github.com/charmbracelet/log"
+	tea "github.com/charmbracelet/bubbletea/v2"
+	"github.com/charmbracelet/lipgloss/v2"
+	"github.com/charmbracelet/log/v2"
 	"github.com/charmbracelet/ssh"
-	"github.com/charmbracelet/wish"
-	"github.com/charmbracelet/wish/activeterm"
-	"github.com/charmbracelet/wish/bubbletea"
-	"github.com/charmbracelet/wish/logging"
+	"github.com/charmbracelet/wish/v2"
+	"github.com/charmbracelet/wish/v2/activeterm"
+	"github.com/charmbracelet/wish/v2/bubbletea"
+	"github.com/charmbracelet/wish/v2/logging"
 	"github.com/charmbracelet/x/editor"
 )
 
@@ -67,15 +67,11 @@ func main() {
 }
 
 func teaHandler(s ssh.Session) (tea.Model, []tea.ProgramOption) {
-	// Create a lipgloss.Renderer for the session
-	renderer := bubbletea.MakeRenderer(s)
 	// Set up the model with the current session and styles.
 	// We'll use the session to call wish.Command, which makes it compatible
 	// with tea.Command.
 	m := model{
-		sess:     s,
-		style:    renderer.NewStyle().Foreground(lipgloss.Color("8")),
-		errStyle: renderer.NewStyle().Foreground(lipgloss.Color("3")),
+		sess: s,
 	}
 	return m, []tea.ProgramOption{tea.WithAltScreen()}
 }
@@ -87,8 +83,10 @@ type model struct {
 	errStyle lipgloss.Style
 }
 
-func (m model) Init() tea.Cmd {
-	return nil
+func (m model) Init() (tea.Model, tea.Cmd) {
+	m.style = lipgloss.NewStyle().Foreground(lipgloss.Color("8"))
+	m.errStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("3"))
+	return m, nil
 }
 
 type cmdFinishedMsg struct{ err error }