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

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

Oleksandr Smirnov refactor, 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 {
		line = strings.TrimPrefix(line, ": ")
		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	line = strings.TrimPrefix(line, ": ")
24	before, after, ok := strings.Cut(line, ":")
25	if !ok {
26	continue
27	}
28
29	key := strings.TrimSpace(before)
30	val := strings.TrimSpace(after)
31	if key != "" {
32	m[key] = val
33	}
34	}
35
36	mashaled, err := json.Marshal(m)
37	if err != nil {
38	return "", fmt.Errorf("demo: %w", err)
39	}
40	return fmt.Sprintf("\| %s\n", string(mashaled)), nil
41	}