mathpix 0.1.0

data/lib/mathpix/mcp/tools/get_usage_tool.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require_relative '../base_tool'
+
+module Mathpix
+  module MCP
+    module Tools
+      # Get Usage Tool
+      #
+      # Retrieves API usage statistics and account limits
+      # Thin delegate to Mathpix::Client (usage endpoint)
+      #
+      # The geodesic path: official SDK + usage tracking
+      class GetUsageTool < BaseTool
+        description "Get Mathpix API usage statistics and remaining credits"
+
+        input_schema(
+          properties: {
+            detailed: {
+              type: "boolean",
+              description: "Include detailed breakdown by operation type (default: false)"
+            }
+          },
+          required: []
+        )
+
+        def self.call(detailed: false, server_context:)
+          safe_execute do
+            client = mathpix_client(server_context)
+
+            begin
+              # Try to call the usage endpoint
+              usage_data = client.get('/usage')
+
+              response_data = {
+                success: true,
+                usage: {
+                  requests_this_month: usage_data['requests_this_month'],
+                  requests_remaining: usage_data['requests_remaining'],
+                  plan: usage_data['plan'],
+                  reset_date: usage_data['reset_date']
+                }
+              }
+
+              if detailed && usage_data['breakdown']
+                response_data[:breakdown] = usage_data['breakdown']
+              end
+
+              json_response(response_data)
+            rescue Mathpix::APIError => e
+              # Handle API-specific errors properly
+              error_response(e)
+            rescue StandardError => e
+              # Handle unexpected errors
+              error_response(e)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mathpix/mcp/tools/list_formats_tool.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+require_relative '../base_tool'
+
+module Mathpix
+  module MCP
+    module Tools
+      # List Formats Tool
+      #
+      # Lists all available output formats for Mathpix OCR
+      # Static data from API documentation
+      #
+      # The geodesic path: official SDK + format catalog
+      class ListFormatsTool < BaseTool
+        description "List all available output formats for Mathpix OCR operations"
+
+        input_schema(
+          properties: {
+            category: {
+              type: "string",
+              description: "Filter by category: image, document, or all (default: all)",
+              enum: ["image", "document", "all"]
+            }
+          },
+          required: []
+        )
+
+        def self.call(category: "all", server_context:)
+          safe_execute do
+            # Static format definitions
+            image_formats = [
+              { name: "latex_styled", description: "LaTeX with styling", type: "image" },
+              { name: "text", description: "Plain text", type: "image" },
+              { name: "latex_list", description: "Array of LaTeX expressions", type: "image" },
+              { name: "mathml", description: "MathML markup", type: "image" },
+              { name: "asciimath", description: "AsciiMath notation", type: "image" },
+              { name: "text_display", description: "Display-style text", type: "image" },
+              { name: "latex_simplified", description: "Simplified LaTeX", type: "image" },
+              { name: "data", description: "Full response data with metadata", type: "image" },
+              { name: "html", description: "HTML markup", type: "image" }
+            ]
+
+            document_formats = [
+              { name: "markdown", description: "Markdown format", type: "document" },
+              { name: "latex", description: "LaTeX document", type: "document" },
+              { name: "html", description: "HTML document", type: "document" },
+              { name: "docx", description: "Microsoft Word document", type: "document" },
+              { name: "tex.zip", description: "LaTeX with figures (zipped)", type: "document" }
+            ]
+
+            # Filter by category
+            formats = case category
+            when "image"
+              image_formats
+            when "document"
+              document_formats
+            when "all"
+              image_formats + document_formats
+            else
+              image_formats + document_formats
+            end
+
+            # Format response
+            response_data = {
+              success: true,
+              category: category,
+              count: formats.length,
+              formats: formats,
+              usage: {
+                image_capture: "Use with snap() or ConvertImageTool",
+                document_conversion: "Use with document() or ConvertDocumentTool"
+              }
+            }
+
+            json_response(response_data)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mathpix/mcp/tools/search_results_tool.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+require_relative '../base_tool'
+
+module Mathpix
+  module MCP
+    module Tools
+      # Search Results Tool
+      #
+      # Searches recent capture results with optional filtering
+      # Thin delegate to Mathpix::Client#recent with search
+      #
+      # The geodesic path: official SDK + search filtering
+      class SearchResultsTool < BaseTool
+        description "Search recent Mathpix capture results with optional text filtering"
+
+        input_schema(
+          properties: {
+            query: {
+              type: "string",
+              description: "Search query to filter results by LaTeX or text content"
+            },
+            limit: {
+              type: "number",
+              description: "Maximum number of results to return (default: 10, max: 100)"
+            },
+            offset: {
+              type: "number",
+              description: "Offset for pagination (default: 0)"
+            },
+            include_content: {
+              type: "boolean",
+              description: "Include full LaTeX/text content in results (default: false)"
+            }
+          },
+          required: []
+        )
+
+        def self.call(query: nil, limit: 10, offset: 0, include_content: false, server_context:)
+          safe_execute do
+            client = mathpix_client(server_context)
+
+            # Validate and constrain limit
+            limit = [[limit.to_i, 1].max, 100].min
+
+            # Get recent results from API
+            recent_results = client.recent(limit: limit + offset)
+
+            # Apply offset
+            results = recent_results.drop(offset)
+
+            # Apply search filter if query provided
+            if query && !query.empty?
+              query_lower = query.downcase
+              results = results.select do |result|
+                (result.latex&.downcase&.include?(query_lower)) ||
+                (result.text&.downcase&.include?(query_lower))
+              end
+            end
+
+            # Limit after filtering
+            results = results.take(limit)
+
+            # Format results
+            formatted_results = results.map do |result|
+              item = {
+                id: result.request_id,
+                created_at: result.timestamp,
+                confidence: result.confidence,
+                is_printed: result.printed?,
+                is_handwritten: result.handwritten?
+              }
+
+              # Include content if requested
+              if include_content
+                item[:latex] = result.latex
+                item[:text] = result.text
+              else
+                # Just include preview
+                item[:latex_preview] = truncate(result.latex, 100)
+                item[:text_preview] = truncate(result.text, 100)
+              end
+
+              item
+            end
+
+            # Format response
+            response_data = {
+              success: true,
+              query: query,
+              limit: limit,
+              offset: offset,
+              count: formatted_results.length,
+              results: formatted_results
+            }
+
+            json_response(response_data)
+          end
+        end
+
+        private
+
+        def self.truncate(text, max_length)
+          return nil unless text
+          return text if text.length <= max_length
+          "#{text[0...max_length]}..."
+        end
+      end
+    end
+  end
+end