Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b6f90b61c5 | ||
|
|
764ceb79ec | ||
|
|
e89805bf8e | ||
|
|
cb7c487b42 | ||
|
|
e1ea9cab1f | ||
|
|
a6dc6f5e56 | ||
|
|
203d1bbf69 | ||
|
|
c504f7b165 | ||
|
|
34564140f5 | ||
|
|
e3ccc56dba |
@@ -29,7 +29,7 @@ func (s channelService) Channel(id string) (geeksbot.Channel, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s channelService) CreateChannel(c geeksbot.Channel) (geeksbot.Channel, error) {
|
func (s channelService) CreateChannel(c geeksbot.Channel) (geeksbot.Channel, error) {
|
||||||
queryString := "INSERT INTO channels (id, guild_id, admin, default_channel, new_patron VALUES ($1, $2, $3, $4, $5))"
|
queryString := "INSERT INTO channels (id, guild_id, admin, default_channel, new_patron) VALUES ($1, $2, $3, $4, $5)"
|
||||||
_, err := s.db.Exec(queryString, c.ID, c.Guild.ID, c.Admin, c.Default, c.NewPatron)
|
_, err := s.db.Exec(queryString, c.ID, c.Guild.ID, c.Admin, c.Default, c.NewPatron)
|
||||||
return c, err
|
return c, err
|
||||||
}
|
}
|
||||||
@@ -70,7 +70,8 @@ func (s channelService) GuildChannels(g geeksbot.Guild) ([]geeksbot.Channel, err
|
|||||||
func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot.Channel, error) {
|
func (s channelService) GetOrCreateChannel(id string, guild_id string) (geeksbot.Channel, error) {
|
||||||
channel, err := s.Channel(id)
|
channel, err := s.Channel(id)
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
guild, err := GuildService.GetOrCreateGuild(guild_id)
|
var guild geeksbot.Guild
|
||||||
|
guild, err = GuildService.GetOrCreateGuild(guild_id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return geeksbot.Channel{}, err
|
return geeksbot.Channel{}, err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -124,6 +124,14 @@ func (s requestService) CreateRequest(r geeksbot.Request) (geeksbot.Request, err
|
|||||||
completed, completed_at, completed_by, message_id, completed_message)
|
completed, completed_at, completed_by, message_id, completed_message)
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10) RETURNING id`
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10) RETURNING id`
|
||||||
var id int64
|
var id int64
|
||||||
|
var completedByID sql.NullString
|
||||||
|
if r.CompletedBy == nil {
|
||||||
|
completedByID.String = ""
|
||||||
|
completedByID.Valid = false
|
||||||
|
} else {
|
||||||
|
completedByID.String = r.CompletedBy.ID
|
||||||
|
completedByID.Valid = true
|
||||||
|
}
|
||||||
err := s.db.QueryRow(queryString,
|
err := s.db.QueryRow(queryString,
|
||||||
r.Author.ID,
|
r.Author.ID,
|
||||||
r.Channel.ID,
|
r.Channel.ID,
|
||||||
@@ -132,7 +140,7 @@ func (s requestService) CreateRequest(r geeksbot.Request) (geeksbot.Request, err
|
|||||||
r.RequestedAt,
|
r.RequestedAt,
|
||||||
r.Completed,
|
r.Completed,
|
||||||
r.CompletedAt,
|
r.CompletedAt,
|
||||||
r.CompletedBy.ID,
|
completedByID,
|
||||||
r.Message.ID,
|
r.Message.ID,
|
||||||
r.CompletedMessage).Scan(&id)
|
r.CompletedMessage).Scan(&id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ type userService struct {
|
|||||||
|
|
||||||
func (s userService) User(id string) (geeksbot.User, error) {
|
func (s userService) User(id string) (geeksbot.User, error) {
|
||||||
var user geeksbot.User
|
var user geeksbot.User
|
||||||
queryString := "SELECT id, steam_id, active, staff, admin WHERE id = $1"
|
queryString := "SELECT id, steam_id, active, staff, admin FROM users WHERE id = $1"
|
||||||
row := s.db.QueryRow(queryString, id)
|
row := s.db.QueryRow(queryString, id)
|
||||||
err := row.Scan(&user.ID, &user.SteamID, &user.IsActive, &user.IsStaff, &user.IsAdmin)
|
err := row.Scan(&user.ID, &user.SteamID, &user.IsActive, &user.IsStaff, &user.IsAdmin)
|
||||||
return user, err
|
return user, err
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package exts
|
|||||||
import (
|
import (
|
||||||
"github.com/dustinpianalto/disgoman"
|
"github.com/dustinpianalto/disgoman"
|
||||||
"github.com/dustinpianalto/geeksbot/internal/exts/guild"
|
"github.com/dustinpianalto/geeksbot/internal/exts/guild"
|
||||||
|
"github.com/dustinpianalto/geeksbot/internal/exts/requests"
|
||||||
"github.com/dustinpianalto/geeksbot/internal/exts/utils"
|
"github.com/dustinpianalto/geeksbot/internal/exts/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -23,4 +24,5 @@ func AddCommandHandlers(g *disgoman.CommandManager) {
|
|||||||
_ = g.AddCommand(utils.PingCommand)
|
_ = g.AddCommand(utils.PingCommand)
|
||||||
_ = g.AddCommand(guild.AddPrefixCommand)
|
_ = g.AddCommand(guild.AddPrefixCommand)
|
||||||
_ = g.AddCommand(guild.RemovePrefixCommand)
|
_ = g.AddCommand(guild.RemovePrefixCommand)
|
||||||
|
_ = g.AddCommand(requests.RequestCommand)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,8 +1,10 @@
|
|||||||
package requests
|
package requests
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/dustinpianalto/disgoman"
|
"github.com/dustinpianalto/disgoman"
|
||||||
"github.com/dustinpianalto/geeksbot"
|
"github.com/dustinpianalto/geeksbot"
|
||||||
@@ -29,12 +31,12 @@ func requestCommandFunc(ctx disgoman.Context, args []string) {
|
|||||||
requestMsg := strings.Join(args, " ")
|
requestMsg := strings.Join(args, " ")
|
||||||
author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
|
author, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
discord_utils.SendErrorMessage(ctx, "Error creating the request", err)
|
discord_utils.SendErrorMessage(ctx, "Error creating the request. Could not get user.", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
channel, err := services.ChannelService.GetOrCreateChannel(ctx.Message.ChannelID, ctx.Guild.ID)
|
channel, err := services.ChannelService.GetOrCreateChannel(ctx.Message.ChannelID, ctx.Guild.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
discord_utils.SendErrorMessage(ctx, "Error creating the request", err)
|
discord_utils.SendErrorMessage(ctx, "Error creating the request. Could not get channel.", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
int64ID, _ := strconv.ParseInt(ctx.Message.ID, 10, 64)
|
int64ID, _ := strconv.ParseInt(ctx.Message.ID, 10, 64)
|
||||||
@@ -60,4 +62,42 @@ func requestCommandFunc(ctx disgoman.Context, args []string) {
|
|||||||
discord_utils.SendErrorMessage(ctx, "Error creating the request", err)
|
discord_utils.SendErrorMessage(ctx, "Error creating the request", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
channels, err := services.ChannelService.GuildChannels(guild)
|
||||||
|
if err == nil {
|
||||||
|
var mentionRolesString string
|
||||||
|
roles, err := services.GuildService.GuildRoles(guild)
|
||||||
|
if err == nil {
|
||||||
|
for _, r := range roles {
|
||||||
|
if r.RoleType == "admin" || r.RoleType == "moderator" {
|
||||||
|
mentionRolesString += fmt.Sprintf("<@&%s> ", r.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, c := range channels {
|
||||||
|
if c.Admin {
|
||||||
|
_, _ = ctx.Session.ChannelMessageSend(c.ID,
|
||||||
|
fmt.Sprintf("%s\n"+
|
||||||
|
"New Request ID %d "+
|
||||||
|
"%s has requested assistance: \n"+
|
||||||
|
"```\n%s\n```\n"+
|
||||||
|
"Requested At: %s\n"+
|
||||||
|
"In: %s",
|
||||||
|
mentionRolesString,
|
||||||
|
request.ID,
|
||||||
|
ctx.Message.Author.Mention,
|
||||||
|
request.Content,
|
||||||
|
request.RequestedAt.UTC().Format(time.UnixDate),
|
||||||
|
ctx.Channel.Mention(),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = ctx.Send(fmt.Sprintf("%s The admin have recieved your request.\n "+
|
||||||
|
"If you would like to close or add a comment to this request please reference ID `%v`",
|
||||||
|
ctx.Message.Author.Mention(), request.ID,
|
||||||
|
))
|
||||||
|
if err != nil {
|
||||||
|
discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was created.", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user