(feat) couchbase doc crud operations (#3)

* Add document management functions to MCP server

- Implemented `get_document` to retrieve a document by ID.
- Added `upsert_document` for inserting or updating documents.
- Created `delete_document` to remove documents by ID.
- Enhanced error handling and logging for each operation.

* Add function to list documents in a specified collection

- Introduced `list_documents_in_collection` to retrieve document IDs and content from a specified collection.
- Implemented error handling for cluster availability and query execution.
- Added logging for query operations to enhance traceability.

* fixed feedback based on comments

* Refactor document management functions in MCP server

- Renamed `get_document` to `get_document_by_id` for clarity.
- Renamed `upsert_document` to `upsert_document_by_id` to specify operation by ID.
- Renamed `delete_document` to `delete_document_by_id` for consistency.
- Updated docstrings to reflect changes and improve documentation.
- Enhanced error logging by removing type information for cleaner output.

* Update README.md to reflect enhanced document management features

- Added descriptions for new functionalities: retrieving, upserting, and deleting documents by ID from specified scopes and collections.

---------

Co-authored-by: Nithish Raghunandanan <[email protected]>

Author: teetangh

README.md

@@ -4,9 +4,12 @@ An [MCP](https://modelcontextprotocol.io/) server implementation of Couchbase th
 
 ## Features
 
-- Get a list of all the scopes and collections in the specified bucket in a Couchbase cluster
-- Get the structure for a collection in a Couchbase cluster
-- Run a [SQL++ query](https://www.couchbase.com/sqlplusplus/) against data in Couchbase cluster bucket
+- Get a list of all the scopes and collections in the specified bucket
+- Get the structure for a collection
+- Get a document by ID from a specified scope and collection
+- Upsert a document by ID to a specified scope and collection
+- Delete a document by ID from a specified scope and collection
+- Run a [SQL++ query](https://www.couchbase.com/sqlplusplus/) on a specified scope
 
 ## Prerequisites
 

src/mcp_server.py

@@ -122,6 +122,59 @@ def get_schema_for_collection(
         raise
 
 
+@mcp.tool()
+def get_document_by_id(
+    ctx: Context, scope_name: str, collection_name: str, document_id: str
+) -> dict[str, Any]:
+    """Get a document by its ID from the specified scope and collection."""
+    bucket = ctx.request_context.lifespan_context.bucket
+    try:
+        collection = bucket.scope(scope_name).collection(collection_name)
+        result = collection.get(document_id)
+        return result.content_as[dict]
+    except Exception as e:
+        logger.error(f"Error getting document {document_id}: {e}")
+        raise
+
+
+@mcp.tool()
+def upsert_document_by_id(
+    ctx: Context,
+    scope_name: str,
+    collection_name: str,
+    document_id: str,
+    document_content: dict[str, Any],
+) -> bool:
+    """Insert or update a document by its ID.
+    Returns True on success, False on failure."""
+    bucket = ctx.request_context.lifespan_context.bucket
+    try:
+        collection = bucket.scope(scope_name).collection(collection_name)
+        collection.upsert(document_id, document_content)
+        logger.info(f"Successfully upserted document {document_id}")
+        return True
+    except Exception as e:
+        logger.error(f"Error upserting document {document_id}: {e}")
+        return False
+
+
+@mcp.tool()
+def delete_document_by_id(
+    ctx: Context, scope_name: str, collection_name: str, document_id: str
+) -> bool:
+    """Delete a document by its ID.
+    Returns True on success, False on failure."""
+    bucket = ctx.request_context.lifespan_context.bucket
+    try:
+        collection = bucket.scope(scope_name).collection(collection_name)
+        collection.remove(document_id)
+        logger.info(f"Successfully deleted document {document_id}")
+        return True
+    except Exception as e:
+        logger.error(f"Error deleting document {document_id}: {e}")
+        return False
+
+
 @mcp.tool()
 def run_sql_plus_plus_query(
     ctx: Context, scope_name: str, query: str