69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
// import file writer
|
|
"os"
|
|
// import websocket
|
|
)
|
|
|
|
func handleCase(caseName, data string) string {
|
|
// caseName = strings.ReplaceAll(caseName, "\n", "")
|
|
caseName = strings.ReplaceAll(caseName, ":", "")
|
|
|
|
log.Println("Handling case", caseName)
|
|
switch caseName {
|
|
case "python", "python3", "py":
|
|
runPython(data)
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func runPython(data string) {
|
|
log.Println("Running python")
|
|
// create a file in the workspace directory
|
|
f, err := os.Create("./workspace/main.py")
|
|
if err != nil {
|
|
log.Println("Error creating file")
|
|
return
|
|
}
|
|
defer f.Close()
|
|
|
|
// write the data to the file
|
|
_, err = f.WriteString(data)
|
|
if err != nil {
|
|
log.Println("Error writing to file")
|
|
return
|
|
}
|
|
|
|
// find python, python3, or py in the system
|
|
fname, err := exec.LookPath("python3")
|
|
if err != nil {
|
|
log.Println("Python not found")
|
|
} else {
|
|
log.Println("Python found at", fname)
|
|
}
|
|
// fname := "C:\\Users\\zachd\\scoop\\shims\\python3.exe"
|
|
|
|
// if main.py does not exist, return
|
|
if _, err := os.Stat("./workspace/main.py"); os.IsNotExist(err) {
|
|
log.Println("File does not exist")
|
|
return
|
|
}
|
|
|
|
cmd := exec.Command(fname, "./main.py")
|
|
cmd.Dir = filepath.ToSlash("./workspace/")
|
|
// set the command's stdout to the pipe writer
|
|
cmd.Stdout = pyWriter
|
|
|
|
// set the command's stderr to the pipe writer
|
|
cmd.Stderr = pyWriter
|
|
|
|
// start the command
|
|
cmd.Run()
|
|
log.Println("Python runtime finished")
|
|
}
|