1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
package main
import (
"flag"
"log/slog"
"os"
"sync"
"github.com/SayaAndy/saya-today-article-metadata-add/config"
"github.com/SayaAndy/saya-today-article-metadata-add/internal/frontmatter"
"github.com/SayaAndy/saya-today-article-metadata-add/internal/storage"
)
var configPath = flag.String("c", "config.json", "Path to the configuration file")
func main() {
flag.Parse()
cfg := &config.Config{}
if err := config.LoadConfig(*configPath, cfg); err != nil {
slog.Error("fail to load configuration", slog.String("error", err.Error()))
os.Exit(1)
}
slog.SetLogLoggerLevel(cfg.LogLevel)
slog.Info("starting metadata extractor...")
storageClient, err := storage.NewStorageClientMap[cfg.Storage.Type](&cfg.Storage, &cfg.DraftMode)
if err != nil {
slog.Error("fail to initialize input client", slog.String("error", err.Error()))
os.Exit(1)
}
generalLogger := slog.With(
slog.String("storage_type", cfg.Storage.Type),
)
generalLogger.Info("initialized storage client")
files, err := storageClient.Scan()
if err != nil {
generalLogger.Error("fail to scan input files", slog.String("error", err.Error()))
os.Exit(1)
}
generalLogger.Info("scanned files", slog.Int("file_count", len(files)))
semaphore := make(chan struct{}, cfg.MaxConcurrentJobs)
var wg sync.WaitGroup
wg.Add(len(files))
for i, file := range files {
semaphore <- struct{}{}
go func(index int, inputName string) {
defer wg.Done()
defer func() { <-semaphore }()
if cfg.DraftMode.Enabled && !storageClient.CompareDraftAndProd(inputName) {
generalLogger.Debug("skip a draft because prod object is identical to it", slog.String("file", inputName))
return
}
generalLogger.Debug("processing a file", slog.String("file", inputName))
reader, sz, err := storageClient.GetReader(inputName)
if err != nil {
generalLogger.Warn("fail to get reader for a file", slog.String("file", inputName), slog.String("error", err.Error()))
return
}
defer reader.Close()
content := make([]byte, sz)
ln, err := reader.Read(content)
if err != nil {
generalLogger.Warn("fail to read content from a file", slog.String("file", inputName), slog.String("error", err.Error()))
return
}
generalLogger.Debug("read content from a file",
slog.String("file", inputName),
slog.Int64("expected_size", sz),
slog.Int("output_size", ln))
metadata, _, err := frontmatter.ParseFrontmatter(content)
if err != nil {
generalLogger.Warn("fail to parse frontmatter of a file", slog.String("file", inputName), slog.String("error", err.Error()))
return
}
if metadata == nil {
generalLogger.Info("skip a file due to it not having metadata", slog.String("file", inputName))
return
}
if err = storageClient.WriteMetadata(inputName, metadata); err != nil {
generalLogger.Warn("fail to write metadata to a file", slog.String("file", inputName), slog.String("error", err.Error()))
}
}(i, file)
}
wg.Wait()
}
|
