work

app/sbin/api

@@ -111,6 +111,10 @@ class ContainerModel(BaseModel):
 class UUIDRequest(BaseModel):
     UUID: str
 
+class CommandRequest(BaseModel):
+    uuid: str
+    method: int 
+
 
 # ---------------------- Routes ----------------------
 @app.get("/", include_in_schema=False)
@@ -315,12 +319,17 @@ def pull_all_images():
     return {"message": run_command([f"{BIN_PATH}/pullAllContainers"])}
 
 
+@app.post("/client/git", dependencies=[Depends(verify_api_key)])
+def git_tool(request: CommandRequest):
+    if request.method == 1:
+        command = [f"{BIN_PATH}/gitPull", request.uuid]
+    elif request.method == 2:
+        command = [f"{BIN_PATH}/gitRevert", request.uuid]
+    else:
+        raise HTTPException(status_code=400, detail="Invalid method")
 
-
-@app.post("/client/revert/{uuid}", dependencies=[Depends(verify_api_key)])
-def revert_changes(uuid: str):
-    return {"message": run_command([f"{BIN_PATH}/gitRevert", uuid ])}
-
+    output = run_command(command)
+    return {"message": output}
 
 @app.get("/client/logs/{uuid}", dependencies=[Depends(verify_api_key)])
 async def get_odoo_log_summary(uuid: str):