From 294d06ea4865e4bae88c1dee6a8a841a7f204189 Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Wed, 19 Feb 2025 20:44:54 +0800 Subject: ssh/recv, users, schema: Create pubkey_only users --- users.go | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 users.go (limited to 'users.go') 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 +} -- cgit v1.2.3