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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
package main
import (
"context"
"errors"
"log"
"log/slog"
"os"
"os/exec"
"path"
"searchix/internal/config"
"searchix/internal/importer"
"searchix/internal/search"
"slices"
"strings"
"time"
)
const timeout = 30 * time.Minute
func main() {
if _, found := os.LookupEnv("DEBUG"); found {
slog.SetLogLoggerLevel(slog.LevelDebug)
}
cfg, err := config.GetConfig()
if err != nil {
log.Fatal(err)
}
enabledSources := slices.DeleteFunc(cfg.Sources, func(s importer.Source) bool {
return !s.Enable
})
if len(enabledSources) == 0 {
slog.Info("No sources enabled")
return
}
indexer, err := search.NewIndexer(cfg.DataPath)
if err != nil {
log.Fatalf("Failed to create indexer: %v", err)
}
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
var imp importer.Importer
var hadErrors bool
for _, source := range enabledSources {
logger := slog.With("name", source.Name, "importer", source.Type.String())
logger.Debug("starting importer")
switch source.Type {
case importer.ChannelNixpkgs:
imp = importer.NewNixpkgsChannelImporter(source, cfg.DataPath, logger)
case importer.Channel:
imp = importer.NewChannelImporter(source, cfg.DataPath, logger)
default:
log.Printf("unsupported importer type %s", source.Type.String())
continue
}
updated, err := imp.FetchIfNeeded(ctx)
if err != nil {
var exerr *exec.ExitError
if errors.As(err, &exerr) {
lines := strings.Split(strings.TrimSpace(string(exerr.Stderr)), "\n")
for _, line := range lines {
logger.Warn("importer fetch failed", "stderr", line, "status", exerr.ExitCode())
}
} else {
logger.Warn("importer fetch failed", "error", err)
}
hadErrors = true
continue
}
logger.Info("importer fetch succeeded", "updated", updated)
if updated {
hadWarnings, err := imp.Import(ctx, indexer)
if err != nil {
msg := err.Error()
for _, line := range strings.Split(strings.TrimSpace(msg), "\n") {
logger.Error("importer init failed", "error", line)
}
continue
}
if hadWarnings {
logger.Warn("importer succeeded, but with warnings/errors")
} else {
logger.Info("importer succeeded")
}
}
}
err = indexer.Close()
if err != nil {
slog.Error("error closing indexer", "error", err)
}
if hadErrors {
os.RemoveAll(path.Join(cfg.DataPath, "index.bleve"))
defer os.Exit(1)
}
}
|