57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package dirwalk
|
|
|
|
import (
|
|
"github.com/maddalax/htmgo/framework/h"
|
|
"io/fs"
|
|
"os"
|
|
"slices"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type Page struct {
|
|
RoutePath string
|
|
FilePath string
|
|
Parts []string
|
|
}
|
|
|
|
func WalkPages(dir string, system fs.FS) []*Page {
|
|
pages := make([]*Page, 0)
|
|
fs.WalkDir(system, dir, func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
name := d.Name()
|
|
if !d.IsDir() && (strings.HasSuffix(name, ".md") || strings.HasSuffix(name, ".go")) {
|
|
fullPath := strings.Replace(path, dir, "", 1)
|
|
fullPath = strings.TrimSuffix(fullPath, ".md")
|
|
pages = append(pages, &Page{
|
|
RoutePath: fullPath,
|
|
FilePath: path,
|
|
Parts: h.Filter(strings.Split(fullPath, string(os.PathSeparator)), func(item string) bool {
|
|
return item != ""
|
|
}),
|
|
})
|
|
}
|
|
return nil
|
|
})
|
|
|
|
var getRouteOrder = func(page *Page) int {
|
|
fileName := page.Parts[len(page.Parts)-1]
|
|
if len(fileName) > 1 && fileName[1] == '_' {
|
|
num, err := strconv.ParseInt(fileName[0:1], 10, 64)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
page.Parts[len(page.Parts)-1] = fileName[2:]
|
|
return int(num)
|
|
}
|
|
return 0
|
|
}
|
|
|
|
slices.SortFunc(pages, func(a *Page, b *Page) int {
|
|
return getRouteOrder(a) - getRouteOrder(b)
|
|
})
|
|
|
|
return pages
|
|
}
|