mirror of
https://github.com/osmarks/mycorrhiza.git
synced 2024-10-30 03:36:16 +00:00
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package hyphae
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/bouncepaw/mycorrhiza/mimetype"
|
|
"github.com/bouncepaw/mycorrhiza/util"
|
|
)
|
|
|
|
// Index finds all hypha files in the full `path` and saves them to the hypha storage.
|
|
func Index(path string) {
|
|
byNames = make(map[string]*Hypha)
|
|
ch := make(chan *Hypha, 5)
|
|
|
|
go func(ch chan *Hypha) {
|
|
indexHelper(path, 0, ch)
|
|
close(ch)
|
|
}(ch)
|
|
|
|
for h := range ch {
|
|
// At this time it is safe to ignore the mutex, because there is only one worker.
|
|
if oh := ByName(h.Name); oh.Exists {
|
|
oh.MergeIn(h)
|
|
} else {
|
|
h.Insert()
|
|
}
|
|
}
|
|
|
|
log.Println("Indexed", Count(), "hyphae")
|
|
}
|
|
|
|
// indexHelper finds all hypha files in the full `path` and sends them to the channel. Handling of duplicate entries and attachment and counting them is up to the caller.
|
|
func indexHelper(path string, nestLevel uint, ch chan *Hypha) {
|
|
nodes, err := os.ReadDir(path)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for _, node := range nodes {
|
|
// If this hypha looks like it can be a hypha path, go deeper. Do not touch the .git and static folders for they have an administrative importance!
|
|
if node.IsDir() &&
|
|
util.IsCanonicalName(node.Name()) &&
|
|
node.Name() != ".git" &&
|
|
!(nestLevel == 0 && node.Name() == "static") {
|
|
indexHelper(filepath.Join(path, node.Name()), nestLevel+1, ch)
|
|
continue
|
|
}
|
|
|
|
var (
|
|
hyphaPartPath = filepath.Join(path, node.Name())
|
|
hyphaName, isText, skip = mimetype.DataFromFilename(hyphaPartPath)
|
|
hypha = &Hypha{Name: hyphaName, Exists: true}
|
|
)
|
|
if !skip {
|
|
if isText {
|
|
hypha.TextPath = hyphaPartPath
|
|
} else {
|
|
hypha.BinaryPath = hyphaPartPath
|
|
}
|
|
ch <- hypha
|
|
}
|
|
}
|
|
}
|