onasty: cmd/seed/main.go (b3a528af9e74a124dbb79687ebb116ce8a8c8e8c)

Smirnov Oleksandr refactor: logger set default interface (#131)..., 12 months ago

package main

import (
	"context"
	"fmt"
	"log/slog"
	"os"

	"github.com/olexsmir/onasty/internal/config"
	"github.com/olexsmir/onasty/internal/hasher"
	"github.com/olexsmir/onasty/internal/logger"
	"github.com/olexsmir/onasty/internal/store/psqlutil"
)

func main() {
	if err := run(context.Background()); err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
}

func run(ctx context.Context) error {
	cfg := config.NewConfig()

	if err := logger.SetDefault(cfg.LogLevel, cfg.LogFormat, cfg.LogShowLine); err != nil {
		return err
	}

	psql, err := psqlutil.Connect(ctx, cfg.PostgresDSN)
	if err != nil {
		return err
	}

	userHasher := hasher.NewSHA256Hasher(cfg.PasswordSalt)
	noteHasher := hasher.NewSHA256Hasher(cfg.NotePasswordSalt)

	if err := seedUsers(ctx, userHasher, psql); err != nil {
		return fmt.Errorf("failed to seed users: %w", err)
	}

	slog.Info("Users seeded successfully")

	if err := seedNotes(ctx, noteHasher, psql); err != nil {
		return fmt.Errorf("failed to seed notes: %w", err)
	}

	slog.Info("Notes seeded successfully")

	return nil
}

1	package main
2
3	import (
4	"context"
5	"fmt"
6	"log/slog"
7	"os"
8
9	"github.com/olexsmir/onasty/internal/config"
10	"github.com/olexsmir/onasty/internal/hasher"
11	"github.com/olexsmir/onasty/internal/logger"
12	"github.com/olexsmir/onasty/internal/store/psqlutil"
13	)
14
15	func main() {
16	if err := run(context.Background()); err != nil {
17	fmt.Fprintf(os.Stderr, "error: %v\n", err)
18	os.Exit(1)
19	}
20	}
21
22	func run(ctx context.Context) error {
23	cfg := config.NewConfig()
24
25	if err := logger.SetDefault(cfg.LogLevel, cfg.LogFormat, cfg.LogShowLine); err != nil {
26	return err
27	}
28
29	psql, err := psqlutil.Connect(ctx, cfg.PostgresDSN)
30	if err != nil {
31	return err
32	}
33
34	userHasher := hasher.NewSHA256Hasher(cfg.PasswordSalt)
35	noteHasher := hasher.NewSHA256Hasher(cfg.NotePasswordSalt)
36
37	if err := seedUsers(ctx, userHasher, psql); err != nil {
38	return fmt.Errorf("failed to seed users: %w", err)
39	}
40
41	slog.Info("Users seeded successfully")
42
43	if err := seedNotes(ctx, noteHasher, psql); err != nil {
44	return fmt.Errorf("failed to seed notes: %w", err)
45	}
46
47	slog.Info("Notes seeded successfully")
48
49	return nil
50	}