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
|
package cli
import (
"context"
"flag"
"fmt"
"os"
"time"
"codeberg.org/snonux/goprecords/internal/storage"
)
func runExclude(args []string) error {
fs := flag.NewFlagSet("exclude", flag.ExitOnError)
dbPath := fs.String("db", "goprecords.db", "SQLite database path")
reason := fs.String("reason", "", "Reason for exclusion")
if err := fs.Parse(args); err != nil {
return err
}
if fs.NArg() < 1 {
fmt.Fprintln(os.Stderr, "exclude: hostname required")
fs.Usage()
return fmt.Errorf("missing hostname")
}
host := fs.Arg(0)
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.AddExcludedHost(ctx, db, host, *reason); err != nil {
return err
}
fmt.Fprintf(os.Stderr, "excluded host %q from alerts\n", host)
return nil
}
func runUnexclude(args []string) error {
fs := flag.NewFlagSet("unexclude", flag.ExitOnError)
dbPath := fs.String("db", "goprecords.db", "SQLite database path")
if err := fs.Parse(args); err != nil {
return err
}
if fs.NArg() < 1 {
fmt.Fprintln(os.Stderr, "unexclude: hostname required")
fs.Usage()
return fmt.Errorf("missing hostname")
}
host := fs.Arg(0)
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.RemoveExcludedHost(ctx, db, host); err != nil {
return err
}
fmt.Fprintf(os.Stderr, "removed host %q from exclusion list\n", host)
return nil
}
func runListExcluded(args []string) error {
fs := flag.NewFlagSet("list-excluded", flag.ExitOnError)
dbPath := fs.String("db", "goprecords.db", "SQLite database path")
if err := fs.Parse(args); err != nil {
return err
}
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)
}
hosts, err := storage.LoadExcludedHosts(ctx, db)
if err != nil {
return err
}
if len(hosts) == 0 {
fmt.Println("no excluded hosts")
return nil
}
for _, h := range hosts {
t := time.Unix(h.ExcludedAt, 0).UTC().Format("2006-01-02")
if h.Reason != "" {
fmt.Printf("%-40s excluded=%s reason=%s\n", h.Host, t, h.Reason)
} else {
fmt.Printf("%-40s excluded=%s\n", h.Host, t)
}
}
return nil
}
|
