viye: internal/plumbing/shell/shell.go (0da33435eec3ff0b661e98fd06fbfb0729b81093)

1

package shell

2

3

import (

4

	"fmt"

5

	"os"

6

	"os/exec"

7

	"path/filepath"

8

	"strconv"

9

	"strings"

10

	"syscall"

11

12

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

13

14

15

type Tool struct{}

16

17

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

18

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

19

	return isCmd(ctx) || isBGCmd(ctx) || isKill(ctx) || isMkdir(ctx)

20

21

22

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

23

	switch {

24

	case isCmd(ctx):

25

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

26

		cmd.Dir = ctx.Dir

27

		out, _ := cmd.CombinedOutput()

28

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

29

30

	case isBGCmd(ctx):

31

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

32

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

33

		cmd.Dir = ctx.Dir

34

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

35

			return "", err

36

37

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

38

39

	case isKill(ctx):

40

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

41

		if err != nil {

42

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

43

44

		p, err := os.FindProcess(pid)

45

		if err != nil {

46

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

47

48

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

49

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

50

51

		return "| done\n", nil

52

53

	case isMkdir(ctx):

54

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

55

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

56

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

57

58

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

59

60

	default:

61

		panic("shouldn't have paniced")

62

63

64

65

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

66

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

67

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

68

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

1	package shell
2
3	import (
4	"fmt"
5	"os"
6	"os/exec"
7	"path/filepath"
8	"strconv"
9	"strings"
10	"syscall"
11
12	"github.com/olexsmir/viye/internal/viye"
13	)
14
15	type Tool struct{}
16
17	func (Tool) Name() string { return "shell($, $$, kill, mkdir)" }
18	func (Tool) Match(ctx *viye.Context) bool {
19	return isCmd(ctx) \|\| isBGCmd(ctx) \|\| isKill(ctx) \|\| isMkdir(ctx)
20	}
21
22	func (Tool) Execute(ctx *viye.Context) (string, error) {
23	switch {
24	case isCmd(ctx):
25	cmd := exec.Command("sh", "-c", strings.Join(ctx.Args, " "))
26	cmd.Dir = ctx.Dir
27	out, _ := cmd.CombinedOutput()
28	return viye.Indent(string(out)), nil
29
30	case isBGCmd(ctx):
31	cmd := exec.Command("sh", "-c", strings.Join(ctx.Args, " "))
32	cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
33	cmd.Dir = ctx.Dir
34	if err := cmd.Start(); err != nil {
35	return "", err
36	}
37	return fmt.Sprintf("\| pid: %d", cmd.Process.Pid), nil
38
39	case isKill(ctx):
40	pid, err := strconv.Atoi(ctx.Args[0])
41	if err != nil {
42	return "", fmt.Errorf("kill: invalid pid %q", ctx.Args[0])
43	}
44	p, err := os.FindProcess(pid)
45	if err != nil {
46	return "", fmt.Errorf("kill: %w", err)
47	}
48	if err := p.Kill(); err != nil {
49	return "", fmt.Errorf("kill: %w", err)
50	}
51	return "\| done\n", nil
52
53	case isMkdir(ctx):
54	path := filepath.Join(append([]string{ctx.Dir}, ctx.Args...)...)
55	if err := os.MkdirAll(path, 0o755); err != nil {
56	return "", fmt.Errorf("mkdir: %w", err)
57	}
58	return "\| created " + path + "\n", nil
59
60	default:
61	panic("shouldn't have paniced")
62	}
63	}
64
65	func isCmd(c *viye.Context) bool { return c.Cmd == "$" }
66	func isBGCmd(c *viye.Context) bool { return c.Cmd == "$$" }
67	func isKill(c *viye.Context) bool { return c.Cmd == "kill" && len(c.Args) == 1 }
68	func isMkdir(c *viye.Context) bool { return c.Cmd == "mkdir" && len(c.Args) >= 1 }