-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathnewmain.go
82 lines (67 loc) · 1.5 KB
/
newmain.go
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
package main
import (
"context"
"database/sql"
_ "embed"
"flag"
"fmt"
"log"
"log/slog"
"os"
"path/filepath"
_ "github.com/joho/godotenv/autoload"
_ "github.com/mattn/go-sqlite3"
"github.com/bupd/digital-wellbeing/internal/database"
)
var loggerLevels = map[string]slog.Level{
"debug": slog.LevelDebug,
"info": slog.LevelInfo,
"warn": slog.LevelWarn,
"error": slog.LevelError,
}
func run() error {
ctx := context.Background()
home, err := os.UserHomeDir()
if err != nil {
slog.Error(fmt.Sprintf("Cannot find user home dir: %v", err))
return err
}
db, err := sql.Open("sqlite3", filepath.Join(home, ".digital-wellbeing/data.db"))
if err != nil {
slog.Error(fmt.Sprintf("Cannot open Database: %v", err))
return err
}
defer db.Close()
queries := database.New(db)
tx, err := db.BeginTx(ctx, nil)
users, err := queries.WithTx(tx).ListUsers(ctx)
if err != nil {
return err
}
log.Println(users)
return nil
}
type opts struct {
logger slog.Level
}
func main() {
var logLevel string
flag.StringVar(&logLevel, "log-level", "debug", "sets logger level")
flag.Parse()
loglevel, ok := loggerLevels[logLevel]
if !ok {
log.Fatalf("invalid log level you dumb shit: %v", logLevel)
}
opts := &slog.HandlerOptions{
Level: loglevel,
}
handler := slog.NewJSONHandler(os.Stdout, opts)
logger := slog.New(handler)
logger.Debug("Debug message")
logger.Info("Info message")
logger.Warn("Warning message")
logger.Error("Error message")
if err := run(); err != nil {
log.Fatal(err)
}
}