certmanager updates

internal/database/account.go

@@ -0,0 +1,100 @@
+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
+}

internal/database/database.go

@@ -36,6 +36,24 @@ const schema = `
     );
     CREATE INDEX IF NOT EXISTS service_index
         ON issuer(id, name);
+
+    DROP TABLE IF EXISTS account;
+    CREATE TABLE IF NOT EXISTS account (
+	id          INT  NOT NULL,
+	username    TEXT NOT NULL,
+	password    TEXT NOT NULL,
+        created_at  TEXT NOT NULL,
+        updated_at  TEXT NOT NULL,
+        disabled    BOOL
+    );
+
+    DROP TABLE IF EXISTS grant;
+    CREATE TABLE IF NOT EXISTS grant (
+	id          INT  NOT NULL,
+        account_id  INT  NOT NULL,
+	operation   TEXT NOT NULL,
+	subject_id  INT  NOT NULL
+    );
 `
 
 type Database struct {

internal/database/grant.go

@@ -0,0 +1,79 @@
+package database
+
+import (
+	"context"
+
+	"certmanager/internal/descriptor"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+//type Grant struct {
+//ID        int64  `json:"id"         yaml:"id"         db:"id"`
+//AccountID int64  `json:"accountID"  yaml:"accountID"  db:"account_id"`
+//Operation string `json:"operation"  yaml:"operation"  db:"operation"`
+//SubjectID int64  `json:"subjectID"  yaml:"subjectID"  db:"subjectID"`
+//}
+
+func (db *Database) InsertGrant(ctx context.Context, grant *descriptor.Grant) error {
+	var err error
+	request := `INSERT INTO grant(id, account_id, operation, subject_id)
+                        VALUES ($1, $2, $3, $4)`
+	_, err = db.db.Exec(request, grant.ID, grant.AccountID, grant.Operation,
+		grant.SubjectID)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (db *Database) ListGrantsByAccountID(ctx context.Context, accountID int64) ([]descriptor.Grant, error) {
+	var err error
+	request := `SELECT * FROM grant WHERE `
+	res := make([]descriptor.Grant, 0)
+	err = db.db.Select(&res, request, accountID)
+	if err != nil {
+		return res, err
+	}
+	return res, err
+}
+
+func (db *Database) GetGrant(ctx context.Context, accountID, subjectID int64) (bool, []*descriptor.Grant, error) {
+	var err error
+	var res []*descriptor.Grant
+	var exists bool
+	request := `SELECT id, operation, grant_id, subject_id FROM grant
+                 WHERE account_id = $1
+                 AND subject_id = $1`
+	dbRes := make([]*descriptor.Grant, 0)
+	err = db.db.Select(&dbRes, request, accountID, subjectID)
+	if err != nil {
+		return exists, res, err
+	}
+	if len(dbRes) == 0 {
+		return false, res, err
+	}
+	exists = true
+	res = dbRes
+	return exists, res, err
+}
+
+func (db *Database) DeleteGrantByAccountID(ctx context.Context, grantID int64) error {
+	var err error
+	request := `DELETE FROM grant WHERE grant_id = $1`
+	_, err = db.db.Exec(request, grantID)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (db *Database) DeleteGrantsBySubjectID(ctx context.Context, subjectID int64) error {
+	var err error
+	request := `DELETE FROM grant WHERE subject_id = $1`
+	_, err = db.db.Exec(request, subjectID)
+	if err != nil {
+		return err
+	}
+	return err
+}