mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 16:34:02 +01:00
* Move main to cmd * Move api to internal * Move logger and manager to internal * Move shell hiding code to separate package * Decouple job from desktop and utils * Decouple session from config * Move static into internal * Decouple config from dlna * Move desktop to internal * Move dlna to internal * Decouple remaining packages from config * Move config into internal * Move jsonschema and paths to models * Make ffmpeg functions private * Move file utility methods into fsutil package * Move symwalk into fsutil * Move single-use util functions into client package * Move slice functions to separate packages * Add env var to suppress windowsgui arg * Move hash functions into separate package * Move identify to internal * Move autotag to internal * Touch UI when generating backend
59 lines
1,019 B
Go
59 lines
1,019 B
Go
package manager
|
|
|
|
import (
|
|
"archive/zip"
|
|
"strings"
|
|
|
|
"github.com/stashapp/stash/internal/manager/config"
|
|
"github.com/stashapp/stash/pkg/file"
|
|
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
)
|
|
|
|
func walkGalleryZip(path string, walkFunc func(file *zip.File) error) error {
|
|
readCloser, err := zip.OpenReader(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer readCloser.Close()
|
|
|
|
excludeImgRegex := generateRegexps(config.GetInstance().GetImageExcludes())
|
|
|
|
for _, f := range readCloser.File {
|
|
if f.FileInfo().IsDir() {
|
|
continue
|
|
}
|
|
|
|
if strings.Contains(f.Name, "__MACOSX") {
|
|
continue
|
|
}
|
|
|
|
if !isImage(f.Name) {
|
|
continue
|
|
}
|
|
|
|
if matchFileRegex(file.ZipFile(path, f).Path(), excludeImgRegex) {
|
|
continue
|
|
}
|
|
|
|
err := walkFunc(f)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func countImagesInZip(path string) int {
|
|
ret := 0
|
|
err := walkGalleryZip(path, func(file *zip.File) error {
|
|
ret++
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
logger.Warnf("Error while walking gallery zip: %v", err)
|
|
}
|
|
|
|
return ret
|
|
}
|