mirror of
https://github.com/wailsapp/wails.git
synced 2025-05-04 07:29:56 +08:00
d4096868e3
* Fix and optimize content type sniffer - Minimize copying and buffering - Ensure it sniffs the full 512-bytes prefix * Fix assorted warnings * Cleanup error formatting - Remove unnecessary formatting calls - Fix invalid format strings - Standardise logging calls * Fix and optimize index fallback method - Pass through non-404 responses correctly - Do not buffer original response * Test content sniffing and index fallback * Update changelog * Remove obsolete check * Add safety checks in sniffer
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package assetserver
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"log/slog"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
HeaderHost = "Host"
|
|
HeaderContentType = "Content-Type"
|
|
HeaderContentLength = "Content-Length"
|
|
HeaderUserAgent = "User-Agent"
|
|
// TODO: Is this needed?
|
|
HeaderCacheControl = "Cache-Control"
|
|
HeaderUpgrade = "Upgrade"
|
|
|
|
WailsUserAgentValue = "wails.io"
|
|
)
|
|
|
|
type assetServerLogger struct{}
|
|
|
|
var assetServerLoggerKey assetServerLogger
|
|
|
|
func ServeFile(rw http.ResponseWriter, filename string, blob []byte) error {
|
|
header := rw.Header()
|
|
header.Set(HeaderContentLength, fmt.Sprintf("%d", len(blob)))
|
|
if mimeType := header.Get(HeaderContentType); mimeType == "" {
|
|
mimeType = GetMimetype(filename, blob)
|
|
header.Set(HeaderContentType, mimeType)
|
|
}
|
|
|
|
rw.WriteHeader(http.StatusOK)
|
|
_, err := io.Copy(rw, bytes.NewReader(blob))
|
|
return err
|
|
}
|
|
|
|
func isWebSocket(req *http.Request) bool {
|
|
upgrade := req.Header.Get(HeaderUpgrade)
|
|
return strings.EqualFold(upgrade, "websocket")
|
|
}
|
|
|
|
func contextWithLogger(ctx context.Context, logger *slog.Logger) context.Context {
|
|
return context.WithValue(ctx, assetServerLoggerKey, logger)
|
|
}
|
|
|
|
func logInfo(ctx context.Context, message string, args ...interface{}) {
|
|
if logger, _ := ctx.Value(assetServerLoggerKey).(*slog.Logger); logger != nil {
|
|
logger.Info(message, args...)
|
|
}
|
|
}
|
|
|
|
func logError(ctx context.Context, message string, args ...interface{}) {
|
|
if logger, _ := ctx.Value(assetServerLoggerKey).(*slog.Logger); logger != nil {
|
|
logger.Error(message, args...)
|
|
}
|
|
}
|