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

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

Oleksandr Smirnov some refactoring again, 1 month ago

package curl

import (
	"context"
	"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 "", viye.ErrTimeout
	}

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

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