mirror of
https://github.com/kovidgoyal/kitty
synced 2026-07-07 00:15:12 +02:00
403 lines
10 KiB
Go
403 lines
10 KiB
Go
// License: GPLv3 Copyright: 2023, Kovid Goyal, <kovid at kovidgoyal.net>
|
|
|
|
package mouse_demo
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"slices"
|
|
"strconv"
|
|
"strings"
|
|
|
|
kitty "github.com/kovidgoyal/kitty"
|
|
"github.com/kovidgoyal/kitty/tools/tui/loop"
|
|
)
|
|
|
|
var _ = fmt.Print
|
|
|
|
const dnd_accepted_mimes = "text/plain text/uri-list"
|
|
|
|
func dnd_escape(metadata, payload string) string {
|
|
b := strings.Builder{}
|
|
b.WriteString(fmt.Sprintf("\x1b]%d;", kitty.DndCode))
|
|
b.WriteString(metadata)
|
|
if payload != "" {
|
|
b.WriteByte(';')
|
|
b.WriteString(payload)
|
|
}
|
|
b.WriteString("\x1b\\")
|
|
return b.String()
|
|
}
|
|
|
|
func dnd_start_accepting() string {
|
|
return dnd_escape("t=a", dnd_accepted_mimes)
|
|
}
|
|
|
|
func dnd_stop_accepting() string {
|
|
return dnd_escape("t=A", "")
|
|
}
|
|
|
|
func dnd_accept_drag(mimes string) string {
|
|
return dnd_escape("t=m:o=1", mimes)
|
|
}
|
|
|
|
func dnd_request_data(mime string) string {
|
|
return dnd_escape("t=r", mime)
|
|
}
|
|
|
|
func dnd_finish() string {
|
|
return dnd_escape("t=r", "")
|
|
}
|
|
|
|
type dnd_state struct {
|
|
dragging bool
|
|
drag_mimes []string
|
|
drop_mimes []string
|
|
collecting string // MIME type currently being collected
|
|
collect_buf strings.Builder
|
|
plain_text string
|
|
uri_list []string
|
|
has_drop_data bool
|
|
}
|
|
|
|
func (d *dnd_state) reset_drag() {
|
|
d.dragging = false
|
|
d.drag_mimes = nil
|
|
}
|
|
|
|
func (d *dnd_state) reset_drop_data() {
|
|
d.drop_mimes = nil
|
|
d.collecting = ""
|
|
d.collect_buf.Reset()
|
|
d.plain_text = ""
|
|
d.uri_list = nil
|
|
d.has_drop_data = false
|
|
}
|
|
|
|
func draw_rounded_box(lp *loop.Loop, width int, lines []string) {
|
|
if width < 4 {
|
|
width = 4
|
|
}
|
|
inner_width := width - 2
|
|
// Top border
|
|
lp.QueueWriteString("╭" + strings.Repeat("─", inner_width) + "╮\r\n")
|
|
for _, line := range lines {
|
|
if len(line) > inner_width {
|
|
line = line[:inner_width]
|
|
}
|
|
padding := inner_width - len(line)
|
|
lp.QueueWriteString("│" + line + strings.Repeat(" ", padding) + "│\r\n")
|
|
}
|
|
// Bottom border
|
|
lp.QueueWriteString("╰" + strings.Repeat("─", inner_width) + "╯\r\n")
|
|
}
|
|
|
|
func Run(args []string) (rc int, err error) {
|
|
all_pointer_shapes := []loop.PointerShape{
|
|
// start all pointer shapes (auto generated by gen-key-constants.py do not edit)
|
|
loop.DEFAULT_POINTER,
|
|
loop.TEXT_POINTER,
|
|
loop.POINTER_POINTER,
|
|
loop.HELP_POINTER,
|
|
loop.WAIT_POINTER,
|
|
loop.PROGRESS_POINTER,
|
|
loop.CROSSHAIR_POINTER,
|
|
loop.CELL_POINTER,
|
|
loop.VERTICAL_TEXT_POINTER,
|
|
loop.MOVE_POINTER,
|
|
loop.E_RESIZE_POINTER,
|
|
loop.NE_RESIZE_POINTER,
|
|
loop.NW_RESIZE_POINTER,
|
|
loop.N_RESIZE_POINTER,
|
|
loop.SE_RESIZE_POINTER,
|
|
loop.SW_RESIZE_POINTER,
|
|
loop.S_RESIZE_POINTER,
|
|
loop.W_RESIZE_POINTER,
|
|
loop.EW_RESIZE_POINTER,
|
|
loop.NS_RESIZE_POINTER,
|
|
loop.NESW_RESIZE_POINTER,
|
|
loop.NWSE_RESIZE_POINTER,
|
|
loop.ZOOM_IN_POINTER,
|
|
loop.ZOOM_OUT_POINTER,
|
|
loop.ALIAS_POINTER,
|
|
loop.COPY_POINTER,
|
|
loop.NOT_ALLOWED_POINTER,
|
|
loop.NO_DROP_POINTER,
|
|
loop.GRAB_POINTER,
|
|
loop.GRABBING_POINTER,
|
|
// end all pointer shapes
|
|
}
|
|
all_pointer_shape_names := make([]string, len(all_pointer_shapes))
|
|
col_width := 0
|
|
for i, p := range all_pointer_shapes {
|
|
all_pointer_shape_names[i] = p.String()
|
|
col_width = max(col_width, len(all_pointer_shape_names[i]))
|
|
}
|
|
col_width += 1
|
|
|
|
lp, err := loop.New()
|
|
if err != nil {
|
|
return 1, err
|
|
}
|
|
lp.MouseTrackingMode(loop.FULL_MOUSE_TRACKING)
|
|
var current_mouse_event *loop.MouseEvent
|
|
var dnd dnd_state
|
|
|
|
draw_screen := func() {
|
|
lp.StartAtomicUpdate()
|
|
defer lp.EndAtomicUpdate()
|
|
lp.AllowLineWrapping(false)
|
|
defer lp.AllowLineWrapping(true)
|
|
lp.ClearScreen()
|
|
|
|
sw := 80
|
|
sh := 24
|
|
if s, err := lp.ScreenSize(); err == nil {
|
|
sw = int(s.WidthCells)
|
|
sh = int(s.HeightCells)
|
|
}
|
|
|
|
if current_mouse_event == nil {
|
|
lp.Println(`Move the mouse or click to see mouse events`)
|
|
lp.Println("Hover the mouse over the names below to see the shapes")
|
|
lp.Println()
|
|
for pos := 0; pos < len(all_pointer_shapes); {
|
|
num_cols := max(1, sw/col_width)
|
|
colfmt := "%-" + strconv.Itoa(col_width) + "s"
|
|
for c := 0; c < num_cols && pos < len(all_pointer_shapes); c++ {
|
|
lp.Printf(colfmt, all_pointer_shape_names[pos])
|
|
pos++
|
|
}
|
|
lp.Println()
|
|
}
|
|
} else if current_mouse_event.Event_type == loop.MOUSE_LEAVE {
|
|
lp.Println("Mouse has left the window")
|
|
} else {
|
|
lp.Printf("Position: %d, %d (pixels)\r\n", current_mouse_event.Pixel.X, current_mouse_event.Pixel.Y)
|
|
lp.Printf("Cell : %d, %d\r\n", current_mouse_event.Cell.X, current_mouse_event.Cell.Y)
|
|
lp.Printf("Type : %s\r\n", current_mouse_event.Event_type)
|
|
y := 3
|
|
if current_mouse_event.Buttons != loop.NO_MOUSE_BUTTON {
|
|
lp.Println(current_mouse_event.Buttons.String())
|
|
y += 1
|
|
}
|
|
if mods := current_mouse_event.Mods.String(); mods != "" {
|
|
lp.Printf("Modifiers: %s\r\n", mods)
|
|
y += 1
|
|
}
|
|
lp.Println("Hover the mouse over the names below to see the shapes")
|
|
y += 1
|
|
|
|
num_cols := max(1, sw/col_width)
|
|
pos := 0
|
|
colfmt := "%-" + strconv.Itoa(col_width) + "s"
|
|
is_on_name := false
|
|
var ps loop.PointerShape
|
|
for y < sh && pos < len(all_pointer_shapes) {
|
|
is_row := y == current_mouse_event.Cell.Y
|
|
for c := 0; c < num_cols && pos < len(all_pointer_shapes); c++ {
|
|
name := all_pointer_shape_names[pos]
|
|
is_hovered := false
|
|
if is_row {
|
|
start_x := c * col_width
|
|
x := current_mouse_event.Cell.X
|
|
if x < start_x+len(name) && x >= start_x {
|
|
is_on_name = true
|
|
is_hovered = true
|
|
ps = all_pointer_shapes[pos]
|
|
}
|
|
}
|
|
if is_hovered {
|
|
lp.QueueWriteString("\x1b[31m")
|
|
}
|
|
lp.Printf(colfmt, name)
|
|
lp.QueueWriteString("\x1b[m")
|
|
pos++
|
|
}
|
|
y += 1
|
|
lp.Println()
|
|
}
|
|
lp.PopPointerShape()
|
|
if is_on_name {
|
|
lp.PushPointerShape(ps)
|
|
}
|
|
}
|
|
|
|
// Draw the drop area below the pointer shapes list
|
|
lp.Println()
|
|
box_width := min(sw, 60)
|
|
if dnd.dragging {
|
|
lines := []string{"Dragging over window. MIME types:"}
|
|
for _, m := range dnd.drag_mimes {
|
|
lines = append(lines, " "+m)
|
|
}
|
|
draw_rounded_box(lp, box_width, lines)
|
|
} else if dnd.has_drop_data {
|
|
lines := []string{}
|
|
if dnd.plain_text != "" {
|
|
lines = append(lines, "text/plain: "+dnd.plain_text)
|
|
}
|
|
for _, u := range dnd.uri_list {
|
|
lines = append(lines, "URL: "+u)
|
|
}
|
|
if len(lines) == 0 {
|
|
lines = []string{"Drop received (no recognized content)"}
|
|
}
|
|
draw_rounded_box(lp, box_width, lines)
|
|
} else {
|
|
draw_rounded_box(lp, box_width, []string{"Drop something onto this window"})
|
|
}
|
|
}
|
|
|
|
handle_dnd_osc := func(raw []byte) error {
|
|
// raw is the OSC payload after ESC ] and before ST
|
|
// Format: 72;metadata;payload or 72;metadata
|
|
prefix := fmt.Sprintf("%d;", kitty.DndCode)
|
|
if !bytes.HasPrefix(raw, []byte(prefix)) {
|
|
return nil
|
|
}
|
|
rest := string(raw[len(prefix):])
|
|
// Split into metadata and optional payload
|
|
meta, payload, _ := strings.Cut(rest, ";")
|
|
// Parse metadata key=value pairs separated by ':'
|
|
meta_map := make(map[string]string)
|
|
for kv := range strings.SplitSeq(meta, ":") {
|
|
k, v, _ := strings.Cut(kv, "=")
|
|
if k != "" {
|
|
meta_map[k] = v
|
|
}
|
|
}
|
|
t := meta_map["t"]
|
|
switch t {
|
|
case "m":
|
|
// Drag move event from terminal
|
|
// Check if drag has left the window (x=-1, y=-1)
|
|
if meta_map["x"] == "-1" || meta_map["y"] == "-1" {
|
|
dnd.reset_drag()
|
|
draw_screen()
|
|
return nil
|
|
}
|
|
mimes := strings.Fields(payload)
|
|
if len(mimes) > 0 {
|
|
dnd.drag_mimes = mimes
|
|
}
|
|
dnd.dragging = true
|
|
// Accept the drag with copy operation
|
|
accepted_mimes := []string{}
|
|
for _, m := range dnd.drag_mimes {
|
|
if m == "text/plain" || m == "text/uri-list" {
|
|
accepted_mimes = append(accepted_mimes, m)
|
|
}
|
|
}
|
|
if len(accepted_mimes) > 0 {
|
|
lp.QueueWriteString(dnd_accept_drag(strings.Join(accepted_mimes, " ")))
|
|
}
|
|
draw_screen()
|
|
case "M":
|
|
// Drop event from terminal
|
|
dnd.dragging = false
|
|
dnd.reset_drop_data()
|
|
mimes := strings.Fields(payload)
|
|
dnd.drop_mimes = mimes
|
|
// Request data for text/plain first, then text/uri-list
|
|
if slices.Contains(mimes, "text/plain") {
|
|
dnd.collecting = "text/plain"
|
|
lp.QueueWriteString(dnd_request_data("text/plain"))
|
|
return nil
|
|
}
|
|
if slices.Contains(mimes, "text/uri-list") {
|
|
dnd.collecting = "text/uri-list"
|
|
lp.QueueWriteString(dnd_request_data("text/uri-list"))
|
|
return nil
|
|
}
|
|
// Nothing to collect, signal done
|
|
lp.QueueWriteString(dnd_finish())
|
|
dnd.has_drop_data = true
|
|
draw_screen()
|
|
case "r":
|
|
// Data response from terminal
|
|
if payload == "" {
|
|
// End of data for current MIME type
|
|
if dnd.collecting == "text/plain" {
|
|
text := dnd.collect_buf.String()
|
|
// Get first line
|
|
if before, _, ok := strings.Cut(text, "\n"); ok {
|
|
dnd.plain_text = before
|
|
} else {
|
|
dnd.plain_text = text
|
|
}
|
|
dnd.collect_buf.Reset()
|
|
// Now request text/uri-list if available
|
|
if slices.Contains(dnd.drop_mimes, "text/uri-list") {
|
|
dnd.collecting = "text/uri-list"
|
|
lp.QueueWriteString(dnd_request_data("text/uri-list"))
|
|
return nil
|
|
}
|
|
} else if dnd.collecting == "text/uri-list" {
|
|
text := dnd.collect_buf.String()
|
|
dnd.collect_buf.Reset()
|
|
// Parse URI list: lines starting with # are comments
|
|
for line := range strings.SplitSeq(text, "\n") {
|
|
line = strings.TrimRight(line, "\r")
|
|
if line != "" && !strings.HasPrefix(line, "#") {
|
|
dnd.uri_list = append(dnd.uri_list, line)
|
|
}
|
|
}
|
|
}
|
|
dnd.collecting = ""
|
|
// Signal done
|
|
lp.QueueWriteString(dnd_finish())
|
|
dnd.has_drop_data = true
|
|
draw_screen()
|
|
} else {
|
|
// Decode base64 payload and append to buffer
|
|
decoded, err := base64.RawStdEncoding.DecodeString(payload)
|
|
if err == nil {
|
|
dnd.collect_buf.Write(decoded)
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
lp.OnInitialize = func() (string, error) {
|
|
lp.SetWindowTitle("kitty mouse features demo")
|
|
lp.SetCursorVisible(false)
|
|
lp.QueueWriteString(dnd_start_accepting())
|
|
draw_screen()
|
|
return "", nil
|
|
}
|
|
lp.OnFinalize = func() string {
|
|
lp.SetCursorVisible(true)
|
|
return dnd_stop_accepting()
|
|
}
|
|
|
|
lp.OnMouseEvent = func(ev *loop.MouseEvent) error {
|
|
current_mouse_event = ev
|
|
draw_screen()
|
|
return nil
|
|
}
|
|
lp.OnKeyEvent = func(ev *loop.KeyEvent) error {
|
|
if ev.MatchesPressOrRepeat("esc") || ev.MatchesPressOrRepeat("ctrl+c") {
|
|
lp.Quit(0)
|
|
}
|
|
return nil
|
|
}
|
|
lp.OnResize = func(old_size loop.ScreenSize, new_size loop.ScreenSize) error {
|
|
draw_screen()
|
|
return nil
|
|
}
|
|
lp.OnEscapeCode = func(etype loop.EscapeCodeType, raw []byte) error {
|
|
if etype == loop.OSC {
|
|
return handle_dnd_osc(raw)
|
|
}
|
|
return nil
|
|
}
|
|
err = lp.Run()
|
|
if err != nil {
|
|
rc = 1
|
|
}
|
|
return
|
|
}
|