90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
_ "github.com/lib/pq"
|
|
"log"
|
|
)
|
|
|
|
func GetUsers(db *sql.DB) ([]string, error) {
|
|
query := `SELECT username FROM accounts;`
|
|
|
|
rows, err := db.Query(query)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var users []string
|
|
for rows.Next() {
|
|
var user string
|
|
if err := rows.Scan(&user); err != nil {
|
|
return nil, err
|
|
}
|
|
users = append(users, user)
|
|
}
|
|
return users, err
|
|
}
|
|
|
|
func CheckUserExists(db *sql.DB, username string) (bool, error) {
|
|
query := `SELECT COUNT(1) FROM accounts WHERE username= $1`
|
|
|
|
var count int
|
|
|
|
err := db.QueryRow(query, username).Scan(&count)
|
|
|
|
if err != nil {
|
|
return false, fmt.Errorf("error checking username exists: %w", err)
|
|
}
|
|
|
|
return count > 0, err
|
|
}
|
|
|
|
func InsertUser(db *sql.DB, username string, passwordHash string) (string, error) {
|
|
query := `
|
|
INSERT INTO Accounts (username, password_hash)
|
|
VALUES ($1, $2)
|
|
RETURNING user_id;
|
|
`
|
|
|
|
var userID string
|
|
err := db.QueryRow(query, username, passwordHash).Scan(&userID)
|
|
if err != nil {
|
|
return "", fmt.Errorf("error inserting user: %w", err)
|
|
}
|
|
|
|
fmt.Printf("Inserted user: %w", username)
|
|
|
|
return userID, err
|
|
}
|
|
|
|
func GetPasswordHash(db *sql.DB, username string) (string, error) {
|
|
query := `
|
|
SELECT password_hash FROM Accounts
|
|
WHERE LOWER(username) = LOWER($1);
|
|
`
|
|
|
|
var passwordHash string
|
|
err := db.QueryRow(query, username).Scan(&passwordHash)
|
|
if err != nil {
|
|
return "", fmt.Errorf("error getting password hash: %w", err)
|
|
}
|
|
|
|
return passwordHash, err
|
|
}
|
|
|
|
func GetUserID(db *sql.DB, username string) (uuid.UUID, error) {
|
|
query := `
|
|
SELECT user_id FROM Accounts
|
|
WHERE LOWER(username) = LOWER($1);
|
|
`
|
|
|
|
var userID uuid.UUID
|
|
err := db.QueryRow(query, username).Scan(&userID)
|
|
if err != nil {
|
|
return uuid.Nil, fmt.Errorf("error getting user id: %w", err)
|
|
}
|
|
return userID, nil
|
|
}
|