working commit

2026-06-05 18:25:03 +02:00
commit 7d8abba003
82 changed files with 21863 additions and 0 deletions
@@ -0,0 +1,114 @@
+package database
+
+import (
+	"context"
+
+	"mbase/app/descriptor"
+)
+
+func (db *Database) InsertAccount(ctx context.Context, account *descriptor.Account) error {
+	var err error
+
+	request := `INSERT INTO account(id, username, passhash, disabled, created_at, updated_at)
+                        VALUES ($1, $2, $3, $4, $5, $6)`
+	_, err = db.db.Exec(request, account.ID, account.Username, account.Passhash,
+		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, passhash = $2, disabled = $3, updated_at = $4 WHERE id = $5`
+	_, err = db.db.Exec(request, account.Username, account.Passhash, account.Disabled, account.UpdatedAt, accountID)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (db *Database) ReducedListAccounts(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) CompletedListAccounts(ctx context.Context) ([]descriptor.Account, error) {
+	var err error
+	request := `SELECT * 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, passhash, 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, passhash, 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
+}
@@ -0,0 +1,124 @@
+package database
+
+import (
+	"context"
+	"path/filepath"
+
+	"mbase/pkg/logger"
+
+	"github.com/jmoiron/sqlx"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+const schema = `
+    --- DROP TABLE IF EXISTS issuer;
+    CREATE TABLE IF NOT EXISTS issuer (
+	id          INT  NOT NULL,
+	name        TEXT NOT NULL,
+	cert        TEXT NOT NULL,
+	key         TEXT,
+	signer_id   INT  NOT NULL,
+	signer_name TEXT NOT NULL,
+	revoked     BOOL
+    );
+    CREATE UNIQUE INDEX IF NOT EXISTS issuer_index01
+        ON issuer(id);
+    CREATE UNIQUE INDEX IF NOT EXISTS issuer_index02
+        ON issuer(name);
+
+    --- DROP TABLE IF EXISTS service;
+    CREATE TABLE IF NOT EXISTS service (
+	id          INT  NOT NULL,
+        issuer_id   INT  NOT NULL,
+        issuer_name TEXT NOT NULL,
+	name        TEXT NOT NULL,
+	cert        TEXT NOT NULL,
+	key         TEXT NOT NULL,
+	revoked     BOOL
+    );
+    CREATE UNIQUE INDEX IF NOT EXISTS service_index01
+        ON service(id);
+    CREATE UNIQUE INDEX IF NOT EXISTS service_index02
+        ON service(name);
+
+    --- DROP TABLE IF EXISTS account;
+    CREATE TABLE IF NOT EXISTS account (
+	id          INT  NOT NULL,
+	username    TEXT NOT NULL,
+	passhash    TEXT NOT NULL,
+        created_at  TEXT NOT NULL,
+        updated_at  TEXT NOT NULL,
+        disabled    BOOL
+    );
+    CREATE UNIQUE INDEX IF NOT EXISTS account_index01
+        ON account(id);
+    CREATE UNIQUE INDEX IF NOT EXISTS account_index02
+        ON account(username);
+
+
+    --- DROP TABLE IF EXISTS grant;
+    CREATE TABLE IF NOT EXISTS grant (
+	id          INT  NOT NULL,
+        account_id  INT  NOT NULL,
+	operation   TEXT NOT NULL,
+        created_at  TEXT NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS grant_index01
+        ON grant(account_id);
+    CREATE UNIQUE INDEX IF NOT EXISTS grant_index02
+        ON grant(account_id, operation);
+`
+
+type Database struct {
+	datapath string
+	db       *sqlx.DB
+	log      *logger.Logger
+}
+
+func NewDatabase(datapath string) (*Database, error) {
+	var err error
+	db := &Database{
+		datapath: datapath,
+	}
+	db.log = logger.NewLogger("database")
+	return db, err
+
+}
+
+func (db *Database) OpenDatabase() error {
+	var err error
+	dbPath := filepath.Join(db.datapath, "certmanager.db")
+	db.db, err = sqlx.Open("sqlite3", dbPath)
+	if err != nil {
+		return err
+	}
+
+	err = db.db.Ping()
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (db *Database) InitDatabase() error {
+	var err error
+	_, err = db.db.Exec(schema)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (db *Database) CleanDatabase(ctx context.Context) error {
+	var err error
+
+	request := `
+                DELETE FROM account;
+                DELETE FROM grant;
+        `
+	_, err = db.db.Exec(request)
+	if err != nil {
+		return err
+	}
+	return err
+}
@@ -0,0 +1,77 @@
+package database
+
+import (
+	"context"
+
+	"mbase/app/descriptor"
+)
+
+func (db *Database) InsertGrant(ctx context.Context, grant *descriptor.Grant) error {
+	var err error
+	request := `INSERT INTO grant(id, account_id, operation, created_at)
+                        VALUES ($1, $2, $3, $4)`
+	_, err = db.db.Exec(request, grant.ID, grant.AccountID, grant.Operation, grant.CreatedAt)
+	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 account_id = $1`
+	res := make([]descriptor.Grant, 0)
+	err = db.db.Select(&res, request, accountID)
+	if err != nil {
+		return res, err
+	}
+	return res, err
+}
+
+func (db *Database) ListGrants(ctx context.Context) ([]descriptor.Grant, error) {
+	var err error
+	request := `SELECT * FROM grant`
+	res := make([]descriptor.Grant, 0)
+	err = db.db.Select(&res, request)
+	if err != nil {
+		return res, err
+	}
+	return res, err
+}
+
+func (db *Database) GetGrant(ctx context.Context, accountID int64, operation string) (bool, *descriptor.Grant, error) {
+	var err error
+	res := &descriptor.Grant{}
+	request := `SELECT * FROM grant WHERE account_id = $1 AND operation = $2 LIMIT 1`
+	dbRes := make([]descriptor.Grant, 0)
+	err = db.db.Select(&dbRes, request, accountID, operation)
+	if err != nil {
+		return false, res, err
+	}
+	if len(dbRes) == 0 {
+		return false, res, err
+
+	}
+	res = &dbRes[0]
+	return true, res, err
+}
+
+func (db *Database) DeleteGrantByAccountID(ctx context.Context, grantID int64, operation string) error {
+	var err error
+	request := `DELETE FROM grant WHERE account_id = $1 AND operation = $2`
+	_, err = db.db.Exec(request, grantID, operation)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (db *Database) DeleteAllGrantsForAccountID(ctx context.Context, grantID int64) error {
+	var err error
+	request := `DELETE FROM grant WHERE account_id = $1`
+	_, err = db.db.Exec(request, grantID)
+	if err != nil {
+		return err
+	}
+	return err
+}