implement JWT tokens
This commit is contained in:
24
api/db.go
24
api/db.go
@@ -7,7 +7,6 @@ import (
|
||||
|
||||
"git.dubyatp.xyz/chat-api-server/db"
|
||||
"github.com/gocql/gocql"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
func dbGetUser(id string) (*User, error) {
|
||||
@@ -28,8 +27,6 @@ func dbGetUser(id string) (*User, error) {
|
||||
}
|
||||
|
||||
func dbGetUserByName(username string) (*User, error) {
|
||||
// This will be deprecated soon after implementing https://git.dubyatp.xyz/williamp/chatservice_concept/issues/1
|
||||
|
||||
query := `SELECT id, name, password FROM users WHERE name = ?`
|
||||
var user User
|
||||
err := db.Session.Query(query, username).Scan(&user.ID, &user.Name, &user.Password)
|
||||
@@ -128,23 +125,24 @@ func dbGetAllMessages() ([]*Message, error) {
|
||||
}
|
||||
|
||||
func dbAddSession(session *Session) error {
|
||||
query := `INSERT INTO sessions (session_token, username) VALUES (?, ?)`
|
||||
err := db.Session.Query(query, session.Token, session.Username).Exec()
|
||||
query := `INSERT INTO sessions (jwttoken, userid, expiry) VALUES (?, ?, ?)`
|
||||
err := db.Session.Query(query, session.Token, session.UserID, session.Expiry).Exec()
|
||||
if err != nil {
|
||||
slog.Error("db: failed to add session", "error", err)
|
||||
return fmt.Errorf("failed to add session")
|
||||
}
|
||||
|
||||
slog.Debug("db: session added", "username", session.Username)
|
||||
slog.Debug("db: session added", "userID", session.UserID)
|
||||
return nil
|
||||
}
|
||||
|
||||
func dbGetSession(id uuid.UUID) (*Session, error) {
|
||||
query := `SELECT session_token, username FROM sessions WHERE session_token = ?`
|
||||
func dbGetSession(jwtToken string) (*Session, error) {
|
||||
query := `SELECT jwttoken, userid, expiry FROM sessions WHERE jwttoken = ?`
|
||||
var session Session
|
||||
err := db.Session.Query(query, id).Scan(
|
||||
err := db.Session.Query(query, jwtToken).Scan(
|
||||
&session.Token,
|
||||
&session.Username)
|
||||
&session.UserID,
|
||||
&session.Expiry)
|
||||
if err == gocql.ErrNotFound {
|
||||
slog.Debug("db: session not found")
|
||||
return nil, errors.New("Session not found")
|
||||
@@ -156,10 +154,10 @@ func dbGetSession(id uuid.UUID) (*Session, error) {
|
||||
return &session, nil
|
||||
}
|
||||
|
||||
func dbDeleteSession(id uuid.UUID) error {
|
||||
query := `DELETE FROM sessions WHERE session_token = ?`
|
||||
func dbDeleteSession(jwtToken string) error {
|
||||
query := `DELETE FROM sessions WHERE jwttoken = ?`
|
||||
|
||||
err := db.Session.Query(query, id).Exec()
|
||||
err := db.Session.Query(query, jwtToken).Exec()
|
||||
|
||||
if err != nil {
|
||||
slog.Error("db: failed to delete session")
|
||||
|
Reference in New Issue
Block a user