langprotect-mcp-gateway 1.3.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

langprotect_mcp_gateway/setup_helper.py

@@ -7,6 +7,9 @@ Automatically configures VS Code for global MCP gateway usage
 import os
 import json
 import sys
+import getpass
+import urllib.request
+import urllib.error
 from pathlib import Path
 
 
@@ -22,24 +25,135 @@ def get_vscode_settings_path():
         return home / ".config/Code/User/settings.json"
 
 
-def create_wrapper_script():
-    """Create the global wrapper script"""
+def validate_credentials(url, email, password):
+    """Validate credentials against the backend API"""
+    try:
+        import json
+        
+        # Prepare the request
+        data = json.dumps({"email": email, "password": password}).encode('utf-8')
+        req = urllib.request.Request(
+            f"{url}/api/auth/login/",
+            data=data,
+            headers={'Content-Type': 'application/json'}
+        )
+        
+        # Make the request
+        with urllib.request.urlopen(req, timeout=10) as response:
+            return response.status in [200, 201]
+            
+    except urllib.error.HTTPError as e:
+        # Parse error message if available
+        try:
+            error_body = e.read().decode('utf-8')
+            error_data = json.loads(error_body)
+            error_msg = error_data.get('detail', error_data.get('message', 'Authentication failed'))
+            print(f"      ✗ {error_msg}")
+        except:
+            print(f"      ✗ Authentication failed (HTTP {e.code})")
+        return False
+    except urllib.error.URLError as e:
+        print(f"      ✗ Cannot connect to {url}")
+        print(f"        Make sure the backend is running and accessible")
+        return False
+    except Exception as e:
+        print(f"      ✗ Error: {e}")
+        return False
+
+
+def prompt_credentials():
+    """Interactively prompt user for credentials with validation"""
+    print()
+    print("═" * 65)
+    print("         🔐 Enter Your LangProtect Credentials")
+    print("═" * 65)
+    print()
+    
+    while True:
+        # Prompt for URL
+        url = input("Backend URL [http://localhost:8000]: ").strip()
+        if not url:
+            url = "http://localhost:8000"
+        
+        # Prompt for email
+        email = input("Email: ").strip()
+        if not email:
+            print("✗ Email cannot be empty!")
+            print()
+            continue
+        
+        # Prompt for password (hidden)
+        password = getpass.getpass("Password: ")
+        if not password:
+            print("✗ Password cannot be empty!")
+            print()
+            continue
+        
+        # Validate credentials
+        print("      Validating credentials...")
+        if validate_credentials(url, email, password):
+            print("      ✓ Credentials validated successfully!")
+            print()
+            return url, email, password
+        else:
+            print()
+            print("Please try again or press Ctrl+C to cancel.")
+            print()
+
+
+def create_wrapper_script(url=None, email=None, password=None):
+    """Create the global wrapper script with credentials"""
     wrapper_dir = Path.home() / ".local/bin"
     wrapper_dir.mkdir(parents=True, exist_ok=True)
     
     wrapper_path = wrapper_dir / "langprotect-mcp-wrapper.sh"
     
-    wrapper_content = """#!/bin/bash
+    # Check if credentials provided via environment variables
+    if not url or not email or not password:
+        url = os.environ.get('LANGPROTECT_URL')
+        email = os.environ.get('LANGPROTECT_EMAIL')
+        password = os.environ.get('LANGPROTECT_PASSWORD')
+    
+    # If still not provided, prompt user
+    if not url or not email or not password:
+        url, email, password = prompt_credentials()
+    else:
+        # Validate environment credentials
+        print("   Using credentials from environment variables...")
+        print("   Validating...")
+        if not validate_credentials(url, email, password):
+            print("   ✗ Environment credentials invalid. Please enter manually:")
+            url, email, password = prompt_credentials()
+        else:
+            print("   ✓ Environment credentials validated!")
+    
+    # Create wrapper with actual credentials
+    wrapper_content = f"""#!/bin/bash
 # LangProtect MCP Gateway Wrapper
