// Copyright (c) 2014-present, b3log.org // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // https://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package session import ( "encoding/json" "net/http" "path/filepath" "runtime" "strings" "sync" "text/template" "time" "github.com/88250/gulu" "github.com/88250/wide/conf" "github.com/88250/wide/i18n" ) const ( // TODO: i18n userExists = "user exists" userCreated = "user created" userCreateError = "user create error" ) // Exclusive lock for adding user. var addUserMutex sync.Mutex // PreferenceHandler handles request of preference page. func PreferenceHandler(w http.ResponseWriter, r *http.Request) { httpSession, _ := HTTPSession.Get(r, CookieName) if httpSession.IsNew { http.Redirect(w, r, "/login", http.StatusFound) return } httpSession.Options.MaxAge = conf.Wide.HTTPSessionMaxAge httpSession.Save(r, w) uid := httpSession.Values["uid"].(string) user := conf.GetUser(uid) if "GET" == r.Method { tmpLinux := user.GoBuildArgsForLinux tmpWindows := user.GoBuildArgsForWindows tmpDarwin := user.GoBuildArgsForDarwin user.GoBuildArgsForLinux = strings.Replace(user.GoBuildArgsForLinux, `"`, `"`, -1) user.GoBuildArgsForWindows = strings.Replace(user.GoBuildArgsForWindows, `"`, `"`, -1) user.GoBuildArgsForDarwin = strings.Replace(user.GoBuildArgsForDarwin, `"`, `"`, -1) model := map[string]interface{}{"conf": conf.Wide, "i18n": i18n.GetAll(user.Locale), "user": user, "ver": conf.WideVersion, "goos": runtime.GOOS, "goarch": runtime.GOARCH, "gover": runtime.Version(), "locales": i18n.GetLocalesNames(), "gofmts": gulu.Go.GetGoFormats(), "themes": conf.GetThemes(), "editorThemes": conf.GetEditorThemes()} t, err := template.ParseFiles("views/preference.html") if nil != err { logger.Error(err) http.Error(w, err.Error(), http.StatusInternalServerError) user.GoBuildArgsForLinux = tmpLinux user.GoBuildArgsForWindows = tmpWindows user.GoBuildArgsForDarwin = tmpDarwin return } t.Execute(w, model) user.GoBuildArgsForLinux = tmpLinux user.GoBuildArgsForWindows = tmpWindows user.GoBuildArgsForDarwin = tmpDarwin return } // non-GET request as save request result := gulu.Ret.NewResult() defer gulu.Ret.RetResult(w, r, result) args := struct { FontFamily string FontSize string GoFmt string GoBuildArgsForLinux string GoBuildArgsForWindows string GoBuildArgsForDarwin string Keymap string Workspace string Username string Locale string Theme string EditorFontFamily string EditorFontSize string EditorLineHeight string EditorTheme string EditorTabSize string }{} if err := json.NewDecoder(r.Body).Decode(&args); err != nil { logger.Error(err) result.Code = -1 return } user.FontFamily = args.FontFamily user.FontSize = args.FontSize user.GoFormat = args.GoFmt user.GoBuildArgsForLinux = args.GoBuildArgsForLinux user.GoBuildArgsForWindows = args.GoBuildArgsForWindows user.GoBuildArgsForDarwin = args.GoBuildArgsForDarwin user.Keymap = args.Keymap // XXX: disallow change workspace at present // user.Workspace = args.Workspace user.Locale = args.Locale user.Theme = args.Theme user.Editor.FontFamily = args.EditorFontFamily user.Editor.FontSize = args.EditorFontSize user.Editor.LineHeight = args.EditorLineHeight user.Editor.Theme = args.EditorTheme user.Editor.TabSize = args.EditorTabSize conf.UpdateCustomizedConf(uid) now := time.Now().UnixNano() user.Lived = now user.Updated = now if user.Save() { result.Code = 0 } else { result.Code = -1 } } // FixedTimeSave saves online users' configurations periodically (1 minute). // // Main goal of this function is to save user session content, for restoring session content while user open Wide next time. func FixedTimeSave() { go func() { defer gulu.Panic.Recover(nil) for _ = range time.Tick(time.Minute) { SaveOnlineUsers() } }() } // CanAccess determines whether the user specified by the given user id can access the specified path. func CanAccess(userId, path string) bool { path = filepath.FromSlash(path) userWorkspace := conf.GetUserWorkspace(userId) workspaces := filepath.SplitList(userWorkspace) for _, workspace := range workspaces { if strings.HasPrefix(path, workspace) { return true } } return false } // SaveOnlineUsers saves online users' configurations at once. func SaveOnlineUsers() { users := getOnlineUsers() for _, u := range users { if u.Save() { logger.Tracef("Saved online user [%s]'s configurations", u.Name) } } } func getOnlineUsers() []*conf.User { ret := []*conf.User{} uids := map[string]string{} // distinct uid for _, s := range WideSessions { uids[s.UserId] = s.UserId } for _, uid := range uids { u := conf.GetUser(uid) if "playground" == uid { // user [playground] is a reserved mock user continue } if nil == u { logger.Warnf("Not found user [%s]", uid) continue } ret = append(ret, u) } return ret }