chore: Adding builder endpoints, WIP

PiperOrigin-RevId: 813489379

Author: wyf7107

src/google/adk/cli/fast_api.py

@@ -252,40 +252,92 @@ def tear_down_observer(observer: Observer, _: AdkWebServer):
 
   @working_in_progress("builder_save is not ready for use.")
   @app.post("/builder/save", response_model_exclude_none=True)
-  async def builder_build(files: list[UploadFile]) -> bool:
+  async def builder_build(
+      files: list[UploadFile], tmp: Optional[bool] = False
+  ) -> bool:
     base_path = Path.cwd() / agents_dir
-
     for file in files:
+      if not file.filename:
+        logger.exception("Agent name is missing in the input files")
+        return False
+      agent_name, filename = file.filename.split("/")
+      agent_dir = os.path.join(base_path, agent_name)
       try:
         # File name format: {app_name}/{agent_name}.yaml
-        if not file.filename:
-          logger.exception("Agent name is missing in the input files")
-          return False
-
-        agent_name, filename = file.filename.split("/")
-
-        agent_dir = os.path.join(base_path, agent_name)
-        os.makedirs(agent_dir, exist_ok=True)
-        file_path = os.path.join(agent_dir, filename)
-
-        with open(file_path, "wb") as buffer:
-          shutil.copyfileobj(file.file, buffer)
-
+        if tmp:
+          agent_dir = os.path.join(agent_dir, "tmp/" + agent_name)
+          os.makedirs(agent_dir, exist_ok=True)
+          file_path = os.path.join(agent_dir, filename)
+          with open(file_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+
+        else:
+          source_dir = os.path.join(agent_dir, "tmp/" + agent_name)
+          destination_dir = agent_dir
+          for item in os.listdir(source_dir):
+            source_item = os.path.join(source_dir, item)
+            destination_item = os.path.join(destination_dir, item)
+            if os.path.isdir(source_item):
+              shutil.copytree(source_item, destination_item, dirs_exist_ok=True)
+            # Check if the item is a file
+            elif os.path.isfile(source_item):
+              shutil.copy2(source_item, destination_item)
       except Exception as e:
         logger.exception("Error in builder_build: %s", e)
         return False
 
     return True
 
+  @working_in_progress("builder_save is not ready for use.")
+  @app.post("/builder/app/{app_name}/cancel", response_model_exclude_none=True)
+  async def builder_cancel(app_name: str) -> bool:
+    base_path = Path.cwd() / agents_dir
+    agent_dir = os.path.join(base_path, app_name)
+    destination_dir = os.path.join(agent_dir, "tmp/" + app_name)
+    source_dir = agent_dir
+    source_items = set(os.listdir(source_dir))
+    try:
+      for item in os.listdir(destination_dir):
+        if item in source_items:
+          continue
+        # If it doesn't exist in the source, delete it from the destination
+        item_path = os.path.join(destination_dir, item)
+        if os.path.isdir(item_path):
+          shutil.rmtree(item_path)
+        elif os.path.isfile(item_path):
+          os.remove(item_path)
+
+      for item in os.listdir(source_dir):
+        source_item = os.path.join(source_dir, item)
+        destination_item = os.path.join(destination_dir, item)
+        if item == "tmp" and os.path.isdir(source_item):
+          continue
+        if os.path.isdir(source_item):
+          shutil.copytree(source_item, destination_item, dirs_exist_ok=True)
+        # Check if the item is a file
+        elif os.path.isfile(source_item):
+          shutil.copy2(source_item, destination_item)
+    except Exception as e:
+      logger.exception("Error in builder_build: %s", e)
+      return False
+    return True
+
   @working_in_progress("builder_get is not ready for use.")
   @app.get(
       "/builder/app/{app_name}",
       response_model_exclude_none=True,
       response_class=PlainTextResponse,
   )
-  async def get_agent_builder(app_name: str, file_path: Optional[str] = None):
+  async def get_agent_builder(
+      app_name: str,
+      file_path: Optional[str] = None,
+      tmp: Optional[bool] = False,
+  ):
     base_path = Path.cwd() / agents_dir
     agent_dir = base_path / app_name
+    if tmp:
+      agent_dir = agent_dir / "tmp"
+      agent_dir = agent_dir / app_name
     if not file_path:
       file_name = "root_agent.yaml"
       root_file_path = agent_dir / file_name