Make some basic pack creating endpoints work

sticker/server/api/setup.py

@@ -13,12 +13,22 @@
 #
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
-from aiohttp import web
+from typing import Any
+import random
+import string
+import json
 
-from ..database import User, AccessToken
+from aiohttp import web
+from pkg_resources import resource_stream
+import jsonschema
+
+from ..database import User, AccessToken, Pack, Sticker
+from .errors import Error
 
 routes = web.RouteTableDef()
 
+pack_schema = json.load(resource_stream("sticker.server.api", "pack.schema.json"))
+
 
 @routes.get("/whoami")
 async def whoami(req: web.Request) -> web.Response:
@@ -30,3 +40,68 @@ async def whoami(req: web.Request) -> web.Response:
         "homeserver_url": user.homeserver_url,
         "last_seen": int(token.last_seen_date.timestamp() / 60) * 60,
     })
+
+
+@routes.get("/packs")
+async def packs(req: web.Request) -> web.Response:
+    user: User = req["user"]
+    packs = await user.get_packs()
+    return web.json_response([pack.to_dict() for pack in packs])
+
+
+async def get_json(req: web.Request, schema: str) -> Any:
+    try:
+        data = await req.json()
+    except json.JSONDecodeError:
+        raise Error.request_not_json
+    try:
+        jsonschema.validate(data, schema)
+    except jsonschema.ValidationError as e:
+        raise Error.schema_error(e.message, e.path)
+    return data
+
+
+@routes.post("/packs/create")
+async def upload_pack(req: web.Request) -> web.Response:
+    data = await get_json(req, pack_schema)
+    user: User = req["user"]
+    title = data.pop("title")
+    raw_stickers = data.pop("stickers")
+    pack_id_suffix = data.pop("id", "".join(random.choices(string.ascii_lowercase, k=12)))
+    pack = Pack(id=f"{user.id}_{pack_id_suffix}", owner=user.id, title=title, meta=data)
+    stickers = [Sticker(pack_id=pack.id, id=sticker.pop("id"), url=sticker.pop("url"),
+                        body=sticker.pop("body"), meta=sticker) for sticker in raw_stickers]
+    await pack.insert()
+    await pack.set_stickers(stickers)
+    await user.add_pack(pack)
+
+    return web.json_response({
+        **pack.to_dict(),
+        "stickers": [sticker.to_dict() for sticker in stickers],
+    })
+
+
+@routes.get("/pack/{pack_id}")
+async def get_pack(req: web.Request) -> web.Response:
+    user: User = req["user"]
+    pack = await user.get_pack(req.match_info["pack_id"])
+    if pack is None:
+        raise Error.pack_not_found
+    return web.json_response({
+        **pack.to_dict(),
+        "stickers": [sticker.to_dict() for sticker in await pack.get_stickers()],
+    })
+
+
+@routes.delete("/pack/{pack_id}")
+async def delete_pack(req: web.Request) -> web.Response:
+    user: User = req["user"]
+    pack = await user.get_pack(req.match_info["pack_id"])
+    if pack is None:
+        raise Error.pack_not_found
+
+    if pack.owner != user.id:
+        await user.remove_pack(pack)
+    else:
+        await pack.delete()
+    return web.Response(status=204)