101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
|
|
"certmanager/internal/descriptor"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func (db *Database) InsertAccount(ctx context.Context, account *descriptor.Account) error {
|
|
var err error
|
|
request := `INSERT INTO account(id, username, password, disabled, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $4, $5, $6)`
|
|
_, err = db.db.Exec(request, account.ID, account.Username, account.Password,
|
|
account.Disabled, account.CreatedAt, account.UpdatedAt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (db *Database) UpdateAccountByID(ctx context.Context, accountID int64, account *descriptor.Account) error {
|
|
var err error
|
|
request := `UPDATE account SET username = $1, password = $2, disabled = $3, updated_at = $4 WHERE id = $6`
|
|
_, err = db.db.Exec(request, account.Username, account.Password, account.Disabled, account.UpdatedAt, accountID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (db *Database) ListAccounts(ctx context.Context) ([]descriptor.Account, error) {
|
|
var err error
|
|
request := `SELECT id, username, disabled, created_at, updated_at FROM account`
|
|
res := make([]descriptor.Account, 0)
|
|
err = db.db.Select(&res, request)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
return res, err
|
|
}
|
|
|
|
func (db *Database) GetAccountByID(ctx context.Context, accountID int64) (bool, *descriptor.Account, error) {
|
|
var err error
|
|
var res *descriptor.Account
|
|
var exists bool
|
|
request := `SELECT id, username, password, disabled, created_at, updated_at
|
|
FROM account WHERE id = $1 LiMIT 1`
|
|
dbRes := make([]descriptor.Account, 0)
|
|
err = db.db.Select(&dbRes, request, accountID)
|
|
if err != nil {
|
|
return exists, res, err
|
|
}
|
|
if len(dbRes) == 0 {
|
|
return exists, res, err
|
|
}
|
|
exists = true
|
|
res = &dbRes[0]
|
|
return exists, res, err
|
|
}
|
|
|
|
func (db *Database) GetAccountByUsername(ctx context.Context, username string) (bool, *descriptor.Account, error) {
|
|
var err error
|
|
var res *descriptor.Account
|
|
var exists bool
|
|
request := `SELECT id, username, password, disabled, created_at, updated_at
|
|
FROM account WHERE username = $1 LIMIT 1`
|
|
dbRes := make([]descriptor.Account, 0)
|
|
err = db.db.Select(&dbRes, request, username)
|
|
if err != nil {
|
|
return exists, res, err
|
|
}
|
|
if len(dbRes) == 0 {
|
|
return false, res, err
|
|
}
|
|
exists = true
|
|
res = &dbRes[0]
|
|
return exists, res, err
|
|
}
|
|
|
|
func (db *Database) DeleteAccountByID(ctx context.Context, accountID int64) error {
|
|
var err error
|
|
request := `DELETE FROM account WHERE id = $1`
|
|
_, err = db.db.Exec(request, accountID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (db *Database) DeleteAccountByUsername(ctx context.Context, username string) error {
|
|
var err error
|
|
request := `DELETE FROM account WHERE username = $1`
|
|
_, err = db.db.Exec(request, username)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|