Add images upload to backend

Author: EvieePy

mystbin/backend/routers/pastes.py

@@ -53,6 +53,9 @@
 del __p, __f  # micro-opt, don't keep unneeded variables in-ram
 
 
+CF_DLURL = f"https://api.cloudflare.com/client/v4/accounts/{__config['cf_images_account']}/images/v2/direct_upload?requireSignedURLs=true"
+
+
 def generate_paste_id():
     """Generate three random words."""
     word_samples = sample(word_list, 3)
@@ -177,6 +180,26 @@ async def put_pastes(
     return UJSONResponse(paste, status_code=201)
 
 
+@router.get(
+    "/images/upload",
+    tags=["pastes"],
+    status_code=201,
+    include_in_schema=False,
+)
+async def get_image_upload_link(request: MystbinRequest, auth: Optional[str] = ''):
+    if auth != __config['cf_images_frontend_secret']:
+        return UJSONResponse({'error': 'Unauthorized.'}, status_code=401)
+
+    headers = {
+    'Authorization': f'Bearer {__config["cf_images_token"]}'}
+
+    async with request.app.state.client.post(CF_DLURL, headers=headers) as resp:
+        data = await resp.json()
+
+    result = data['result']['uploadURL']
+    return UJSONResponse({'url': result}, status_code=201)
+
+
 desc = f"""Get a paste by ID.
 
 This endpoint falls under the `getpaste` ratelimit bucket.

mystbin/backend/utils/db.py

@@ -968,3 +968,12 @@ async def put_log(self, request: Request, response: Response) -> None:
             response.status_code,
             resp,
         )
+
+        async def put_paste_images(self, parent_id, tab_id) -> None:
+            query = """INSERT INTO images VALUES($1, $2)"""
+
+            await self._do_query(
+            query,
+            parent_id,
+            tab_id
+            )

mystbin/database/schema.sql

@@ -35,6 +35,14 @@ CREATE TABLE IF NOT EXISTS files (
     PRIMARY KEY (parent_id, index)
 );
 
+CREATE TABLE IF NOT EXISTS images (
+    parent_id TEXT REFERENCES pastes(id) ON DELETE CASCADE,
+    tab_id BIGINT,
+    index SERIAL NOT NULL,
+    PRIMARY KEY (parent_id, index)
+
+);
+
 CREATE TABLE IF NOT EXISTS bookmarks (
     userid bigint not null references users(id) ON DELETE CASCADE,
     paste text not null references pastes(id) ON DELETE CASCADE,