Nikolai Rodionov
68e166d90f
All checks were successful
ci/woodpecker/push/build Pipeline was successful
Signed-off-by: Nikolai Rodionov <allanger@badhouseplants.net>
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"time"
|
|
)
|
|
|
|
type TokenData struct {
|
|
UUID string
|
|
Decsription string
|
|
TokenHash string
|
|
UserID string
|
|
CreatedAt time.Time
|
|
GeneratedAt time.Time
|
|
ExpiresAt time.Time
|
|
RevokedAt time.Time
|
|
LastUsedAt time.Time
|
|
Scope string
|
|
}
|
|
|
|
// CreateTokens adds a new token to a database
|
|
func CreateToken(ctx context.Context, db *sql.DB, data *TokenData) error {
|
|
query := `
|
|
INSERT INTO tokens
|
|
(uuid, description, token_hash, user_id, scopes, created_at, generated_at, expires_at)
|
|
VALUES
|
|
($1, $2, $3, $4, $5, $6, $7, $8);`
|
|
if _, err := db.QueryContext(
|
|
ctx,
|
|
query,
|
|
data.UUID,
|
|
data.Decsription,
|
|
data.TokenHash,
|
|
data.UserID,
|
|
data.Scope,
|
|
data.CreatedAt,
|
|
data.GeneratedAt,
|
|
data.ExpiresAt,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// UpdateToken updates token description and scope
|
|
func UpdateToken(ctx context.Context, db *sql.DB, data *TokenData) error {
|
|
query := "UPDATE tokens SET description = $1, scopes = $2 WHERE uuid = $3;"
|
|
if _, err := db.QueryContext(ctx, query, data.Decsription, data.Scope, data.UUID); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetToken(ctx context.Context, db *sql.DB, tokenID, userID string) (*TokenData, error) {
|
|
query := `
|
|
SELECT
|
|
uuid, description, generated_at, expires_at,
|
|
last_used_at, revoked_at, created_at, scopes
|
|
FROM tokens
|
|
WHERE uuid = $1 AND user_id = $2;`
|
|
|
|
var generatedAt sql.NullTime
|
|
var expiresAt sql.NullTime
|
|
var revokedAt sql.NullTime
|
|
var lastUsedAt sql.NullTime
|
|
var createdAt sql.NullTime
|
|
|
|
result := &TokenData{}
|
|
|
|
if err := db.QueryRowContext(ctx, query, tokenID, userID).Scan(
|
|
&result.UUID,
|
|
&result.Decsription,
|
|
&generatedAt,
|
|
&expiresAt,
|
|
&lastUsedAt,
|
|
&revokedAt,
|
|
&createdAt,
|
|
&result.Scope,
|
|
); err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, ErrNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
result.GeneratedAt = generatedAt.Time
|
|
result.ExpiresAt = expiresAt.Time
|
|
result.RevokedAt = revokedAt.Time
|
|
result.LastUsedAt = lastUsedAt.Time
|
|
result.CreatedAt = createdAt.Time
|
|
|
|
return result, nil
|
|
}
|