Merge pull request #107 from Hestia-Homes/feature/tasks

added example to simulate work just need to provoke this sqs once it'…

deployment/lambda/lambda_example/docker/app.py

@@ -1,19 +1,181 @@
-"""
-A quick example of lambda working a function in python
-"""
+import os
+import json
+import time
+import requests
+
+
+# ============================================================
+# CONFIG
+# ============================================================
+
+API_BASE = "https://api.dev.hestia.homes/v1/tasks/"  # e.g. https://api.myserver.com/tasks
+API_TOKEN = "4QPwbB6hEdUloDVtbBJCUTfGBdBgWwpeavWQ7t5Z"
+
+
+# ============================================================
+# Low-level API wrapper
+# ============================================================
+
+def api(method, path, body=None):
+    """Generic function to call your backend Task API."""
+    url = f"{API_BASE}{path}"
+
+    headers = {
+        "Authorization": f"Bearer {API_TOKEN}",
+        "Content-Type": "application/json",
+    }
+
+    print(f"[API] {method} {url}")
+    if body:
+        print(f"[API] Payload: {json.dumps(body)}")
+
+    response = requests.request(
+        method,
+        url,
+        headers=headers,
+        data=json.dumps(body) if body else None
+    )
+
+    print(f"[API] Response: {response.status_code}")
+
+    if response.status_code >= 400:
+        print(f"[API ERROR] {response.text}")
+        raise Exception(f"API Error {response.status_code}: {response.text}")
+
+    return response.json()
+
+
+# ============================================================
+# CloudWatch Logs URL generator (with time range)
+# ============================================================
+
+def build_cloudwatch_url(context, start_ms, end_ms):
+    region = context.invoked_function_arn.split(":")[3]
+    log_group = context.log_group_name
+    log_stream = context.log_stream_name
+
+    log_group_enc = log_group.replace("/", "$252F")
+    log_stream_enc = log_stream.replace("/", "$252F")
+
+    return (
+        f"https://{region}.console.aws.amazon.com/cloudwatch/home"
+        f"?region={region}#logsV2:log-groups/log-group/{log_group_enc}"
+        f"/log-events/{log_stream_enc}"
+        f"$3Fstart$3D{start_ms}$26end$3D{end_ms}"
+    )
+
+
+# ============================================================
+# Subtask lifecycle action helpers
+# ============================================================
+
+def mark_in_progress(subtask_id):
+    print(f"[SUBTASK] Marking {subtask_id} as IN PROGRESS")
+    return api("PUT", f"/subtasks/{subtask_id}/status", {"status": "in progress"})
+
+
+def mark_complete(subtask_id, outputs, cloud_logs):
+    print(f"[SUBTASK] Marking {subtask_id} as COMPLETE")
+    return api("POST", f"/subtasks/{subtask_id}/finalize", {
+        "status": "complete",
+        "outputs": outputs,
+        "cloud_logs_url": cloud_logs
+    })
+
+
+def mark_failed(subtask_id, error_message, cloud_logs):
+    print(f"[SUBTASK] Marking {subtask_id} as FAILED")
+    return api("POST", f"/subtasks/{subtask_id}/finalize", {
+        "status": "failed",
+        "outputs": {"error": error_message},
+        "cloud_logs_url": cloud_logs
+    })
+
+
+# ============================================================
+# Lambda Handler
+# ============================================================
 
 def handler(event, context):
+    """
+    Example Lambda that:
+    - marks subtask in progress
+    - prints simulated work steps
+    - marks complete or failed
+    - attaches CloudWatch Log URL
+    """
+
+    subtask_id = event["subtask_id"]
+
+    print("=====================================")
+    print(" 🚀 LAMBDA INVOCATION STARTED")
+    print(f" 🧩 Subtask ID: {subtask_id}")
+    print("=====================================")
+
+    start_ms = int(time.time() * 1000)
+
     try:
-        print("Printing from lambda example")
+        # ---- Step 1: Mark subtask IN PROGRESS ----
+        mark_in_progress(subtask_id)
+
+        print("\n---- Simulating Work ----")
+
+        print("Step 1: Loading resources...")
+        time.sleep(0.5)
+
+        print("Step 2: Calling an external service...")
+        time.sleep(0.7)
+
+        print("Step 3: Processing data...")
+        time.sleep(1.0)
+
+        # UNCOMMENT TO TEST FAILURE
+        # raise Exception("Simulated failure for testing")
+
+        print("Step 4: Finalizing outputs...")
+        time.sleep(0.5)
+
+        outputs = {
+            "result": "success",
+            "calculated_value": 123,
+            "details": "This is example output from the handler.",
+            "input_params": event.get("params", {})
+        }
+
+        print("Work completed successfully.")
+
+        end_ms = int(time.time() * 1000)
+        cloud_logs_url = build_cloudwatch_url(context, start_ms, end_ms)
+
+        print(f"Generated CloudWatch URL:\n{cloud_logs_url}\n")
+
+        # ---- Step 2: Mark COMPLETE ----
+        mark_complete(subtask_id, outputs, cloud_logs_url)
+
+        print("=====================================")
+        print(" ✅ LAMBDA EXECUTION COMPLETE")
+        print("=====================================")
 
         return {
             "statusCode": 200,
-            "body": f"s3 uri {s3_uri}"
+            "body": outputs,
+            "cloudwatch_logs": cloud_logs_url
         }
 
     except Exception as e:
-        print(f"❌ Error: {e}")
+        print("=====================================")
+        print(" ❌ ERROR IN LAMBDA EXECUTION")
+        print("=====================================")
+        print(f"Error: {e}")
+
+        end_ms = int(time.time() * 1000)
+        cloud_logs_url = build_cloudwatch_url(context, start_ms, end_ms)
+
+        # ---- Step 3: Mark FAILED ----
+        mark_failed(subtask_id, str(e), cloud_logs_url)
+
         return {
             "statusCode": 500,
-            "body": str(e)
-        }
+            "error": str(e),
+            "cloudwatch_logs": cloud_logs_url
+        }