From 0adcee59a57178dc7706a43a15aa5b90fe1cb1ee Mon Sep 17 00:00:00 2001
From: Thibaut Horel <thibaut.horel@gmail.com>
Date: Thu, 27 Dec 2018 10:13:29 -0500
Subject: WIP: add login template

---
 store.go | 35 ++++++++++++++++++++++++-----------
 1 file changed, 24 insertions(+), 11 deletions(-)

(limited to 'store.go')

diff --git a/store.go b/store.go
index 55d808b..1db6411 100644
--- a/store.go
+++ b/store.go
@@ -15,32 +15,45 @@ type Session struct {
 type User struct {
 	Id       int64
 	UserName string
+	Password string
 }
 
 type Store interface {
 	GetSession(id string) (*Session, bool)
-	GetUser(id int64) (*User, bool)
+	GetUser(name string) (*User, bool)
 }
 
 type PgStore struct {
 	*sql.DB
-	cache map[string]*Session
+	sessionCache map[string]*Session
 }
 
-func NewPgStore() *PgSessionStore {
+func NewPgStore() *PgStore {
 	db, err := sql.Open("postgres", "postgres://auth_master:pass@localhost/authdb")
 	if err != nil {
 		log.Panic(err)
 	}
-	return &PgSessionStore{db, make(map[string]*Session)}
+	return &PgStore{db, make(map[string]*Session)}
 }
 
-func (store *PgSessionStore) Get(id string) (*Session, bool) {
-	s, ok := store.cache[id]
-	if !ok {
-		row := store.QueryRow("SELECT id, user_id FROM session WHERE id = ?", id)
-		var s Session
-		row.Scan(&s.
+func (store *PgStore) GetSession(id string) (*Session, bool) {
+	s, ok := store.sessionCache[id]
+	if ok {
+		return s, true
 	}
-	return s, ok
+	s = new(Session)
+	row := store.QueryRow("SELECT id, user_id FROM session WHERE id = $1", id)
+	if err := row.Scan(s.Id, s.UserId); err != nil {
+		return nil, false
+	}
+	return s, true
+}
+
+func (store *PgStore) GetUser(name string) (*User, bool) {
+	u := new(User)
+	row := store.QueryRow("SELECT id, user_name, password FROM user WHERE user_name = $1", name)
+	if err := row.Scan(u.Id, u.UserName, u.Password); err != nil {
+		return nil, false
+	}
+	return u, true
 }
-- 
cgit v1.2.3-70-g09d2