package main import ( "encoding/json" "flag" "html/template" "math/rand" "net/http" "runtime" "strconv" "github.com/b3log/wide/conf" "github.com/b3log/wide/editor" "github.com/b3log/wide/event" "github.com/b3log/wide/file" "github.com/b3log/wide/i18n" "github.com/b3log/wide/notification" "github.com/b3log/wide/output" "github.com/b3log/wide/session" "github.com/b3log/wide/shell" "github.com/b3log/wide/util" "github.com/golang/glog" ) // Wide 中唯一一个 init 函数. func init() { // TODO:默认启动参数 flag.Set("logtostderr", "true") flag.Set("v", "3") flag.Parse() // 加载事件处理 event.Load() // 加载配置 conf.Load() // 定时检查运行环境 conf.FixedTimeCheckEnv() // 定时保存配置 conf.FixedTimeSave() } // 登录. 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) var args map[string]interface{} if err := json.NewDecoder(r.Body).Decode(&args); err != nil { glog.Error(err) succ = true return } username := args["username"].(string) password := args["password"].(string) for _, user := range conf.Wide.Users { if user.Name == username && user.Password == password { succ = true } } if !succ { return } // 创建 HTTP 会话 httpSession, _ := session.HTTPSession.Get(r, "wide-session") httpSession.Values["username"] = username httpSession.Values["id"] = strconv.Itoa(rand.Int()) httpSession.Options.MaxAge = 60 * 60 * 24 // 一天过期 httpSession.Save(r, w) glog.Infof("Created a HTTP session [%s] for user [%s]", httpSession.Values["id"].(string), username) } // Wide 首页. func indexHandler(w http.ResponseWriter, r *http.Request) { i18n.Load() httpSession, _ := session.HTTPSession.Get(r, "wide-session") if httpSession.IsNew { http.Redirect(w, r, "/login", http.StatusForbidden) return } httpSession.Save(r, w) // 创建一个 Wide 会话 wideSession := session.WideSessions.New(httpSession) model := map[string]interface{}{"conf": conf.Wide, "i18n": i18n.GetAll(r), "locale": i18n.GetLocale(r), "session": wideSession} wideSessions := session.WideSessions.GetByHTTPSession(httpSession) username := httpSession.Values["username"].(string) glog.V(3).Infof("User [%s] has [%d] sessions", username, len(wideSessions)) t, err := template.ParseFiles("view/index.html") if nil != err { glog.Error(err) http.Error(w, err.Error(), 500) return } t.Execute(w, model) } // favicon.ico 请求处理. func faviconHandler(w http.ResponseWriter, r *http.Request) { // TODO: favicon.ico 请求处理. } // 主程序入口. func main() { runtime.GOMAXPROCS(conf.Wide.MaxProcs) defer glog.Flush() // 静态资源 http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static")))) http.HandleFunc("/favicon.ico", faviconHandler) // 库资源 http.Handle("/data/", http.StripPrefix("/data/", http.FileServer(http.Dir("data")))) // IDE http.HandleFunc("/login", loginHandler) http.HandleFunc("/", indexHandler) http.HandleFunc("/session/ws", session.WSHandler) http.HandleFunc("/session/save", session.SaveContent) // 运行相关 http.HandleFunc("/build", output.BuildHandler) http.HandleFunc("/run", output.RunHandler) http.HandleFunc("/stop", output.StopHandler) http.HandleFunc("/go/get", output.GoGetHandler) http.HandleFunc("/go/install", output.GoInstallHandler) http.HandleFunc("/output/ws", output.WSHandler) // 文件树 http.HandleFunc("/files", file.GetFiles) http.HandleFunc("/file", file.GetFile) http.HandleFunc("/file/save", file.SaveFile) http.HandleFunc("/file/new", file.NewFile) http.HandleFunc("/file/remove", file.RemoveFile) // 编辑器 http.HandleFunc("/editor/ws", editor.WSHandler) http.HandleFunc("/go/fmt", editor.GoFmtHandler) http.HandleFunc("/autocomplete", editor.AutocompleteHandler) http.HandleFunc("/find/decl", editor.FindDeclarationHandler) http.HandleFunc("/find/usages", editor.FindUsagesHandler) http.HandleFunc("/html/fmt", editor.HTMLFmtHandler) http.HandleFunc("/json/fmt", editor.JSONFmtHandler) // Shell http.HandleFunc("/shell/ws", shell.WSHandler) http.HandleFunc("/shell", shell.IndexHandler) // 通知 http.HandleFunc("/notification/ws", notification.WSHandler) // 用户 http.HandleFunc("/user/new", session.AddUser) http.HandleFunc("/user/repos/init", session.InitGitRepos) // 文档 http.Handle("/doc/", http.StripPrefix("/doc/", http.FileServer(http.Dir("doc")))) glog.V(0).Infof("Wide is running [%s]", conf.Wide.Server) err := http.ListenAndServe(conf.Wide.Server, nil) if err != nil { glog.Fatal(err) } }