Merge pull request #17 from xEsk/main

Implemented a new NoCompress option

Author: NoOne7135

src/app/app.go

@@ -55,6 +55,16 @@ func NewApp(params *param.Params) App {
 	return App{params: params, server: nil, cache: cache}
 }
 
+func (app *App) ShouldSkipCompression(filePath string) bool {
+	ext := strings.ToLower(filepath.Ext(filePath))
+	for _, blocked := range app.params.NoCompress {
+		if strings.ToLower(blocked) == ext {
+			return true
+		}
+	}
+	return false
+}
+
 func (app *App) CompressFiles() {
 	if !app.params.Gzip && !app.params.Brotli {
 		return
@@ -73,6 +83,10 @@ func (app *App) CompressFiles() {
 			return nil
 		}
 
+		if app.ShouldSkipCompression(filePath) {
+			return nil
+		}
+
 		if info.Size() > app.params.Threshold {
 			data, _ := os.ReadFile(filePath)
 
@@ -231,6 +245,14 @@ func (app *App) HandlerFuncNew(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	if r.Header.Get("Range") != "" || app.ShouldSkipCompression(requestedPath) {
+		if responseItem.ContentType != "" {
+			w.Header().Set("Content-Type", responseItem.ContentType)
+		}
+		http.ServeContent(w, r, responseItem.Name, responseItem.ModTime, bytes.NewReader(responseItem.Content))
+		return
+	}
+
 	if slices.Contains(app.params.IgnoreCacheControlPaths, r.URL.Path) || path.Ext(responseItem.Name) == ".html" {
 		w.Header().Set("Cache-Control", "no-store")
 	} else {

src/param/param.go

@@ -80,6 +80,11 @@ var Flags = []cli.Flag{
 		Name:    "log-pretty",
 		Value:   false,
 	},
+	&cli.StringSliceFlag{
+		EnvVars: []string{"NO_COMPRESS"},
+		Name:    "no-compress",
+		Value:   nil,
+	},
 }
 
 type Params struct {
@@ -96,6 +101,7 @@ type Params struct {
 	CacheBuffer             int
 	Logger                  bool
 	LogPretty               bool
+	NoCompress              []string
 	//DirectoryListing        bool
 }
 
@@ -119,6 +125,7 @@ func ContextToParams(c *cli.Context) (*Params, error) {
 		CacheBuffer:             c.Int("cache-buffer"),
 		Logger:                  c.Bool("logger"),
 		LogPretty:               c.Bool("log-pretty"),
+		NoCompress:              c.StringSlice("no-compress"),
 		//DirectoryListing:        c.Bool("directory-listing"),
 	}, nil
 }