debug, deleting assets

handler/CMakeLists.txt

@@ -10,6 +10,7 @@ set(HEADERS
     poll.h
     listassets.h
     addasset.h
+    deleteasset.cpp
 )
 
 set(SOURCES
@@ -21,6 +22,7 @@ set(SOURCES
     poll.cpp
     listassets.cpp
     addasset.cpp
+    deleteasset.cpp
 )
 
 target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})

handler/addasset.cpp

@@ -68,8 +68,3 @@ void Handler::AddAsset::handle (Request& request) {
         return error(request, Response::Status::internalError);
     }
 }
-
-void Handler::AddAsset::error (Request& request, Response::Status status) {
-    Response& res = request.createResponse(status);
-    res.send();
-}

handler/addasset.h

@@ -14,7 +14,6 @@ public:
     AddAsset (std::weak_ptr<Server> server);
     virtual void handle (Request& request) override;
 
-    static void error (Request& request, Response::Status status);
 private:
     std::weak_ptr<Server> server;
 };

handler/deleteasset.cpp

@@ -0,0 +1,59 @@
+//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
+//SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "deleteasset.h"
+
+#include "server/server.h"
+#include "server/session.h"
+#include "database/exceptions.h"
+
+Handler::DeleteAsset::DeleteAsset (std::weak_ptr<Server> server):
+    Handler("deleteAsset", Request::Method::post),
+    server(server)
+{}
+
+void Handler::DeleteAsset::handle (Request& request) {
+    std::string access = request.getAuthorizationToken();
+    if (access.empty())
+        return error(request, Response::Status::unauthorized);
+
+    if (access.size() != 32)
+        return error(request, Response::Status::badRequest);
+
+    std::shared_ptr<Server> srv = server.lock();
+    if (!srv)
+        return error(request, Response::Status::internalError);
+
+    std::map form = request.getForm();
+    std::map<std::string, std::string>::const_iterator itr = form.find("id");
+    if (itr == form.end())
+        return error(request, Response::Status::badRequest);
+
+    unsigned int assetId;
+    try {
+        assetId = std::stoul(itr->second);
+    } catch (const std::exception& e) {
+        return error(request, Response::Status::badRequest);
+    }
+
+    try {
+        Session& session = srv->getSession(access);
+        bool success = srv->getDatabase()->deleteAsset(assetId, session.owner);
+        if (!success)
+            return error(request, Response::Status::forbidden);
+
+        Response& res = request.createResponse(Response::Status::ok);
+        res.send();
+
+        session.assetRemoved(assetId);
+
+    } catch (const DB::NoSession& e) {
+        return error(request, Response::Status::unauthorized);
+    } catch (const std::exception& e) {
+        std::cerr << "Exception on poll:\n\t" << e.what() << std::endl;
+        return error(request, Response::Status::internalError);
+    }  catch (...) {
+        std::cerr << "Unknown exception on poll" << std::endl;
+        return error(request, Response::Status::internalError);
+    }
+}

handler/deleteasset.h

@@ -0,0 +1,21 @@
+//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
+//SPDX-License-Identifier: GPL-3.0-or-later
+
+#pragma once
+
+#include <memory>
+
+#include "handler.h"
+
+class Server;
+namespace Handler {
+class DeleteAsset : public Handler {
+public:
+    DeleteAsset (std::weak_ptr<Server> server);
+
+    virtual void handle (Request& request) override;
+
+private:
+    std::weak_ptr<Server> server;
+};
+}

handler/handler.cpp

@@ -9,3 +9,8 @@ Handler::Handler::Handler(const std::string& path, Request::Method method):
 {}
 
 Handler::Handler::~Handler() {}
+
+void Handler::Handler::error (Request& request, Response::Status status) {
+    Response& res = request.createResponse(status);
+    res.send();
+}

handler/handler.h

@@ -15,6 +15,9 @@ class Handler {
 protected:
     Handler(const std::string& path, Request::Method method);
 
+protected:
+    static void error (Request& request, Response::Status status);
+
 public:
     virtual ~Handler();
 

handler/listassets.cpp

@@ -49,8 +49,3 @@ void Handler::ListAssets::handle (Request& request) {
         return error(request, Response::Status::internalError);
     }
 }
-
-void Handler::ListAssets::error (Request& request, Response::Status status) {
-    Response& res = request.createResponse(status);
-    res.send();
-}

handler/listassets.h

@@ -14,7 +14,6 @@ public:
     ListAssets (std::weak_ptr<Server> server);
     void handle (Request& request) override;
 
-    static void error (Request& request, Response::Status status);
 private:
     std::weak_ptr<Server> server;
 };