Add debug logging for list_drafts issue

Author: ty13r

src/handlers/post_handler.py

@@ -240,10 +240,36 @@ async def list_drafts(self, limit: int = 10) -> List[Dict[str, Any]]:
         # python-substack returns a generator, convert to list
         # The API returns all posts, so we need to filter for drafts only
         try:
-            all_posts = list(self.client.get_drafts(limit=min(limit * 3, 50)))  # Get more to ensure we have enough drafts, but cap at 50
-            logger.debug(f"Retrieved {len(all_posts)} posts from API")
+            # Debug: Let's see what client we're using
+            logger.info(f"Client type: {type(self.client)}")
+            logger.info(f"Client class name: {self.client.__class__.__name__}")
+            
+            # Check publication info
+            if hasattr(self.client, 'client') and hasattr(self.client.client, 'publication_id'):
+                logger.info(f"Publication ID: {self.client.client.publication_id}")
+            if hasattr(self.client, 'client') and hasattr(self.client.client, 'subdomain'):
+                logger.info(f"Subdomain: {self.client.client.subdomain}")
+            
+            # Try to get the raw API response
+            raw_result = self.client.get_drafts(limit=min(limit * 3, 50))
+            logger.info(f"Raw result type: {type(raw_result)}")
+            
+            all_posts = list(raw_result)  # Get more to ensure we have enough drafts, but cap at 50
+            logger.info(f"Retrieved {len(all_posts)} posts from API")
+            
+            # If empty, let's try a different approach
+            if len(all_posts) == 0:
+                logger.warning("get_drafts returned empty, trying direct API call")
+                # Check if we have the underlying client
+                if hasattr(self.client, 'client'):
+                    logger.info("Found underlying client, checking its properties")
+                    inner_client = self.client.client
+                    logger.info(f"Inner client type: {type(inner_client)}")
+                    logger.info(f"Inner client dir: {[x for x in dir(inner_client) if not x.startswith('_')][:10]}")
         except Exception as e:
-            logger.error(f"Error getting drafts: {e}")
+            logger.error(f"Error getting drafts: {type(e).__name__}: {e}")
+            import traceback
+            logger.error(traceback.format_exc())
             return []
 
         drafts = []
@@ -256,17 +282,10 @@ async def list_drafts(self, limit: int = 10) -> List[Dict[str, Any]]:
             logger.debug(f"Post has title: {post.get('title') is not None}")
             logger.debug(f"Post has post_date: {post.get('post_date') is not None}")
 
-            # Check if it's actually a draft (not published)
-            # Drafts have type='draft' or no post_date
-            is_draft = (
-                post.get('type') == 'draft' or 
-                (not post.get('post_date') and (post.get('draft_title') or post.get('title')))
-            )
-            
-            if is_draft:
-                drafts.append(post)
-                if len(drafts) >= limit:
-                    break
+            # For debugging: add ALL posts to see what we're getting
+            drafts.append(post)
+            if len(drafts) >= limit:
+                break
 
         logger.info(f"Returning {len(drafts)} drafts")
         return drafts

src/server.py

@@ -440,20 +440,25 @@ async def handle_call_tool(name: str, arguments: Optional[Dict[str, Any]]) -> Li
                     )]
 
                 elif name == "list_drafts":
+                    logger.info(f"list_drafts called with arguments: {arguments}")
                     post_handler = PostHandler(client)
                     drafts = await post_handler.list_drafts(
                         limit=arguments.get("limit", 10)
                     )
+                    logger.info(f"list_drafts returned {len(drafts)} drafts")
 
                     draft_list = []
                     for draft in drafts:
                         title = draft.get('draft_title') or draft.get('title') or 'Untitled'
                         draft_id = draft.get('id')
                         draft_list.append(f"- {title} (ID: {draft_id})")
 
+                    response_text = f"Found {len(drafts)} drafts:\n" + "\n".join(draft_list)
+                    logger.info(f"Returning response: {response_text[:100]}...")
+                    
                     return [TextContent(
                         type="text",
-                        text=f"Found {len(drafts)} drafts:\n" + "\n".join(draft_list)
+                        text=response_text
                     )]
 
                 elif name == "upload_image":
@@ -814,7 +819,9 @@ async def handle_call_tool(name: str, arguments: Optional[Dict[str, Any]]) -> Li
 
     async def run(self):
         """Run the MCP server using stdio transport"""
+        logger.info("Starting MCP server...")
         async with stdio_server() as (read_stream, write_stream):
+            logger.info("Stdio transport established")
             await self.server.run(
                 read_stream,
                 write_stream,
@@ -824,6 +831,7 @@ async def run(self):
                     capabilities={}
                 )
             )
+            logger.info("Server run completed")
 
 
 def main():

test_simple_mcp.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+# Simple test of MCP server with initialization
+
+export SUBSTACK_PUBLICATION_URL="https://neroaugustus.substack.com/"
+
+# Test with proper MCP initialization sequence
+echo '{"jsonrpc":"2.0","method":"initialize","params":{"protocolVersion":"0.1.0","capabilities":{},"clientInfo":{"name":"test","version":"1.0.0"}},"id":1}
+{"jsonrpc":"2.0","method":"initialized","params":{}}
+{"jsonrpc":"2.0","method":"tools/call","params":{"name":"list_drafts","arguments":{"limit":5}},"id":2}' | /opt/homebrew/bin/substack-mcp-plus