Locazia: v1 auth (not tested yet)

app/api/routes/auth.py

@@ -0,0 +1,45 @@
+from sanic import response
+from app.core._config import TELEGRAM_API_KEY
+from app.core.models.user import User
+from app.core.logger import make_log
+from aiogram.utils.web_app import safe_parse_webapp_init_data
+from datetime import datetime
+import os
+import hashlib
+
+
+async def s_api_v1_auth_twa(request):
+    if not request.json:
+        return response.json({"error": "No data provided"}, status=400)
+
+    if not request.json.get('twa_data'):
+        return response.json({"error": "No TWA data provided"}, status=400)
+
+    twa_data = request.json['twa_data']
+    twa_data = safe_parse_webapp_init_data(token=TELEGRAM_API_KEY, init_data=twa_data)
+    assert twa_data
+
+    known_user = request.ctx.db_session.query(User).filter(User.telegram_id == twa_data.user.id).first()
+    if not known_user:
+        new_user = User(
+            telegram_id=twa_data.user.id,
+            username=twa_data.user.username,
+            meta={
+                "first_name": twa_data.user.first_name,
+                "last_name": twa_data.user.last_name,
+                "photo_url": twa_data.user.photo_url
+            },
+            lang_code=twa_data.user.language_code,
+            last_use=datetime.now(),
+            created=datetime.now()
+        )
+        request.ctx.db_session.add(new_user)
+        request.ctx.db_session.commit()
+
+        known_user = request.ctx.db_session.query(User).filter(User.telegram_id == twa_data.user.id).first()
+        assert known_user, "User not created"
+
+    return response.json({
+        'user': known_user.json_format(),
+        'auth_v1_token': known_user.create_api_token_v1(request.ctx.db_session, "USER_API_V1")['auth_v1_token']
+    })

app/api/routes/custodial.py

@@ -0,0 +1,36 @@
+from sanic import response
+from app.core._config import TELEGRAM_API_KEY
+from app.core.models.user import User
+from app.core.logger import make_log
+from datetime import datetime
+import os
+import hashlib
+
+
+async def s_api_v1_custodial_upload_content(request):
+    if not request.json:
+        return response.json({"error": "No data provided"}, status=400)
+
+    if not request.json.get('content'):
+        return response.json({"error": "No content provided"}, status=400)
+
+    if not request.json.get('content_hash'):
+        return response.json({"error": "No content hash provided"}, status=400)
+
+    content = request.json['content']
+    content_hash = request.json['content_hash']
+
+    known_user = request.ctx.db_session.query(User).filter(User.telegram_id == request.ctx.user.telegram_id).first()
+    if not known_user:
+        return response.json({"error": "User not found"}, status=400)
+
+    content_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), f"../../../content/{content_hash}")
+    if os.path.exists(content_path):
+        return response.json({"error": "Content already exists"}, status=400)
+
+    with open(content_path, "wb") as f:
+        f.write(content)
+
+    return response.json({
+        'content_hash': content_hash
+    })

app/api/routes/node_storage.py

@@ -0,0 +1,58 @@
+from sanic import response
+from app.core._config import UPLOADS_DIR
+from app.core.storage import db_session
+from app.core.models.node_storage import StoredContent
+from app.core.logger import make_log
+from datetime import datetime
+from base58 import b58encode, b58decode
+import os
+import hashlib
+
+
+async def s_api_v1_storage_post(request):
+    if not request.files and not request.json:
+        return response.json({"error": "No file provided"}, status=400)
+
+    file_param = list(request.files.values())[0][0] if request.files else None
+    file_name_json = request.json.get("filename") if request.json else None
+
+    if file_param:
+        file_content = file_param.body
+        file_name = file_name_json or file_param.name
+    else:
+        return response.json({"error": "No file provided"}, status=400)
+
+    try:
+        file_hash = b58encode(hashlib.sha256(file_content).digest()).decode()
+        new_content = StoredContent(
+            hash=file_hash,
+            filename=file_name,
+            user_id=None,
+            meta={},
+            created=datetime.now(),
+            key_id=None
+        )
+        request.ctx.db_session.add(new_content)
+        request.ctx.db_session.commit()
+
+        file_path = os.path.join(UPLOADS_DIR, file_hash)
+        with open(file_path, "wb") as file:
+            file.write(file_content)
+
+        return response.json({"content_sha256": file_hash})
+    except BaseException as e:
+        return response.json({"error": f"Error: {e}"}, status=500)
+
+
+async def s_api_v1_storage_get(request, file_hash):
+    content = request.ctx.db_session.query(StoredContent).filter(StoredContent.hash == file_hash).first()
+    if not content:
+        return response.json({"error": "File not found"}, status=404)
+
+    make_log(f"File {file_hash} requested by {request.ip}")
+
+    file_path = os.path.join(UPLOADS_DIR, file_hash)
+    if not os.path.exists(file_path):
+        return response.json({"error": "File not found"}, status=404)
+
+    return await response.file(file_path)