Adding the Get and Save Snippets

Author: swapnil-nagar

src/function_app.py

@@ -1,15 +1,16 @@
 import azure.functions as func
 import logging
 import json
-
+ 
 app = func.FunctionApp(http_auth_level=func.AuthLevel.ANONYMOUS)
-
-# Constants for the Azure Blob Storage container and file name
-_CONTAINER_NAME = "test"
-_FILE_NAME = "test.txt"
+ 
+# Constants for the Azure Blob Storage container, file, and blob path
+_SNIPPET_NAME_PROPERTY_NAME = "snippetname"
+_SNIPPET_PROPERTY_NAME = "snippet"
+_BLOB_PATH = "snippets/{mcptoolargs." + _SNIPPET_NAME_PROPERTY_NAME + "}.json"
 
 @app.generic_trigger(arg_name="context", type="mcpToolTrigger", toolName="hello", 
-                     description="Gets code snippets from your snippet collection.", 
+                     description="Hello world.", 
                      toolProperties="[]")
 def hello_mcp(context) -> None:
     """
@@ -23,43 +24,61 @@ def hello_mcp(context) -> None:
     """
     return "Hello I am MCPTool!"
 
-@app.generic_trigger(arg_name="context", type="mcpToolTrigger", toolName="savesnippets", 
-                     description="Save code snippets.", 
-                     toolProperties="[{\"propertyName\":\"codeSnippet\",\"propertyType\":\"string\",\"description\":\"The name of the snippet.\"}]")
-@app.generic_output_binding(arg_name="file", type="blob", connection="AzureWebJobsStorage", path=f"{_CONTAINER_NAME}/{_FILE_NAME}")
-def save_snippets(context, file: func.Out[str]):
-    """
-    Saves a code snippet to Azure Blob Storage.
 
+@app.generic_trigger(
+    arg_name="context",
+    type="mcpToolTrigger",
+    toolName="getsnippet",
+    description="Retrieve a snippet by name.",
+    toolProperties=f"[{{\"propertyName\":\"{_SNIPPET_NAME_PROPERTY_NAME}\",\"propertyType\":\"string\",\"description\":\"The name of the snippet.\"}}]"
+)
+@app.generic_input_binding(
+    arg_name="file",
+    type="blob",
+    connection="AzureWebJobsStorage",
+    path=_BLOB_PATH
+)
+def get_snippet(file: func.InputStream, context) -> str:
+    """
+    Retrieves a snippet by name from Azure Blob Storage.
+ 
     Args:
-        context: The trigger context containing the input data as JSON.
-        file (func.Out[str]): The output binding to write the snippet to Azure Blob Storage.
-
-    Raises:
-        KeyError: If the "codeSnippet" key is missing in the input JSON.
-        json.JSONDecodeError: If the context is not valid JSON.
+        file (func.InputStream): The input binding to read the snippet from Azure Blob Storage.
+        context: The trigger context containing the input arguments.
+ 
+    Returns:
+        str: The content of the snippet or an error message.
     """
-    content = json.loads(context)
-    msg = content["arguments"]["codeSnippet"]
-    logging.info(msg)
-    file.set(msg)  # Write the snippet to the specified blob
+    snippet_content = file.read().decode("utf-8")
+    logging.info(f"Retrieved snippet: {snippet_content}")
+    return snippet_content
 
-@app.generic_trigger(arg_name="context", type="mcpToolTrigger", toolName="getsnippets", 
-                     description="Gets code snippets from your snippet collection.", 
-                     toolProperties="[]")
-@app.generic_input_binding(arg_name="file", type="blob", connection="AzureWebJobsStorage", path=f"{_CONTAINER_NAME}/{_FILE_NAME}")
-def get_snippets(file: func.InputStream, context) -> None:
-    """
-    Retrieves a saved code snippet from Azure Blob Storage.
 
-    Args:
-        file (func.InputStream): The input binding to read the snippet from Azure Blob Storage.
-        context: The trigger context (not used in this function).
+@app.generic_trigger(
+    arg_name="context",
+    type="mcpToolTrigger",
+    toolName="savesnippet",
+    description="Save a snippet with a name.",
+    toolProperties=f"[{{\"propertyName\":\"{_SNIPPET_NAME_PROPERTY_NAME}\",\"propertyType\":\"string\",\"description\":\"The name of the snippet.\"}},"
+                   f"{{\"propertyName\":\"{_SNIPPET_PROPERTY_NAME}\",\"propertyType\":\"string\",\"description\":\"The content of the snippet.\"}}]"
+)
+@app.generic_output_binding(
+    arg_name="file",
+    type="blob",
+    connection="AzureWebJobsStorage",
+    path=_BLOB_PATH
+)
+def save_snippet(file: func.Out[str], context) -> str:
+    content = json.loads(context)
+    snippet_name_from_args = content["arguments"][_SNIPPET_NAME_PROPERTY_NAME]
+    snippet_content_from_args = content["arguments"][_SNIPPET_PROPERTY_NAME]
 
-    Returns:
-        str: The content of the blob as a UTF-8 decoded string.
+    if not snippet_name_from_args:
+        return "No snippet name provided"
 
-    Raises:
-        Exception: If the blob does not exist or cannot be read.
-    """
-    return file.read().decode('utf-8')  # Read and decode the blob content
+    if not snippet_content_from_args:
+        return "No snippet content provided"
+ 
+    file.set(snippet_content_from_args)
+    logging.info(f"Saved snippet: {snippet_content_from_args}")
+    return f"Snippet '{snippet_content_from_args}' saved successfully"