Improve onboarding

services/tenant/app/services/subscription_limit_service.py

@@ -155,20 +155,20 @@ class SubscriptionLimitService:
         try:
             async with self.database_manager.get_session() as db_session:
                 await self._init_repositories(db_session)
-                
+
                 # Get subscription limits
                 subscription = await self.subscription_repo.get_active_subscription(tenant_id)
                 if not subscription:
                     return {"can_add": False, "reason": "No active subscription"}
-                
+
                 # Check if unlimited users (-1)
                 if subscription.max_users == -1:
                     return {"can_add": True, "reason": "Unlimited users allowed"}
-                
+
                 # Count current active members
                 members = await self.member_repo.get_tenant_members(tenant_id, active_only=True)
                 current_users = len(members)
-                
+
                 can_add = current_users < subscription.max_users
                 return {
                     "can_add": can_add,
@@ -176,12 +176,80 @@ class SubscriptionLimitService:
                     "max_allowed": subscription.max_users,
                     "reason": "Within limits" if can_add else f"Maximum {subscription.max_users} users allowed for {subscription.plan} plan"
                 }
-                
+
         except Exception as e:
             logger.error("Failed to check user limits",
                         tenant_id=tenant_id,
                         error=str(e))
             return {"can_add": False, "reason": "Error checking limits"}
+
+    async def can_add_recipe(self, tenant_id: str) -> Dict[str, Any]:
+        """Check if tenant can add another recipe"""
+        try:
+            async with self.database_manager.get_session() as db_session:
+                await self._init_repositories(db_session)
+
+                subscription = await self.subscription_repo.get_active_subscription(tenant_id)
+                if not subscription:
+                    return {"can_add": False, "reason": "No active subscription"}
+
+                # Get recipe limit from plan
+                recipes_limit = await self._get_limit_from_plan(subscription.plan, 'recipes')
+
+                # Check if unlimited (-1 or None)
+                if recipes_limit is None or recipes_limit == -1:
+                    return {"can_add": True, "reason": "Unlimited recipes allowed"}
+
+                # Count current recipes from recipes service
+                current_recipes = await self._get_recipe_count(tenant_id)
+
+                can_add = current_recipes < recipes_limit
+                return {
+                    "can_add": can_add,
+                    "current_count": current_recipes,
+                    "max_allowed": recipes_limit,
+                    "reason": "Within limits" if can_add else f"Maximum {recipes_limit} recipes allowed for {subscription.plan} plan"
+                }
+
+        except Exception as e:
+            logger.error("Failed to check recipe limits",
+                        tenant_id=tenant_id,
+                        error=str(e))
+            return {"can_add": False, "reason": "Error checking limits"}
+
+    async def can_add_supplier(self, tenant_id: str) -> Dict[str, Any]:
+        """Check if tenant can add another supplier"""
+        try:
+            async with self.database_manager.get_session() as db_session:
+                await self._init_repositories(db_session)
+
+                subscription = await self.subscription_repo.get_active_subscription(tenant_id)
+                if not subscription:
+                    return {"can_add": False, "reason": "No active subscription"}
+
+                # Get supplier limit from plan
+                suppliers_limit = await self._get_limit_from_plan(subscription.plan, 'suppliers')
+
+                # Check if unlimited (-1 or None)
+                if suppliers_limit is None or suppliers_limit == -1:
+                    return {"can_add": True, "reason": "Unlimited suppliers allowed"}
+
+                # Count current suppliers from suppliers service
+                current_suppliers = await self._get_supplier_count(tenant_id)
+
+                can_add = current_suppliers < suppliers_limit
+                return {
+                    "can_add": can_add,
+                    "current_count": current_suppliers,
+                    "max_allowed": suppliers_limit,
+                    "reason": "Within limits" if can_add else f"Maximum {suppliers_limit} suppliers allowed for {subscription.plan} plan"
+                }
+
+        except Exception as e:
+            logger.error("Failed to check supplier limits",
+                        tenant_id=tenant_id,
+                        error=str(e))
+            return {"can_add": False, "reason": "Error checking limits"}
     
     async def has_feature(self, tenant_id: str, feature: str) -> Dict[str, Any]:
         """Check if tenant has access to a specific feature"""