95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package hyphae
|
||
|
||
import (
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"github.com/bouncepaw/mycorrhiza/mimetype"
|
||
)
|
||
|
||
// Index finds all hypha files in the full `path` and saves them to the hypha storage.
|
||
func Index(path string) {
|
||
byNames = make(map[string]ExistingHypha)
|
||
ch := make(chan ExistingHypha, 5)
|
||
|
||
go func(ch chan ExistingHypha) {
|
||
indexHelper(path, 0, ch)
|
||
close(ch)
|
||
}(ch)
|
||
|
||
for foundHypha := range ch {
|
||
switch storedHypha := ByName(foundHypha.CanonicalName()).(type) {
|
||
case *EmptyHypha:
|
||
Insert(foundHypha)
|
||
|
||
case *TextualHypha:
|
||
switch foundHypha := foundHypha.(type) {
|
||
case *TextualHypha: // conflict! overwrite
|
||
storedHypha.mycoFilePath = foundHypha.mycoFilePath
|
||
log.Printf(
|
||
"File collision for hypha ‘%s’, using ‘%s’ rather than ‘%s’\n",
|
||
foundHypha.CanonicalName(),
|
||
foundHypha.TextFilePath(),
|
||
storedHypha.TextFilePath(),
|
||
)
|
||
case *MediaHypha: // no conflict
|
||
Insert(ExtendTextualToMedia(storedHypha, foundHypha.mediaFilePath))
|
||
}
|
||
|
||
case *MediaHypha:
|
||
switch foundHypha := foundHypha.(type) {
|
||
case *TextualHypha: // no conflict
|
||
storedHypha.mycoFilePath = foundHypha.mycoFilePath
|
||
case *MediaHypha: // conflict! overwrite
|
||
storedHypha.mediaFilePath = foundHypha.mediaFilePath
|
||
log.Printf(
|
||
"File collision for hypha ‘%s’, using ‘%s’ rather than ‘%s’\n",
|
||
foundHypha.CanonicalName(),
|
||
foundHypha.MediaFilePath(),
|
||
storedHypha.MediaFilePath(),
|
||
)
|
||
}
|
||
}
|
||
}
|
||
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 media and counting them is
|
||
// up to the caller.
|
||
func indexHelper(path string, nestLevel uint, ch chan ExistingHypha) {
|
||
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 folders for it has an administrative importance!
|
||
if node.IsDir() && IsValidName(node.Name()) && node.Name() != ".git" {
|
||
indexHelper(filepath.Join(path, node.Name()), nestLevel+1, ch)
|
||
continue
|
||
}
|
||
|
||
var (
|
||
hyphaPartPath = filepath.Join(path, node.Name())
|
||
hyphaName, isText, skip = mimetype.DataFromFilename(hyphaPartPath)
|
||
)
|
||
if !skip {
|
||
if isText {
|
||
ch <- &TextualHypha{
|
||
canonicalName: hyphaName,
|
||
mycoFilePath: hyphaPartPath,
|
||
}
|
||
} else {
|
||
ch <- &MediaHypha{
|
||
canonicalName: hyphaName,
|
||
mycoFilePath: "",
|
||
mediaFilePath: hyphaPartPath,
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|