roulette/cmd/files.go

213 lines
3.6 KiB
Go
Raw Normal View History

2022-09-08 15:57:59 +00:00
/*
Copyright © 2022 Seednode <seednode@seedno.de>
*/
package cmd
import (
"errors"
"fmt"
2022-09-08 15:57:59 +00:00
"math/rand"
"os"
"path/filepath"
"regexp"
"strconv"
2022-09-08 15:57:59 +00:00
"time"
"github.com/h2non/filetype"
2022-09-08 15:57:59 +00:00
)
func getFirstFile(path string) (string, error) {
re := regexp.MustCompile("(.+)([0-9]{3})(\\..+)")
split := re.FindAllStringSubmatch(path, -1)
if len(split) < 1 || len(split[0]) < 3 {
return "", nil
}
base := split[0][1]
number := 1
extension := split[0][3]
fileName := fmt.Sprintf("%v%.3d%v", base, number, extension)
nextFile, err := checkNextFile(fileName)
if err != nil {
return "", err
}
if !nextFile {
return "", nil
}
return fileName, nil
}
func getNextFile(path string) (string, error) {
re := regexp.MustCompile("(.+)([0-9]{3})(\\..+)")
split := re.FindAllStringSubmatch(path, -1)
2022-09-17 18:12:22 +00:00
if len(split) < 1 || len(split[0]) < 3 {
2022-09-17 17:42:25 +00:00
return "", nil
}
base := split[0][1]
number, err := strconv.Atoi(split[0][2])
if err != nil {
2022-09-17 17:42:25 +00:00
return "", err
}
extension := split[0][3]
incremented := number + 1
fileName := fmt.Sprintf("%v%.3d%v", base, incremented, extension)
nextFile, err := checkNextFile(fileName)
if err != nil {
return "", err
}
if !nextFile {
return "", nil
}
return fileName, nil
}
func checkNextFile(path string) (bool, error) {
if _, err := os.Stat(path); err == nil {
return true, nil
} else if errors.Is(err, os.ErrNotExist) {
return false, nil
} else {
return false, err
}
}
func checkIfImage(path string) (bool, error) {
file, err := os.Open(path)
if err != nil {
return false, err
}
defer file.Close()
head := make([]byte, 261)
file.Read(head)
if filetype.IsImage(head) {
return true, nil
}
return false, nil
}
func getFiles(path string) ([]string, error) {
var paths []string
2022-09-08 15:57:59 +00:00
err := filepath.WalkDir(path, func(p string, info os.DirEntry, err error) error {
if info.IsDir() && p != path {
return filepath.SkipDir
} else {
2022-09-08 20:30:51 +00:00
absolutePath, err := filepath.Abs(p)
if err != nil {
return err
}
paths = append(paths, absolutePath)
}
2022-09-08 20:30:51 +00:00
return err
})
2022-09-08 15:57:59 +00:00
if err != nil {
return nil, err
2022-09-08 15:57:59 +00:00
}
return paths, nil
2022-09-08 15:57:59 +00:00
}
2022-09-08 17:12:58 +00:00
func getFilesRecursive(path string) ([]string, error) {
var paths []string
err := filepath.WalkDir(path, func(p string, info os.DirEntry, err error) error {
if !info.IsDir() {
2022-09-08 20:30:51 +00:00
absolutePath, err := filepath.Abs(p)
if err != nil {
return err
}
paths = append(paths, absolutePath)
2022-09-08 17:12:58 +00:00
}
return err
})
if err != nil {
return nil, err
}
2022-09-08 17:12:58 +00:00
return paths, nil
2022-09-08 17:12:58 +00:00
}
func getFileList(paths []string) ([]string, error) {
2022-09-08 17:12:58 +00:00
fileList := []string{}
2022-09-08 15:57:59 +00:00
for i := 0; i < len(paths); i++ {
2022-09-08 17:12:58 +00:00
if Recursive {
f, err := getFilesRecursive(paths[i])
2022-09-08 17:12:58 +00:00
if err != nil {
return nil, err
2022-09-08 17:12:58 +00:00
}
2022-09-08 20:30:51 +00:00
2022-09-08 17:12:58 +00:00
fileList = append(fileList, f...)
} else {
f, err := getFiles(paths[i])
2022-09-08 17:12:58 +00:00
if err != nil {
return nil, err
2022-09-08 17:12:58 +00:00
}
2022-09-08 20:30:51 +00:00
2022-09-08 17:12:58 +00:00
fileList = append(fileList, f...)
}
2022-09-08 15:57:59 +00:00
}
return fileList, nil
2022-09-08 17:12:58 +00:00
}
2022-09-08 15:57:59 +00:00
func pickFile(args []string) (string, error) {
fileList, err := getFileList(args)
if err != nil {
return "", err
}
rand.Seed(time.Now().UnixNano())
rand.Shuffle(len(fileList), func(i, j int) { fileList[i], fileList[j] = fileList[j], fileList[i] })
for i := 0; i < len(fileList); i++ {
filePath := fileList[i]
isImage, err := checkIfImage(filePath)
if err != nil {
return "", err
}
if isImage {
return filePath, nil
}
}
2022-09-08 17:12:58 +00:00
err = errors.New("no images found")
2022-09-08 15:57:59 +00:00
return "", err
2022-09-08 20:30:51 +00:00
}
func normalizePaths(args []string) ([]string, error) {
2022-09-08 20:30:51 +00:00
var paths []string
for i := 0; i < len(args); i++ {
absolutePath, err := filepath.Abs(args[i])
if err != nil {
return nil, err
2022-09-08 20:30:51 +00:00
}
paths = append(paths, absolutePath)
}
return paths, nil
2022-09-08 15:57:59 +00:00
}