81 lines
1.8 KiB
Go
Executable file
81 lines
1.8 KiB
Go
Executable file
package main
|
|
|
|
import (
|
|
"embed"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"atri.dad/api"
|
|
"atri.dad/lib"
|
|
"atri.dad/pages"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/labstack/echo/v4/middleware"
|
|
)
|
|
|
|
//go:embed public/*
|
|
var PublicFS embed.FS
|
|
|
|
func main() {
|
|
// Load environment variables
|
|
godotenv.Load(".env")
|
|
|
|
// Initialize Echo router
|
|
e := echo.New()
|
|
|
|
// Middleware
|
|
e.Use(middleware.Logger())
|
|
e.Use(middleware.Recover())
|
|
e.Pre(middleware.RemoveTrailingSlash())
|
|
e.Use(middleware.RequestID())
|
|
e.Use(middleware.Secure())
|
|
e.Use(middleware.GzipWithConfig(middleware.GzipConfig{
|
|
Level: 5,
|
|
}))
|
|
e.Use(middleware.RateLimiter(middleware.NewRateLimiterMemoryStore(50)))
|
|
|
|
// Static server
|
|
fs := http.FS(PublicFS)
|
|
e.GET("/public/*", echo.WrapHandler(http.FileServer(fs)))
|
|
|
|
// Page routes
|
|
e.GET("/", pages.Home)
|
|
e.GET("/projects", pages.Projects)
|
|
e.GET("/talks", pages.Talks)
|
|
e.GET("/posts", pages.Posts)
|
|
e.GET("/posts/:post", pages.Post)
|
|
e.GET("/tools", pages.Tools)
|
|
e.GET("/tools/resize", pages.Resize)
|
|
e.GET("/tools/ssedemo", pages.SSEDemo)
|
|
|
|
// API Routes:
|
|
apiGroup := e.Group("/api")
|
|
apiGroup.GET("/ping", api.Ping)
|
|
apiGroup.GET("/rss", api.RSSFeedHandler)
|
|
apiGroup.GET("/post/copy", api.PostCopy)
|
|
|
|
apiGroup.GET("/sse", func(c echo.Context) error {
|
|
return api.SSE(c)
|
|
})
|
|
|
|
apiGroup.POST("/tools/sendsse", func(c echo.Context) error {
|
|
return api.SSEDemoSend(c)
|
|
})
|
|
|
|
apiGroup.POST("/tools/resize", api.ResizeHandler)
|
|
|
|
// Parse command-line arguments for IP and port
|
|
ip := flag.String("ip", "", "IP address to bind the server to")
|
|
port := flag.String("port", "3000", "Port to bind the server to")
|
|
flag.Parse()
|
|
|
|
// Start server with HTTP/2 support
|
|
s := &http.Server{
|
|
Addr: fmt.Sprintf("%s:%s", *ip, *port),
|
|
Handler: e,
|
|
}
|
|
e.Logger.Fatal(e.StartServer(s))
|
|
lib.LogSuccess.Println("Server started on port", *port)
|
|
}
|