clipboard_images.go RawHistoryFolder

1package modes
2
3import (
4	"fmt"
5	"os"
6	"strings"
7
8	"github.com/patriceckhart/zot/packages/provider"
9)
10
11type clipboardImageAttachment struct {
12	Marker string
13	Image  provider.ImageBlock
14}
15
16type clipboardImageReader func() (string, []byte, bool, error)
17
18func readClipboardImageAndCleanup(read clipboardImageReader) ([]byte, bool, error) {
19	path, data, ok, err := read()
20	if path != "" {
21		if removeErr := os.Remove(path); removeErr != nil && err == nil {
22			err = fmt.Errorf("remove temporary clipboard image: %w", removeErr)
23		}
24	}
25	return data, ok, err
26}
27
28func preparePromptWithClipboardImages(text string, pending []clipboardImageAttachment) (string, []provider.ImageBlock) {
29	if len(pending) == 0 {
30		return text, nil
31	}
32
33	out := text
34	images := make([]provider.ImageBlock, 0, len(pending))
35	for _, item := range pending {
36		if item.Marker == "" || !strings.Contains(out, item.Marker) {
37			continue
38		}
39		out = removeClipboardMarker(out, item.Marker)
40		images = append(images, item.Image)
41	}
42	return strings.TrimSpace(out), images
43}
44
45func removeClipboardMarker(text, marker string) string {
46	for {
47		idx := strings.Index(text, marker)
48		if idx < 0 {
49			return text
50		}
51		end := idx + len(marker)
52		prevInline := idx > 0 && isInlineWhitespace(text[idx-1])
53		nextInline := end < len(text) && isInlineWhitespace(text[end])
54		prevLineBreak := idx == 0 || idx > 0 && isLineBreak(text[idx-1])
55		nextLineBreak := end == len(text) || end < len(text) && isLineBreak(text[end])
56		switch {
57		case prevInline && nextInline:
58			text = text[:idx] + text[end+1:]
59		case prevInline && nextLineBreak:
60			text = text[:idx-1] + text[end:]
61		case prevLineBreak && nextInline:
62			text = text[:idx] + text[end+1:]
63		default:
64			text = text[:idx] + text[end:]
65		}
66	}
67}
68
69func isInlineWhitespace(b byte) bool {
70	return b == ' ' || b == '\t'
71}
72
73func isLineBreak(b byte) bool {
74	return b == '\n' || b == '\r'
75}