From 9c4348d402441dc8bac2cb3ecc41efc825307f0b Mon Sep 17 00:00:00 2001
From: Andrew <saintruler@gmail.com>
Date: Wed, 28 Apr 2021 14:37:37 +0400
Subject: Added Added view for getting keys of registered users.

---
 http-server/database.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'http-server/database.go')

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
 
-- 
cgit v1.2.3