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

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

Oleksandr Smirnov some refactoring again, 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)
		if key != "" {
			m[key] = strings.TrimSpace(after)
		}
	}

	marshaled, err := json.Marshal(m)
	if err != nil {
		return "", fmt.Errorf("demo: %w", err)
	}
	return viye.FormatOutput(marshaled), 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	if key != "" {
30	m[key] = strings.TrimSpace(after)
31	}
32	}
33
34	marshaled, err := json.Marshal(m)
35	if err != nil {
36	return "", fmt.Errorf("demo: %w", err)
37	}
38	return viye.FormatOutput(marshaled), nil
39	}