-# This wrapper allows global configuration for all VS Code workspaces
+# Auto-configured by langprotect-gateway-setup
+
+# ============================================================
+# Backend Connection
+# ============================================================
+export LANGPROTECT_URL="${{LANGPROTECT_URL:-{url}}}"
+export LANGPROTECT_EMAIL="${{LANGPROTECT_EMAIL:-{email}}}"
+export LANGPROTECT_PASSWORD="${{LANGPROTECT_PASSWORD:-{password}}}"
+
+# ============================================================
+# Security Controls (v1.3.1+)
+# ============================================================
+export LANGPROTECT_ENABLE_MASKING="${{LANGPROTECT_ENABLE_MASKING:-true}}"
+export LANGPROTECT_FAIL_CLOSED="${{LANGPROTECT_FAIL_CLOSED:-false}}"
+export LANGPROTECT_SCAN_TIMEOUT="${{LANGPROTECT_SCAN_TIMEOUT:-5.0}}"
+export LANGPROTECT_ENTROPY_DETECTION="${{LANGPROTECT_ENTROPY_DETECTION:-true}}"
 
-# Configure these environment variables with your LangProtect credentials
-export LANGPROTECT_URL="${LANGPROTECT_URL:-http://localhost:8000}"
-export LANGPROTECT_EMAIL="${LANGPROTECT_EMAIL:-your.email@company.com}"
-export LANGPROTECT_PASSWORD="${LANGPROTECT_PASSWORD:-your-password}"
-export MCP_SERVER_COMMAND="${MCP_SERVER_COMMAND:-npx}"
-export MCP_SERVER_ARGS="${MCP_SERVER_ARGS:--y,@modelcontextprotocol/server-filesystem,.}"
+# ============================================================
+# MCP Server Configuration
+# ============================================================
+export MCP_SERVER_COMMAND="${{MCP_SERVER_COMMAND:-npx}}"
+export MCP_SERVER_ARGS="${{MCP_SERVER_ARGS:--y,@modelcontextprotocol/server-filesystem,.}}"
 
