109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"os"
|
|
"os/signal"
|
|
|
|
wh "go-eve-pi/webhook"
|
|
|
|
logger "git.site.quack-lab.dev/dave/cylogger"
|
|
|
|
"github.com/fasthttp/router"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
var webhook wh.Webhook
|
|
|
|
func main() {
|
|
// Add flag for generating .env.example
|
|
help := flag.Bool("h", false, "Show help")
|
|
flag.Parse()
|
|
|
|
if *help {
|
|
flag.PrintDefaults()
|
|
if err := GenerateEnvExample(); err != nil {
|
|
logger.Error("Failed to generate .env.example: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
logger.Info("Generated .env.example")
|
|
os.Exit(0)
|
|
}
|
|
|
|
logger.Init(logger.ParseLevel(options.LogLevel))
|
|
logger.Info("Starting Eve PI")
|
|
|
|
// Create SSO instance
|
|
sso, err := NewSSO(
|
|
options.ClientID,
|
|
options.RedirectURI,
|
|
options.Scopes,
|
|
)
|
|
if err != nil {
|
|
logger.Error("Failed to create SSO instance %v", err)
|
|
return
|
|
}
|
|
|
|
webhook = wh.NewZulipWebhook(options.WebhookURL, options.WebhookEmail, options.WebhookToken)
|
|
|
|
// Setup fasthttp router
|
|
r := router.New()
|
|
|
|
// Configure SSO to use existing fasthttp router
|
|
sso.SetRouter(r)
|
|
|
|
// Add your own routes
|
|
r.GET("/", func(ctx *fasthttp.RequestCtx) {
|
|
ctx.SetStatusCode(fasthttp.StatusOK)
|
|
ctx.WriteString("EVE PI Server Running")
|
|
})
|
|
|
|
r.GET("/login/{character}", func(ctx *fasthttp.RequestCtx) {
|
|
charName := ctx.UserValue("character")
|
|
charNameStr, ok := charName.(string)
|
|
if !ok || charNameStr == "" {
|
|
ctx.SetStatusCode(fasthttp.StatusBadRequest)
|
|
ctx.WriteString("Missing character parameter")
|
|
return
|
|
}
|
|
logger.Info("Login requested for character %s", charNameStr)
|
|
// Trigger the auth flow (will register callback if needed)
|
|
token, err := sso.GetToken(context.Background(), charNameStr)
|
|
if err != nil {
|
|
logger.Error("Failed to authenticate character %s: %v", charNameStr, err)
|
|
ctx.SetStatusCode(fasthttp.StatusInternalServerError)
|
|
ctx.WriteString("Authentication failed")
|
|
return
|
|
}
|
|
logger.Info("Successfully authenticated character %s", charNameStr)
|
|
ctx.SetContentType("text/plain")
|
|
ctx.SetStatusCode(fasthttp.StatusOK)
|
|
ctx.WriteString("Authenticated! Access token: " + token)
|
|
})
|
|
|
|
logger.Info("Starting web server on 0.0.0.0:%s", options.Port)
|
|
go func() {
|
|
if err := fasthttp.ListenAndServe("0.0.0.0:"+options.Port, r.Handler); err != nil {
|
|
logger.Error("Server failed: %v", err)
|
|
}
|
|
}()
|
|
|
|
// for i := 0; i < 100; i++ {
|
|
// logger.Info("Sending test webhook %d", i)
|
|
// go func() {
|
|
// err := webhook.Post("EvePI", "test", "test" + strconv.Itoa(i))
|
|
// if err != nil {
|
|
// logger.Error("Failed to send test webhook %d: %v", i, err)
|
|
// }
|
|
// }()
|
|
// }
|
|
|
|
// Listen for SIGINT and gracefully shut down the server
|
|
sigCh := make(chan os.Signal, 1)
|
|
signal.Notify(sigCh, os.Interrupt)
|
|
<-sigCh
|
|
logger.Info("SIGINT received, shutting down web server gracefully...")
|
|
logger.Info("Web server shut down cleanly")
|
|
}
|