viye: internal/plumbing/curl/curl.go (089144fdd711d2f3e016714c6a1a7942330a1d36)

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

Oleksandr Smirnov add curl tool, 1 month ago

package curl

import (
	"context"
	"fmt"
	"os/exec"
	"strings"
	"time"

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

type Tool struct{}

func (Tool) Name() string               { return "curl" }
func (Tool) Match(c *viye.Context) bool { return c.Cmd == "curl" }
func (Tool) Execute(c *viye.Context) (string, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	args := append([]string{"-sSL"}, c.Args...)
	args = append(args, viye.ParseData(c.Body)...)
	cmd := exec.CommandContext(ctx, "curl", args...)
	cmd.Dir = c.Dir

	output, err := cmd.CombinedOutput()
	if err != nil && ctx.Err() == context.DeadlineExceeded {
		return "", fmt.Errorf("timeout: command exceeded 5s")
	}

	out := strings.ReplaceAll(string(output), "\r", "")
	return viye.FormatOutput(out), nil
}

1	package curl
2
3	import (
4	"context"
5	"fmt"
6	"os/exec"
7	"strings"
8	"time"
9
10	"github.com/olexsmir/viye/internal/viye"
11	)
12
13	type Tool struct{}
14
15	func (Tool) Name() string { return "curl" }
16	func (Tool) Match(c *viye.Context) bool { return c.Cmd == "curl" }
17	func (Tool) Execute(c *viye.Context) (string, error) {
18	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
19	defer cancel()
20
21	args := append([]string{"-sSL"}, c.Args...)
22	args = append(args, viye.ParseData(c.Body)...)
23	cmd := exec.CommandContext(ctx, "curl", args...)
24	cmd.Dir = c.Dir
25
26	output, err := cmd.CombinedOutput()
27	if err != nil && ctx.Err() == context.DeadlineExceeded {
28	return "", fmt.Errorf("timeout: command exceeded 5s")
29	}
30
31	out := strings.ReplaceAll(string(output), "\r", "")
32	return viye.FormatOutput(out), nil
33	}