attendance.go

package slash

import (
	"fmt"

	"github.com/bwmarrin/discordgo"
	"github.com/ritsec/ops-bot-iii/commands/slash/permission"
	"github.com/ritsec/ops-bot-iii/data"
	"github.com/ritsec/ops-bot-iii/helpers"
	"github.com/ritsec/ops-bot-iii/logging"
	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace"
	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
)

// Attendance slash command
func Attendance() (*discordgo.ApplicationCommand, func(s *discordgo.Session, i *discordgo.InteractionCreate)) {
	return &discordgo.ApplicationCommand{
			Name:                     "attendance",
			Description:              "Get signin history",
			DefaultMemberPermissions: &permission.Member,
		},
		func(s *discordgo.Session, i *discordgo.InteractionCreate) {
			span := tracer.StartSpan(
				"commands.slash.attendance:Attendance",
				tracer.ResourceName("/attendance"),
			)
			defer span.Finish()

			logging.Debug(s, "Attendance command received", i.Member.User, span)

			err := s.InteractionRespond(
				i.Interaction,
				&discordgo.InteractionResponse{
					Type: discordgo.InteractionResponseChannelMessageWithSource,
					Data: &discordgo.InteractionResponseData{
						Content: attendanceMessage(i.Member.User.ID, span.Context()),
						Flags:   discordgo.MessageFlagsEphemeral,
					},
				},
			)
			if err != nil {
				logging.Error(s, err.Error(), i.Member.User, span)
			} else {
				logging.Debug(s, "Signin History Given", i.Member.User, span)
			}
		}
}

// returns the message sent to the user by the Attendance command
func attendanceMessage(userID string, ctx ddtrace.SpanContext) (message string) {
	span := tracer.StartSpan(
		"commands.slash.attendance:attendanceMessage",
		tracer.ResourceName("/attendance:attendanceMessage"),
		tracer.ChildOf(ctx),
	)
	defer span.Finish()

	message = "**Your Signins:**"
	signinTypes := helpers.SigninTypeArray()

	totalSignins, err := data.Signin.GetSignins(userID, span.Context())
	if err != nil {
		logging.Error(nil, err.Error(), nil, span)
		totalSignins = 0
	}
	message += fmt.Sprintf("\n\tTotal Signins: `%d`", totalSignins)

	for _, signinType := range signinTypes {
		entSigninType := helpers.StringToType(signinType)
		signins, err := data.Signin.GetSigninsByType(userID, entSigninType, span.Context())
		if err != nil {
			logging.Error(nil, err.Error(), nil, span)
			signins = 0
		}
		if signins != 0 {
			message += fmt.Sprintf("\n\t%s: `%d`", signinType, signins)
		}
	}

	return message
}