wide/notification/notifications.go

155 lines
3.4 KiB
Go
Raw Normal View History

2014-09-15 09:03:44 +04:00
// 通知.
2014-09-15 10:24:40 +04:00
package notification
2014-09-15 09:03:44 +04:00
import (
"net/http"
"os"
"os/exec"
"runtime"
2014-09-15 14:03:52 +04:00
"time"
2014-09-15 09:03:44 +04:00
2014-09-15 19:20:01 +04:00
"strconv"
2014-09-15 09:03:44 +04:00
"github.com/b3log/wide/conf"
2014-09-15 14:03:52 +04:00
"github.com/b3log/wide/event"
2014-09-15 19:20:01 +04:00
"github.com/b3log/wide/i18n"
2014-09-15 09:03:44 +04:00
"github.com/b3log/wide/user"
"github.com/golang/glog"
"github.com/gorilla/websocket"
)
2014-09-15 19:20:01 +04:00
const (
Error = "ERROR" // 通知.严重程度ERROR
Warn = "WARN" // 通知.严重程度WARN
Info = "INFO" // 通知.严重程度INFO
Setup = "Setup" // 通知.类型:安装
)
2014-09-15 09:03:44 +04:00
// 通知结构.
type Notification struct {
2014-09-15 19:20:01 +04:00
event *event.Event
Type string `json:"type"`
Severity string `json:"severity"`
Message string `json:"message"`
2014-09-15 09:03:44 +04:00
}
2014-09-15 14:03:52 +04:00
// 一个用户会话的 WebSocket 通道结构.
type WSChannel struct {
2014-09-15 19:20:01 +04:00
Conn *websocket.Conn // WebSocket 连接
Request *http.Request // 关联的 HTTP 请求
Time time.Time // 该通道最近一次使用时间
2014-09-15 14:03:52 +04:00
}
2014-09-15 10:24:40 +04:00
// 通知通道.
2014-09-15 19:20:01 +04:00
// <sid, WSChannel>
var notificationWSs = map[string]*WSChannel{}
func event2Notification(e *event.Event) {
if nil == notificationWSs[e.Sid] {
return
}
wsChannel := notificationWSs[e.Sid]
var notification Notification
switch e.Code {
case event.EvtCodeGocodeNotFound:
notification = Notification{event: e, Type: Setup, Severity: Error}
case event.EvtCodeIDEStubNotFound:
notification = Notification{event: e, Type: Setup, Severity: Error}
default:
glog.Warningf("Can't handle event[code=%d]", e.Code)
return
}
msgs := i18n.GetLangs(wsChannel.Request)
notification.Message = msgs["notification_"+strconv.Itoa(e.Code)].(string)
wsChannel.Conn.WriteJSON(&notification)
}
2014-09-15 09:03:44 +04:00
func WSHandler(w http.ResponseWriter, r *http.Request) {
session, _ := user.Session.Get(r, "wide-session")
sid := session.Values["id"].(string)
2014-09-15 14:03:52 +04:00
conn, _ := websocket.Upgrade(w, r, nil, 1024, 1024)
2014-09-15 19:20:01 +04:00
wsChan := WSChannel{Conn: conn, Request: r, Time: time.Now()}
2014-09-15 14:03:52 +04:00
2014-09-15 19:20:01 +04:00
notificationWSs[sid] = &wsChan
2014-09-15 09:03:44 +04:00
ret := map[string]interface{}{"output": "Notification initialized", "cmd": "init-notification"}
2014-09-15 14:03:52 +04:00
wsChan.Conn.WriteJSON(&ret)
2014-09-15 19:20:01 +04:00
glog.Infof("Open a new [Notification] with session [%s], %d", sid, len(notificationWSs))
2014-09-15 09:03:44 +04:00
2014-09-15 19:20:01 +04:00
// 初始化用户事件队列
event.InitUserQueue(sid, event.HandleFunc(event2Notification))
2014-09-15 09:03:44 +04:00
input := map[string]interface{}{}
for {
2014-09-15 14:03:52 +04:00
if err := wsChan.Conn.ReadJSON(&input); err != nil {
2014-09-15 09:03:44 +04:00
if err.Error() == "EOF" {
return
}
if err.Error() == "unexpected EOF" {
return
}
2014-09-15 14:03:52 +04:00
glog.Error("Notification WS ERROR: " + err.Error())
2014-09-15 09:03:44 +04:00
return
}
output := ""
2014-09-15 14:03:52 +04:00
ret = map[string]interface{}{"output": output, "cmd": "notification-output"}
2014-09-15 09:03:44 +04:00
2014-09-15 14:03:52 +04:00
if err := wsChan.Conn.WriteJSON(&ret); err != nil {
glog.Error("Notification WS ERROR: " + err.Error())
2014-09-15 09:03:44 +04:00
return
}
}
}
func pipeCommands(username string, commands ...*exec.Cmd) string {
for i, command := range commands[:len(commands)-1] {
setCmdEnv(command, username)
stdout, err := command.StdoutPipe()
if nil != err {
return err.Error()
}
command.Start()
commands[i+1].Stdin = stdout
}
last := commands[len(commands)-1]
setCmdEnv(last, username)
out, err := last.CombinedOutput()
if err != nil {
return err.Error()
}
return string(out)
}
func setCmdEnv(cmd *exec.Cmd, username string) {
userWorkspace := conf.Wide.GetUserWorkspace(username)
cmd.Env = append(cmd.Env,
"TERM="+os.Getenv("TERM"),
"GOPATH="+userWorkspace,
"GOOS="+runtime.GOOS,
"GOARCH="+runtime.GOARCH,
"GOROOT="+runtime.GOROOT(),
"PATH="+os.Getenv("PATH"))
cmd.Dir = userWorkspace
}