viye: internal/plumbing/json/json.go (00a2383)

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

Oleksandr Smirnov use olexsmir.xyz/x/is for asserts; test missing plumbings, 2 months 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(ctx *viye.Context) bool { return ctx.Cmd == "json" }
func (Tool) Execute(ctx *viye.Context) (string, error) {
	if len(ctx.Body) == 0 {
		return ": name: your name\n" +
			": age: 0\n", nil
	}

	m := make(map[string]string)
	for _, line := range ctx.Body {
		line = strings.TrimPrefix(line, ": ")
		col := strings.Index(line, ":")
		if col < 0 {
			continue
		}

		key := strings.TrimSpace(line[:col])
		val := strings.TrimSpace(line[col+1:])
		if key != "" {
			m[key] = val
		}
	}

	b, err := json.Marshal(m)
	if err != nil {
		return "", fmt.Errorf("demo: %w", err)
	}
	return fmt.Sprintf("| %s\n", string(b)), 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(ctx *viye.Context) bool { return ctx.Cmd == "json" }
15	func (Tool) Execute(ctx *viye.Context) (string, error) {
16	if len(ctx.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 ctx.Body {
23	line = strings.TrimPrefix(line, ": ")
24	col := strings.Index(line, ":")
25	if col < 0 {
26	continue
27	}
28
29	key := strings.TrimSpace(line[:col])
30	val := strings.TrimSpace(line[col+1:])
31	if key != "" {
32	m[key] = val
33	}
34	}
35
36	b, err := json.Marshal(m)
37	if err != nil {
38	return "", fmt.Errorf("demo: %w", err)
39	}
40	return fmt.Sprintf("\| %s\n", string(b)), nil
41	}