Compare commits
5 commits
Author | SHA1 | Date | |
---|---|---|---|
|
cba38255a5 | ||
|
13ab5b1902 | ||
|
873110f5ce | ||
|
97de2fd06c | ||
|
1b06291e8d |
2 changed files with 19 additions and 5 deletions
22
tree.go
22
tree.go
|
@ -3,6 +3,7 @@ package zord_tree
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"cmp"
|
||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
@ -13,10 +14,10 @@ import (
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"slices"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
abcex "g.arns.lt/zordsdavini/abcex/v4"
|
abcex "g.arns.lt/zordsdavini/abcex/v4"
|
||||||
"golang.org/x/exp/slices"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
|
@ -24,10 +25,10 @@ type Config struct {
|
||||||
AttachmentDirName string
|
AttachmentDirName string
|
||||||
CustomMeta map[string]func() string
|
CustomMeta map[string]func() string
|
||||||
Excludes []string
|
Excludes []string
|
||||||
Apps map[string]appCallback
|
Apps map[string]AppCallback
|
||||||
}
|
}
|
||||||
|
|
||||||
type appCallback func(dir string, content string, info os.FileInfo) (string, error)
|
type AppCallback func(dir string, content string, info os.FileInfo) (string, error)
|
||||||
|
|
||||||
type File struct {
|
type File struct {
|
||||||
Id string
|
Id string
|
||||||
|
@ -150,12 +151,14 @@ func NewConfig(
|
||||||
attachmentDirName string,
|
attachmentDirName string,
|
||||||
customMeta map[string]func() string,
|
customMeta map[string]func() string,
|
||||||
excludes []string,
|
excludes []string,
|
||||||
|
apps map[string]AppCallback,
|
||||||
) Config {
|
) Config {
|
||||||
return Config{
|
return Config{
|
||||||
ReadableFormats: readableFormats,
|
ReadableFormats: readableFormats,
|
||||||
AttachmentDirName: attachmentDirName,
|
AttachmentDirName: attachmentDirName,
|
||||||
CustomMeta: customMeta,
|
CustomMeta: customMeta,
|
||||||
Excludes: excludes,
|
Excludes: excludes,
|
||||||
|
Apps: apps,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -260,7 +263,7 @@ func moveAttachments(dir string, config Config) (map[string]string, error) {
|
||||||
return attachmentRegistry, err
|
return attachmentRegistry, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func applyApp(app appCallback, dir string, config Config) error {
|
func applyApp(app AppCallback, dir string, config Config) error {
|
||||||
err := filepath.Walk(dir, func(fullPath string, info os.FileInfo, e error) error {
|
err := filepath.Walk(dir, func(fullPath string, info os.FileInfo, e error) error {
|
||||||
if e != nil {
|
if e != nil {
|
||||||
return e
|
return e
|
||||||
|
@ -293,7 +296,7 @@ func applyApp(app appCallback, dir string, config Config) error {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
content, err = app(dir, content, info)
|
content, err = app(path.Dir(fullPath), content, info)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -566,7 +569,16 @@ FILE_LOOP:
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return tree, err
|
return tree, err
|
||||||
}
|
}
|
||||||
|
if nextFile.Id == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
tree.Files = append(tree.Files, nextFile)
|
tree.Files = append(tree.Files, nextFile)
|
||||||
|
slices.SortFunc(tree.Files, func(a, b File) int {
|
||||||
|
if len(a.Id) == len(b.Id) {
|
||||||
|
return strings.Compare(a.Id, b.Id)
|
||||||
|
}
|
||||||
|
return cmp.Compare(len(a.Id), len(b.Id))
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
return tree, nil
|
return tree, nil
|
||||||
|
|
|
@ -22,6 +22,7 @@ func prepare(t *testing.T) Config {
|
||||||
"__a",
|
"__a",
|
||||||
make(map[string]func() string),
|
make(map[string]func() string),
|
||||||
[]string{"exclude_[0-9]+.bin"},
|
[]string{"exclude_[0-9]+.bin"},
|
||||||
|
map[string]AppCallback{},
|
||||||
)
|
)
|
||||||
|
|
||||||
err = PopulateTree("./testdata/sunny1", []string{}, config)
|
err = PopulateTree("./testdata/sunny1", []string{}, config)
|
||||||
|
@ -38,6 +39,7 @@ func TestFromNotExistingDirectory(t *testing.T) {
|
||||||
"__a",
|
"__a",
|
||||||
make(map[string]func() string),
|
make(map[string]func() string),
|
||||||
[]string{},
|
[]string{},
|
||||||
|
map[string]AppCallback{},
|
||||||
)
|
)
|
||||||
|
|
||||||
_, err := BuildTree("./testing/i_dont_exist", []string{}, config)
|
_, err := BuildTree("./testing/i_dont_exist", []string{}, config)
|
||||||
|
|
Loading…
Add table
Reference in a new issue