update core engine

receiver/main.py

@@ -11,6 +11,12 @@ from challenges.Sheesh import Sheesh
 from challenges.Warmup import Warmup
 
 import os
+import asyncio
+import logging
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 security = HTTPBasic()
@@ -25,6 +31,50 @@ challenges = {
     "warmup": Warmup(15000),
 }
 
+async def run_challenge_checks():
+    """Run check function on all challenges at startup"""
+    logger.info("\n" + "="*60)
+    logger.info("Running challenge checks...")
+    logger.info("="*60 + "\n")
+    
+    results = {}
+    
+    for name, challenge in challenges.items():
+        logger.info(f"\n[{name}] Starting check...")
+        try:
+            # Give service time between checks
+            await asyncio.sleep(2)
+            
+            result = challenge.check()
+            results[name] = result
+            
+            if result:
+                logger.info(f"[{name}] ✓ Check PASSED")
+            else:
+                logger.warning(f"[{name}] ✗ Check FAILED")
+        except Exception as e:
+            logger.error(f"[{name}] ✗ Check ERROR: {e}")
+            results[name] = False
+    
+    # Print summary
+    logger.info("\n" + "="*60)
+    logger.info("Challenge Check Summary:")
+    logger.info("="*60)
+    passed = sum(1 for r in results.values() if r)
+    total = len(results)
+    for name, result in results.items():
+        status = "✓ PASS" if result else "✗ FAIL"
+        logger.info(f"  {name:20} {status}")
+    logger.info(f"\nTotal: {passed}/{total} passed")
+    logger.info("="*60 + "\n")
+    
+    return results
+
+@app.on_event("startup")
+async def startup_event():
+    """Run challenge checks on application startup"""
+    asyncio.create_task(run_challenge_checks())
+
 class Flag(BaseModel):
     flag: str
     challenge: str
@@ -32,7 +82,6 @@ class Flag(BaseModel):
 class History(BaseModel):
     log: str
 
-
 @app.get("/")
 def read_root():
     return {"service": "receiver-service"}