feat: Add coins field to Problem

src/pwncore/config.py

@@ -56,9 +56,9 @@ class Config:
 config = Config(
     development=True,
     db_url="sqlite://:memory:",
-    # docker_url=None,  # None for default system docker
+    docker_url=None,  # None for default system docker
     # Or set it to an arbitrary URL for testing without Docker
-    docker_url="http://google.com",
+    # docker_url="http://google.com",
     flag="C0D",
     max_containers_per_team=3,
     jwt_secret="mysecret",

src/pwncore/models/ctf.py

@@ -22,6 +22,8 @@ class Problem(Model):
     points = fields.IntField()
     author = fields.TextField()
 
+    coins = fields.IntField(default=0)
+
     image_name = fields.TextField()
     image_config: fields.Field[dict[str, list]] = fields.JSONField()  # type: ignore[assignment]
 

src/pwncore/routes/ctf/__init__.py

@@ -39,10 +39,11 @@ async def ctf_list():
     return problems
 
 
+@atomic()
 @router.post("/{ctf_id}/flag")
 async def flag_post(ctf_id: int, flag: Flag, response: Response, jwt: RequireJwt):
     team_id = jwt["team_id"]
-    problem = await Problem.exists(id=ctf_id)
+    problem = await Problem.get_or_none(id=ctf_id)
     if not problem:
         response.status_code = 404
         return {"msg_code": config.msg_codes["ctf_not_found"]}
@@ -57,6 +58,11 @@ async def flag_post(ctf_id: int, flag: Flag, response: Response, jwt: RequireJwt
     )
     if check_solved:
         await SolvedProblem.create(team_id=team_id, problem_id=ctf_id)
+
+        team = await Team.get(id=team_id)
+        team.coins += problem.coins
+        await team.save()
+
         return {"status": True}
     return {"status": False}
 

src/pwncore/routes/ctf/start.py

@@ -4,7 +4,7 @@
 import uuid
 from tortoise.transactions import atomic
 
-from pwncore.models import Problem, Container, Ports, Team
+from pwncore.models import Problem, Container, Ports
 from pwncore.container import docker_client
 from pwncore.config import config
 from pwncore.routes.auth import RequireJwt
@@ -25,18 +25,6 @@ async def start_docker_container(ctf_id: int, response: Response, jwt: RequireJw
         }
     }
     """
-    if config.development:
-        await Problem.create(
-            name="Invisible-Incursion",
-            description="Chod de tujhe se na ho paye",
-            author="Meetesh Saini",
-            points=300,
-            image_name="key:latest",
-            image_config={"PortBindings": {"22/tcp": [{}]}},
-        )
-        await Team.create(
-            name="CID Squad" + uuid.uuid4().hex, secret_hash="veryverysecret"
-        )
 
     ctf = await Problem.get_or_none(id=ctf_id)
     if not ctf: