minitap-mcp 0.4.3__py3-none-any.whl → 0.5.0__py3-none-any.whl

minitap/mcp/core/agents/extract_figma_assets.py

@@ -1,15 +1,9 @@
-"""Agent to extract Figma asset URLs from design context code."""
+"""Agent to extract Figma asset URLs from design context code using regex."""
 
 import re
-import uuid
-from pathlib import Path
 
-from jinja2 import Template
-from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
 from pydantic import BaseModel, Field
 
-from minitap.mcp.core.llm import get_minitap_llm
-
 
 class FigmaAsset(BaseModel):
     """Represents a single Figma asset."""
@@ -27,70 +21,49 @@ class ExtractedAssets(BaseModel):
         description="List of all extracted assets from the Figma design context",
     )
     code_implementation: str = Field(
-        description=(
-            "The React/TypeScript code\n"
-            "with the local url declarations turned into const declarations"
-        )
+        description="The React/TypeScript code with imports instead of const declarations"
     )
 
 
-def sanitize_unicode_for_llm(text: str) -> str:
-    """Remove or replace problematic Unicode characters that increase token consumption.
-
-    Characters outside the Basic Multilingual Plane (BMP) like emoji and special symbols
-    get escaped as \\U sequences when sent to LLMs, dramatically increasing token count
-    and processing time.
+def extract_figma_assets(design_context_code: str) -> ExtractedAssets:
+    """Extract asset URLs from Figma design context code using regex.
 
     Args:
-        text: The text to sanitize
+        design_context_code: The React/TypeScript code from get_design_context
 
     Returns:
-        Text with problematic Unicode characters replaced with placeholders
+        ExtractedAssets with list of assets and transformed code
     """
+    # Regex captures: (1) variable name, (2) full URL, (4) extension
+    # Supports http/https, any domain, query strings, optional semicolon
+    pattern = r'const\s+(\w+)\s*=\s*["\']((https?://[^"\']+?)\.(\w+)(?:\?[^"\']*)?)["\'];?'
+    matches = re.finditer(pattern, design_context_code)
+
+    assets = []
+    asset_lines = []
+
+    for match in matches:
+        var_name = match.group(1)
+        url = match.group(2)
+        extension = match.group(4)
+
+        assets.append(FigmaAsset(variable_name=var_name, url=url, extension=extension))
+        asset_lines.append(match.group(0))
+
+    import_statements = []
+    for asset in assets:
+        import_statements.append(
+            f"import {asset.variable_name} from './{asset.variable_name}.{asset.extension}';"
+        )
 
-    # Replace characters outside BMP (U+10000 and above) with a placeholder
-    # These are typically emoji, special symbols, or rare characters
-    def replace_high_unicode(match):
-        char = match.group(0)
-        codepoint = ord(char)
-        # Return a descriptive placeholder
-        return f"[U+{codepoint:X}]"
-
-    # Pattern matches characters with codepoints >= U+10000
-    pattern = re.compile(r"[\U00010000-\U0010FFFF]")
-    sanitized = pattern.sub(replace_high_unicode, text)
-
-    return sanitized
-
+    transformed_code = design_context_code
+    for line in asset_lines:
+        transformed_code = transformed_code.replace(line, "")
 
-async def extract_figma_assets(design_context_code: str) -> ExtractedAssets:
-    """Extract asset URLs from Figma design context code.
+    lines = transformed_code.split("\n")
+    while lines and not lines[0].strip():
+        lines.pop(0)
 
-    Args:
-        design_context_code: The React/TypeScript code from get_design_context
+    final_code = "\n".join(import_statements) + "\n\n" + "\n".join(lines)
 
