wide/main.go

259 lines
6.5 KiB
Go
Raw Normal View History

2014-08-18 17:45:43 +04:00
package main
import (
2014-09-21 16:31:36 +04:00
"encoding/json"
2014-09-02 18:57:30 +04:00
"flag"
2014-09-07 13:07:25 +04:00
"html/template"
"math/rand"
"net/http"
"runtime"
"strconv"
2014-09-24 10:36:34 +04:00
"time"
2014-09-07 13:07:25 +04:00
2014-08-18 17:51:03 +04:00
"github.com/b3log/wide/conf"
"github.com/b3log/wide/editor"
2014-09-15 10:24:40 +04:00
"github.com/b3log/wide/event"
2014-08-18 17:51:03 +04:00
"github.com/b3log/wide/file"
2014-08-25 18:17:02 +04:00
"github.com/b3log/wide/i18n"
2014-09-15 14:03:52 +04:00
"github.com/b3log/wide/notification"
2014-08-18 17:51:03 +04:00
"github.com/b3log/wide/output"
2014-09-17 10:35:48 +04:00
"github.com/b3log/wide/session"
2014-08-18 17:51:03 +04:00
"github.com/b3log/wide/shell"
2014-09-21 16:31:36 +04:00
"github.com/b3log/wide/util"
2014-08-18 17:45:43 +04:00
"github.com/golang/glog"
)
2014-09-02 18:57:30 +04:00
// Wide 中唯一一个 init 函数.
func init() {
2014-09-19 17:10:04 +04:00
// TODO:默认启动参数
2014-09-02 18:57:30 +04:00
flag.Set("logtostderr", "true")
2014-09-16 13:32:55 +04:00
flag.Set("v", "3")
2014-09-15 10:24:40 +04:00
flag.Parse()
2014-09-13 12:50:18 +04:00
2014-09-15 10:24:40 +04:00
// 加载事件处理
event.Load()
2014-09-02 18:57:30 +04:00
2014-09-15 10:24:40 +04:00
// 加载配置
conf.Load()
2014-09-15 14:03:52 +04:00
2014-09-19 17:10:04 +04:00
// 定时检查运行环境
2014-09-22 19:13:07 +04:00
conf.FixedTimeCheckEnv()
// 定时保存配置
conf.FixedTimeSave()
2014-09-23 17:03:44 +04:00
// 定时检查无效会话
session.FixedTimeRelease()
2014-09-02 18:57:30 +04:00
}
2014-09-21 16:31:36 +04:00
// 登录.
func loginHandler(w http.ResponseWriter, r *http.Request) {
i18n.Load()
if r.Method == "GET" {
// 展示登录页面
model := map[string]interface{}{"conf": conf.Wide, "i18n": i18n.GetAll(r), "locale": i18n.GetLocale(r)}
t, err := template.ParseFiles("view/login.html")
if nil != err {
glog.Error(err)
http.Error(w, err.Error(), 500)
return
}
t.Execute(w, model)
return
}
// 非 GET 请求当作是登录请求
succ := false
data := map[string]interface{}{"succ": &succ}
defer util.RetJSON(w, r, data)
2014-09-23 18:29:53 +04:00
args := struct {
Username string
Password string
}{}
2014-09-21 16:31:36 +04:00
if err := json.NewDecoder(r.Body).Decode(&args); err != nil {
glog.Error(err)
succ = true
return
}
for _, user := range conf.Wide.Users {
2014-09-23 18:29:53 +04:00
if user.Name == args.Username && user.Password == args.Password {
2014-09-21 16:31:36 +04:00
succ = true
}
}
if !succ {
return
}
// 创建 HTTP 会话
httpSession, _ := session.HTTPSession.Get(r, "wide-session")
2014-09-23 18:29:53 +04:00
httpSession.Values["username"] = args.Username
2014-09-21 16:31:36 +04:00
httpSession.Values["id"] = strconv.Itoa(rand.Int())
httpSession.Options.MaxAge = 60 * 60 * 24 // 一天过期
httpSession.Save(r, w)
2014-09-23 18:29:53 +04:00
glog.Infof("Created a HTTP session [%s] for user [%s]", httpSession.Values["id"].(string), args.Username)
2014-09-21 16:31:36 +04:00
}
2014-09-12 13:10:58 +04:00
// Wide 首页.
2014-08-18 17:45:43 +04:00
func indexHandler(w http.ResponseWriter, r *http.Request) {
2014-09-02 18:57:30 +04:00
i18n.Load()
2014-09-17 10:35:48 +04:00
httpSession, _ := session.HTTPSession.Get(r, "wide-session")
2014-08-18 17:45:43 +04:00
2014-09-16 19:58:52 +04:00
if httpSession.IsNew {
2014-09-21 16:31:36 +04:00
http.Redirect(w, r, "/login", http.StatusForbidden)
2014-09-01 14:55:11 +04:00
2014-09-21 16:31:36 +04:00
return
2014-09-01 14:55:11 +04:00
}
2014-08-18 17:45:43 +04:00
2014-09-16 19:58:52 +04:00
httpSession.Save(r, w)
2014-08-18 17:45:43 +04:00
2014-09-17 10:35:48 +04:00
// 创建一个 Wide 会话
wideSession := session.WideSessions.New(httpSession)
2014-09-21 16:31:36 +04:00
username := httpSession.Values["username"].(string)
2014-09-24 07:00:33 +04:00
wideSessions := session.WideSessions.GetByUsername(username)
2014-09-23 18:29:53 +04:00
userConf := conf.Wide.GetUser(username)
model := map[string]interface{}{"conf": conf.Wide, "i18n": i18n.GetAll(r), "locale": i18n.GetLocale(r),
"session": wideSession, "latestSessionContent": userConf.LatestSessionContent}
2014-09-17 07:17:25 +04:00
glog.V(3).Infof("User [%s] has [%d] sessions", username, len(wideSessions))
2014-09-01 06:47:59 +04:00
t, err := template.ParseFiles("view/index.html")
2014-08-18 17:45:43 +04:00
if nil != err {
glog.Error(err)
http.Error(w, err.Error(), 500)
return
}
2014-08-25 18:17:02 +04:00
t.Execute(w, model)
2014-08-18 17:45:43 +04:00
}
2014-09-17 07:17:25 +04:00
// favicon.ico 请求处理.
func faviconHandler(w http.ResponseWriter, r *http.Request) {
2014-09-24 10:53:05 +04:00
// TODO: favicon.ico 请求处理
2014-09-17 07:17:25 +04:00
}
2014-09-16 11:06:52 +04:00
// 主程序入口.
2014-08-18 17:45:43 +04:00
func main() {
2014-09-02 18:57:30 +04:00
runtime.GOMAXPROCS(conf.Wide.MaxProcs)
defer glog.Flush()
2014-09-01 06:47:59 +04:00
// 静态资源
2014-08-18 17:45:43 +04:00
http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))
2014-09-24 10:36:34 +04:00
http.HandleFunc("/favicon.ico", handlerWrapper(faviconHandler))
2014-08-18 17:45:43 +04:00
2014-09-01 17:40:10 +04:00
// 库资源
http.Handle("/data/", http.StripPrefix("/data/", http.FileServer(http.Dir("data"))))
2014-09-21 16:31:36 +04:00
// IDE
2014-09-24 10:36:34 +04:00
http.HandleFunc("/login", handlerWrapper(loginHandler))
http.HandleFunc("/", handlerWrapper(indexHandler))
2014-09-23 05:45:56 +04:00
// 会话
2014-09-24 10:36:34 +04:00
http.HandleFunc("/session/ws", handlerWrapper(session.WSHandler))
http.HandleFunc("/session/save", handlerWrapper(session.SaveContent))
2014-08-18 17:45:43 +04:00
2014-09-01 06:47:59 +04:00
// 运行相关
2014-09-24 10:36:34 +04:00
http.HandleFunc("/build", handlerWrapper(output.BuildHandler))
http.HandleFunc("/run", handlerWrapper(output.RunHandler))
http.HandleFunc("/stop", handlerWrapper(output.StopHandler))
http.HandleFunc("/go/get", handlerWrapper(output.GoGetHandler))
http.HandleFunc("/go/install", handlerWrapper(output.GoInstallHandler))
http.HandleFunc("/output/ws", handlerWrapper(output.WSHandler))
2014-08-18 17:45:43 +04:00
2014-09-01 06:47:59 +04:00
// 文件树
2014-09-24 10:36:34 +04:00
http.HandleFunc("/files", handlerWrapper(file.GetFiles))
http.HandleFunc("/file", handlerWrapper(file.GetFile))
http.HandleFunc("/file/save", handlerWrapper(file.SaveFile))
http.HandleFunc("/file/new", handlerWrapper(file.NewFile))
http.HandleFunc("/file/remove", handlerWrapper(file.RemoveFile))
2014-08-18 17:45:43 +04:00
2014-09-01 06:47:59 +04:00
// 编辑器
2014-09-24 10:36:34 +04:00
http.HandleFunc("/editor/ws", handlerWrapper(editor.WSHandler))
http.HandleFunc("/go/fmt", handlerWrapper(editor.GoFmtHandler))
http.HandleFunc("/autocomplete", handlerWrapper(editor.AutocompleteHandler))
http.HandleFunc("/find/decl", handlerWrapper(editor.FindDeclarationHandler))
http.HandleFunc("/find/usages", handlerWrapper(editor.FindUsagesHandler))
http.HandleFunc("/html/fmt", handlerWrapper(editor.HTMLFmtHandler))
http.HandleFunc("/json/fmt", handlerWrapper(editor.JSONFmtHandler))
2014-08-18 17:45:43 +04:00
2014-09-01 06:47:59 +04:00
// Shell
2014-09-24 10:36:34 +04:00
http.HandleFunc("/shell/ws", handlerWrapper(shell.WSHandler))
http.HandleFunc("/shell", handlerWrapper(shell.IndexHandler))
2014-08-18 17:45:43 +04:00
2014-09-15 14:03:52 +04:00
// 通知
2014-09-24 10:36:34 +04:00
http.HandleFunc("/notification/ws", handlerWrapper(notification.WSHandler))
2014-09-15 14:03:52 +04:00
2014-09-01 06:47:59 +04:00
// 用户
2014-09-24 10:36:34 +04:00
http.HandleFunc("/user/new", handlerWrapper(session.AddUser))
http.HandleFunc("/user/repos/init", handlerWrapper(session.InitGitRepos))
2014-08-31 14:50:38 +04:00
2014-09-01 06:47:59 +04:00
// 文档
http.Handle("/doc/", http.StripPrefix("/doc/", http.FileServer(http.Dir("doc"))))
2014-09-08 07:37:34 +04:00
glog.V(0).Infof("Wide is running [%s]", conf.Wide.Server)
2014-08-31 13:23:14 +04:00
2014-08-18 17:45:43 +04:00
err := http.ListenAndServe(conf.Wide.Server, nil)
if err != nil {
glog.Fatal(err)
}
}
2014-09-24 07:58:02 +04:00
2014-09-24 10:36:34 +04:00
// HTTP Handler 包装.
// 完成共性处理:
// 1. panic recover
// 2. 请求计时
func handlerWrapper(f func(w http.ResponseWriter, r *http.Request)) func(w http.ResponseWriter, r *http.Request) {
handler := panicRecover(f)
handler = stopwatch(handler)
return handler
}
// Handler 包装请求计时.
func stopwatch(handler func(w http.ResponseWriter, r *http.Request)) func(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) {
start := time.Now()
defer func() {
glog.V(5).Infof("[%s] [%s]", r.RequestURI, time.Since(start))
}()
// Handler 处理
handler(w, r)
}
}
// Handler 包装 recover panic.
func panicRecover(handler func(w http.ResponseWriter, r *http.Request)) func(w http.ResponseWriter, r *http.Request) {
2014-09-24 07:58:02 +04:00
return func(w http.ResponseWriter, r *http.Request) {
defer func() {
if r := recover(); r != nil {
glog.Errorf("PANIC RECOVERED: %v", r)
}
}()
// Handler 处理
2014-09-24 10:36:34 +04:00
handler(w, r)
2014-09-24 07:58:02 +04:00
}
}