Compare commits
1 Commits
codex/ensu
...
codex/refa
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a0efdb5001 |
@@ -16,22 +16,15 @@ class RouterManager:
|
|||||||
self.app = app
|
self.app = app
|
||||||
self.included_routers = []
|
self.included_routers = []
|
||||||
self.failed_routers = []
|
self.failed_routers = []
|
||||||
self._included_router_names = set()
|
|
||||||
|
|
||||||
def include_router_safely(self, router, router_name: str = None) -> bool:
|
def include_router_safely(self, router, router_name: str = None) -> bool:
|
||||||
"""Include a router safely with error handling."""
|
"""Include a router safely with error handling."""
|
||||||
verbose = os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
verbose = os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
||||||
router_name = router_name or getattr(router, 'prefix', 'unknown')
|
|
||||||
|
|
||||||
if router_name in self._included_router_names:
|
|
||||||
if verbose:
|
|
||||||
logger.info(f"↩️ Router already included, skipping duplicate: {router_name}")
|
|
||||||
return True
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
self.app.include_router(router)
|
self.app.include_router(router)
|
||||||
|
router_name = router_name or getattr(router, 'prefix', 'unknown')
|
||||||
self.included_routers.append(router_name)
|
self.included_routers.append(router_name)
|
||||||
self._included_router_names.add(router_name)
|
|
||||||
if verbose:
|
if verbose:
|
||||||
logger.info(f"✅ Router included successfully: {router_name}")
|
logger.info(f"✅ Router included successfully: {router_name}")
|
||||||
return True
|
return True
|
||||||
@@ -47,21 +40,19 @@ class RouterManager:
|
|||||||
# Import os locally to avoid UnboundLocalError if it's shadowed
|
# Import os locally to avoid UnboundLocalError if it's shadowed
|
||||||
import os
|
import os
|
||||||
verbose = os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
verbose = os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
||||||
demo_mode = os.getenv("ALWRITY_DEMO_MODE", "false").lower() == "true"
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if verbose:
|
if verbose:
|
||||||
logger.info(f"Including core routers (demo_mode={demo_mode})...")
|
logger.info("Including core routers...")
|
||||||
|
|
||||||
# Subscription router MUST always be included (including demo mode) so
|
|
||||||
# payment/preflight/subscription endpoints remain available.
|
|
||||||
from api.subscription import router as subscription_router
|
|
||||||
self.include_router_safely(subscription_router, "subscription")
|
|
||||||
|
|
||||||
# Component logic router
|
# Component logic router
|
||||||
from api.component_logic import router as component_logic_router
|
from api.component_logic import router as component_logic_router
|
||||||
self.include_router_safely(component_logic_router, "component_logic")
|
self.include_router_safely(component_logic_router, "component_logic")
|
||||||
|
|
||||||
|
# Subscription router
|
||||||
|
from api.subscription import router as subscription_router
|
||||||
|
self.include_router_safely(subscription_router, "subscription")
|
||||||
|
|
||||||
# Step 3 Research router (core onboarding functionality)
|
# Step 3 Research router (core onboarding functionality)
|
||||||
from api.onboarding_utils.step3_routes import router as step3_research_router
|
from api.onboarding_utils.step3_routes import router as step3_research_router
|
||||||
self.include_router_safely(step3_research_router, "step3_research")
|
self.include_router_safely(step3_research_router, "step3_research")
|
||||||
|
|||||||
@@ -260,9 +260,6 @@ async def onboarding_status():
|
|||||||
|
|
||||||
# Include routers using modular utilities
|
# Include routers using modular utilities
|
||||||
router_manager.include_core_routers()
|
router_manager.include_core_routers()
|
||||||
# Safety net: keep subscription routes available even if core inclusion flow changes
|
|
||||||
# in special modes (e.g., demo mode). De-dup is handled by RouterManager.
|
|
||||||
router_manager.include_router_safely(subscription_router, "subscription")
|
|
||||||
router_manager.include_optional_routers()
|
router_manager.include_optional_routers()
|
||||||
|
|
||||||
# Include assets serving router (must be mounted to serve generated images)
|
# Include assets serving router (must be mounted to serve generated images)
|
||||||
|
|||||||
@@ -244,9 +244,6 @@ async def onboarding_status():
|
|||||||
|
|
||||||
# Include routers using modular utilities
|
# Include routers using modular utilities
|
||||||
router_manager.include_core_routers()
|
router_manager.include_core_routers()
|
||||||
# Safety net: keep subscription routes available even if core inclusion flow changes
|
|
||||||
# in special modes (e.g., demo mode). De-dup is handled by RouterManager.
|
|
||||||
router_manager.include_router_safely(subscription_router, "subscription")
|
|
||||||
router_manager.include_optional_routers()
|
router_manager.include_optional_routers()
|
||||||
|
|
||||||
# SEO Dashboard endpoints
|
# SEO Dashboard endpoints
|
||||||
|
|||||||
@@ -280,10 +280,11 @@ const InitialRouteHandler: React.FC = () => {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Loading state - only wait for onboarding init, not subscription check
|
// Only block on onboarding initialization once we know the user has an active subscription.
|
||||||
// Subscription check is non-blocking and happens in background
|
// This allows no-subscription/inactive flows to continue even when onboarding data is still null.
|
||||||
const waitingForOnboardingInit = loading || !data;
|
const isActiveSubscriber = Boolean(subscription && subscription.active && subscription.plan !== 'none');
|
||||||
if (loading || waitingForOnboardingInit) {
|
const waitingForOnboardingInit = isActiveSubscriber && (loading || !data);
|
||||||
|
if (waitingForOnboardingInit) {
|
||||||
return (
|
return (
|
||||||
<Box
|
<Box
|
||||||
display="flex"
|
display="flex"
|
||||||
@@ -295,7 +296,7 @@ const InitialRouteHandler: React.FC = () => {
|
|||||||
>
|
>
|
||||||
<CircularProgress size={60} />
|
<CircularProgress size={60} />
|
||||||
<Typography variant="h6" color="textSecondary">
|
<Typography variant="h6" color="textSecondary">
|
||||||
{subscriptionLoading ? 'Checking subscription...' : 'Preparing your workspace...'}
|
Preparing your workspace...
|
||||||
</Typography>
|
</Typography>
|
||||||
</Box>
|
</Box>
|
||||||
);
|
);
|
||||||
|
|||||||
Reference in New Issue
Block a user