71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"go.uber.org/fx"
|
|
"go.uber.org/fx/fxevent"
|
|
|
|
"git.asdf.cafe/abs3nt/gspot/src/components/cache"
|
|
"git.asdf.cafe/abs3nt/gspot/src/components/commands"
|
|
"git.asdf.cafe/abs3nt/gspot/src/components/daemon"
|
|
"git.asdf.cafe/abs3nt/gspot/src/components/logger"
|
|
"git.asdf.cafe/abs3nt/gspot/src/services"
|
|
)
|
|
|
|
func main() {
|
|
var s fx.Shutdowner
|
|
app := fx.New(
|
|
fx.WithLogger(func(logger *slog.Logger) fxevent.Logger {
|
|
l := &fxevent.SlogLogger{Logger: logger}
|
|
l.UseLogLevel(slog.LevelDebug)
|
|
return l
|
|
}),
|
|
fx.Populate(&s),
|
|
services.Config,
|
|
fx.Provide(
|
|
Context,
|
|
cache.NewCache,
|
|
commands.NewCommander,
|
|
logger.NewLogger,
|
|
),
|
|
fx.Invoke(
|
|
daemon.Run,
|
|
),
|
|
)
|
|
app.Run()
|
|
}
|
|
|
|
type AsyncInit func(func(ctx context.Context) error)
|
|
|
|
var ErrContextShutdown = errors.New("shutdown")
|
|
|
|
func Context(
|
|
lc fx.Lifecycle,
|
|
s fx.Shutdowner,
|
|
log *slog.Logger,
|
|
) (context.Context, AsyncInit) {
|
|
if log == nil {
|
|
log = slog.Default()
|
|
}
|
|
ctx, cn := context.WithCancelCause(context.Background())
|
|
lc.Append(fx.Hook{
|
|
OnStop: func(ctx context.Context) error {
|
|
cn(fmt.Errorf("%w: %w", context.Canceled, ErrContextShutdown))
|
|
return nil
|
|
},
|
|
})
|
|
return ctx, func(fn func(ctx context.Context) error) {
|
|
go func() {
|
|
err := fn(ctx)
|
|
if err != nil {
|
|
log.Error("Failed to run async hook", "err", err)
|
|
s.Shutdown()
|
|
}
|
|
}()
|
|
}
|
|
}
|