128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package sign
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/fs"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/proxy"
|
|
"github.com/gofiber/fiber/v2/utils"
|
|
"github.com/samber/lo"
|
|
"github.com/spf13/viper"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
func makeHypertextResponse(c *fiber.Ctx, upstream *UpstreamInstance) error {
|
|
timeout := time.Duration(viper.GetInt64("performance.network_timeout")) * time.Millisecond
|
|
return proxy.Do(c, upstream.MakeURI(c), &fasthttp.Client{
|
|
ReadTimeout: timeout,
|
|
WriteTimeout: timeout,
|
|
})
|
|
}
|
|
|
|
func makeFileResponse(c *fiber.Ctx, upstream *UpstreamInstance) error {
|
|
uri, queries := upstream.GetRawURI()
|
|
root := http.Dir(uri)
|
|
|
|
method := c.Method()
|
|
|
|
// We only serve static assets for GET and HEAD methods
|
|
if method != fiber.MethodGet && method != fiber.MethodHead {
|
|
return c.Next()
|
|
}
|
|
|
|
// Strip prefix
|
|
prefix := c.Route().Path
|
|
path := strings.TrimPrefix(c.Path(), prefix)
|
|
if !strings.HasPrefix(path, "/") {
|
|
path = "/" + path
|
|
}
|
|
|
|
// Add prefix
|
|
if queries.Get("prefix") != "" {
|
|
path = queries.Get("prefix") + path
|
|
}
|
|
|
|
if len(path) > 1 {
|
|
path = utils.TrimRight(path, '/')
|
|
}
|
|
|
|
file, err := root.Open(path)
|
|
if err != nil && errors.Is(err, fs.ErrNotExist) {
|
|
if queries.Get("suffix") != "" {
|
|
file, err = root.Open(path + queries.Get("suffix"))
|
|
}
|
|
if err != nil && queries.Get("fallback") != "" {
|
|
file, err = root.Open(queries.Get("fallback"))
|
|
}
|
|
}
|
|
if err != nil {
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
return fiber.ErrNotFound
|
|
}
|
|
return fmt.Errorf("failed to open: %w", err)
|
|
}
|
|
|
|
stat, err := file.Stat()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to stat: %w", err)
|
|
}
|
|
|
|
// Serve index if path is directory
|
|
if stat.IsDir() {
|
|
indexFile := lo.Ternary(len(queries.Get("index")) > 0, queries.Get("index"), "index.html")
|
|
indexPath := utils.TrimRight(path, '/') + indexFile
|
|
index, err := root.Open(indexPath)
|
|
if err == nil {
|
|
indexStat, err := index.Stat()
|
|
if err == nil {
|
|
file = index
|
|
stat = indexStat
|
|
}
|
|
}
|
|
}
|
|
|
|
c.Status(fiber.StatusOK)
|
|
|
|
modTime := stat.ModTime()
|
|
contentLength := int(stat.Size())
|
|
|
|
// Set Content-Type header
|
|
if queries.Get("charset") == "" {
|
|
c.Type(filepath.Ext(stat.Name()))
|
|
} else {
|
|
c.Type(filepath.Ext(stat.Name()), queries.Get("charset"))
|
|
}
|
|
|
|
// Set Last-Modified header
|
|
if !modTime.IsZero() {
|
|
c.Set(fiber.HeaderLastModified, modTime.UTC().Format(http.TimeFormat))
|
|
}
|
|
|
|
if method == fiber.MethodGet {
|
|
maxAge, err := strconv.Atoi(queries.Get("maxAge"))
|
|
if lo.Ternary(err != nil, maxAge, 0) > 0 {
|
|
c.Set(fiber.HeaderCacheControl, "public, max-age="+queries.Get("maxAge"))
|
|
}
|
|
c.Response().SetBodyStream(file, contentLength)
|
|
return nil
|
|
}
|
|
if method == fiber.MethodHead {
|
|
c.Request().ResetBody()
|
|
c.Response().SkipBody = true
|
|
c.Response().Header.SetContentLength(contentLength)
|
|
if err := file.Close(); err != nil {
|
|
return fmt.Errorf("failed to close: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
return fiber.ErrNotFound
|
|
}
|