Function Index

package main

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"os"

	"github.com/GoogleCloudPlatform/functions-framework-go/funcframework"
	_ "github.com/joho/godotenv/autoload"
	"k8s.io/klog"

	botjoin "github.com/roleypoly/roleypoly/src/functions/bot-join"
	createsession "github.com/roleypoly/roleypoly/src/functions/create-session"
	loginbounce "github.com/roleypoly/roleypoly/src/functions/login-bounce"
	loginhandler "github.com/roleypoly/roleypoly/src/functions/login-handler"
	sessiondata "github.com/roleypoly/roleypoly/src/functions/session-data"
)

var mappings map[string]http.HandlerFunc = map[string]http.HandlerFunc{
	"/session-data":   sessiondata.SessionData,
	"/bot-join":       botjoin.BotJoin,
	"/login-bounce":   loginbounce.LoginBounce,
	"/login-handler":  loginhandler.LoginHandler,
	"/create-session": createsession.CreateSession,
}

var port string

func main() {
	ctx := context.Background()

	err := funcframework.RegisterHTTPFunctionContext(ctx, "/", rootHandler)
	if err != nil {
		log.Fatalf("funcframework.RegisterHTTPFunctionContext: %v\n", err)
	}

	for path, handler := range mappings {
		err := funcframework.RegisterHTTPFunctionContext(ctx, path, wrapLogging(handler))
		if err != nil {
			log.Fatalf("funcframework.RegisterHTTPFunctionContext: %v\n", err)
		}

		fmt.Println("Added", path)
	}

	// Use PORT environment variable, or default to 6600.
	port = "6600"
	if envPort := os.Getenv("PORT"); envPort != "" {
		port = envPort
	}

	fmt.Printf("Starting on http://localhost:%s\n", port)

	if err := funcframework.Start(port); err != nil {
		log.Fatalf("funcframework.Start: %v\n", err)
	}
}

func rootHandler(rw http.ResponseWriter, r *http.Request) {
	body := `<!doctype html><meta charset="utf-8"><title>Roleypoly Functions</title><style>body{font-family: sans-serif; font-size: 1.4em}</style><h1>Function Index</h1>`

	for path := range mappings {
		body += fmt.Sprintf(`<a href="http://localhost:%s%s">%s</a><br>`, port, path, path)
	}

	fmt.Fprintln(rw, body)
}

func wrapLogging(fn func(rw http.ResponseWriter, r *http.Request)) func(rw http.ResponseWriter, r *http.Request) {
	return func(rw http.ResponseWriter, r *http.Request) {
		klog.Info(r.Method, " ", r.RequestURI)
		fn(rw, r)
	}
}