wide/session/users.go

174 lines
3.8 KiB
Go
Raw Normal View History

2014-09-17 10:35:48 +04:00
package session
2014-08-31 14:50:38 +04:00
import (
"encoding/json"
"math/rand"
2014-09-12 13:10:58 +04:00
"net/http"
"path/filepath"
"strconv"
"text/template"
2014-09-12 13:10:58 +04:00
2014-08-31 14:50:38 +04:00
"github.com/b3log/wide/conf"
"github.com/b3log/wide/i18n"
2014-09-01 16:50:51 +04:00
"github.com/b3log/wide/util"
2014-08-31 14:50:38 +04:00
"github.com/golang/glog"
)
const (
UserExists = "user exists"
UserCreated = "user created"
UserCreateError = "user create error"
2014-08-31 14:50:38 +04:00
)
// LoginHandler handles request of user login.
func LoginHandler(w http.ResponseWriter, r *http.Request) {
if "GET" == r.Method {
// show the login page
model := map[string]interface{}{"conf": conf.Wide, "i18n": i18n.GetAll(conf.Wide.Locale),
"locale": conf.Wide.Locale, "ver": conf.WideVersion}
t, err := template.ParseFiles("views/login.html")
if nil != err {
glog.Error(err)
http.Error(w, err.Error(), 500)
return
}
t.Execute(w, model)
return
}
// non-GET request as login request
succ := false
data := map[string]interface{}{"succ": &succ}
2014-09-01 16:50:51 +04:00
defer util.RetJSON(w, r, data)
args := struct {
Username string
Password string
}{}
2014-08-31 14:50:38 +04:00
2014-10-28 16:32:19 +03:00
if err := json.NewDecoder(r.Body).Decode(&args); err != nil {
2014-08-31 14:50:38 +04:00
glog.Error(err)
succ = true
2014-08-31 14:50:38 +04:00
return
}
for _, user := range conf.Wide.Users {
if user.Name == args.Username && user.Password == args.Password {
succ = true
}
}
2014-08-31 14:50:38 +04:00
if !succ {
return
2014-08-31 14:50:38 +04:00
}
// create a HTTP session
httpSession, _ := HTTPSession.Get(r, "wide-session")
httpSession.Values["username"] = args.Username
httpSession.Values["id"] = strconv.Itoa(rand.Int())
httpSession.Options.MaxAge = conf.Wide.HTTPSessionMaxAge
httpSession.Save(r, w)
glog.Infof("Created a HTTP session [%s] for user [%s]", httpSession.Values["id"].(string), args.Username)
2014-08-31 14:50:38 +04:00
}
// LogoutHandler handles request of user logout (exit).
func LogoutHandler(w http.ResponseWriter, r *http.Request) {
2014-09-01 16:50:51 +04:00
data := map[string]interface{}{"succ": true}
defer util.RetJSON(w, r, data)
httpSession, _ := HTTPSession.Get(r, "wide-session")
httpSession.Options.MaxAge = -1
httpSession.Save(r, w)
}
// SignUpUser handles request of registering user.
func SignUpUser(w http.ResponseWriter, r *http.Request) {
if "GET" == r.Method {
// show the user sign up page
firstUserWorkspace := conf.Wide.GetUserWorkspace(conf.Wide.Users[0].Name)
dir := filepath.Dir(firstUserWorkspace)
model := map[string]interface{}{"conf": conf.Wide, "i18n": i18n.GetAll(conf.Wide.Locale),
2014-10-30 13:21:44 +03:00
"locale": conf.Wide.Locale, "ver": conf.WideVersion, "dir": dir,
"pathSeparator": conf.PathSeparator}
t, err := template.ParseFiles("views/sign_up.html")
if nil != err {
glog.Error(err)
http.Error(w, err.Error(), 500)
return
}
t.Execute(w, model)
return
}
// non-GET request as add user request
succ := true
data := map[string]interface{}{"succ": &succ}
defer util.RetJSON(w, r, data)
var args map[string]interface{}
2014-08-31 14:50:38 +04:00
if err := json.NewDecoder(r.Body).Decode(&args); err != nil {
glog.Error(err)
succ = false
2014-08-31 14:50:38 +04:00
return
}
2014-08-31 14:50:38 +04:00
username := args["username"].(string)
password := args["password"].(string)
msg := addUser(username, password)
if UserCreated != msg {
succ = false
data["msg"] = msg
}
2014-08-31 14:50:38 +04:00
}
func addUser(username, password string) string {
// XXX: validate
2014-08-31 14:50:38 +04:00
for _, user := range conf.Wide.Users {
if user.Name == username {
return UserExists
2014-08-31 14:50:38 +04:00
}
}
firstUserWorkspace := conf.Wide.GetUserWorkspace(conf.Wide.Users[0].Name)
dir := filepath.Dir(firstUserWorkspace)
workspace := filepath.Join(dir, username)
2014-11-02 05:48:34 +03:00
conf.Wide.Users = append(conf.Wide.Users,
&conf.User{Name: username, Password: password, Workspace: workspace,
Locale: conf.Wide.Locale, GoFormat: "gofmt", FontFamily: "Helvetica", FontSize: "inherit",
Editor: &conf.Editor{FontFamily: "Consolas, 'Courier New', monospace", FontSize: "inherit"}})
2014-08-31 14:50:38 +04:00
if !conf.Save() {
return UserCreateError
2014-08-31 14:50:38 +04:00
}
conf.CreateWorkspaceDir(workspace)
2014-11-01 08:35:02 +03:00
conf.UpdateCustomizedConf(username)
2014-08-31 14:50:38 +04:00
glog.Infof("Created a user [%s]", username)
return UserCreated
2014-08-31 14:50:38 +04:00
}