viye: internal/plumbing/json/json.go (fbda676c3bdbee9f984c64d5e09cbeb8555cda89)

viye/internal/plumbing/json/json.go (view raw)

Oleksandr Smirnov strip body of ": ", 1 month ago

package json

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/olexsmir/viye/internal/viye"
)

type Tool struct{}

func (Tool) Name() string               { return "json" }
func (Tool) Match(c *viye.Context) bool { return c.Cmd == "json" }
func (Tool) Execute(c *viye.Context) (string, error) {
	if len(c.Body) == 0 {
		return ": name: your name\n" +
			": age: 0\n", nil
	}

	m := make(map[string]string)
	for _, line := range c.Body {
		before, after, ok := strings.Cut(line, ":")
		if !ok {
			continue
		}

		key := strings.TrimSpace(before)
		val := strings.TrimSpace(after)
		if key != "" {
			m[key] = val
		}
	}

	mashaled, err := json.Marshal(m)
	if err != nil {
		return "", fmt.Errorf("demo: %w", err)
	}
	return fmt.Sprintf("| %s\n", string(mashaled)), nil
}

1	package json
2
3	import (
4	"encoding/json"
5	"fmt"
6	"strings"
7
8	"github.com/olexsmir/viye/internal/viye"
9	)
10
11	type Tool struct{}
12
13	func (Tool) Name() string { return "json" }
14	func (Tool) Match(c *viye.Context) bool { return c.Cmd == "json" }
15	func (Tool) Execute(c *viye.Context) (string, error) {
16	if len(c.Body) == 0 {
17	return ": name: your name\n" +
18	": age: 0\n", nil
19	}
20
21	m := make(map[string]string)
22	for _, line := range c.Body {
23	before, after, ok := strings.Cut(line, ":")
24	if !ok {
25	continue
26	}
27
28	key := strings.TrimSpace(before)
29	val := strings.TrimSpace(after)
30	if key != "" {
31	m[key] = val
32	}
33	}
34
35	mashaled, err := json.Marshal(m)
36	if err != nil {
37	return "", fmt.Errorf("demo: %w", err)
38	}
39	return fmt.Sprintf("\| %s\n", string(mashaled)), nil
40	}