Add basic user page

1 files changed, 18 insertions, 8 deletions

diff --git a/store.go b/store.go
index 9723b66..89698ea 100644
--- a/store.go
+++ b/store.go
@@ -3,13 +3,15 @@ package main
 import (
 	"database/sql"
 	"log"
+	"time"
 
 	_ "github.com/lib/pq"
 )
 
 type Session struct {
-	Id     string
-	UserId int64
+	Id      string
+	UserId  int64
+	Created time.Time
 }
 
 type User struct {
@@ -43,8 +45,11 @@ func (store *PgStore) GetSession(id string) (*Session, bool) {
 		return s, true
 	}
 	s = new(Session)
-	row := store.QueryRow("SELECT id, user_id FROM sessions WHERE id = $1", id)
-	if err := row.Scan(&s.Id, &s.UserId); err != nil {
+	row := store.QueryRow(
+		"SELECT id, user_id, created_at FROM sessions WHERE id = $1",
+		id,
+	)
+	if err := row.Scan(&s.Id, &s.UserId, &s.Created); err != nil {
 		return nil, false
 	}
 	store.sessionCache[s.Id] = s
@@ -52,16 +57,21 @@ func (store *PgStore) GetSession(id string) (*Session, bool) {
 }
 
 func (store *PgStore) NewSession(userId int64) *Session {
-	var id string
-	store.QueryRow("INSERT INTO sessions(user_id) VALUES ($1) RETURNING id", userId).Scan(&id)
-	s := &Session{id, userId}
+	s := &Session{UserId: userId}
+	store.QueryRow(
+		"INSERT INTO sessions(user_id) VALUES ($1) RETURNING id, created_at",
+		userId,
+	).Scan(&s.Id, &s.Created)
 	store.sessionCache[s.Id] = s
 	return s
 }
 
 func (store *PgStore) GetUser(name string) (*User, bool) {
 	u := &User{Name: name}
-	row := store.QueryRow("SELECT id, password FROM users WHERE name = $1", name)
+	row := store.QueryRow(
+		"SELECT id, password FROM users WHERE name = $1",
+		name,
+	)
 	if err := row.Scan(&u.Id, &u.Password); err != nil {
 		return nil, false
 	}