mirror of
https://github.com/cwinfo/matterbridge.git
synced 2025-07-21 00:56:47 +00:00
Update vendor
This commit is contained in:
97
vendor/github.com/labstack/echo/cookbook/twitter/handler/user.go
generated
vendored
Normal file
97
vendor/github.com/labstack/echo/cookbook/twitter/handler/user.go
generated
vendored
Normal file
@ -0,0 +1,97 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
jwt "github.com/dgrijalva/jwt-go"
|
||||
"github.com/labstack/echo"
|
||||
"github.com/labstack/echo/cookbook/twitter/model"
|
||||
mgo "gopkg.in/mgo.v2"
|
||||
"gopkg.in/mgo.v2/bson"
|
||||
)
|
||||
|
||||
func (h *Handler) Signup(c echo.Context) (err error) {
|
||||
// Bind
|
||||
u := &model.User{ID: bson.NewObjectId()}
|
||||
if err = c.Bind(u); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Validate
|
||||
if u.Email == "" || u.Password == "" {
|
||||
return &echo.HTTPError{Code: http.StatusBadRequest, Message: "invalid email or password"}
|
||||
}
|
||||
|
||||
// Save user
|
||||
db := h.DB.Clone()
|
||||
defer db.Close()
|
||||
if err = db.DB("twitter").C("users").Insert(u); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
return c.JSON(http.StatusCreated, u)
|
||||
}
|
||||
|
||||
func (h *Handler) Login(c echo.Context) (err error) {
|
||||
// Bind
|
||||
u := new(model.User)
|
||||
if err = c.Bind(u); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// Find user
|
||||
db := h.DB.Clone()
|
||||
defer db.Close()
|
||||
if err = db.DB("twitter").C("users").
|
||||
Find(bson.M{"email": u.Email, "password": u.Password}).One(u); err != nil {
|
||||
if err == mgo.ErrNotFound {
|
||||
return &echo.HTTPError{Code: http.StatusUnauthorized, Message: "invalid email or password"}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
//-----
|
||||
// JWT
|
||||
//-----
|
||||
|
||||
// Create token
|
||||
token := jwt.New(jwt.SigningMethodHS256)
|
||||
|
||||
// Set claims
|
||||
claims := token.Claims.(jwt.MapClaims)
|
||||
claims["id"] = u.ID
|
||||
claims["exp"] = time.Now().Add(time.Hour * 72).Unix()
|
||||
|
||||
// Generate encoded token and send it as response
|
||||
u.Token, err = token.SignedString([]byte(Key))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
u.Password = "" // Don't send password
|
||||
return c.JSON(http.StatusOK, u)
|
||||
}
|
||||
|
||||
func (h *Handler) Follow(c echo.Context) (err error) {
|
||||
userID := userIDFromToken(c)
|
||||
id := c.Param("id")
|
||||
|
||||
// Add a follower to user
|
||||
db := h.DB.Clone()
|
||||
defer db.Close()
|
||||
if err = db.DB("twitter").C("users").
|
||||
UpdateId(bson.ObjectIdHex(id), bson.M{"$addToSet": bson.M{"followers": userID}}); err != nil {
|
||||
if err == mgo.ErrNotFound {
|
||||
return echo.ErrNotFound
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func userIDFromToken(c echo.Context) string {
|
||||
user := c.Get("user").(*jwt.Token)
|
||||
claims := user.Claims.(jwt.MapClaims)
|
||||
return claims["id"].(string)
|
||||
}
|
Reference in New Issue
Block a user