package output import ( "encoding/json" "github.com/b3log/wide/conf" "github.com/b3log/wide/user" "github.com/b3log/wide/util" "github.com/golang/glog" "github.com/gorilla/websocket" "io" "math/rand" "net/http" "os" "os/exec" "runtime" "strings" ) var outputWS = map[string]*websocket.Conn{} func WSHandler(w http.ResponseWriter, r *http.Request) { session, _ := user.Session.Get(r, "wide-session") sid := session.Values["id"].(string) outputWS[sid], _ = websocket.Upgrade(w, r, nil, 1024, 1024) ret := map[string]interface{}{"output": "Ouput initialized", "cmd": "init-output"} outputWS[sid].WriteJSON(&ret) glog.Infof("Open a new [Output] with session [%s], %d", sid, len(outputWS)) } func RunHandler(w http.ResponseWriter, r *http.Request) { data := map[string]interface{}{"succ": true} defer util.RetJSON(w, r, data) session, _ := user.Session.Get(r, "wide-session") sid := session.Values["id"].(string) decoder := json.NewDecoder(r.Body) var args map[string]interface{} if err := decoder.Decode(&args); err != nil { glog.Error(err) data["succ"] = false return } filePath := args["executable"].(string) curDir := filePath[:strings.LastIndex(filePath, string(os.PathSeparator))] cmd := exec.Command(filePath) cmd.Dir = curDir stdout, err := cmd.StdoutPipe() if nil != err { glog.Error(err) data["succ"] = false return } stderr, err := cmd.StderrPipe() if nil != err { glog.Error(err) data["succ"] = false return } reader := io.MultiReader(stdout, stderr) cmd.Start() channelRet := map[string]interface{}{} go func(runningId int) { glog.Infof("Session [%s] is running [id=%d, file=%s]", sid, runningId, filePath) for { buf := make([]byte, 1024) count, err := reader.Read(buf) if nil != err || 0 == count { glog.Infof("Session [%s] 's running [id=%d, file=%s] has done", sid, runningId, filePath) break } else { channelRet["output"] = string(buf[:count]) channelRet["cmd"] = "run" if nil != outputWS[sid] { err := outputWS[sid].WriteJSON(&channelRet) if nil != err { glog.Error(err) break } } } } }(rand.Int()) } func BuildHandler(w http.ResponseWriter, r *http.Request) { data := map[string]interface{}{"succ": true} defer util.RetJSON(w, r, data) session, _ := user.Session.Get(r, "wide-session") sid := session.Values["id"].(string) username := session.Values["username"].(string) decoder := json.NewDecoder(r.Body) var args map[string]interface{} if err := decoder.Decode(&args); err != nil { glog.Error(err) data["succ"] = false return } filePath := args["file"].(string) curDir := filePath[:strings.LastIndex(filePath, string(os.PathSeparator))] fout, err := os.Create(filePath) if nil != err { glog.Error(err) data["succ"] = false return } code := args["code"].(string) fout.WriteString(code) if err := fout.Close(); nil != err { glog.Error(err) data["succ"] = false return } suffix := "" if "windows" == runtime.GOOS { suffix = ".exe" } executable := "main" + suffix argv := []string{"build", "-o", executable, filePath} cmd := exec.Command("go", argv...) cmd.Dir = curDir setCmdEnv(cmd, username) glog.Infof("go build -o %s %s", executable, filePath) executable = curDir + string(os.PathSeparator) + executable // 先把可执行文件删了 err = os.RemoveAll(executable) if nil != err { glog.Info(err) data["succ"] = false return } stdout, err := cmd.StdoutPipe() if nil != err { glog.Error(err) data["succ"] = false return } stderr, err := cmd.StderrPipe() if nil != err { glog.Error(err) data["succ"] = false return } if data["succ"].(bool) { reader := io.MultiReader(stdout, stderr) cmd.Start() go func(runningId int) { glog.Infof("Session [%s] is building [id=%d, file=%s]", sid, runningId, filePath) // 一次性读取 buf := make([]byte, 1024*8) count, _ := reader.Read(buf) if 0 == count { // 说明构建成功,没有错误信息输出 go func() { // 运行 go install,生成的库用于 gocode lib-path cmd := exec.Command("go", "install") cmd.Dir = curDir setCmdEnv(cmd, username) out, _ := cmd.CombinedOutput() if len(out) > 0 { glog.Warning(string(out)) } }() } channelRet := map[string]interface{}{} channelRet["output"] = string(buf[:count]) channelRet["cmd"] = "build" channelRet["nextCmd"] = "run" channelRet["executable"] = executable if nil != outputWS[sid] { glog.Infof("Session [%s] 's build [id=%d, file=%s] has done", sid, runningId, filePath) err := outputWS[sid].WriteJSON(&channelRet) if nil != err { glog.Error(err) } } }(rand.Int()) } } func GoGetHandler(w http.ResponseWriter, r *http.Request) { data := map[string]interface{}{"succ": true} defer util.RetJSON(w, r, data) session, _ := user.Session.Get(r, "wide-session") sid := session.Values["id"].(string) username := session.Values["username"].(string) decoder := json.NewDecoder(r.Body) var args map[string]interface{} if err := decoder.Decode(&args); err != nil { glog.Error(err) data["succ"] = false return } filePath := args["file"].(string) curDir := filePath[:strings.LastIndex(filePath, string(os.PathSeparator))] cmd := exec.Command("go", "get") cmd.Dir = curDir setCmdEnv(cmd, username) stdout, err := cmd.StdoutPipe() if nil != err { glog.Error(err) data["succ"] = false return } stderr, err := cmd.StderrPipe() if nil != err { glog.Error(err) data["succ"] = false return } reader := io.MultiReader(stdout, stderr) cmd.Start() channelRet := map[string]interface{}{} go func(runningId int) { glog.Infof("Session [%s] is running [go get] [runningId=%d]", sid, runningId) for { buf := make([]byte, 1024) count, err := reader.Read(buf) if nil != err || 0 == count { glog.Infof("Session [%s] 's running [go get] [runningId=%d] has done", sid, runningId) break } else { channelRet["output"] = string(buf[:count]) channelRet["cmd"] = "go get" if nil != outputWS[sid] { err := outputWS[sid].WriteJSON(&channelRet) if nil != err { glog.Error(err) break } } } } }(rand.Int()) } func setCmdEnv(cmd *exec.Cmd, username string) { userWorkspace := conf.Wide.GetUserWorkspace(username) cmd.Env = append(cmd.Env, "GOPATH="+userWorkspace, "GOOS="+runtime.GOOS, "GOARCH="+runtime.GOARCH, "GOROOT="+runtime.GOROOT(), "PATH="+os.Getenv("PATH")) //"TERM="+os.Getenv("COMSPEC"), //"ComSpec="+os.Getenv("ComSpec") }