framework

var Module = fx.Module("framework",
	fx.Provide(func(ev *env.Env) *fm {
		return &fm{Env: ev}
	}),
	mongoDb.NewMongoClientFx[*fm](),

	fx.Provide(func(f *fm) (app.RealVectorGrpcClient, error) {
		return grpc.NewGrpcClient(f.VectorGrpcAddr)
	}),

	fx.Provide(func(ev *env.Env, logger *slog.Logger) (*nats.JetstreamClient, error) {
		nc, err := nats.NewClient(ev.NatsUrl, nats.ClientOpts{
			Name:   "message-offfice",
			Logger: logger,
		})
		if err != nil {
			return nil, errors.NewE(err)
		}

		return nats.NewJetstreamClient(nc)
	}),

	app.Module,

	fx.Provide(func(logger *slog.Logger) (app.InternalGrpcServer, error) {
		return grpc.NewGrpcServer(grpc.ServerOpts{Logger: logger.With(slog.String("component", "internal-grpc"))})
	}),

	fx.Invoke(func(lf fx.Lifecycle, logr logging.Logger, server app.InternalGrpcServer, ev *env.Env) {
		lf.Append(fx.Hook{
			OnStart: func(context.Context) error {
				go func() {
					if err := server.Listen(fmt.Sprintf(":%d", ev.InternalGrpcPort)); err != nil {
						logr.Errorf(err, "while starting internal grpc server")
					}
				}()
				return nil
			},
			OnStop: func(context.Context) error {
				server.Stop()
				return nil
			},
		})
	}),

	fx.Provide(func(ev *env.Env) (app.InfraGRPCClient, error) {
		return grpc.NewGrpcClient(ev.InfraGRPCAddr)
	}),

	fx.Provide(func(logger *slog.Logger) (app.ExternalGrpcServer, error) {
		return grpc.NewGrpcServer(grpc.ServerOpts{Logger: logger.With(slog.String("component", "external-grpc"))})
	}),

	fx.Invoke(func(lf fx.Lifecycle, logr logging.Logger, server app.ExternalGrpcServer, ev *env.Env) {
		lf.Append(fx.Hook{
			OnStart: func(context.Context) error {
				go func() {
					if err := server.Listen(fmt.Sprintf(":%d", ev.ExternalGrpcPort)); err != nil {
						logr.Errorf(err, "while starting external grpc server")
					}
				}()
				return nil
			},
			OnStop: func(context.Context) error {
				server.Stop()
				return nil
			},
		})
	}),

	fx.Provide(func(logger logging.Logger, e *env.Env) httpServer.Server {
		return httpServer.NewServer(httpServer.ServerArgs{Logger: logger, IsDev: e.IsDev})
	}),

	fx.Invoke(func(lf fx.Lifecycle, server httpServer.Server, ev *env.Env) {
		lf.Append(fx.Hook{
			OnStart: func(context.Context) error {
				return server.Listen(fmt.Sprintf(":%d", ev.HttpPort))
			},
			OnStop: func(context.Context) error {
				return server.Close()
			},
		})
	}),
)