login.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. package api
  2. import (
  3. "errors"
  4. "log"
  5. "net/http"
  6. "time"
  7. jwt "github.com/dgrijalva/jwt-go"
  8. "gogs.carducci-dante.gov.it/karmen/config"
  9. "gogs.carducci-dante.gov.it/karmen/core/renderer"
  10. )
  11. func logoutHandler() http.Handler {
  12. fn := func(w http.ResponseWriter, r *http.Request) {
  13. session, err := store.Get(r, "login-session")
  14. if err != nil {
  15. http.Error(w, err.Error(), http.StatusInternalServerError)
  16. return
  17. }
  18. session.Values["token"] = []uint8{}
  19. session.Save(r, w)
  20. http.Redirect(w, r, "/", http.StatusSeeOther)
  21. }
  22. return http.HandlerFunc(fn)
  23. }
  24. func loginHandler() http.Handler {
  25. fn := func(w http.ResponseWriter, r *http.Request) {
  26. if r.Method == "GET" {
  27. renderer.Render["html"](w, r, nil, "login", "login")
  28. }
  29. if r.Method == "POST" {
  30. r.ParseForm()
  31. token, err := getToken(r.FormValue("username"), r.FormValue("password"))
  32. if err != nil {
  33. panic(err)
  34. } else {
  35. session, err := store.Get(r, "login-session")
  36. if err != nil {
  37. panic(err)
  38. }
  39. session.Values["token"] = token
  40. session.Save(r, w)
  41. r.Method = "GET"
  42. http.Redirect(w, r, "/teachers", http.StatusSeeOther)
  43. }
  44. }
  45. }
  46. return http.HandlerFunc(fn)
  47. }
  48. func queryDB(username string, password string) (*User, error) {
  49. log.Println(username, config.Config.Admin.Username, password, config.Config.Admin.Password)
  50. if username == config.Config.Admin.Username && password == config.Config.Admin.Password {
  51. return &User{username, true}, nil
  52. }
  53. return nil, errors.New("Authentication failed!")
  54. }
  55. func getToken(username string, password string) ([]byte, error) {
  56. user, err := queryDB(username, password)
  57. if err != nil {
  58. return nil, err
  59. }
  60. /* Set token claims */
  61. claims := make(map[string]interface{})
  62. claims["admin"] = user.Admin
  63. claims["name"] = user.Name
  64. claims["exp"] = time.Now().Add(time.Hour * 24).Unix()
  65. /* Create the token */
  66. token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims(claims))
  67. /* Sign the token with our secret */
  68. tokenString, err := token.SignedString(signingKey)
  69. if err != nil {
  70. return nil, err
  71. }
  72. return []byte(tokenString), nil
  73. }