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
|
package cli
import (
"context"
"flag"
"fmt"
"os"
"codeberg.org/snonux/goprecords/internal/storage"
)
func runImport(args []string) error {
fs := flag.NewFlagSet("import", flag.ExitOnError)
statsDir := fs.String("stats-dir", "", "Directory containing .records files (required)")
dbPath := fs.String("db", "goprecords.db", "SQLite database path")
if err := fs.Parse(args); err != nil {
return err
}
if *statsDir == "" {
fmt.Fprintln(os.Stderr, "import: missing required flag: -stats-dir")
fs.Usage()
return fmt.Errorf("missing -stats-dir")
}
ctx := context.Background()
db, err := storage.Open(ctx, *dbPath)
if err != nil {
return fmt.Errorf("open db: %w", err)
}
defer db.Close()
if err := storage.CreateSchema(ctx, db); err != nil {
return fmt.Errorf("schema: %w", err)
}
if err := storage.ImportFromDir(ctx, db, *statsDir); err != nil {
return fmt.Errorf("import: %w", err)
}
fmt.Fprintf(os.Stderr, "imported %s into %s\n", *statsDir, *dbPath)
return nil
}
|
