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

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

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

package curl

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

	"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(), viye.Timeout)
	defer cancel()

	args := append([]string{"-sSL"}, c.Args...)
	args = append(args, 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 %s", viye.Timeout)
	}

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