handle static routes

2hmad · 2hmad · commit 7dda68f47090 · 2023-04-10T00:47:33.000+02:00
diff --git a/router.go b/router.go
@@ -1,8 +1,10 @@
 package routing
 
 import (
+	"fmt"
 	"github.com/valyala/fasthttp"
 	"strings"
+	"time"
 )
 
 type Handler func(ctx *Context) error
@@ -19,6 +21,14 @@ type Router struct {
 	middlewares     map[string][]Middleware
 }
 
+type Static struct {
+	Root          string
+	Compress      bool
+	ByteRange     bool
+	IndexName     string
+	CacheDuration time.Duration
+}
+
 func New() *Router {
 	return &Router{
 		routes:      make(map[string][]*Route),
@@ -132,3 +142,60 @@ func (r *Route) match(path string) (bool, map[string]string) {
 
 	return true, params
 }
+
+func (options *Static) notFoundHandler(ctx *fasthttp.RequestCtx) {
+	ctx.SetStatusCode(fasthttp.StatusNotFound)
+	ctx.SetContentType("text/plain; charset=utf-8")
+	_, err := fmt.Fprintf(ctx, "404 Not Found")
+	if err != nil {
+		return
+	}
+}
+
+func (options *Static) pathRewrite(ctx *fasthttp.RequestCtx) []byte {
+	path := ctx.Path()
+
+	if len(path) > 1 && path[len(path)-1] == '/' {
+		path = path[:len(path)-1]
+	}
+
+	// Remove the last part of the path
+	parts := strings.Split(string(path), "/")
+	if len(parts) > 1 {
+		parts = parts[:len(parts)-1]
+	}
+	path = []byte(strings.Join(parts, "/"))
+
+	if options.IndexName != "" {
+		// Append the index file name to the path
+		path = append(path, '/')
+		path = append(path, options.IndexName...)
+	}
+
+	return path
+}
+
+func (r *Router) Static(prefix, root string, options *Static) {
+	if options == nil {
+		options = &Static{}
+	}
+	if options.Root == "" {
+		options.Root = root
+	}
+	fs := fasthttp.FS{
+		Root:               options.Root,
+		IndexNames:         []string{options.IndexName},
+		PathRewrite:        options.pathRewrite,
+		GenerateIndexPages: false,
+		Compress:           options.Compress,
+		AcceptByteRange:    options.ByteRange,
+		CacheDuration:      options.CacheDuration,
+		PathNotFound:       options.notFoundHandler, // Set custom error handler for undefined routes
+	}
+
+	r.Get(prefix, func(c *Context) error {
+		fsHandler := fs.NewRequestHandler()
+		fsHandler(c.RequestCtx)
+		return nil
+	})
+}
diff --git a/router_test.go b/router_test.go
@@ -91,3 +91,19 @@ func TestContext_SetHeader(t *testing.T) {
 		return
 	}
 }
+
+func TestRouter_Static(t *testing.T) {
+	router := New()
+
+	router.Static("/", "./static", &Static{
+		Compress:      true,
+		ByteRange:     false,
+		IndexName:     "index.html",
+		CacheDuration: 24 * time.Hour,
+	})
+
+	err := fasthttp.ListenAndServe(":8083", RouterHandler(router))
+	if err != nil {
+		return
+	}
+}
diff --git a/static/index.html b/static/index.html
@@ -0,0 +1,9 @@
+<html>
+<head>
+    <title>My First Web Page</title>
+</head>
+<body>
+    <h1>My First Web Page</h1>
+    <p>My first paragraph.</p>
+</body>
+</html>
