You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"io/fs"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"code.ndumas.com/ndumas/obsidian-pipeline/fsm"
|
|
)
|
|
|
|
func main() {
|
|
var (
|
|
source, target, attachmentsDir, blogDir string
|
|
dev bool
|
|
)
|
|
l, _ := zap.NewProduction()
|
|
|
|
flag.BoolVar(&dev, "dev", false, "developer mode")
|
|
flag.StringVar(&source, "source", "", "source directory containing your vault")
|
|
flag.StringVar(&target, "target", "", "target directory containing your hugo site")
|
|
flag.StringVar(&attachmentsDir, "attachments", "", "directory containing your vault's attachments")
|
|
flag.StringVar(&blogDir, "blog", "", "vault directory containing blog posts to-be-published")
|
|
|
|
flag.Parse()
|
|
|
|
fileNames := make(chan string)
|
|
|
|
walkFunc := func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if d.IsDir() {
|
|
return nil
|
|
}
|
|
fileNames <- path
|
|
|
|
return nil
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
go func() {
|
|
defer wg.Done()
|
|
wg.Add(1)
|
|
for fn := range fileNames {
|
|
if !strings.HasSuffix(fn, ".md") {
|
|
continue
|
|
}
|
|
m := fsm.NewStateMachine(&fsm.NoteFound)
|
|
ctx := context.WithValue(context.Background(), "note", fn)
|
|
_, err := m.Transition(ctx, "CopyPost")
|
|
if err != nil {
|
|
l.Fatal("could not transition from NoteFound to CopyPost", zap.Error(err))
|
|
}
|
|
}
|
|
}()
|
|
|
|
root := os.DirFS(source)
|
|
err := fs.WalkDir(root, ".", walkFunc)
|
|
if err != nil {
|
|
l.Fatal("error walking for files", zap.Error(err))
|
|
}
|
|
close(fileNames)
|
|
|
|
wg.Wait()
|
|
}
|