ssh/recv, users, schema: Create pubkey_only users

1 files changed, 26 insertions, 0 deletions

diff --git a/users.go b/users.go
new file mode 100644
index 0000000..9ac0798
--- /dev/null
+++ b/users.go
@@ -0,0 +1,26 @@
+package main
+
+import (
+	"context"
+)
+
+func add_user_ssh(ctx context.Context, pubkey string) (user_id int, err error) {
+	tx, err := database.Begin(ctx)
+	if err != nil {
+		return
+	}
+	defer tx.Rollback(ctx)
+
+	err = tx.QueryRow(ctx, `INSERT INTO users (type) VALUES ('pubkey_only') RETURNING id`).Scan(&user_id)
+	if err != nil {
+		return
+	}
+
+	_, err = tx.Exec(ctx, `INSERT INTO ssh_public_keys (key_string, user_id) VALUES ($1, $2)`, pubkey, user_id)
+	if err != nil {
+		return
+	}
+
+	err = tx.Commit(ctx)
+	return
+}