add http-api

http-api/routes/users/__init__.py

@@ -0,0 +1,84 @@
+import json
+from http import HTTPStatus
+from typing import Annotated
+
+import boto3
+from aws_lambda_powertools.event_handler.api_gateway import (
+    Response,
+    Router,
+)
+from elasticsearch import Elasticsearch
+from pydantic import UUID4, BaseModel, StringConstraints
+from layercake.dynamodb import DynamoDBPersistenceLayer
+from dynamodb import KeyLoc, get_records
+
+import elastic
+from http_models import RecordResponse, SearchResponse
+from settings import ELASTIC_CONN, USER_TABLE
+
+router = Router()
+dynamodb_client = boto3.client('dynamodb')
+user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
+elastic_client = Elasticsearch(**ELASTIC_CONN)
+
+
+@router.get('/', compress=True, tags=['User'], summary='Get users')
+def get_users() -> SearchResponse:
+    event = router.current_event
+    query = event.get_query_string_value('query', '{}')
+    page_size = event.get_query_string_value('page_size', '25')
+
+    return elastic.search(  # type: ignore
+        index=USER_TABLE,
+        page_size=int(page_size),
+        query=json.loads(query),
+        elastic_client=elastic_client,
+    )
+
+
+class ResetPasswordPayload(BaseModel):
+    cognito_sub: UUID4
+    new_password: Annotated[str, StringConstraints(min_length=6)]
+
+
+@router.patch('/<id>', compress=True, tags=['User'])
+def patch_reset(id: str, payload: ResetPasswordPayload):
+    return Response(status_code=HTTPStatus.OK)
+
+
+@router.get('/<id>/emails', compress=True, tags=['User'], summary='Get user emails')
+def get_emails(id: str) -> RecordResponse:
+    start_key = router.current_event.get_query_string_value('start_key', None)
+
+    return get_records(  # type: ignore
+        pk=id,
+        prefix='emails',
+        start_key=start_key,
+        persistence_layer=user_layer,
+    )
+
+
+@router.get('/<id>/logs', compress=True, tags=['User'], summary='Get user logs')
+def get_logs(id: str) -> RecordResponse:
+    start_key = router.current_event.get_query_string_value('start_key', None)
+
+    return get_records(  # type: ignore
+        pk=id,
+        prefix='log',
+        delimiter=':',
+        keyloc_prefix=KeyLoc.PK,
+        start_key=start_key,
+        persistence_layer=user_layer,
+    )
+
+
+@router.get('/<id>/orgs', compress=True, tags=['User'], summary='Get user orgs')
+def get_orgs(id: str) -> RecordResponse:
+    start_key = router.current_event.get_query_string_value('start_key', None)
+
+    return get_records(  # type: ignore
+        pk=id,
+        prefix='orgs',
+        start_key=start_key,
+        persistence_layer=user_layer,
+    )