check if a user exists before creating it

1 files changed, 26 insertions, 7 deletions

diff --git a/user.go b/user.go
index e262151..d1ed261 100644
--- a/user.go
+++ b/user.go
@@ -1,6 +1,5 @@
 package main
 
-import "fmt"
 import "github.com/mxk/go-sqlite/sqlite3"
 
 type User struct {
@@ -11,13 +10,16 @@ type User struct {
 
 func createUser(db *sqlite3.Conn, name string, isAdmin bool) {
 
-    args := sqlite3.NamedArgs{"$a": name, "$b": isAdmin}
-    sql := "INSERT INTO users (name, admin) VALUES ($a, $b)"
-
-    db.Exec(sql, args)
+    //Check if a user with this name already exists
+    if (getUserForName(db, name).id == 0) {
+        args := sqlite3.NamedArgs{"$a": name, "$b": isAdmin}
+        sql := "INSERT INTO users (name, admin) VALUES ($a, $b)"
 
+        db.Exec(sql, args)
+    }
 
 }
+
 func getUserForId(db *sqlite3.Conn, id int) User {
 
     args := sqlite3.NamedArgs{"$a": id}
@@ -28,8 +30,6 @@ func getUserForId(db *sqlite3.Conn, id int) User {
     for ; err == nil ; err = s.Next() {
         var rowid int
         s.Scan(&rowid, row)     // Assigns 1st column to rowid, the rest to row
-        fmt.Println(rowid, row) // Prints "1 map[a:1 b:demo c:<nil>]"
-        fmt.Println(row["name"])
 
         user := User{id:rowid, name:row["name"].(string), isAdmin:row["admin"].(bool)}
         return user
@@ -38,5 +38,24 @@ func getUserForId(db *sqlite3.Conn, id int) User {
     //If we get here there are no matching users
     return User{id:0, name:"", isAdmin:false}
 
+}
+
+func getUserForName(db *sqlite3.Conn, name string) User {
+
+    args := sqlite3.NamedArgs{"$a": name}
+    sql := "SELECT * FROM users WHERE name = $a"
+    s, err := db.Query(sql, args)
+    row := make(sqlite3.RowMap)
+    
+    for ; err == nil ; err = s.Next() {
+        var rowid int
+        s.Scan(&rowid, row)
+
+        user := User{id:rowid, name:row["name"].(string), isAdmin:row["admin"].(bool)}
+        return user
+    }
+
+    //If we get here there are no matching users
+    return User{id:0, name:"", isAdmin:false}
 
 }
\ No newline at end of file