lmao.go

package main

import (
	"context"
	"errors"
	"log"
	"net/http"
	"runtime/debug"
	"strconv"
	"time"

	"github.com/aws/aws-lambda-go/lambda"
	"github.com/diamondburned/arikawa/v3/discord"
	"github.com/sirupsen/logrus"
	"github.com/uav-gaming/lmao/lmao"
)

// Constants
const BUILD_TIME_KEY string = "build_time"

// Globals
var (
	// Variables populated by build script.
	BuildTime string

	// Variables populated by `init`s and other runtime functions.
	BuildInfo map[string]interface{}
	bot       *lmao.LMAO
)

// Initialize build info
func init() {
	BuildInfo = make(map[string]interface{})

	// Get build info from binary.
	info, ok := debug.ReadBuildInfo()
	if !ok {
		logrus.Fatal("Failed to read build info.")
	}
	for _, setting := range info.Settings {
		BuildInfo[setting.Key] = setting.Value
	}

	// Parse and inject build time
	build_epoch, err := strconv.ParseInt(BuildTime, 10, 64)
	if err != nil {
		log.Fatal("Failed to parse timestamp to int64: ", err)
	}
	BuildInfo[BUILD_TIME_KEY] = time.Unix(build_epoch, 0)
}

func HandleRequest(ctx context.Context, request lmao.Request) (lmao.Response, error) {
	logrus.Infof("Received request %+v", request)

	// Verify request signature.
	if !bot.VerifyRequest(request) {
		logrus.Warn("Request verification failed.")
		return lmao.Response{
			StatusCode: http.StatusUnauthorized,
		}, nil
	}

	// Parse and handle request.
	var event discord.InteractionEvent
	if err := event.UnmarshalJSON([]byte(request.Body)); err != nil {
		logrus.Error("Failed to unmarshal request body: ", request.Body)
		return lmao.Response{}, errors.New("invalid request format")
	}

	response := bot.HandleInteraction(event)
	return lmao.ToHttpResponse(response)
}

func main() {
	// Setup logger.
	logrus.SetFormatter(&logrus.JSONFormatter{
		DisableTimestamp: true, // Logged by AWS.
	})
	logrus.SetReportCaller(true) // Log caller.

	logrus.Info("Starting up with build info: ", BuildInfo)

	var err error
	bot, err = lmao.NewLMAO(lmao.GetenvMustStr("DISCORD_TOKEN"), lmao.GetenvMustEd25519PubKey("DISCORD_PUBLIC_KEY"), lmao.GetenvMustValidSnowflake[discord.AppID]("DISCORD_APPLICATION_ID"))
	if err != nil {
		logrus.Fatal("Failed to init bot: ", err)
	}

	lambda.Start(HandleRequest)
}