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

1

package shell

2

3

import (

4

	"context"

5

	"fmt"

6

	"os"

7

	"os/exec"

8

	"path/filepath"

9

	"strconv"

10

	"strings"

11

	"syscall"

12

13

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

14

15

16

type Tool struct{}

17

18

func (Tool) Name() string { return "shell($, $$, ls, kill, mkdir)" }

19

func (Tool) Match(c *viye.Context) bool {

20

	return isCmd(c) || isBGCmd(c) || isLs(c) || isKill(c) || isMkdir(c)

21

22

23

func (Tool) Execute(c *viye.Context) (string, error) {

24

	switch {

25

	case isCmd(c):

26

		ctx, cancel := context.WithTimeout(context.Background(), viye.Timeout)

27

		defer cancel()

28

29

		cmd := exec.CommandContext(ctx, "sh", "-c", strings.Join(c.Args, " "))

30

		cmd.Dir = c.Dir

31

		out, err := cmd.CombinedOutput()

32

		if err != nil {

33

			if ctx.Err() == context.DeadlineExceeded {

34

				return "", fmt.Errorf("timeout: command exceeded 5s")

35

36

37

		return viye.FormatOutput(string(out)), nil

38

39

	case isBGCmd(c):

40

		cmd := exec.Command("sh", "-c", strings.Join(c.Args, " "))

41

		cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}

42

		cmd.Dir = c.Dir

43

		if err := cmd.Start(); err != nil {

44

			return "", err

45

46

		return fmt.Sprintf("| pid: %d", cmd.Process.Pid), nil

47

48

	case isLs(c):

49

		entries, err := os.ReadDir(c.Dir)

50

		if err != nil {

51

			return "", err

52

53

		var res []string

54

		for _, e := range entries {

55

			name := e.Name()

56

			if e.IsDir() {

57

				name += "/"

58

59

			res = append(res, name)

60

61

		return viye.FormatOutputList(res), nil

62

63

	case isKill(c):

64

		pid, err := strconv.Atoi(c.Args[0])

65

		if err != nil {

66

			return "", fmt.Errorf("kill: invalid pid %q", c.Args[0])

67

68

		p, err := os.FindProcess(pid)

69

		if err != nil {

70

			return "", fmt.Errorf("kill: %w", err)

71

72

		if err := p.Kill(); err != nil {

73

			return "", fmt.Errorf("kill: %w", err)

74

75

		return "| done\n", nil

76

77

	case isMkdir(c):

78

		path := filepath.Join(append([]string{c.Dir}, c.Args...)...)

79

		if err := os.MkdirAll(path, 0o755); err != nil {

80

			return "", fmt.Errorf("mkdir: %w", err)

81

82

		return "| created " + path + "\n", nil

83

84

	default:

85

		panic("shouldn't have paniced")

86

87

88

89

func isCmd(c *viye.Context) bool   { return c.Cmd == "$" }

90

func isBGCmd(c *viye.Context) bool { return c.Cmd == "$$" }

91

func isLs(c *viye.Context) bool    { return c.Cmd == "ls" }

92

func isKill(c *viye.Context) bool  { return c.Cmd == "kill" && len(c.Args) == 1 }

93

func isMkdir(c *viye.Context) bool { return c.Cmd == "mkdir" && len(c.Args) >= 1 }

1	package shell
2
3	import (
4	"context"
5	"fmt"
6	"os"
7	"os/exec"
8	"path/filepath"
9	"strconv"
10	"strings"
11	"syscall"
12
13	"github.com/olexsmir/viye/internal/viye"
14	)
15
16	type Tool struct{}
17
18	func (Tool) Name() string { return "shell($, $$, ls, kill, mkdir)" }
19	func (Tool) Match(c *viye.Context) bool {
20	return isCmd(c) \|\| isBGCmd(c) \|\| isLs(c) \|\| isKill(c) \|\| isMkdir(c)
21	}
22
23	func (Tool) Execute(c *viye.Context) (string, error) {
24	switch {
25	case isCmd(c):
26	ctx, cancel := context.WithTimeout(context.Background(), viye.Timeout)
27	defer cancel()
28
29	cmd := exec.CommandContext(ctx, "sh", "-c", strings.Join(c.Args, " "))
30	cmd.Dir = c.Dir
31	out, err := cmd.CombinedOutput()
32	if err != nil {
33	if ctx.Err() == context.DeadlineExceeded {
34	return "", fmt.Errorf("timeout: command exceeded 5s")
35	}
36	}
37	return viye.FormatOutput(string(out)), nil
38
39	case isBGCmd(c):
40	cmd := exec.Command("sh", "-c", strings.Join(c.Args, " "))
41	cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
42	cmd.Dir = c.Dir
43	if err := cmd.Start(); err != nil {
44	return "", err
45	}
46	return fmt.Sprintf("\| pid: %d", cmd.Process.Pid), nil
47
48	case isLs(c):
49	entries, err := os.ReadDir(c.Dir)
50	if err != nil {
51	return "", err
52	}
53	var res []string
54	for _, e := range entries {
55	name := e.Name()
56	if e.IsDir() {
57	name += "/"
58	}
59	res = append(res, name)
60	}
61	return viye.FormatOutputList(res), nil
62
63	case isKill(c):
64	pid, err := strconv.Atoi(c.Args[0])
65	if err != nil {
66	return "", fmt.Errorf("kill: invalid pid %q", c.Args[0])
67	}
68	p, err := os.FindProcess(pid)
69	if err != nil {
70	return "", fmt.Errorf("kill: %w", err)
71	}
72	if err := p.Kill(); err != nil {
73	return "", fmt.Errorf("kill: %w", err)
74	}
75	return "\| done\n", nil
76
77	case isMkdir(c):
78	path := filepath.Join(append([]string{c.Dir}, c.Args...)...)
79	if err := os.MkdirAll(path, 0o755); err != nil {
80	return "", fmt.Errorf("mkdir: %w", err)
81	}
82	return "\| created " + path + "\n", nil
83
84	default:
85	panic("shouldn't have paniced")
86	}
87	}
88
89	func isCmd(c *viye.Context) bool { return c.Cmd == "$" }
90	func isBGCmd(c *viye.Context) bool { return c.Cmd == "$$" }
91	func isLs(c *viye.Context) bool { return c.Cmd == "ls" }
92	func isKill(c *viye.Context) bool { return c.Cmd == "kill" && len(c.Args) == 1 }
93	func isMkdir(c *viye.Context) bool { return c.Cmd == "mkdir" && len(c.Args) >= 1 }