🎉 Init Project

Author: powduck

.gitignore

@@ -13,3 +13,5 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+.idea
+a.exe

go.mod

@@ -0,0 +1,3 @@
+module github.com/Pow-Duck/processes
+
+go 1.14

processes.go

@@ -0,0 +1,142 @@
+package processes
+
+import (
+	"fmt"
+	"os/exec"
+	"runtime"
+	"strings"
+	"time"
+)
+
+type Process struct {
+	Cmd *exec.Cmd
+}
+
+func Command(exe string, args ...string) (*Process, error) {
+	command := exec.Command(exe, args...)
+	errChan := make(chan error)
+	go func() {
+		if err := command.Start(); err != nil {
+			errChan <- err
+		}
+	}()
+
+	for {
+		select {
+		case <-time.After(time.Millisecond * 50):
+			return &Process{Cmd: command}, nil
+
+		case err := <-errChan:
+			return nil, err
+		}
+	}
+}
+
+func ProcessSnapshot() (map[string]string, error) {
+	if runtime.GOOS == "windows" {
+		return processSnapshotWindows()
+	}
+
+	return processSnapshotLinux()
+}
+
+func GetPid(serverName string) (string, error) {
+	snapshot, err := ProcessSnapshot()
+	if err != nil {
+		return "", err
+	}
+	for k, v := range snapshot {
+		if strings.Index(v, serverName) != -1 {
+			return k, nil
+		}
+	}
+	return "", fmt.Errorf("NOT FUND")
+}
+
+func KillByPid(pid string) error {
+	if runtime.GOOS == "windows" {
+		return killPidWindows(pid)
+	}
+	return killPidLinux(pid)
+}
+
+func killPidLinux(pid string) error {
+	_, err := RunCommand(fmt.Sprintf("kill -9 %s", pid))
+	return err
+}
+
+func killPidWindows(pid string) error {
+	c := fmt.Sprintf(`taskkill /F /pid %s`, pid)
+	_, err := RunCommand(c)
+	return err
+}
+
+func processSnapshotLinux() (map[string]string, error) {
+	rp := map[string]string{}
+	a := `ps -ef |awk '{print $2,$8}'`
+	command, err := RunCommand(a)
+	if err != nil {
+		return nil, err
+	}
+	split := strings.Split(command, "\n")
+	for _, v := range split {
+		sp2 := strings.Split(v, " ")
+		if len(sp2) >= 2 {
+			rp[sp2[0]] = sp2[1]
+		}
+	}
+	return rp, nil
+}
+
+func processSnapshotWindows() (map[string]string, error) {
+	rp := map[string]string{}
+	a := "tasklist"
+	command, err := RunCommand(a)
+	if err != nil {
+		return nil, err
+	}
+	split := strings.Split(command, "\n")
+	for _, v := range split {
+		sp2 := strings.Split(v, " ")
+		sp2 = clearNone(sp2)
+		if len(sp2) >= 2 {
+			rp[sp2[1]] = sp2[0]
+		}
+	}
+	return rp, nil
+}
+
+func runInWindows(cmd string) (string, error) {
+	result, err := exec.Command("cmd", "/c", cmd).Output()
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSpace(string(result)), err
+}
+
+func RunCommand(cmd string) (string, error) {
+	if runtime.GOOS == "windows" {
+		return runInWindows(cmd)
+	} else {
+		return runInLinux(cmd)
+	}
+}
+
+func runInLinux(cmd string) (string, error) {
+	//fmt.Println("Running Linux cmd:" + cmd)
+	result, err := exec.Command("/bin/sh", "-c", cmd).Output()
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSpace(string(result)), err
+}
+
+func clearNone(sp []string) []string {
+	var result []string
+	for _, v := range sp {
+		if v != "" {
+			result = append(result, v)
+		}
+	}
+	return result
+}

processes_test.go

@@ -0,0 +1,29 @@
+package processes
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"os/exec"
+	"testing"
+)
+
+func TestBaseCommand(t *testing.T) {
+	command := exec.Command("a.exe")
+	command.Stdout = os.Stdout
+	command.Stderr = os.Stderr
+
+	err := command.Start()
+	if err != nil {
+		log.Fatalln(err)
+	}
+}
+
+func TestQuery(t *testing.T) {
+	command, err := RunCommand(`C:\Users\Github/AppData/Local/chia-blockchain/app-1.1.6/resources/app.asar.unpacked/daemon/chia.exe version`)
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	fmt.Println(command)
+}

query.go

@@ -0,0 +1,48 @@
+package processes
+
+import (
+	"bufio"
+	"bytes"
+	"os/exec"
+	"time"
+)
+
+type ProcessQuery struct {
+	stdout *bytes.Buffer
+	stderr *bytes.Buffer
+}
+
+func Query(exe string, args ...string) (*ProcessQuery, error) {
+	process := ProcessQuery{
+		stdout: bytes.NewBuffer([]byte{}),
+		stderr: bytes.NewBuffer([]byte{}),
+	}
+
+	command := exec.Command(exe, args...)
+
+	process.stdout = bytes.NewBuffer([]byte{})
+	stdoutWrite := bufio.NewWriter(process.stdout)
+
+	process.stderr = bytes.NewBuffer([]byte{})
+	stderrWrite := bufio.NewWriter(process.stderr)
+
+	command.Stdout = stdoutWrite
+	command.Stderr = stderrWrite
+
+	errChan := make(chan error)
+	go func() {
+		if err := command.Start(); err != nil {
+			errChan <- err
+		}
+	}()
+
+	for {
+		select {
+		case <-time.After(time.Millisecond * 50):
+			return &process, nil
+
+		case err := <-errChan:
+			return nil, err
+		}
+	}
+}

test/a.go

@@ -0,0 +1,15 @@
+package main
+
+import (
+	"fmt"
+	"time"
+)
+
+func main() {
+	i := 0
+	for {
+		i += 1
+		time.Sleep(time.Second)
+		fmt.Println("hello world: ", i)
+	}
+}