102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
|
|
"certmanager/internal/descriptor"
|
|
"certmanager/pkg/auxid"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func (db *Database) InsertIssuer(ctx context.Context, issuer *descriptor.Issuer) (int64, error) {
|
|
var err error
|
|
var res int64
|
|
issuer.ID = auxid.GenID()
|
|
request := `INSERT INTO issuer(id, name, cert, key, revoked)
|
|
VALUES ($1, $2, $3, $4, $5)`
|
|
_, err = db.db.Exec(request, issuer.ID, issuer.Name, issuer.Cert, issuer.Key, issuer.Revoked)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
res = issuer.ID
|
|
return res, err
|
|
}
|
|
|
|
func (db *Database) UpdateIssuerByID(ctx context.Context, issuerID int64, issuer *descriptor.Issuer) error {
|
|
var err error
|
|
request := `UPDATE issuer SET revoked = $1 WHERE id = $2`
|
|
_, err = db.db.Exec(request, issuer.Revoked, issuerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (db *Database) ListIssuers(ctx context.Context) ([]descriptor.Issuer, error) {
|
|
var err error
|
|
request := `SELECT id, name, revoked FROM issuer`
|
|
res := make([]descriptor.Issuer, 0)
|
|
err = db.db.Select(&res, request)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
return res, err
|
|
}
|
|
|
|
func (db *Database) GetIssuerByID(ctx context.Context, issuerID int64) (bool, *descriptor.Issuer, error) {
|
|
var err error
|
|
var res *descriptor.Issuer
|
|
var exists bool
|
|
request := `SELECT id, name, cert, key, revoked FROM issuer WHERE id = $1 LiMIT 1`
|
|
dbRes := make([]descriptor.Issuer, 0)
|
|
err = db.db.Select(&dbRes, request, issuerID)
|
|
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) GetIssuerByName(ctx context.Context, issuerName string) (bool, *descriptor.Issuer, error) {
|
|
var err error
|
|
var res *descriptor.Issuer
|
|
var exists bool
|
|
request := `SELECT id, name, cert, key, revoked FROM issuer WHERE name = $1 LIMIT 1`
|
|
dbRes := make([]descriptor.Issuer, 0)
|
|
err = db.db.Select(&dbRes, request, issuerName)
|
|
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) DeleteIssuerByID(ctx context.Context, issuerID int64) error {
|
|
var err error
|
|
request := `DELETE FROM issuer WHERE id = $1`
|
|
_, err = db.db.Exec(request, issuerID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (db *Database) DeleteIssuerByName(ctx context.Context, issuerName string) error {
|
|
var err error
|
|
request := `DELETE FROM issuer WHERE name = $1`
|
|
_, err = db.db.Exec(request, issuerName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|