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,