Add view command and update utils

development v0.0.87
DustyP 5 years ago
parent 128011c79a
commit 0bf31a3e5d

@ -2,10 +2,10 @@ package discord_utils
import "github.com/dustinpianalto/disgoman" import "github.com/dustinpianalto/disgoman"
func GetChannelName(ctx disgoman.Context, id string) (string, error) { func GetChannelName(ctx disgoman.Context, id string) string {
channel, err := ctx.Session.Channel(id) channel, err := ctx.Session.Channel(id)
if err != nil { if err != nil {
return "", err return ""
} }
return channel.Name, nil return channel.Name
} }

@ -2,13 +2,13 @@ package discord_utils
import "github.com/dustinpianalto/disgoman" import "github.com/dustinpianalto/disgoman"
func GetDisplayName(ctx disgoman.Context, id string) (string, error) { func GetDisplayName(ctx disgoman.Context, id string) string {
member, err := ctx.Session.GuildMember(ctx.Guild.ID, id) member, err := ctx.Session.GuildMember(ctx.Guild.ID, id)
if err != nil { if err != nil {
return "", err return ""
} }
if member.Nick != "" { if member.Nick != "" {
return member.Nick, nil return member.Nick
} }
return member.User.Username, nil return member.User.Username
} }

@ -34,4 +34,6 @@ func AddCommandHandlers(g *disgoman.CommandManager) {
_ = g.AddCommand(requests.RequestCommand) _ = g.AddCommand(requests.RequestCommand)
_ = g.AddCommand(requests.CloseCommand) _ = g.AddCommand(requests.CloseCommand)
_ = g.AddCommand(requests.ListCommand) _ = g.AddCommand(requests.ListCommand)
_ = g.AddCommand(requests.ViewCommand)
_ = g.AddCommand(requests.CommentCommand)
} }

