main.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"compress/flate"
 	"fmt"
 	_ "github.com/lib/pq"
 	"log"
@@ -62,13 +61,11 @@ func Routes() *chi.Mux {
 		MaxAge:           360,
 	})
 
-	compressor := middleware.NewCompressor(flate.DefaultCompression)
-
 	router.Use(
-		compressor.Handler,
 		cors.Handler,
 		render.SetContentType(render.ContentTypeJSON),
 		middleware.Logger,
+		middleware.DefaultCompress,
 		middleware.RedirectSlashes,
 		middleware.Recoverer,
 	)

packages/auth/auth.go

@@ -8,6 +8,7 @@ import (
 	"git.minhas.io/asara/sudoscientist-go-backend/packages/middleware"
 	"git.minhas.io/asara/sudoscientist-go-backend/packages/users"
 	"github.com/badoux/checkmail"
+	"github.com/dgrijalva/jwt-go"
 	"github.com/go-chi/chi"
 	"github.com/go-chi/jwtauth"
 	"github.com/go-chi/render"
@@ -65,6 +66,7 @@ type Claims struct {
 	Username string `json:"username", db:"username"`
 	Admin    bool   `json:"admin", db:"admin"`
 	Verified bool   `json:"verified", db:"verified"`
+	jwt.StandardClaims
 }
 
 type JWT struct {
@@ -173,12 +175,14 @@ func register(w http.ResponseWriter, r *http.Request) {
 	}
 	users.CreateProfile(creds.Username, creds.Email)
 	expirationTime := time.Now().Add(24 * time.Hour)
-	claims := map[string]interface{}{
-		"username": creds.Username,
-		"admin":    false,
-		"verified": false,
+	claims := &Claims{
+		Username: creds.Username,
+		Admin:    false,
+		Verified: false,
+		StandardClaims: jwt.StandardClaims{
+			ExpiresAt: expirationTime.Unix(),
+		},
 	}
-	jwtauth.SetExpiry(claims, expirationTime)
 	if PostalEnabled {
 		_, emailToken, _ := EmailAuth.Encode(claims)
 		returnMessage, ok := sendEmailToken(w, emailToken, creds.Username, creds.Email)
@@ -223,12 +227,14 @@ func signin(w http.ResponseWriter, r *http.Request) {
 	user_claims := &Claims{}
 	user_claims_query := DB.QueryRow("SELECT username, admin, verified FROM users WHERE username=$1", creds.Username)
 	err = user_claims_query.Scan(&user_claims.Username, &user_claims.Admin, &user_claims.Verified)
-	claims := map[string]interface{}{
-		"username": user_claims.Username,
-		"admin":    user_claims.Admin,
-		"verified": user_claims.Verified,
+	claims := &Claims{
+		Username: user_claims.Username,
+		Admin:    user_claims.Admin,
+		Verified: user_claims.Verified,
+		StandardClaims: jwt.StandardClaims{
+			ExpiresAt: expirationTime.Unix(),
+		},
 	}
-	jwtauth.SetExpiry(claims, expirationTime)
 	_, tokenString, _ := TokenAuth.Encode(claims)
 	setCookies(w, tokenString, expirationTime)
 	w.WriteHeader(http.StatusOK)
@@ -252,12 +258,14 @@ func refresh(w http.ResponseWriter, r *http.Request) {
 		render.JSON(w, r, returnMessage)
 		return
 	}
-	newClaims := map[string]interface{}{
-		"username": user_claims.Username,
-		"admin":    user_claims.Admin,
-		"verified": user_claims.Verified,
+	newClaims := &Claims{
+		Username: user_claims.Username,
+		Admin:    user_claims.Admin,
+		Verified: user_claims.Verified,
+		StandardClaims: jwt.StandardClaims{
+			ExpiresAt: expirationTime.Unix(),
+		},
 	}
-	jwtauth.SetExpiry(newClaims, expirationTime)
 	_, tokenString, _ := TokenAuth.Encode(newClaims)
 	setCookies(w, tokenString, expirationTime)
 	w.WriteHeader(http.StatusOK)