implement core foundation

go/internal/network/service.go

@@ -9,8 +9,11 @@ import (
 	"strings"
 
 	pbaero "github.com/flowy-live/llink/genproto/aero"
+	"github.com/flowy-live/llink/internal/billing"
 	"github.com/flowy-live/llink/internal/utils"
+	"github.com/jackc/pgx/v5"
 	"github.com/jackc/pgx/v5/pgxpool"
+	"slices"
 )
 
 var ErrNotFound = errors.New("network not found")
@@ -19,17 +22,16 @@ var ErrCapacityExceeded = errors.New("active stream capacity exceeded")
 var ErrInvalidRetentionHours = errors.New("message retention hours must be between 24 and 336")
 
 type Service interface {
-	// Create creates a network and adds adminHumanId as the first member. Returns ErrInvalidName if name is empty.
 	Create(ctx context.Context, name, adminHumanId string) (*Network, error)
-	// GetByID returns ErrNotFound if network doesn't exist.
 	GetByID(ctx context.Context, id string) (*Network, error)
-	// SetName returns ErrNotFound or ErrInvalidName.
 	SetName(ctx context.Context, id, name string) error
+	// AddMembers inserts members and syncs the new seat count to billing
+	// atomically; a Stripe failure rolls the insert back.
 	AddMembers(ctx context.Context, networkID string, humanIds []string) error
 	RemoveMember(ctx context.Context, networkID, humanId string) error
+	CountSeats(ctx context.Context, networkID string) (int, error)
 	ListForHuman(ctx context.Context, humanId string) ([]*Network, error)
 	IsMember(ctx context.Context, networkID, humanId string) (bool, error)
-	// ListAll returns all networks with their members
 	ListAll(ctx context.Context) ([]*Network, error)
 
 	// Invitations (email-based, for users who haven't registered yet)
@@ -41,12 +43,19 @@ type Service interface {
 }
 
 type serviceImpl struct {
-	repo    repository
-	aeroSvc pbaero.PrimaryClient
+	pool       *pgxpool.Pool
+	repo       repository
+	aeroSvc    pbaero.PrimaryClient
+	billingSvc billing.Service
 }
 
-func NewService(pool *pgxpool.Pool, aeroSvc pbaero.PrimaryClient) Service {
-	return &serviceImpl{repo: newRepository(pool), aeroSvc: aeroSvc}
+func NewService(pool *pgxpool.Pool, aeroSvc pbaero.PrimaryClient, billingSvc billing.Service) Service {
+	return &serviceImpl{
+		pool:       pool,
+		repo:       newRepository(pool),
+		aeroSvc:    aeroSvc,
+		billingSvc: billingSvc,
+	}
 }
 
 func (s *serviceImpl) Create(ctx context.Context, name, adminHumanId string) (*Network, error) {
@@ -60,8 +69,7 @@ func (s *serviceImpl) Create(ctx context.Context, name, adminHumanId string) (*N
 		return nil, err
 	}
 
-	err = s.AddMembers(ctx, network.ID, []string{adminHumanId})
-	if err != nil {
+	if err := s.AddMembers(ctx, network.ID, []string{adminHumanId}); err != nil {
 		return nil, err
 	}
 
@@ -90,22 +98,58 @@ func (s *serviceImpl) SetName(ctx context.Context, id, name string) error {
 }
 
 func (s *serviceImpl) AddMembers(ctx context.Context, networkID string, humanIds []string) error {
-	for _, humanId := range humanIds {
-		if humanId == "" {
-			return fmt.Errorf("invalid humanId")
-		}
-		if err := s.repo.addMember(ctx, networkID, humanId); err != nil {
-			return err
-		}
+	if slices.Contains(humanIds, "") {
+		return fmt.Errorf("invalid humanId")
 	}
-	return nil
+	return s.mutateMembers(ctx, networkID, func(tx pgx.Tx) error {
+		for _, humanId := range humanIds {
+			if err := s.repo.addMember(ctx, tx, networkID, humanId); err != nil {
+				return err
+			}
+		}
+		return nil
+	})
 }
 
 func (s *serviceImpl) RemoveMember(ctx context.Context, networkID, humanId string) error {
 	if humanId == "" {
 		return fmt.Errorf("invalid humanId")
 	}
-	return s.repo.removeMember(ctx, networkID, humanId)
+	return s.mutateMembers(ctx, networkID, func(tx pgx.Tx) error {
+		return s.repo.removeMember(ctx, tx, networkID, humanId)
+	})
+}
+
+// mutateMembers runs fn in a tx, recounts seats, calls billing.SyncSeats,
+// and commits. Any error rolls the membership change back.
+func (s *serviceImpl) mutateMembers(ctx context.Context, networkID string, fn func(pgx.Tx) error) error {
+	tx, err := s.pool.Begin(ctx)
+	if err != nil {
+		return fmt.Errorf("begin tx: %w", err)
+	}
+	defer tx.Rollback(ctx)
+
+	if err := fn(tx); err != nil {
+		return err
+	}
+
+	seats, err := s.repo.countSeats(ctx, tx, networkID)
+	if err != nil {
+		return fmt.Errorf("count seats: %w", err)
+	}
+
+	if err := s.billingSvc.SyncSeats(ctx, networkID, seats); err != nil {
+		return fmt.Errorf("sync billing seats: %w", err)
+	}
+
+	if err := tx.Commit(ctx); err != nil {
+		return fmt.Errorf("commit tx: %w", err)
+	}
+	return nil
+}
+
+func (s *serviceImpl) CountSeats(ctx context.Context, networkID string) (int, error) {
+	return s.repo.countSeats(ctx, s.pool, networkID)
 }
 
 func (s *serviceImpl) ListForHuman(ctx context.Context, humanId string) ([]*Network, error) {
@@ -126,8 +170,6 @@ func (s *serviceImpl) ListAll(ctx context.Context) ([]*Network, error) {
 	return s.repo.listAll(ctx)
 }
 
-// Invitation methods
-
 func (s *serviceImpl) InviteByEmail(ctx context.Context, networkID string, emails []string) error {
 	network, err := s.repo.getByID(ctx, networkID)
 	if err != nil {
@@ -180,10 +222,13 @@ func (s *serviceImpl) AcceptInvitation(ctx context.Context, networkID, email, hu
 		return fmt.Errorf("invalid humanId")
 	}
 
-	if err := s.repo.deleteInvitation(ctx, networkID, normalized); err != nil {
-		return err
-	}
-	return s.repo.addMember(ctx, networkID, humanId)
+	return s.mutateMembers(ctx, networkID, func(tx pgx.Tx) error {
+		err := s.repo.deleteInvitation(ctx, tx, networkID, normalized)
+		if err != nil {
+			return err
+		}
+		return s.repo.addMember(ctx, tx, networkID, humanId)
+	})
 }
 
 func (s *serviceImpl) RevokeInvitation(ctx context.Context, networkID, email string) error {
@@ -191,7 +236,7 @@ func (s *serviceImpl) RevokeInvitation(ctx context.Context, networkID, email str
 	if err != nil {
 		return fmt.Errorf("invalid email: %w", err)
 	}
-	return s.repo.deleteInvitation(ctx, networkID, normalized)
+	return s.repo.deleteInvitation(ctx, s.pool, networkID, normalized)
 }
 
 func buildInvitationHTML(networkName string) string {