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.
obsidian-pipeline/main.go

154 lines
3.3 KiB
Go

2 years ago
package main
import (
"flag"
"fmt"
"io/fs"
"os"
// "path/filepath"
"strings"
"go.uber.org/zap"
)
func NewAttachmentMover() *AttachmentMover {
var am AttachmentMover
l, _ := zap.NewProduction()
am.L = l
am.Attachments = make(map[string]bool)
am.Posts = make([]string, 0)
return &am
}
type AttachmentMover struct {
Source, Target string
Attachments map[string]bool
Posts []string
L *zap.Logger
}
func (am *AttachmentMover) Walk() error {
notesRoot := os.DirFS(am.Source)
blogRoot := os.DirFS(am.Target)
err := fs.WalkDir(notesRoot, ".", am.findAttachments)
if err != nil {
return fmt.Errorf("error scanning for attachments: %q", err)
}
err = fs.WalkDir(notesRoot, ".", am.findNotes)
if err != nil {
return fmt.Errorf("error scanning vault for posts: %q", err)
}
err = fs.WalkDir(blogRoot, ".", am.findPosts)
if err != nil {
return fmt.Errorf("error scanning blog for posts: %q", err)
}
2 years ago
return nil
}
func (am *AttachmentMover) findNotes(path string, d fs.DirEntry, err error) error {
2 years ago
if err != nil {
return err
}
if d.IsDir() {
return nil
}
walkLogger := am.L.Named("FindNotes")
2 years ago
if strings.HasSuffix(path, ".md") && strings.Contains(path, "blog/published") {
walkLogger.Info("found blog post to publish, adding to index", zap.String("path", path))
am.Attachments[path] = true
}
return nil
}
func (am *AttachmentMover) findAttachments(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
2 years ago
}
if d.IsDir() {
return nil
}
walkLogger := am.L.Named("FindAttachments")
if strings.Contains(path, "attachments") {
walkLogger.Info("found attachment file, adding to index", zap.String("path", path))
2 years ago
am.Attachments[path] = true
}
return nil
}
func (am *AttachmentMover) findPosts(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if d.IsDir() {
return nil
}
walkLogger := am.L.Named("FindPosts")
if strings.HasSuffix(path, "index.md") {
walkLogger.Info("found index.md, adding to index", zap.String("path", path))
am.Posts = append(am.Posts, path)
}
return nil
}
2 years ago
func (am *AttachmentMover) Move() error {
moveLogger := am.L.Named("Move")
2 years ago
moveLogger.Info("scanning posts", zap.Strings("posts", am.Posts))
for _, post := range am.Posts {
// log.Printf("scanning %q for attachment links", post)
linkedAttachments, err := extractAttachments(post)
if err != nil {
return fmt.Errorf("could not extract attachment links from %q: %w", post, err)
}
for _, attachment := range linkedAttachments {
moveAttachment(post, attachment)
}
}
return nil
}
func moveAttachment(post, attachment string) error {
return nil
}
func extractAttachments(fn string) ([]string, error) {
attachments := make([]string, 0)
return attachments, nil
}
func main() {
am := NewAttachmentMover()
defer am.L.Sync()
2 years ago
flag.StringVar(&am.Source, "source", "", "source directory containing your vault")
flag.StringVar(&am.Target, "target", "", "target directory containing your hugo site")
flag.Parse()
if am.Source == "" || am.Target == "" {
am.L.Fatal("flags not provided")
}
err := am.Walk()
if err != nil {
am.L.Fatal("error walking blog or notes dir to gather file names", zap.Error(err))
2 years ago
}
err = am.Move()
if err != nil {
am.L.Fatal("error moving notes", zap.Error(err))
2 years ago
}
}