@ -180,7 +180,17 @@ func closeCommandFunc(ctx disgoman.Context, args []string) {
if err != nil { if err != nil {
discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was closed.", err) discord_utils.SendErrorMessage(ctx, "There was an error sending the message. The request was closed.", err)
} }
dmChannel, err := ctx.Session.UserChannelCreate(request.Author.ID)
if err != nil {
return
}
_, _ = ctx.Session.ChannelMessageSend(dmChannel.ID,
fmt.Sprintf("%s has closed request %d which you opened in the %s channel.\n```%s```\n",
discord_utils.GetDisplayName(ctx, request.CompletedBy.ID),
request.ID,
discord_utils.GetChannelName(ctx, request.Channel.ID),
request.Content,
))
} }
} }
@ -233,18 +243,22 @@ func listCommandFunc(ctx disgoman.Context, args []string) {
} else { } else {
channelName = channel.Name channelName = channel.Name
} }
commentCount, err := services.RequestService.RequestCommentCount(request)
if err != nil {
commentCount = 0
}
_, _ = ctx.Send(fmt.Sprintf("```md\n"+ _, _ = ctx.Send(fmt.Sprintf("```md\n"+
"< Request ID Requested By >\n"+ "< Request ID Requested By >\n"+
"< %-11d %23s >\n"+ "< %-11d %23s >\n"+
"%s\n\n"+ "%s\n\n"+
"Comments: Not Implemented Yet\n"+ "Comments: %d\n"+
"Requested At: %s\n"+ "Requested At: %s\n"+
"In: %s\n"+ "In: %s\n"+
"```", "```",
request.ID, request.ID,
authorName, authorName,
request.Content, request.Content,
commentCount,
request.RequestedAt.Format("2006-01-02 15:04:05 MST"), request.RequestedAt.Format("2006-01-02 15:04:05 MST"),
channelName, channelName,
)) ))
@ -303,8 +317,6 @@ func commentCommandFunc(ctx disgoman.Context, args []string) {
comments, _ := services.RequestService.RequestComments(request) comments, _ := services.RequestService.RequestComments(request)
var commentString string var commentString string
var commentStrings []string var commentStrings []string
displayName, _ := discord_utils.GetDisplayName(ctx, request.Author.ID)
channelName, _ := discord_utils.GetChannelName(ctx, request.Channel.ID)
commentString = fmt.Sprintf("Comment added:\n```md\n"+ commentString = fmt.Sprintf("Comment added:\n```md\n"+
"< Request ID Requested By >\n"+ "< Request ID Requested By >\n"+
"< %-11d %23s >\n"+ "< %-11d %23s >\n"+
@ -314,20 +326,19 @@ func commentCommandFunc(ctx disgoman.Context, args []string) {
"In: %s\n"+ "In: %s\n"+
"```", "```",
request.ID, request.ID,
displayName, discord_utils.GetDisplayName(ctx, request.Author.ID),
request.Content, request.Content,
request.RequestedAt.Format("2006-01-02 15:04:05 MST"), request.RequestedAt.Format("2006-01-02 15:04:05 MST"),
channelName, discord_utils.GetChannelName(ctx, request.Channel.ID),
) )
for _, c := range comments { for _, c := range comments {
cAuthorName, err := discord_utils.GetDisplayName(ctx, c.Author.ID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
continue continue
} }
cs := fmt.Sprintf("```md\n%s\n- %s At %s\n```\n", cs := fmt.Sprintf("```md\n%s\n- %s At %s\n```\n",
c.Content, c.Content,
cAuthorName, discord_utils.GetDisplayName(ctx, c.Author.ID),
c.CommentAt.Format("2006-01-02 15:04:05 MST"), c.CommentAt.Format("2006-01-02 15:04:05 MST"),
) )
if len(commentString+cs) >= 2000 { if len(commentString+cs) >= 2000 {
@ -344,4 +355,91 @@ func commentCommandFunc(ctx disgoman.Context, args []string) {
} }
} }
_, err = ctx.Send(fmt.Sprintf("%s your comment has been added.", ctx.Message.Author.Mention())) _, err = ctx.Send(fmt.Sprintf("%s your comment has been added.", ctx.Message.Author.Mention()))
dmChannel, err := ctx.Session.UserChannelCreate(request.Author.ID)
if err != nil {
return
}
_, _ = ctx.Session.ChannelMessageSend(dmChannel.ID,
fmt.Sprintf("%s has add a comment to request %d which you opened in the %s channel.\n```%s```\n```%s```",
discord_utils.GetDisplayName(ctx, author.ID),
request.ID,
discord_utils.GetChannelName(ctx, ctx.Channel.ID),
request.Content,
comment.Content,
))
}
var ViewCommand = &disgoman.Command{
Name: "view",
Aliases: nil,
Description: "View the details about a request.",
OwnerOnly: false,
Hidden: false,
RequiredPermissions: 0,
Invoke: viewCommandFunc,
}
func viewCommandFunc(ctx disgoman.Context, args []string) {
guild, err := services.GuildService.GetOrCreateGuild(ctx.Guild.ID)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Error getting Guild from the database", err)
return
}
id, err := strconv.ParseInt(args[0], 10, 64)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Please include the ID of the request to view.", err)
return
}
request, err := services.RequestService.Request(id)
if err != nil || request.Guild.ID != guild.ID {
discord_utils.SendErrorMessage(ctx, fmt.Sprintf("%d is not a valid request in this guild", id), err)
return
}
requestor, err := services.UserService.GetOrCreateUser(ctx.Message.Author.ID)
if err != nil {
discord_utils.SendErrorMessage(ctx, "Sorry, there was an issue finding your user account", err)
return
}
if request.Author.ID != ctx.Message.Author.ID &&
!discord_utils.IsGuildMod(ctx, requestor) &&
!discord_utils.IsGuildAdmin(ctx, requestor) {
discord_utils.SendErrorMessage(ctx, "You are not authorized to view that request", nil)
return
}
comments, _ := services.RequestService.RequestComments(request)
var commentString string
var commentStrings []string
commentString = fmt.Sprintf("Comment added:\n```md\n"+
"< Request ID Requested By >\n"+
"< %-11d %23s >\n"+
"%s\n\n"+
"Comments: Not Implemented Yet\n"+
"Requested At: %s\n"+
"In: %s\n"+
"```",
request.ID,
discord_utils.GetDisplayName(ctx, request.Author.ID),
request.Content,
request.RequestedAt.Format("2006-01-02 15:04:05 MST"),
discord_utils.GetChannelName(ctx, request.Channel.ID),
)
for _, c := range comments {
if err != nil {
log.Println(err)
continue
}
cs := fmt.Sprintf("```md\n%s\n- %s At %s\n```\n",
c.Content,
discord_utils.GetDisplayName(ctx, c.Author.ID),
c.CommentAt.Format("2006-01-02 15:04:05 MST"),
)
if len(commentString+cs) >= 2000 {
commentStrings = append(commentStrings, commentString)
commentString = ""
}
}
for _, c := range commentStrings {
_, _ = ctx.Send(c)
}
} }

Loading…
Cancel
Save