+# Start the gateway
 exec langprotect-gateway "$@"
 """
     
@@ -130,7 +244,7 @@ def setup():
     print("🚀 Setting up LangProtect MCP Gateway...")
     print()
     
-    # Create wrapper script
+    # Create wrapper script (will prompt for credentials if needed)
     print("📝 Creating global wrapper script...")
     wrapper_path = create_wrapper_script()
     print(f"   ✅ Created: {wrapper_path}")
@@ -145,7 +259,7 @@ def setup():
         print(f"   ⚠️  Could not update VS Code settings: {e}")
     
     # Update Claude Desktop config
-    print("🍏 Configuring Claude Desktop (for high compatibility)...")
+    print("🍏 Configuring Claude Desktop...")
     try:
         claude_path = update_claude_config(wrapper_path)
         print(f"   ✅ Updated: {claude_path}")
@@ -158,25 +272,28 @@ def setup():
     print()
     print("📋 Next steps:")
     print()
-    print("1. Configure your credentials:")
-    print(f"   Edit: {wrapper_path}")
-    print("   Set LANGPROTECT_URL, LANGPROTECT_EMAIL, and LANGPROTECT_PASSWORD")
-    print()
-    print("2. Reload VS Code:")
+    print("1. Reload VS Code:")
     print("   Press Ctrl+Shift+P → 'Developer: Reload Window'")
     print()
-    print("3. Verify it's working:")
+    print("2. Verify it's working:")
     print("   Press Ctrl+Shift+P → 'MCP: List Servers'")
     print("   You should see 'langprotect-gateway' listed")
     print()
-    print("🎉 LangProtect will now protect ALL your VS Code workspaces!")
+    print("3. Test the protection:")
+    print("   Ask AI to read a file with secrets - they'll be masked!")
+    print()
+    print("🎉 LangProtect is now protecting ALL your VS Code workspaces!")
     print()
-    print("💡 Tip: You can also set credentials via environment variables:")
-    print("   export LANGPROTECT_URL=http://localhost:8000")
-    print("   export LANGPROTECT_EMAIL=your.email@company.com")
-    print("   export LANGPROTECT_PASSWORD=your-password")
+    print("💡 Configuration file:", wrapper_path)
     print()
 
 
 if __name__ == "__main__":
-    setup()
+    try:
+        setup()
+    except KeyboardInterrupt:
+        print("\n\n⚠️  Setup cancelled by user.")
+        sys.exit(1)
+    except Exception as e:
+        print(f"\n\n❌ Setup failed: {e}")
+        sys.exit(1)

{langprotect_mcp_gateway-1.3.0.dist-info → langprotect_mcp_gateway-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langprotect-mcp-gateway
-Version: 1.3.0
+Version: 1.3.1
 Summary: Security gateway for Model Context Protocol (MCP) to protect AI tool interactions
 Author-email: LangProtect Security Team <security@langprotect.com>
 License: MIT

langprotect_mcp_gateway-1.3.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+langprotect_mcp_gateway/__init__.py,sha256=PedabfF6wZ_6KxuN60A4qz8T1gD9MszuXwhmrHlGH7I,510
+langprotect_mcp_gateway/gateway.py,sha256=YIggDJ7n0ctUsyyI1s567QFbH7cq5-6CAAdI1J8gQkY,30921
+langprotect_mcp_gateway/response_masker.py,sha256=ui1JusuPwuOKSfrDtt0FxLEGs_y512RcTG4gSz2-MT8,14702
+langprotect_mcp_gateway/setup_helper.py,sha256=alkIyR3jB-RuiD32oNrATc-IskYkTgWUNjy8SLDz5HQ,9746
+langprotect_mcp_gateway-1.3.1.dist-info/licenses/LICENSE,sha256=aoVP65gKtirVmFPToow5L9IKN4FNjfM6Sejq_5b4cbM,1082
+langprotect_mcp_gateway-1.3.1.dist-info/METADATA,sha256=U3u6uT_-AZwBOio_blaZBwRv41aQos9sMMgXrp-ba4w,11787
+langprotect_mcp_gateway-1.3.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+langprotect_mcp_gateway-1.3.1.dist-info/entry_points.txt,sha256=HpnUUuYLQva8b6gazUX0UJO9dFHq86e9gifQfLKpyWc,140
+langprotect_mcp_gateway-1.3.1.dist-info/top_level.txt,sha256=UjNlX13ma4nwJXuEyi9eMX251c5rooeEao4zajX6ZHk,24
+langprotect_mcp_gateway-1.3.1.dist-info/RECORD,,

langprotect_mcp_gateway-1.3.0.dist-info/RECORD

@@ -1,10 +0,0 @@
-langprotect_mcp_gateway/__init__.py,sha256=PedabfF6wZ_6KxuN60A4qz8T1gD9MszuXwhmrHlGH7I,510
-langprotect_mcp_gateway/gateway.py,sha256=YIggDJ7n0ctUsyyI1s567QFbH7cq5-6CAAdI1J8gQkY,30921
-langprotect_mcp_gateway/response_masker.py,sha256=ui1JusuPwuOKSfrDtt0FxLEGs_y512RcTG4gSz2-MT8,14702
-langprotect_mcp_gateway/setup_helper.py,sha256=ghErneMTua9wPATMq8eatnviVAYJMi2bf2UUt8fnXE8,5639
-langprotect_mcp_gateway-1.3.0.dist-info/licenses/LICENSE,sha256=aoVP65gKtirVmFPToow5L9IKN4FNjfM6Sejq_5b4cbM,1082
-langprotect_mcp_gateway-1.3.0.dist-info/METADATA,sha256=f8QWABfqpzyuO_UgtCxC9LI6jAuuy0fsPluZIgNna54,11787
-langprotect_mcp_gateway-1.3.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-langprotect_mcp_gateway-1.3.0.dist-info/entry_points.txt,sha256=HpnUUuYLQva8b6gazUX0UJO9dFHq86e9gifQfLKpyWc,140
-langprotect_mcp_gateway-1.3.0.dist-info/top_level.txt,sha256=UjNlX13ma4nwJXuEyi9eMX251c5rooeEao4zajX6ZHk,24
-langprotect_mcp_gateway-1.3.0.dist-info/RECORD,,