Move db and services to pkg

pkg/database/user.go

@@ -0,0 +1,51 @@
+package database
+
+import (
+	"database/sql"
+
+	"github.com/dustinpianalto/geeksbot"
+)
+
+type userService struct {
+	db *sql.DB
+}
+
+func (s userService) User(id string) (geeksbot.User, error) {
+	var user geeksbot.User
+	queryString := "SELECT id, steam_id, active, staff, admin FROM users WHERE id = $1"
+	row := s.db.QueryRow(queryString, id)
+	err := row.Scan(&user.ID, &user.SteamID, &user.IsActive, &user.IsStaff, &user.IsAdmin)
+	return user, err
+}
+
+func (s userService) CreateUser(u geeksbot.User) (geeksbot.User, error) {
+	queryString := "INSERT INTO users (id, steam_id, active, staff, admin) VALUES ($1, $2, $3, $4, $5)"
+	var err error
+	_, err = s.db.Exec(queryString, u.ID, u.SteamID, u.IsActive, u.IsStaff, u.IsAdmin)
+	return u, err
+}
+
+func (s userService) DeleteUser(u geeksbot.User) error {
+	queryString := "DELETE FROM users WHERE id = $1"
+	_, err := s.db.Exec(queryString, u.ID)
+	return err
+}
+
+func (s userService) UpdateUser(u geeksbot.User) (geeksbot.User, error) {
+	queryString := "UPDATE users SET steam_id = $2, active = $3, staff = $4, admin = $5 WHERE id = $1"
+	_, err := s.db.Exec(queryString, u.ID, u.SteamID, u.IsActive, u.IsStaff, u.IsAdmin)
+	return u, err
+}
+
+func (s userService) GetOrCreateUser(id string) (geeksbot.User, error) {
+	user, err := s.User(id)
+	if err == sql.ErrNoRows {
+		user, err = s.CreateUser(geeksbot.User{
+			ID:       id,
+			IsActive: true,
+			IsAdmin:  false,
+			IsStaff:  false,
+		})
+	}
+	return user, err
+}