feat: add context.SetValue

context.go

@@ -3,6 +3,7 @@ package tea
 import (
 	"context"
 	"image/color"
+	"sync"
 
 	"github.com/charmbracelet/lipgloss"
 )
@@ -13,6 +14,11 @@ import (
 type Context interface {
 	context.Context
 
+	// SetValue sets a value on the context. This is useful for storing values
+	// that needs to be accessed across multiple functions.
+	// You can access the value later using Value.
+	SetValue(key, value interface{})
+
 	// BackgroundColor returns the current background color of the terminal.
 	// It returns nil if the terminal's doesn't support querying the background
 	// color.
@@ -46,12 +52,16 @@ type teaContext struct {
 	kittyFlags      int
 	backgroundColor color.Color
 	hasLightBg      bool // cached value
+
+	values map[interface{}]interface{}
+	mtx    sync.Mutex
 }
 
 func newContext(ctx context.Context) *teaContext {
 	c := new(teaContext)
 	c.Context = ctx
 	c.kittyFlags = -1
+	c.values = make(map[interface{}]interface{})
 	return c
 }
 
@@ -74,3 +84,18 @@ func (c *teaContext) NewStyle() lipgloss.Style {
 func (c *teaContext) ColorProfile() lipgloss.Profile {
 	return c.profile
 }
+
+func (ctx *teaContext) Value(key interface{}) interface{} {
+	ctx.mtx.Lock()
+	defer ctx.mtx.Unlock()
+	if v, ok := ctx.values[key]; ok {
+		return v
+	}
+	return ctx.Context.Value(key)
+}
+
+func (ctx *teaContext) SetValue(key, value interface{}) {
+	ctx.mtx.Lock()
+	defer ctx.mtx.Unlock()
+	ctx.values[key] = value
+}