From 74edd574b3426fff70891fa25b653706000683bd Mon Sep 17 00:00:00 2001 From: Dustin Pianalto Date: Sat, 23 Jan 2021 14:46:25 -0900 Subject: [PATCH] fix bug in user service --- internal/database/user.go | 7 ++++++- internal/exts/utils/utils.go | 1 + 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/internal/database/user.go b/internal/database/user.go index ced3758..21f1b9b 100644 --- a/internal/database/user.go +++ b/internal/database/user.go @@ -20,7 +20,12 @@ func (s userService) User(id string) (geeksbot.User, error) { 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)" - _, err := s.db.Exec(queryString, u.ID, u.SteamID, u.IsActive, u.IsStaff, u.IsAdmin) + var err error + if u.SteamID.Valid { + _, err = s.db.Exec(queryString, u.ID, u.SteamID.String, u.IsActive, u.IsStaff, u.IsAdmin) + } else { + _, err = s.db.Exec(queryString, u.ID, nil, u.IsActive, u.IsStaff, u.IsAdmin) + } return u, err } diff --git a/internal/exts/utils/utils.go b/internal/exts/utils/utils.go index 942b311..292413f 100644 --- a/internal/exts/utils/utils.go +++ b/internal/exts/utils/utils.go @@ -263,6 +263,7 @@ func addUserCommandFunc(ctx disgoman.Context, args []string) { if ctx.Message.Author.ID == ctx.CommandManager.Owners[0] { user := geeksbot.User{ ID: ctx.Message.Author.ID, + SteamID: "", IsActive: true, IsStaff: true, IsAdmin: true,