feat(FastAPI): add /pkgbase/{name}/[un]vote (post)

Signed-off-by: Kevin Morris <kevr@0cost.org>

aurweb/routers/packages.py

@@ -769,3 +769,41 @@ async def pkgbase_unnotify(request: Request, name: str):
 
     return RedirectResponse(f"/pkgbase/{name}",
                             status_code=int(HTTPStatus.SEE_OTHER))
+
+
+@router.post("/pkgbase/{name}/vote")
+@auth_required(True, redirect="/pkgbase/{name}")
+async def pkgbase_vote(request: Request, name: str):
+    pkgbase = get_pkg_or_base(name, PackageBase)
+
+    vote = pkgbase.package_votes.filter(
+        PackageVote.UsersID == request.user.ID
+    ).first()
+    has_cred = request.user.has_credential("CRED_PKGBASE_VOTE")
+    if has_cred and not vote:
+        now = int(datetime.utcnow().timestamp())
+        with db.begin():
+            db.create(PackageVote,
+                      User=request.user,
+                      PackageBase=pkgbase,
+                      VoteTS=now)
+
+    return RedirectResponse(f"/pkgbase/{name}",
+                            status_code=int(HTTPStatus.SEE_OTHER))
+
+
+@router.post("/pkgbase/{name}/unvote")
+@auth_required(True, redirect="/pkgbase/{name}")
+async def pkgbase_unvote(request: Request, name: str):
+    pkgbase = get_pkg_or_base(name, PackageBase)
+
+    vote = pkgbase.package_votes.filter(
+        PackageVote.UsersID == request.user.ID
+    ).first()
+    has_cred = request.user.has_credential("CRED_PKGBASE_VOTE")
+    if has_cred and vote:
+        with db.begin():
+            db.session.delete(vote)
+
+    return RedirectResponse(f"/pkgbase/{name}",
+                            status_code=int(HTTPStatus.SEE_OTHER))

test/test_packages_routes.py

@@ -1753,3 +1753,30 @@ def test_pkgbase_notify(client: TestClient, user: User, package: Package):
         PackageNotification.UserID == user.ID
     ).first()
     assert notif is None
+
+
+def test_pkgbase_vote(client: TestClient, user: User, package: Package):
+    pkgbase = package.PackageBase
+
+    # We haven't voted yet.
+    vote = pkgbase.package_votes.filter(PackageVote.UsersID == user.ID).first()
+    assert vote is None
+
+    # Vote for the package.
+    cookies = {"AURSID": user.login(Request(), "testPassword")}
+    endpoint = f"/pkgbase/{pkgbase.Name}/vote"
+    with client as request:
+        resp = request.post(endpoint, cookies=cookies)
+    assert resp.status_code == int(HTTPStatus.SEE_OTHER)
+
+    vote = pkgbase.package_votes.filter(PackageVote.UsersID == user.ID).first()
+    assert vote is not None
+
+    # Remove vote.
+    endpoint = f"/pkgbase/{pkgbase.Name}/unvote"
+    with client as request:
+        resp = request.post(endpoint, cookies=cookies)
+    assert resp.status_code == int(HTTPStatus.SEE_OTHER)
+
+    vote = pkgbase.package_votes.filter(PackageVote.UsersID == user.ID).first()
+    assert vote is None