clipboard_images.go RawHistoryFolder

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