-    Returns:
-        List of dictionaries containing variable_name, url, and extension
-    """
-    system_message = Template(
-        Path(__file__).parent.joinpath("extract_figma_assets.md").read_text(encoding="utf-8")
-    ).render()
-
-    sanitized_code = sanitize_unicode_for_llm(design_context_code)
-
-    messages: list[BaseMessage] = [
-        SystemMessage(content=system_message),
-        HumanMessage(
-            content=f"Here is the code to analyze:\n\n```typescript\n{sanitized_code}\n```"
-        ),
-    ]
-
-    llm = get_minitap_llm(
-        model="openai/gpt-5",
-        temperature=0,
-        trace_id=str(uuid.uuid4()),
-        remote_tracing=True,
-    ).with_structured_output(ExtractedAssets)
-    result: ExtractedAssets = await llm.ainvoke(messages)  # type: ignore
-
-    return result
+    return ExtractedAssets(assets=assets, code_implementation=final_code)

minitap/mcp/tools/save_figma_assets.py

@@ -41,9 +41,9 @@ logger = get_logger(__name__)
     
     This tool:
     1. Calls get_design_context from Figma MCP to get the React/TypeScript code
-    2. Extracts all asset URLs and code implementation from the code
+    2. Extracts asset URLs and transforms const declarations to import statements
     3. Downloads each asset to .mobile-use/figma_assets/<node-id>/ folder
-    4. Saves the code implementation to .mobile-use/figma_assets/<node-id>/code_implementation.ts
+    4. Saves the transformed code to .mobile-use/figma_assets/<node-id>/code_implementation.ts
     5. Returns a list of downloaded files
     """,
 )
