123 lines
3.0 KiB
Go
123 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"git.vakhrushev.me/av/remembos/internal/config"
|
|
"git.vakhrushev.me/av/remembos/internal/memory"
|
|
"git.vakhrushev.me/av/remembos/internal/memos"
|
|
"git.vakhrushev.me/av/remembos/internal/search"
|
|
"git.vakhrushev.me/av/remembos/internal/storage"
|
|
"git.vakhrushev.me/av/remembos/internal/telegram"
|
|
"git.vakhrushev.me/av/remembos/internal/web"
|
|
)
|
|
|
|
func main() {
|
|
configPath := flag.String("config", "config.toml", "path to config file")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.Load(*configPath)
|
|
if err != nil {
|
|
slog.Error("failed to load config", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Logger
|
|
var logLevel slog.Level
|
|
switch cfg.General.LogLevel {
|
|
case "debug":
|
|
logLevel = slog.LevelDebug
|
|
case "warn":
|
|
logLevel = slog.LevelWarn
|
|
case "error":
|
|
logLevel = slog.LevelError
|
|
default:
|
|
logLevel = slog.LevelInfo
|
|
}
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel}))
|
|
slog.SetDefault(logger)
|
|
|
|
// Timezone
|
|
loc, err := time.LoadLocation(cfg.General.Timezone)
|
|
if err != nil {
|
|
logger.Error("invalid timezone", "timezone", cfg.General.Timezone, "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Storage
|
|
store, err := storage.Open(cfg.Database.Path)
|
|
if err != nil {
|
|
logger.Error("failed to open storage", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
defer store.Close()
|
|
|
|
// Memos client
|
|
client := memos.NewClient(cfg.Memos.URL, cfg.Memos.Token)
|
|
|
|
// Search selector
|
|
selector := search.NewSelector(client, store, &cfg.Search, loc, logger)
|
|
|
|
// Memory service
|
|
memorySvc := memory.NewService(selector, store, client, loc, logger)
|
|
|
|
// Web handler
|
|
handler := web.NewHandler(memorySvc, cfg.Memos.URL, cfg.Memos.PublicURL, cfg.General.AllowLoadMore, logger)
|
|
|
|
// Telegram bot
|
|
var tgBot *telegram.Bot
|
|
if cfg.Telegram.Enabled {
|
|
var err error
|
|
tgBot, err = telegram.NewBot(cfg.Telegram, memorySvc, client, cfg.Memos.URL, cfg.Memos.PublicURL, cfg.General.AllowLoadMore, loc, logger)
|
|
if err != nil {
|
|
logger.Error("failed to create telegram bot", "error", err)
|
|
store.Close()
|
|
os.Exit(1) //nolint:gocritic // store.Close() called above; linter doesn't track manual cleanup
|
|
}
|
|
}
|
|
|
|
// HTTP server
|
|
srv := &http.Server{
|
|
Addr: cfg.Web.Listen,
|
|
Handler: handler,
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 30 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
// Graceful shutdown
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
if tgBot != nil {
|
|
go tgBot.Run(ctx)
|
|
}
|
|
|
|
go func() {
|
|
logger.Info("starting server", "addr", cfg.Web.Listen)
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
logger.Error("server error", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
<-ctx.Done()
|
|
logger.Info("shutting down")
|
|
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
if err := srv.Shutdown(shutdownCtx); err != nil {
|
|
logger.Error("shutdown error", "error", err)
|
|
}
|
|
|
|
logger.Info("stopped")
|
|
}
|