cerman updated

internal/database/database.go

@@ -1,6 +1,7 @@
 package database
 
 import (
+	"context"
 	"path/filepath"
 
 	"certmanager/pkg/logger"
@@ -10,18 +11,20 @@ import (
 )
 
 const schema = `
-    --- DROP TABLE IF EXISTS issuer;
+    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,
-	revoked BOOL
+	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 INDEX IF NOT EXISTS issuer_index
         ON issuer(id, name);
 
-    --- DROP TABLE IF EXISTS service;
+    DROP TABLE IF EXISTS service;
     CREATE TABLE IF NOT EXISTS service (
 	id          INT  NOT NULL,
         issuer_id   INT  NOT NULL,
@@ -71,3 +74,16 @@ func (db *Database) InitDatabase() error {
 	}
 	return err
 }
+
+func (db *Database) CleanDatabase(ctx context.Context) error {
+	var err error
+	request := `
+                DELETE FROM issuer;
+                DELETE FROM service;
+        `
+	_, err = db.db.Exec(request)
+	if err != nil {
+		return err
+	}
+	return err
+}