package svc

import (
	"github.com/zeromicro/go-zero/rest"
	"github.com/zeromicro/go-zero/zrpc"
	"tianyuan-api/apps/admin/internal/config"
	"tianyuan-api/apps/admin/internal/middleware"
	"tianyuan-api/apps/sentinel/sentinel"
	"tianyuan-api/apps/user/user"
)

type ServiceContext struct {
	Config          config.Config
	AuthInterceptor rest.Middleware
	EntRpc          user.EnterpriseClient
	UserRpc         user.UserClient
	ProductRpc      sentinel.ProductClient
}

func NewServiceContext(c config.Config) *ServiceContext {
	return &ServiceContext{
		Config:          c,
		AuthInterceptor: middleware.NewAuthInterceptorMiddleware(c).Handle,
		EntRpc:          user.NewEnterpriseClient(zrpc.MustNewClient(c.UserRpc).Conn()),
		UserRpc:         user.NewUserClient(zrpc.MustNewClient(c.UserRpc).Conn()),
		ProductRpc:      sentinel.NewProductClient(zrpc.MustNewClient(c.SentinelRpc).Conn()),
	}
}