aws-cost-calculator-cli 1.4.0__py3-none-any.whl → 1.6.2__py3-none-any.whl

backend/handlers/profiles.py

@@ -0,0 +1,148 @@
+"""
+Lambda handler for profile CRUD operations.
+"""
+import json
+import os
+import boto3
+from datetime import datetime
+
+
+def handler(event, context):
+    """Handle profile CRUD operations."""
+    
+    # Parse request
+    try:
+        if isinstance(event.get('body'), str):
+            body = json.loads(event['body'])
+        else:
+            body = event.get('body', {})
+    except:
+        body = event
+    
+    # Validate API secret
+    headers = event.get('headers', {})
+    api_secret = headers.get('X-API-Secret') or headers.get('x-api-secret')
+    
+    secret_name = os.environ.get('SECRET_NAME', 'cost-calculator-api-secret')
+    secrets_client = boto3.client('secretsmanager')
+    
+    try:
+        secret_response = secrets_client.get_secret_value(SecretId=secret_name)
+        expected_secret = secret_response['SecretString']
+        
+        if api_secret != expected_secret:
+            return {
+                'statusCode': 401,
+                'headers': {'Content-Type': 'application/json'},
+                'body': json.dumps({'error': 'Unauthorized'})
+            }
+    except Exception as e:
+        return {
+            'statusCode': 500,
+            'headers': {'Content-Type': 'application/json'},
+            'body': json.dumps({'error': f'Secret validation failed: {str(e)}'})
+        }
+    
+    # Get operation
+    operation = body.get('operation')  # list, get, create, update, delete
+    profile_name = body.get('profile_name')
+    
+    # DynamoDB table
+    table_name = os.environ.get('PROFILES_TABLE', 'cost-calculator-profiles')
+    dynamodb = boto3.resource('dynamodb')
+    table = dynamodb.Table(table_name)
+    
+    try:
+        if operation == 'list':
+            # List all profiles
+            response = table.scan()
+            profiles = response.get('Items', [])
+            return {
+                'statusCode': 200,
+                'headers': {'Content-Type': 'application/json'},
+                'body': json.dumps({'profiles': profiles})
+            }
+        
+        elif operation == 'get':
+            # Get specific profile
+            if not profile_name:
+                return {
+                    'statusCode': 400,
+                    'headers': {'Content-Type': 'application/json'},
+                    'body': json.dumps({'error': 'profile_name required'})
+                }
+            
+            response = table.get_item(Key={'profile_name': profile_name})
+            if 'Item' not in response:
+                return {
+                    'statusCode': 404,
+                    'headers': {'Content-Type': 'application/json'},
+                    'body': json.dumps({'error': 'Profile not found'})
+                }
+            
+            return {
+                'statusCode': 200,
+                'headers': {'Content-Type': 'application/json'},
+                'body': json.dumps({'profile': response['Item']})
+            }
+        
+        elif operation == 'create' or operation == 'update':
+            # Create or update profile
+            if not profile_name:
+                return {
+                    'statusCode': 400,
+                    'headers': {'Content-Type': 'application/json'},
+                    'body': json.dumps({'error': 'profile_name required'})
+                }
+            
+            accounts = body.get('accounts', [])
+            description = body.get('description', '')
+            
+            item = {
+                'profile_name': profile_name,
+                'accounts': accounts,
+                'description': description,
+                'updated_at': datetime.utcnow().isoformat()
+            }
+            
+            if operation == 'create':
+                item['created_at'] = datetime.utcnow().isoformat()
+            
+            table.put_item(Item=item)
+            
+            return {
+                'statusCode': 200,
+                'headers': {'Content-Type': 'application/json'},
+                'body': json.dumps({'message': 'Profile saved', 'profile': item})
+            }
+        
+        elif operation == 'delete':
+            # Delete profile
+            if not profile_name:
+                return {
+                    'statusCode': 400,
+                    'headers': {'Content-Type': 'application/json'},
+                    'body': json.dumps({'error': 'profile_name required'})
+                }
+            
+            table.delete_item(Key={'profile_name': profile_name})
+            
+            return {
+                'statusCode': 200,
+                'headers': {'Content-Type': 'application/json'},
+                'body': json.dumps({'message': 'Profile deleted'})
+            }
+        
+        else:
+            return {
+                'statusCode': 400,
+                'headers': {'Content-Type': 'application/json'},
+                'body': json.dumps({'error': 'Invalid operation. Use: list, get, create, update, delete'})
+            }
+    
+    except Exception as e:
+        return {
+            'statusCode': 500,
+            'headers': {'Content-Type': 'application/json'},
+            'body': json.dumps({'error': str(e)})
+        }

backend/handlers/trends.py