@@ -75,10 +75,8 @@ async def save_figma_assets(
     # Step 1: Get design context from Figma MCP
     design_context = await get_design_context(node_id, file_key)
 
-    # Step 2: Extract asset URLs using LLM agent
-    extracted_context: ExtractedAssets = await extract_figma_assets(
-        design_context.code_implementation
-    )
+    # Step 2: Extract asset URLs and transform code
+    extracted_context: ExtractedAssets = extract_figma_assets(design_context.code_implementation)
     if not extracted_context.assets:
         raise ToolError("No assets found in the Figma design context.")
 

{minitap_mcp-0.4.3.dist-info → minitap_mcp-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: minitap-mcp
-Version: 0.4.3
+Version: 0.5.0
 Summary: Model Context Protocol server for controlling Android & iOS devices with natural language
 Author: Pierre-Louis Favreau, Jean-Pierre Lo, Clément Guiguet
 Requires-Dist: fastmcp>=2.12.4

{minitap_mcp-0.4.3.dist-info → minitap_mcp-0.5.0.dist-info}/RECORD

@@ -1,8 +1,7 @@
 minitap/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 minitap/mcp/core/agents/compare_screenshots.md,sha256=Gt27HVzXzu71BxcanKPokz1dFPvq90vXbjE2HOn5X0I,3559
 minitap/mcp/core/agents/compare_screenshots.py,sha256=Yb7kR8Cv0gWzXyNf-6IS7_9l1npfqYmL-SONJJGgzM4,2060
-minitap/mcp/core/agents/extract_figma_assets.md,sha256=JrXuWF8-2PeQpVix-kf-p6zmu2gQVf9Z6ptTK1cedDk,2413
-minitap/mcp/core/agents/extract_figma_assets.py,sha256=4XwPi7JRGPPvW9P4igQFdKYG94cM5Y8qc3pWRkQ8BsU,3095
+minitap/mcp/core/agents/extract_figma_assets.py,sha256=L5aAHm59mrRYaqrwMJSM24SSdZPu2yVg-wsHTF3L8vk,2310
 minitap/mcp/core/config.py,sha256=_rIH31treZlM2RVnTz5cPXhV9Bu4D-w4TmbPh5_mxxM,1026
 minitap/mcp/core/decorators.py,sha256=kMx_mlaa-2U1AgCoYkgPoLOa-iOoKUF1OjcNV7x59Ds,2940
 minitap/mcp/core/device.py,sha256=sEO3Z-8F325hDOObdH1YBhZE60f17FmIclt5UlhY_nU,7875
@@ -17,9 +16,9 @@ minitap/mcp/server/poller.py,sha256=Qakq4yO3EJ9dXmRqtE3sJjyk0ij7VBU-NuupHhTf37g,
 minitap/mcp/tools/analyze_screen.py,sha256=fjcjf3tTZDlxzmiQFHFNgw38bxPz4eisw57zuxshN2A,1984
 minitap/mcp/tools/compare_screenshot_with_figma.py,sha256=G69F6vRFI2tE2wW-oFYPjnY8oFMD9nRZH0H-yvtD4gE,4575
 minitap/mcp/tools/execute_mobile_command.py,sha256=qY3UfcDq1BtYcny1YlEF4WV9LwUJxLAmLJCm1VBzxS8,2442
-minitap/mcp/tools/save_figma_assets.py,sha256=T5a_7wi1aLoyCrn4FWwXEE_m8dXqUu2ZFGDKst_sKHI,9985
+minitap/mcp/tools/save_figma_assets.py,sha256=6GRaaBQfU4h7H_q69Xoy8qdzkZXtBMlFUeV5kCAeBMc,9978
 minitap/mcp/tools/screen_analyzer.md,sha256=TTO80JQWusbA9cKAZn-9cqhgVHm6F_qJh5w152hG3YM,734
-minitap_mcp-0.4.3.dist-info/WHEEL,sha256=5w2T7AS2mz1-rW9CNagNYWRCaB0iQqBMYLwKdlgiR4Q,78
-minitap_mcp-0.4.3.dist-info/entry_points.txt,sha256=rYVoXm7tSQCqQTtHx4Lovgn1YsjwtEEHfddKrfEVHuY,55
-minitap_mcp-0.4.3.dist-info/METADATA,sha256=-3t9Ox1LrXQfWtYQRyar6Bneyp_T1FQkBV0djX1lJBY,8641
-minitap_mcp-0.4.3.dist-info/RECORD,,
+minitap_mcp-0.5.0.dist-info/WHEEL,sha256=DpNsHFUm_gffZe1FgzmqwuqiuPC6Y-uBCzibcJcdupM,78
+minitap_mcp-0.5.0.dist-info/entry_points.txt,sha256=rYVoXm7tSQCqQTtHx4Lovgn1YsjwtEEHfddKrfEVHuY,55
+minitap_mcp-0.5.0.dist-info/METADATA,sha256=j4TMgzWoC3WHoDBEavfB-SSzQVdpUV10E0HZW30njDg,8641
+minitap_mcp-0.5.0.dist-info/RECORD,,

{minitap_mcp-0.4.3.dist-info → minitap_mcp-0.5.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.7
+Generator: uv 0.9.8
 Root-Is-Purelib: true
 Tag: py3-none-any

minitap/mcp/core/agents/extract_figma_assets.md

@@ -1,64 +0,0 @@
-You are an expert at parsing React/TypeScript code to extract asset URLs and generate clean, documented code implementations.
-
-Your task is to:
-
-1. Extract all asset URLs from the provided code snippet
-2. Generate a clean `code_implementation` output that includes the React code with embedded comments referencing implementation and node guidelines
-
-**Instructions:**
-
-## Part 1: Extract Asset URLs
-
-1. Look for all constant declarations that contain URLs pointing to assets (images, SVGs, etc.)
-2. These constants typically follow patterns like:
-
-   - `const imgVariableName = "http://localhost:3845/assets/[hash].[extension]";`
-   - The variable names usually start with `img` followed by a descriptive name in camelCase
-
-3. For each asset URL found, extract:
-   - The **variable name** (e.g., `imgSignal`, `imgBatteryThreeQuarters`)
-   - The **full URL** (e.g., `http://localhost:3845/assets/685c5ac58caa29556e29737cf8f8c9605d9c8571.svg`)
-   - The **file extension** from the URL (e.g., `svg`, `png`, `jpg`)
-
-## Part 2: Generate Code Implementation
-
-The `code_implementation` field should contain:
-
-1. The React/TypeScript code with **LOCAL asset imports** instead of HTTP URLs:
-
-   - Convert `const imgSignal = "http://localhost:3845/assets/[hash].svg";`
-   - To `import imgSignal from './assets/imgSignal.svg';` (or appropriate relative path)
-   - Use the **exact same variable names** as in the original const declarations
-   - **CRITICAL**: Preserve the variable naming convention
-
-2. Preserve all `data-node-id` attributes and other metadata in the code
-
-## Part 3: Return Format
-
-Return a JSON object with two fields:
-
-- `assets`: Array of extracted asset objects
-- `code_implementation`: String containing the React code with embedded guideline comments
-
-```json
-{
-  "assets": [
-    {
-      "variable_name": "imgSignal",
-      "url": "http://localhost:3845/assets/685c5ac58caa29556e29737cf8f8c9605d9c8571.svg",
-      "extension": "svg"
-    },
-    ...
-  ],
-  "code_implementation": "import ... function ..."
-}
-```
-
-**Important:**
-
-- Only extract asset URLs
-- Preserve the exact variable names as they appear in the code
-- DO NOT MISS any assets
-- If no assets are found, return an empty array for `assets`
-- Return ONLY the JSON object with both `assets` and `code_implementation` fields
-- Do NOT include the const declarations of the assets in the code_implementation output - convert them to imports.