Added Added view for getting keys of registered users.

1 files changed, 17 insertions, 0 deletions

diff --git a/http-server/database.go b/http-server/database.go
index d13e8c1..e79b3e7 100644
--- a/http-server/database.go
+++ b/http-server/database.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"database/sql"
+	"errors"
 	_ "github.com/mattn/go-sqlite3"
 )
 
@@ -32,6 +33,22 @@ func (conn *SQLConnection) checkUserRegistered(username string) (bool, error) {
 	return false, nil
 }
 
+func (conn *SQLConnection) getUserKey(username string) (string, error) {
+	query := "SELECT key FROM users WHERE username = ?"
+	result, err := conn.db.Query(query, username)
+	if err != nil {
+		return "", err
+	}
+	if result.Next() {
+		var key string
+		_ = result.Scan(&key)
+		_ = result.Close()
+		return key, nil
+	} else {
+		return "", errors.New("user not found")
+	}
+}
+
 func (conn *SQLConnection) saveMessage(message Message) error {
 	var err error