@@ -0,0 +1,106 @@
+"""
+Lambda handler for trends analysis.
+"""
+import json
+import boto3
+import os
+from algorithms.trends import analyze_trends
+
+# Get API secret from Secrets Manager
+secrets_client = boto3.client('secretsmanager')
+api_secret_arn = os.environ['API_SECRET_ARN']
+api_secret = secrets_client.get_secret_value(SecretId=api_secret_arn)['SecretString']
+
+
+def handler(event, context):
+    """
+    Lambda handler for trends analysis.
+    
+    Expected event:
+    {
+        "credentials": {
+            "access_key": "AKIA...",
+            "secret_key": "...",
+            "session_token": "..." (optional)
+        },
+        "accounts": ["123456789012", "987654321098"],
+        "weeks": 4
+    }
+    """
+    # Handle OPTIONS for CORS
+    if event.get('requestContext', {}).get('http', {}).get('method') == 'OPTIONS':
+        return {
+            'statusCode': 200,
+            'headers': {
+                'Access-Control-Allow-Origin': '*',
+                'Access-Control-Allow-Headers': '*',
+                'Access-Control-Allow-Methods': 'POST, OPTIONS'
+            },
+            'body': ''
+        }
+    
+    try:
+        # Validate API secret
+        headers = event.get('headers', {})
+        provided_secret = headers.get('x-api-secret') or headers.get('X-API-Secret')
+        
+        if provided_secret != api_secret:
+            return {
+                'statusCode': 401,
+                'headers': {'Access-Control-Allow-Origin': '*'},
+                'body': json.dumps({'error': 'Unauthorized'})
+            }
+        
+        # Parse request body
+        body = json.loads(event.get('body', '{}'))
+        
+        credentials = body.get('credentials', {})
+        accounts = body.get('accounts', [])
+        weeks = body.get('weeks', 3)
+        
+        if not credentials or not accounts:
+            return {
+                'statusCode': 400,
+                'headers': {'Access-Control-Allow-Origin': '*'},
+                'body': json.dumps({'error': 'Missing credentials or accounts'})
+            }
+        
+        # Create Cost Explorer client with provided credentials
+        ce_client = boto3.client(
+            'ce',
+            region_name='us-east-1',
+            aws_access_key_id=credentials['access_key'],
+            aws_secret_access_key=credentials['secret_key'],
+            aws_session_token=credentials.get('session_token')
+        )
+        
+        # Run analysis
+        trends_data = analyze_trends(ce_client, accounts, weeks)
+        
+        # Convert datetime objects to strings for JSON serialization
+        def convert_dates(obj):
+            if isinstance(obj, dict):
+                return {k: convert_dates(v) for k, v in obj.items()}
+            elif isinstance(obj, list):
+                return [convert_dates(item) for item in obj]
+            elif hasattr(obj, 'isoformat'):
+                return obj.isoformat()
+            return obj
+        
+        trends_data = convert_dates(trends_data)
+        
+        return {
+            'statusCode': 200,
+            'headers': {
+                'Access-Control-Allow-Origin': '*',
+                'Content-Type': 'application/json'
+            },
+            'body': json.dumps(trends_data)
+        }
+        
+    except Exception as e:
+        return {
+            'statusCode': 500,
+            'headers': {'Access-Control-Allow-Origin': '*'},
+            'body': json.dumps({'error': str(e)})
+        }

cost_calculator/api_client.py

@@ -10,33 +10,15 @@ from pathlib import Path
 
 def get_api_config():
     """
-    Get API configuration from environment or config file.
+    Get API configuration.
     
     Returns:
-        dict with 'base_url' and 'api_secret', or None if not configured
+        dict: API configuration with api_secret, or None if not configured
     """
-    # Try environment variables first
-    base_url = os.environ.get('COST_API_URL')
-    api_secret = os.environ.get('COST_API_SECRET')
+    api_secret = os.environ.get('COST_API_SECRET', '')
     
-    if base_url and api_secret:
-        return {
-            'base_url': base_url.rstrip('/'),
-            'api_secret': api_secret
-        }
-    
-    # Try config file
-    config_dir = Path.home() / '.config' / 'cost-calculator'
-    api_config_file = config_dir / 'api_config.json'
-    
-    if api_config_file.exists():
-        with open(api_config_file, 'r') as f:
-            config = json.load(f)
-            if 'base_url' in config and 'api_secret' in config:
-                return {
-                    'base_url': config['base_url'].rstrip('/'),
-                    'api_secret': config['api_secret']
-                }
+    if api_secret:
+        return {'api_secret': api_secret}
     
     return None
 
@@ -60,21 +42,18 @@ def call_lambda_api(endpoint, credentials, accounts, **kwargs):
     api_config = get_api_config()
     
     if not api_config:
-        raise Exception("API not configured. Set COST_API_URL and COST_API_SECRET environment variables.")
+        raise Exception("API not configured. Set COST_API_SECRET environment variable.")
     
-    # Map endpoint names to URLs
+    # Map endpoint names to Lambda URLs
     endpoint_urls = {
-        'trends': f"{api_config['base_url']}/trends",
-        'monthly': f"{api_config['base_url']}/monthly",
-        'drill': f"{api_config['base_url']}/drill"
+        'trends': 'https://pq3mqntc6vuwi4zw5flulsoleq0yiqtl.lambda-url.us-east-1.on.aws/',
+        'monthly': 'https://6aueebodw6q4zdeu3aaexb6tle0fqhhr.lambda-url.us-east-1.on.aws/',
+        'drill': 'https://3ncm2gzxrsyptrhud3ua3x5lju0akvsr.lambda-url.us-east-1.on.aws/',
+        'analyze': 'https://y6npmidtxwzg62nrqzkbacfs5q0edwgs.lambda-url.us-east-1.on.aws/',
+        'profiles': 'https://64g7jq7sjygec2zmll5lsghrpi0txrzo.lambda-url.us-east-1.on.aws/'
     }
     
-    # For the actual Lambda URLs (no path)
-    if '/trends' not in api_config['base_url']:
-        # Base URL is the function URL itself
-        url = api_config['base_url']
-    else:
-        url = endpoint_urls.get(endpoint)
+    url = endpoint_urls.get(endpoint)
     
     if not url:
         raise Exception(f"Unknown endpoint: {endpoint}")