draft

internal/modules/domains/module.go

@@ -0,0 +1,69 @@
+package domains
+
+import (
+	"context"
+	"strings"
+
+	"github.com/google/uuid"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"stream.api/internal/database/model"
+	"stream.api/internal/modules/common"
+)
+
+type Module struct {
+	runtime *common.Runtime
+}
+
+func New(runtime *common.Runtime) *Module {
+	return &Module{runtime: runtime}
+}
+
+func (m *Module) ListDomains(ctx context.Context, queryValue ListDomainsQuery) (*ListDomainsResult, error) {
+	var rows []model.Domain
+	if err := m.runtime.DB().WithContext(ctx).Where("user_id = ?", queryValue.UserID).Order("created_at DESC").Find(&rows).Error; err != nil {
+		m.runtime.Logger().Error("Failed to list domains", "error", err)
+		return nil, status.Error(codes.Internal, "Failed to load domains")
+	}
+	items := make([]DomainView, 0, len(rows))
+	for i := range rows {
+		items = append(items, DomainView{Domain: &rows[i]})
+	}
+	return &ListDomainsResult{Items: items}, nil
+}
+
+func (m *Module) CreateDomain(ctx context.Context, cmd CreateDomainCommand) (*DomainView, error) {
+	name := common.NormalizeDomain(cmd.Name)
+	if name == "" || !strings.Contains(name, ".") || strings.ContainsAny(name, "/ ") {
+		return nil, status.Error(codes.InvalidArgument, "Invalid domain")
+	}
+	var count int64
+	if err := m.runtime.DB().WithContext(ctx).Model(&model.Domain{}).Where("user_id = ? AND name = ?", cmd.UserID, name).Count(&count).Error; err != nil {
+		m.runtime.Logger().Error("Failed to validate domain", "error", err)
+		return nil, status.Error(codes.Internal, "Failed to create domain")
+	}
+	if count > 0 {
+		return nil, status.Error(codes.InvalidArgument, "Domain already exists")
+	}
+	item := &model.Domain{ID: uuid.New().String(), UserID: cmd.UserID, Name: name}
+	if err := m.runtime.DB().WithContext(ctx).Create(item).Error; err != nil {
+		m.runtime.Logger().Error("Failed to create domain", "error", err)
+		return nil, status.Error(codes.Internal, "Failed to create domain")
+	}
+	return &DomainView{Domain: item}, nil
+}
+
+func (m *Module) DeleteDomain(ctx context.Context, cmd DeleteDomainCommand) error {
+	if cmd.ID == "" {
+		return status.Error(codes.NotFound, "Domain not found")
+	}
+	res := m.runtime.DB().WithContext(ctx).Where("id = ? AND user_id = ?", cmd.ID, cmd.UserID).Delete(&model.Domain{})
+	if res.Error != nil {
+		m.runtime.Logger().Error("Failed to delete domain", "error", res.Error)
+		return status.Error(codes.Internal, "Failed to delete domain")
+	}
+	if res.RowsAffected == 0 {
+		return status.Error(codes.NotFound, "Domain not found")
+	}
+	return nil
+}