reface to openteam

wire/wire.go

@@ -5,9 +5,15 @@ package wire
 
 import (
 	"context"
-	"opencatd-open/team/dao"
-	handler "opencatd-open/team/handler/team"
-	"opencatd-open/team/service"
+	"opencatd-open/internal/controller"
+	proxy "opencatd-open/internal/controller/proxy"
+	team_controller "opencatd-open/internal/controller/team"
+	"opencatd-open/internal/dao"
+	"opencatd-open/pkg/config"
+	"sync"
+
+	service "opencatd-open/internal/service"
+	teamService "opencatd-open/internal/service/team"
 
 	"github.com/google/wire"
 	"gorm.io/gorm"
@@ -17,19 +23,19 @@ import (
 var userSet = wire.NewSet(
 	dao.NewUserDAO,
 	wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
-	service.NewUserService,
+	teamService.NewUserService,
 )
 
 var keySet = wire.NewSet(
 	dao.NewApiKeyDAO,
 	wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
-	service.NewApiKeyService,
+	teamService.NewApiKeyService,
 )
 
 var tokenSet = wire.NewSet(
 	dao.NewTokenDAO,
 	wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
-	service.NewTokenService,
+	teamService.NewTokenService,
 )
 
 var usageSet = wire.NewSet(
@@ -37,11 +43,79 @@ var usageSet = wire.NewSet(
 	wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)),
 	dao.NewDailyUsageDAO,
 	wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)),
-	service.NewUsageService,
+	teamService.NewUsageService,
 )
 
 // 初始化 TeamHandler
-func InitTeamHandler(ctx context.Context, db *gorm.DB) (*handler.TeamHandler, error) {
-	wire.Build(userSet, keySet, tokenSet, usageSet, handler.NewTeamHandler)
+func InitTeamHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*team_controller.Team, error) {
+	wire.Build(userSet, keySet, tokenSet, usageSet, team_controller.NewTeam)
+	return nil, nil
+}
+
+// var userApi = wire.NewSet(
+// 	dao.NewUserDAO,
+// 	wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
+// 	service.NewUserService,
+// )
+
+// var keyApi = wire.NewSet(
+// 	dao.NewApiKeyDAO,
+// 	wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
+// 	service.NewApiKeyService,
+// )
+
+// var tokenApi = wire.NewSet(
+// 	dao.NewTokenDAO,
+// 	wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
+// 	service.NewTokenService,
+// )
+
+// func InitAPIHandler(ctx context.Context, db *gorm.DB) (*controller.Api, error) {
+// 	wire.Build(userApi, keyApi, tokenApi, controller.NewApi)
+// 	return nil, nil
+// }
+
+var repositorySet = wire.NewSet(
+	dao.NewUserDAO,
+	wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
+	dao.NewTokenDAO,
+	wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
+	dao.NewApiKeyDAO,
+	wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
+
+	dao.NewUsageDAO,
+	wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)),
+	dao.NewDailyUsageDAO,
+	wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)),
+)
+
+var serviceSet = wire.NewSet(
+	service.NewUserService,
+	service.NewTokenService,
+	service.NewApiKeyService,
+	service.NewWebAuthnService,
+
+	service.NewUsageService,
+)
+
+func InitAPIHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*controller.Api, error) {
+	wire.Build(
+		repositorySet,
+		serviceSet,
+		controller.NewApi,
+	)
+	return nil, nil
+}
+
+var proxySet = wire.NewSet(
+	dao.NewUserDAO,
+	dao.NewApiKeyDAO,
+	dao.NewTokenDAO,
+	dao.NewUsageDAO,
+	dao.NewDailyUsageDAO,
+)
+
+func InitProxyHandler(ctx context.Context, cfg *config.Config, db *gorm.DB, wg *sync.WaitGroup) (*proxy.Proxy, error) {
+	wire.Build(proxySet, proxy.NewProxy)
 	return nil, nil
 }