Remove timestamp from error output, allow min/max file count value of 0

This commit is contained in:
Seednode 2023-10-03 15:00:32 -05:00
parent 97f5bddafa
commit 5dc218c60e
4 changed files with 45 additions and 46 deletions

View File

@ -17,7 +17,7 @@ import (
var (
ErrInvalidFileCountRange = errors.New("maximum file count limit must be greater than or equal to minimum file count limit")
ErrInvalidFileCountValue = errors.New("file count limits must be positive integers no greater than 2147483647")
ErrInvalidFileCountValue = errors.New("file count limits must be non-negative integers no greater than 2147483647")
ErrInvalidPort = errors.New("listen port must be an integer between 1 and 65535 inclusive")
ErrNoMediaFound = errors.New("no supported media formats found which match all criteria")
)

View File

@ -7,6 +7,7 @@ package cmd
import (
"errors"
"fmt"
"io/fs"
"math/big"
"regexp"
@ -229,8 +230,6 @@ func hasSupportedFiles(path string, formats *types.Types) (bool, error) {
}
func walkPath(path string, fileChannel chan<- string, stats *scanStatsChannels, formats *types.Types) error {
var wg sync.WaitGroup
errorChannel := make(chan error)
done := make(chan bool, 1)
@ -239,8 +238,7 @@ func walkPath(path string, fileChannel chan<- string, stats *scanStatsChannels,
return err
}
var directories = 0
var files = 0
var directories, files = 0, 0
for _, node := range nodes {
if node.IsDir() {
@ -261,50 +259,48 @@ func walkPath(path string, fileChannel chan<- string, stats *scanStatsChannels,
skipFiles = true
}
for _, node := range nodes {
fullPath := filepath.Join(path, node.Name())
var wg sync.WaitGroup
switch {
case node.IsDir() && Recursive:
wg.Add(1)
go func() {
defer wg.Done()
for _, node := range nodes {
wg.Add(1)
go func() {
defer func() {
wg.Done()
}()
err := walkPath(fullPath, fileChannel, stats, formats)
if err != nil {
errorChannel <- err
go func(node fs.DirEntry) {
defer wg.Done()
return
fullPath := filepath.Join(path, node.Name())
switch {
case node.IsDir() && Recursive:
err := walkPath(fullPath, fileChannel, stats, formats)
if err != nil {
errorChannel <- err
return
}
case !node.IsDir() && !skipFiles:
path, err := normalizePath(fullPath)
if err != nil {
errorChannel <- err
return
}
if formats.Validate(path) || Fallback {
fileChannel <- path
stats.filesMatched <- 1
return
}
stats.filesSkipped <- 1
}
}()
case !node.IsDir() && !skipFiles:
wg.Add(1)
go func() {
defer func() {
wg.Done()
}()
path, err := normalizePath(fullPath)
if err != nil {
errorChannel <- err
return
}
if formats.Validate(path) || Fallback {
fileChannel <- path
stats.filesMatched <- 1
return
}
stats.filesSkipped <- 1
}()
}(node)
}
}
}()
go func() {
wg.Wait()

View File

@ -12,7 +12,7 @@ import (
)
const (
ReleaseVersion string = "2.3.2"
ReleaseVersion string = "2.4.0"
)
var (
@ -52,7 +52,7 @@ var (
Args: cobra.MinimumNArgs(1),
PreRunE: func(cmd *cobra.Command, args []string) error {
switch {
case MaxFileCount < 1 || MinFileCount < 1 || MaxFileCount > math.MaxInt32 || MinFileCount > math.MaxInt32:
case MaxFileCount < 0 || MinFileCount < 0 || MaxFileCount > math.MaxInt32 || MinFileCount > math.MaxInt32:
return ErrInvalidFileCountValue
case MinFileCount > MaxFileCount:
return ErrInvalidFileCountRange
@ -97,7 +97,7 @@ func init() {
rootCmd.Flags().StringVar(&IndexFile, "index-file", "", "path to optional persistent index file")
rootCmd.Flags().BoolVarP(&Info, "info", "i", false, "expose informational endpoints")
rootCmd.Flags().IntVar(&MaxFileCount, "max-file-count", math.MaxInt32, "skip directories with file counts above this value")
rootCmd.Flags().IntVar(&MinFileCount, "min-file-count", 1, "skip directories with file counts below this value")
rootCmd.Flags().IntVar(&MinFileCount, "min-file-count", 0, "skip directories with file counts below this value")
rootCmd.Flags().IntVar(&PageLength, "page-length", 0, "pagination length for info pages")
rootCmd.Flags().IntVarP(&Port, "port", "p", 8080, "port to listen on")
rootCmd.Flags().StringVar(&Prefix, "prefix", "/", "root path for http handlers (for reverse proxying)")

View File

@ -10,6 +10,7 @@ import (
"errors"
"fmt"
"io"
"log"
"net"
"net/http"
"os"
@ -404,6 +405,8 @@ func redirectRoot() httprouter.Handle {
}
func ServePage(args []string) error {
log.SetFlags(0)
timeZone := os.Getenv("TZ")
if timeZone != "" {
var err error