1 files changed, 108 insertions, 1 deletions
diff --git a/database/users.go b/database/users.go
index 6fb2601..1ba4ebb 100644
--- a/database/users.go
+++ b/database/users.go
@@ -1,5 +1,112 @@
 package database
 
-func getUsers() {
+import (
+	"database/sql"
+	"log"
+	"time"
 
+	_ "github.com/mattn/go-sqlite3"
+)
+
+const (
+	ExpiryDuration = time.Hour * 24
+)
+
+type User struct {
+	ID          string    `json:"sub"`
+	Mail        string    `json:"email"`
+	Username    string    `json:"preferred_username"`
+	DisplayName string    `json:"name"`
+	CreatedAt   time.Time `json:"created_at"`
+}
+
+type UserSession struct {
+	ID       string    `json:"id"`
+	UserID   string    `json:"user_id"`
+	ExpireAt time.Time `json:"expire_at"`
+}
+
+func GetUser(dbConn *sql.DB, id string) (*User, error) {
+	row := dbConn.QueryRow(`SELECT id, mail, username, display_name, created_at FROM users WHERE id = ?;`, id)
+
+	var user User
+	err := row.Scan(&user.ID, &user.Mail, &user.Username, &user.DisplayName, &user.CreatedAt)
+	if err != nil {
+		log.Println(err)
+		return nil, err
+	}
+
+	return &user, nil
+}
+
+func FindOrSaveUser(dbConn *sql.DB, user *User) (*User, error) {
+	_, err := dbConn.Exec(`INSERT OR REPLACE INTO users (id, mail, username, display_name) VALUES (?, ?, ?, ?);`, user.ID, user.Mail, user.Username, user.DisplayName)
+	if err != nil {
+		return nil, err
+	}
+
+	return user, nil
+}
+
+func MakeUserSessionFor(dbConn *sql.DB, user *User) (*UserSession, error) {
+	expireAt := time.Now().Add(time.Hour * 12)
+
+	_, err := dbConn.Exec(`INSERT OR REPLACE INTO user_sessions (id, user_id, expire_at) VALUES (?, ?, ?);`, user.ID, user.ID, time.Now().Add(ExpiryDuration))
+
+	if err != nil {
+		log.Println(err)
+		return nil, err
+	}
+
+	return &UserSession{
+		ID:       user.ID,
+		UserID:   user.ID,
+		ExpireAt: expireAt,
+	}, nil
+}
+
+func GetSession(dbConn *sql.DB, sessionId string) (*UserSession, error) {
+	row := dbConn.QueryRow(`SELECT id, user_id, expire_at FROM user_sessions WHERE id = ?;`, sessionId)
+
+	var id, userId string
+	var expireAt time.Time
+	err := row.Scan(&id, &userId, &expireAt)
+	if err != nil {
+		log.Println(err)
+		return nil, err
+	}
+
+	return &UserSession{
+		ID:       id,
+		UserID:   userId,
+		ExpireAt: expireAt,
+	}, nil
+}
+
+func DeleteSession(dbConn *sql.DB, sessionId string) error {
+	_, err := dbConn.Exec(`DELETE FROM user_sessions WHERE id = ?;`, sessionId)
+	if err != nil {
+		log.Println(err)
+		return err
+	}
+
+	return nil
+}
+
+func RefreshSession(dbConn *sql.DB, sessionId string) (*UserSession, error) {
+	newExpireAt := time.Now().Add(ExpiryDuration)
+
+	_, err := dbConn.Exec(`UPDATE user_sessions SET expire_at = ? WHERE id = ?;`, newExpireAt, sessionId)
+	if err != nil {
+		log.Println(err)
+		return nil, err
+	}
+
+	session, err := GetSession(dbConn, sessionId)
+	if err != nil {
+		log.Println(err)
+		return nil, err
+	}
+
+	return session, nil
 }