Compare commits
28 Commits
69a2c70fd9
...
498501332d
Author | SHA1 | Date | |
---|---|---|---|
498501332d | |||
ccb5f00f69 | |||
7c4adaf450 | |||
e1d5b6c76c | |||
c2d4bf5ccb | |||
973deaefd9 | |||
4914a467e5 | |||
a9f46b2ab0 | |||
07003c2fe6 | |||
db37abacd2 | |||
a2c2c2a883 | |||
19d786631a | |||
2e01fe8d67 | |||
4df8d4319e | |||
d33ec5def8 | |||
a1ab1339e3 | |||
5d765958e5 | |||
544db92b6e | |||
26114aad5f | |||
f1a2006b4b | |||
fe2fbb9ad0 | |||
59c1ffd027 | |||
4b87b560ac | |||
534c282226 | |||
99a9fd507e | |||
0c50cfa639 | |||
3fe6d25448 | |||
f0d205dee7 |
@ -1,3 +1,6 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
cmake_minimum_required(VERSION 3.5)
|
cmake_minimum_required(VERSION 3.5)
|
||||||
project(pica
|
project(pica
|
||||||
VERSION 0.0.1
|
VERSION 0.0.1
|
||||||
@ -13,28 +16,59 @@ include(GNUInstallDirs)
|
|||||||
set(CMAKE_CXX_STANDARD 17)
|
set(CMAKE_CXX_STANDARD 17)
|
||||||
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
||||||
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake")
|
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake")
|
||||||
set(PICA_BIN_DIR ${CMAKE_CURRENT_BINARY_DIR})
|
|
||||||
|
if (NOT CMAKE_BUILD_TYPE)
|
||||||
|
set(CMAKE_BUILD_TYPE Debug)
|
||||||
|
endif()
|
||||||
|
message("Build type: ${CMAKE_BUILD_TYPE}")
|
||||||
|
|
||||||
|
set(COMPILE_OPTIONS -fno-sized-deallocation)
|
||||||
|
if (CMAKE_BUILD_TYPE STREQUAL Release)
|
||||||
|
list(APPEND COMPILE_OPTIONS -O3)
|
||||||
|
elseif (CMAKE_BUILD_TYPE STREQUAL Debug)
|
||||||
|
list(APPEND COMPILE_OPTIONS -g)
|
||||||
|
list(APPEND COMPILE_OPTIONS -Wall)
|
||||||
|
list(APPEND COMPILE_OPTIONS -Wextra)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
set(COMPILE_OPTIONS_STRING "")
|
||||||
|
foreach(element IN LISTS COMPILE_OPTIONS)
|
||||||
|
if(NOT COMPILE_OPTIONS_STRING STREQUAL "")
|
||||||
|
set(COMPILE_OPTIONS_STRING "${COMPILE_OPTIONS_STRING} ")
|
||||||
|
endif()
|
||||||
|
set(COMPILE_OPTIONS_STRING "${COMPILE_OPTIONS_STRING}${element}")
|
||||||
|
endforeach()
|
||||||
|
message("Compile options: " ${COMPILE_OPTIONS_STRING})
|
||||||
|
|
||||||
find_package(nlohmann_json REQUIRED)
|
find_package(nlohmann_json REQUIRED)
|
||||||
find_package(FCGI REQUIRED)
|
find_package(FCGI REQUIRED)
|
||||||
|
find_package(Argon2 REQUIRED)
|
||||||
|
find_package(Threads REQUIRED)
|
||||||
|
|
||||||
add_executable(pica main.cpp)
|
add_executable(${PROJECT_NAME} main.cpp)
|
||||||
target_include_directories(pica PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
|
target_include_directories(${PROJECT_NAME} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
|
||||||
target_include_directories(pica PRIVATE ${CMAKE_CURRENT_BINARY_DIR})
|
target_include_directories(${PROJECT_NAME} PRIVATE ${CMAKE_CURRENT_BINARY_DIR})
|
||||||
|
target_compile_options(${PROJECT_NAME} PRIVATE ${COMPILE_OPTIONS})
|
||||||
|
|
||||||
add_subdirectory(server)
|
add_subdirectory(server)
|
||||||
|
add_subdirectory(handler)
|
||||||
add_subdirectory(request)
|
add_subdirectory(request)
|
||||||
add_subdirectory(response)
|
add_subdirectory(response)
|
||||||
add_subdirectory(stream)
|
add_subdirectory(stream)
|
||||||
add_subdirectory(database)
|
add_subdirectory(database)
|
||||||
add_subdirectory(utils)
|
add_subdirectory(utils)
|
||||||
|
add_subdirectory(taskmanager)
|
||||||
|
|
||||||
configure_file(config.h.in config.h @ONLY)
|
configure_file(config.h.in config.h @ONLY)
|
||||||
|
configure_file(run.sh.in run.sh @ONLY)
|
||||||
|
execute_process(COMMAND chmod +x ${CMAKE_CURRENT_BINARY_DIR}/run.sh)
|
||||||
|
|
||||||
target_link_libraries(pica PRIVATE
|
target_link_libraries(${PROJECT_NAME} PRIVATE
|
||||||
FCGI::FCGI
|
FCGI::FCGI
|
||||||
FCGI::FCGI++
|
FCGI::FCGI++
|
||||||
nlohmann_json::nlohmann_json
|
nlohmann_json::nlohmann_json
|
||||||
|
Argon2::Argon2
|
||||||
|
Threads::Threads
|
||||||
)
|
)
|
||||||
|
|
||||||
install(TARGETS pica RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
|
install(TARGETS ${PROJECT_NAME} RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
- fcgi
|
- fcgi
|
||||||
- nlohmann_json
|
- nlohmann_json
|
||||||
- mariadb-client
|
- mariadb-client
|
||||||
|
- argon2
|
||||||
|
|
||||||
|
|
||||||
### Building
|
### Building
|
||||||
|
29
cmake/FindArgon2.cmake
Normal file
29
cmake/FindArgon2.cmake
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
find_library(Argon2_LIBRARIES argon2)
|
||||||
|
find_path(Argon2_INCLUDE_DIR argon2.h)
|
||||||
|
|
||||||
|
if (Argon2_LIBRARIES AND Argon2_INCLUDE_DIR)
|
||||||
|
set(Argon2_FOUND TRUE)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
if (Argon2_FOUND)
|
||||||
|
add_library(Argon2::Argon2 SHARED IMPORTED)
|
||||||
|
set_target_properties(Argon2::Argon2 PROPERTIES
|
||||||
|
IMPORTED_LOCATION "${Argon2_LIBRARIES}"
|
||||||
|
INTERFACE_LINK_LIBRARIES "${Argon2_LIBRARIES}"
|
||||||
|
INTERFACE_INCLUDE_DIRECTORIES ${Argon2_INCLUDE_DIR}
|
||||||
|
)
|
||||||
|
|
||||||
|
if (NOT Argon2_FIND_QUIETLY)
|
||||||
|
message(STATUS "Found Argon2 includes: ${Argon2_INCLUDE_DIR}")
|
||||||
|
message(STATUS "Found Argon2 library: ${Argon2_LIBRARIES}")
|
||||||
|
endif ()
|
||||||
|
else ()
|
||||||
|
if (Argon2_FIND_REQUIRED)
|
||||||
|
message(FATAL_ERROR "Could NOT find Argon2 development files")
|
||||||
|
endif ()
|
||||||
|
endif ()
|
||||||
|
|
||||||
|
|
@ -1,3 +1,6 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
find_library(FCGI_LIBRARIES fcgi NAMES FCGI libfcgi)
|
find_library(FCGI_LIBRARIES fcgi NAMES FCGI libfcgi)
|
||||||
find_library(FCGI++_LIBRARIES fcgi++ NAMES FCGI++ libfcgi++)
|
find_library(FCGI++_LIBRARIES fcgi++ NAMES FCGI++ libfcgi++)
|
||||||
|
|
||||||
|
@ -1,3 +1,6 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-laterlater
|
||||||
|
|
||||||
find_library(MariaDB_CLIENT_LIBRARIES mysqlclient NAMES mariadbclient)
|
find_library(MariaDB_CLIENT_LIBRARIES mysqlclient NAMES mariadbclient)
|
||||||
find_path(MariaDB_INCLUDE_DIR mysql/mysql.h)
|
find_path(MariaDB_INCLUDE_DIR mysql/mysql.h)
|
||||||
|
|
||||||
|
@ -1,3 +1,6 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#define FULL_DATA_DIR "@CMAKE_INSTALL_FULL_DATADIR@"
|
#define FULL_DATA_DIR "@CMAKE_INSTALL_FULL_DATADIR@"
|
||||||
@ -7,3 +10,4 @@
|
|||||||
#define BIN_DIR "@CMAKE_INSTALL_BINDIR@"
|
#define BIN_DIR "@CMAKE_INSTALL_BINDIR@"
|
||||||
|
|
||||||
#define PROJECT_NAME "@PROJECT_NAME@"
|
#define PROJECT_NAME "@PROJECT_NAME@"
|
||||||
|
#define PROJECT_VERSION "@PROJECT_VERSION@"
|
||||||
|
@ -1,12 +1,22 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
set(HEADERS
|
set(HEADERS
|
||||||
dbinterface.h
|
interface.h
|
||||||
|
exceptions.h
|
||||||
|
pool.h
|
||||||
|
resource.h
|
||||||
)
|
)
|
||||||
|
|
||||||
set(SOURCES
|
set(SOURCES
|
||||||
dbinterface.cpp
|
interface.cpp
|
||||||
|
exceptions.cpp
|
||||||
|
pool.cpp
|
||||||
|
resource.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
target_sources(pica PRIVATE ${SOURCES})
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
||||||
|
|
||||||
add_subdirectory(mysql)
|
add_subdirectory(mysql)
|
||||||
add_subdirectory(migrations)
|
add_subdirectory(migrations)
|
||||||
|
add_subdirectory(schema)
|
||||||
|
@ -1,26 +0,0 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
|
|
||||||
#include "dbinterface.h"
|
|
||||||
|
|
||||||
#include "mysql/mysql.h"
|
|
||||||
|
|
||||||
DBInterface::DBInterface(Type type):
|
|
||||||
type(type),
|
|
||||||
state(State::disconnected)
|
|
||||||
{}
|
|
||||||
|
|
||||||
DBInterface::~DBInterface() {}
|
|
||||||
|
|
||||||
std::unique_ptr<DBInterface> DBInterface::create(Type type) {
|
|
||||||
switch (type) {
|
|
||||||
case Type::mysql:
|
|
||||||
return std::make_unique<MySQL>();
|
|
||||||
}
|
|
||||||
|
|
||||||
throw std::runtime_error("Unexpected database type: " + std::to_string((uint8_t)type));
|
|
||||||
}
|
|
||||||
|
|
||||||
DBInterface::State DBInterface::currentState() const {
|
|
||||||
return state;
|
|
||||||
}
|
|
@ -1,44 +0,0 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
#include <stdexcept>
|
|
||||||
#include <string>
|
|
||||||
#include <memory>
|
|
||||||
#include <stdint.h>
|
|
||||||
|
|
||||||
class DBInterface {
|
|
||||||
public:
|
|
||||||
enum class Type {
|
|
||||||
mysql
|
|
||||||
};
|
|
||||||
enum class State {
|
|
||||||
disconnected,
|
|
||||||
connecting,
|
|
||||||
connected
|
|
||||||
};
|
|
||||||
static std::unique_ptr<DBInterface> create(Type type);
|
|
||||||
|
|
||||||
virtual ~DBInterface();
|
|
||||||
|
|
||||||
State currentState() const;
|
|
||||||
|
|
||||||
const Type type;
|
|
||||||
|
|
||||||
public:
|
|
||||||
virtual void connect(const std::string& path) = 0;
|
|
||||||
virtual void disconnect() = 0;
|
|
||||||
virtual void setDatabase(const std::string& newDatabase) = 0;
|
|
||||||
virtual void setCredentials(const std::string& login, const std::string& password) = 0;
|
|
||||||
|
|
||||||
virtual void migrate(uint8_t targetVersion) = 0;
|
|
||||||
virtual uint8_t getVersion() = 0;
|
|
||||||
virtual void setVersion(uint8_t version) = 0;
|
|
||||||
|
|
||||||
protected:
|
|
||||||
DBInterface(Type type);
|
|
||||||
|
|
||||||
protected:
|
|
||||||
State state;
|
|
||||||
};
|
|
24
database/exceptions.cpp
Normal file
24
database/exceptions.cpp
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "exceptions.h"
|
||||||
|
|
||||||
|
DB::Duplicate::Duplicate(const std::string& text):
|
||||||
|
std::runtime_error(text)
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::DuplicateLogin::DuplicateLogin(const std::string& text):
|
||||||
|
Duplicate(text)
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::EmptyResult::EmptyResult(const std::string& text):
|
||||||
|
std::runtime_error(text)
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::NoLogin::NoLogin(const std::string& text):
|
||||||
|
EmptyResult(text)
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::NoSession::NoSession (const std::string& text):
|
||||||
|
EmptyResult(text)
|
||||||
|
{}
|
34
database/exceptions.h
Normal file
34
database/exceptions.h
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <stdexcept>
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Duplicate : public std::runtime_error {
|
||||||
|
public:
|
||||||
|
explicit Duplicate(const std::string& text);
|
||||||
|
};
|
||||||
|
|
||||||
|
class DuplicateLogin : public Duplicate {
|
||||||
|
public:
|
||||||
|
explicit DuplicateLogin(const std::string& text);
|
||||||
|
};
|
||||||
|
|
||||||
|
class EmptyResult : public std::runtime_error {
|
||||||
|
public:
|
||||||
|
explicit EmptyResult(const std::string& text);
|
||||||
|
};
|
||||||
|
|
||||||
|
class NoLogin : public EmptyResult {
|
||||||
|
public:
|
||||||
|
explicit NoLogin(const std::string& text);
|
||||||
|
};
|
||||||
|
|
||||||
|
class NoSession : public EmptyResult {
|
||||||
|
public:
|
||||||
|
explicit NoSession(const std::string& text);
|
||||||
|
};
|
||||||
|
}
|
26
database/interface.cpp
Normal file
26
database/interface.cpp
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "interface.h"
|
||||||
|
|
||||||
|
#include "mysql/mysql.h"
|
||||||
|
|
||||||
|
DB::Interface::Interface(Type type):
|
||||||
|
type(type),
|
||||||
|
state(State::disconnected)
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Interface::~Interface() {}
|
||||||
|
|
||||||
|
std::unique_ptr<DB::Interface> DB::Interface::create(Type type) {
|
||||||
|
switch (type) {
|
||||||
|
case Type::mysql:
|
||||||
|
return std::make_unique<MySQL>();
|
||||||
|
}
|
||||||
|
|
||||||
|
throw std::runtime_error("Unexpected database type: " + std::to_string((uint8_t)type));
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Interface::State DB::Interface::currentState() const {
|
||||||
|
return state;
|
||||||
|
}
|
70
database/interface.h
Normal file
70
database/interface.h
Normal file
@ -0,0 +1,70 @@
|
|||||||
|
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <string>
|
||||||
|
#include <memory>
|
||||||
|
#include <vector>
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
|
#include "schema/session.h"
|
||||||
|
#include "schema/asset.h"
|
||||||
|
#include "schema/currency.h"
|
||||||
|
#include "schema/transaction.h"
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Interface {
|
||||||
|
public:
|
||||||
|
enum class Type {
|
||||||
|
mysql
|
||||||
|
};
|
||||||
|
enum class State {
|
||||||
|
disconnected,
|
||||||
|
connecting,
|
||||||
|
connected
|
||||||
|
};
|
||||||
|
static std::unique_ptr<Interface> create(Type type);
|
||||||
|
|
||||||
|
virtual ~Interface();
|
||||||
|
|
||||||
|
State currentState() const;
|
||||||
|
|
||||||
|
const Type type;
|
||||||
|
|
||||||
|
public:
|
||||||
|
virtual void connect(const std::string& path) = 0;
|
||||||
|
virtual void disconnect() = 0;
|
||||||
|
virtual void setDatabase(const std::string& newDatabase) = 0;
|
||||||
|
virtual void setCredentials(const std::string& login, const std::string& password) = 0;
|
||||||
|
|
||||||
|
virtual void migrate(uint8_t targetVersion) = 0;
|
||||||
|
virtual uint8_t getVersion() = 0;
|
||||||
|
virtual void setVersion(uint8_t version) = 0;
|
||||||
|
|
||||||
|
virtual uint32_t registerAccount(const std::string& login, const std::string& hash) = 0;
|
||||||
|
virtual std::string getAccountHash(const std::string& login) = 0;
|
||||||
|
|
||||||
|
virtual Session createSession(const std::string& login, const std::string& access, const std::string& renew) = 0;
|
||||||
|
virtual Session findSession(const std::string& accessToken) = 0;
|
||||||
|
|
||||||
|
virtual std::vector<Asset> listAssets(uint32_t owner) = 0;
|
||||||
|
virtual Asset addAsset(const Asset& asset) = 0;
|
||||||
|
virtual void updateAsset(const Asset& asset) = 0;
|
||||||
|
virtual bool deleteAsset(uint32_t assetId, uint32_t actorId) = 0;
|
||||||
|
|
||||||
|
virtual std::vector<Currency> listUsedCurrencies(uint32_t owner) = 0;
|
||||||
|
|
||||||
|
virtual DB::Transaction addTransaction(const DB::Transaction& transaction) = 0;
|
||||||
|
virtual std::vector<DB::Transaction> listTransactions(uint32_t owner) = 0;
|
||||||
|
virtual void updateTransaction(const DB::Transaction& transaction) = 0;
|
||||||
|
virtual bool deleteTransaction(uint32_t id, uint32_t actorId) = 0;
|
||||||
|
|
||||||
|
protected:
|
||||||
|
Interface(Type type);
|
||||||
|
|
||||||
|
protected:
|
||||||
|
State state;
|
||||||
|
};
|
||||||
|
}
|
@ -1,5 +1,8 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-laterater
|
||||||
|
|
||||||
set(MIGRATIONS migrations)
|
set(MIGRATIONS migrations)
|
||||||
configure_file(m0.sql ${PICA_BIN_DIR}/${CMAKE_INSTALL_DATADIR}/${MIGRATIONS}/m0.sql COPYONLY)
|
configure_file(m0.sql ${PROJECT_BINARY_DIR}/${CMAKE_INSTALL_DATADIR}/${MIGRATIONS}/m0.sql COPYONLY)
|
||||||
|
|
||||||
install(
|
install(
|
||||||
FILES
|
FILES
|
||||||
|
@ -1,6 +1,192 @@
|
|||||||
|
--creating system table
|
||||||
CREATE TABLE IF NOT EXISTS system (
|
CREATE TABLE IF NOT EXISTS system (
|
||||||
`key` VARCHAR(32) PRIMARY KEY,
|
`key` VARCHAR(32) PRIMARY KEY,
|
||||||
`value` TEXT
|
`value` TEXT
|
||||||
);
|
);
|
||||||
|
|
||||||
INSERT INTO system (`key`, `value`) VALUES ('version', '0');
|
--creating roles table
|
||||||
|
CREATE TABLE IF NOT EXISTS roles (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`name` VARCHAR(256) UNIQUE NOT NULL,
|
||||||
|
`color` INTEGER UNSIGNED DEFAULT 0
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating accounts table
|
||||||
|
CREATE TABLE IF NOT EXISTS accounts (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`login` VARCHAR(256) UNIQUE NOT NULL,
|
||||||
|
`nick` VARCHAR(256),
|
||||||
|
`type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`password` VARCHAR(128),
|
||||||
|
`created` TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating role bindings table
|
||||||
|
CREATE TABLE IF NOT EXISTS roleBindings (
|
||||||
|
`account` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`role` INTEGER UNSIGNED NOT NULL,
|
||||||
|
|
||||||
|
PRIMARY KEY (account, role),
|
||||||
|
FOREIGN KEY (account) REFERENCES accounts(id),
|
||||||
|
FOREIGN KEY (role) REFERENCES roles(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating sessions table
|
||||||
|
CREATE TABLE IF NOT EXISTS sessions (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`owner` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`started` TIMESTAMP,
|
||||||
|
`latest` TIMESTAMP,
|
||||||
|
`access` CHAR(32) NOT NULL UNIQUE,
|
||||||
|
`renew` CHAR(32),
|
||||||
|
`persist` BOOLEAN NOT NULL,
|
||||||
|
`device` TEXT,
|
||||||
|
|
||||||
|
FOREIGN KEY (owner) REFERENCES accounts(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating currencies table
|
||||||
|
CREATE TABLE IF NOT EXISTS currencies (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`code` VARCHAR(16) NOT NULL UNIQUE,
|
||||||
|
`title` VARCHAR(256),
|
||||||
|
`manual` BOOLEAN NOT NULL,
|
||||||
|
`created` TIMESTAMP,
|
||||||
|
`type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`value` DECIMAL (20, 5) NOT NULL,
|
||||||
|
`source` TEXT,
|
||||||
|
`description` TEXT,
|
||||||
|
`icon` VARCHAR(256),
|
||||||
|
|
||||||
|
INDEX manual_idx (manual)
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating assets table
|
||||||
|
CREATE TABLE IF NOT EXISTS assets (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`owner` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`currency` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`title` VARCHAR(256),
|
||||||
|
`icon` VARCHAR(256),
|
||||||
|
`color` INTEGER UNSIGNED DEFAULT 0,
|
||||||
|
`balance` DECIMAL (20, 5) DEFAULT 0,
|
||||||
|
`type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`archived` BOOLEAN DEFAULT FALSE,
|
||||||
|
`created` TIMESTAMP,
|
||||||
|
|
||||||
|
INDEX owner_idx (owner),
|
||||||
|
INDEX archived_idx (archived),
|
||||||
|
|
||||||
|
FOREIGN KEY (owner) REFERENCES accounts(id),
|
||||||
|
FOREIGN KEY (currency) REFERENCES currencies(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating parties table
|
||||||
|
CREATE TABLE IF NOT EXISTS parties (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`title` VARCHAR(256) NOT NULL UNIQUE
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating transactions table
|
||||||
|
CREATE TABLE IF NOT EXISTS transactions (
|
||||||
|
`id` INTEGER UNSIGNED AUTO_INCREMENT PRIMARY KEY,
|
||||||
|
`initiator` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`asset` INTEGER UNSIGNED NOT NULL,
|
||||||
|
`parent` INTEGER UNSIGNED,
|
||||||
|
`value` DECIMAL (20, 5) NOT NULL,
|
||||||
|
`state` INTEGER UNSIGNED DEFAULT 0,
|
||||||
|
`modified` TIMESTAMP,
|
||||||
|
`performed` TIMESTAMP,
|
||||||
|
`party` INTEGER UNSIGNED,
|
||||||
|
`notes` TEXT,
|
||||||
|
|
||||||
|
INDEX initiator_idx (initiator),
|
||||||
|
INDEX parent_idx (parent),
|
||||||
|
INDEX asset_idx (asset),
|
||||||
|
INDEX performed_idx (performed),
|
||||||
|
INDEX modified_idx (modified),
|
||||||
|
INDEX party_idx (party),
|
||||||
|
|
||||||
|
FOREIGN KEY (initiator) REFERENCES accounts(id),
|
||||||
|
FOREIGN KEY (asset) REFERENCES assets(id),
|
||||||
|
FOREIGN KEY (parent) REFERENCES transactions(id),
|
||||||
|
FOREIGN KEY (party) REFERENCES parties(id)
|
||||||
|
);
|
||||||
|
|
||||||
|
--creating trigger before insert accounts
|
||||||
|
CREATE TRIGGER before_insert_accounts
|
||||||
|
BEFORE INSERT ON accounts
|
||||||
|
FOR EACH ROW
|
||||||
|
BEGIN
|
||||||
|
SET NEW.created = UTC_TIMESTAMP();
|
||||||
|
END;
|
||||||
|
|
||||||
|
--creating trigger before insert sessions
|
||||||
|
CREATE TRIGGER before_insert_sessions
|
||||||
|
BEFORE INSERT ON sessions
|
||||||
|
FOR EACH ROW
|
||||||
|
BEGIN
|
||||||
|
SET NEW.started = UTC_TIMESTAMP();
|
||||||
|
SET NEW.latest = UTC_TIMESTAMP();
|
||||||
|
END;
|
||||||
|
|
||||||
|
--creating trigger before insert currencies
|
||||||
|
CREATE TRIGGER before_insert_currencies
|
||||||
|
BEFORE INSERT ON currencies
|
||||||
|
FOR EACH ROW
|
||||||
|
BEGIN
|
||||||
|
IF NEW.created IS NULL THEN
|
||||||
|
SET NEW.created = UTC_TIMESTAMP();
|
||||||
|
END IF;
|
||||||
|
END;
|
||||||
|
|
||||||
|
--creating trigger before insert assets
|
||||||
|
CREATE TRIGGER before_insert_assets
|
||||||
|
BEFORE INSERT ON assets
|
||||||
|
FOR EACH ROW
|
||||||
|
BEGIN
|
||||||
|
IF NEW.created IS NULL THEN
|
||||||
|
SET NEW.created = UTC_TIMESTAMP();
|
||||||
|
END IF;
|
||||||
|
END;
|
||||||
|
|
||||||
|
--creating trigger before insert transactions
|
||||||
|
CREATE TRIGGER before_insert_transactions
|
||||||
|
BEFORE INSERT ON transactions
|
||||||
|
FOR EACH ROW
|
||||||
|
BEGIN
|
||||||
|
SET NEW.modified = UTC_TIMESTAMP();
|
||||||
|
IF NEW.performed IS NULL THEN
|
||||||
|
SET NEW.performed = UTC_TIMESTAMP();
|
||||||
|
END IF;
|
||||||
|
END;
|
||||||
|
|
||||||
|
--creating trigger before update transactions
|
||||||
|
CREATE TRIGGER before_update_transactions
|
||||||
|
BEFORE UPDATE ON transactions
|
||||||
|
FOR EACH ROW
|
||||||
|
BEGIN
|
||||||
|
SET NEW.modified = UTC_TIMESTAMP();
|
||||||
|
END;
|
||||||
|
|
||||||
|
--creating default roles
|
||||||
|
INSERT IGNORE INTO
|
||||||
|
roles (`name`)
|
||||||
|
VALUES ('root'),
|
||||||
|
('default');
|
||||||
|
|
||||||
|
--inserting initial version
|
||||||
|
INSERT IGNORE INTO
|
||||||
|
system (`key`, `value`)
|
||||||
|
VALUES ('version', '0');
|
||||||
|
|
||||||
|
--recording initial time
|
||||||
|
INSERT IGNORE INTO
|
||||||
|
system (`key`, `value`)
|
||||||
|
VALUES ('created', UTC_TIMESTAMP());
|
||||||
|
|
||||||
|
--creating default currencies
|
||||||
|
INSERT IGNORE INTO
|
||||||
|
currencies (`code`, `title`, `manual`, `description`, `type`, `value`, `icon`)
|
||||||
|
VALUES ('USD', 'United States Dollar', TRUE, 'Base currency', 1, 1, 'currency-usd');
|
||||||
|
@ -1,15 +1,20 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-laterr
|
||||||
|
|
||||||
set(HEADERS
|
set(HEADERS
|
||||||
mysql.h
|
mysql.h
|
||||||
statement.h
|
statement.h
|
||||||
|
transaction.h
|
||||||
)
|
)
|
||||||
|
|
||||||
set(SOURCES
|
set(SOURCES
|
||||||
mysql.cpp
|
mysql.cpp
|
||||||
statement.cpp
|
statement.cpp
|
||||||
|
transaction.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
find_package(MariaDB REQUIRED)
|
find_package(MariaDB REQUIRED)
|
||||||
|
|
||||||
target_sources(pica PRIVATE ${SOURCES})
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
||||||
|
|
||||||
target_link_libraries(pica PRIVATE MariaDB::client)
|
target_link_libraries(${PROJECT_NAME} PRIVATE MariaDB::client)
|
||||||
|
@ -1,27 +1,63 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "mysql.h"
|
#include "mysql.h"
|
||||||
|
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
#include <chrono>
|
||||||
|
|
||||||
#include "mysqld_error.h"
|
#include "mysqld_error.h"
|
||||||
|
|
||||||
#include "statement.h"
|
#include "statement.h"
|
||||||
|
#include "transaction.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
constexpr const char* versionQuery = "SELECT value FROM system WHERE `key` = 'version'";
|
||||||
constexpr const char* updateQuery = "UPDATE system SET `value` = ? WHERE `key` = 'version'";
|
constexpr const char* updateQuery = "UPDATE system SET `value` = ? WHERE `key` = 'version'";
|
||||||
|
constexpr const char* registerQuery = "INSERT INTO accounts (`login`, `type`, `password`) VALUES (?, 1, ?)";
|
||||||
|
constexpr const char* lastIdQuery = "SELECT LAST_INSERT_ID() AS id";
|
||||||
|
constexpr const char* assignRoleQuery = "INSERT INTO roleBindings (`account`, `role`) SELECT ?, roles.id FROM roles WHERE roles.name = ?";
|
||||||
|
constexpr const char* selectHash = "SELECT password FROM accounts where login = ?";
|
||||||
|
constexpr const char* createSessionQuery = "INSERT INTO sessions (`owner`, `access`, `renew`, `persist`, `device`)"
|
||||||
|
" SELECT accounts.id, ?, ?, true, ? FROM accounts WHERE accounts.login = ?"
|
||||||
|
" RETURNING id, owner";
|
||||||
|
constexpr const char* selectSession = "SELECT id, owner, access, renew FROM sessions where access = ?";
|
||||||
|
constexpr const char* selectAssets = "SELECT id, owner, currency, title, icon, color, archived FROM assets where owner = ?";
|
||||||
|
constexpr const char* insertAsset = "INSERT INTO assets (`owner`, `currency`, `title`, `icon`, `color`, `archived`, `type`)"
|
||||||
|
" VALUES (?, ?, ?, ?, ?, ?, 1)";
|
||||||
|
constexpr const char* updateAssetQuery = "UPDATE assets SET `owner` = ?, `currency` = ?, `title` = ?, `icon` = ?, `color` = ?, `archived` = ?"
|
||||||
|
" WHERE `id` = ?";
|
||||||
|
constexpr const char* removeAsset = "DELETE FROM assets where `id` = ? AND `owner` = ?";
|
||||||
|
constexpr const char* selectUsedCurrencies = "SELECT DISTINCT c.id, c.code, c.title, c.manual, c.icon FROM currencies c"
|
||||||
|
" JOIN assets a ON c.id = a.currency"
|
||||||
|
" WHERE a.owner = ?";
|
||||||
|
constexpr const char* addTransactionQuery = "INSERT INTO transactions"
|
||||||
|
" (`initiator`, `type`, `asset`, `parent`, `value`, `performed`)"
|
||||||
|
" VALUES (?, 1, ?, ?, ?, ?)";
|
||||||
|
constexpr const char* updateTransactionQuery = "UPDATE transactions SET"
|
||||||
|
" `initiator` = ?, `type` = 1, `asset` = ?,"
|
||||||
|
" `parent` = ?, `value` = ?, `performed` = ?"
|
||||||
|
" WHERE `id` = ?";
|
||||||
|
constexpr const char* deleteTransactionQuery = "DELETE FROM transactions where (`id` = ? OR `parent` = ?) AND `initiator` = ?";
|
||||||
|
constexpr const char* selectAllTransactions = "WITH RECURSIVE AllTransactions AS ("
|
||||||
|
" SELECT t.id, t.initiator, t.asset, t.parent, t.value, t.modified, t.performed t.notes FROM transactions t"
|
||||||
|
" JOIN assets a ON t.asset = a.id"
|
||||||
|
" WHERE a.owner = ?"
|
||||||
|
" UNION ALL"
|
||||||
|
|
||||||
|
" SELECT t.id, t.initiator, t.asset, t.parent, t.value, t.modified, t.performed t.notes FROM transactions t"
|
||||||
|
" JOIN AllTransactions at ON t.id = at.parent)"
|
||||||
|
|
||||||
|
" SELECT DISTINCT id, initiator, asset, parent, value, modified, performed notes FROM AllTransactions"
|
||||||
|
" ORDER BY performed"
|
||||||
|
" LIMIT 100 OFFSET 0;";
|
||||||
|
|
||||||
|
|
||||||
static const std::filesystem::path buildSQLPath = "database";
|
static const std::filesystem::path buildSQLPath = "database";
|
||||||
|
|
||||||
struct ResDeleter {
|
DB::MySQL::MySQL ():
|
||||||
void operator () (MYSQL_RES* res) {
|
Interface(Type::mysql),
|
||||||
mysql_free_result(res);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
MySQL::MySQL():
|
|
||||||
DBInterface(Type::mysql),
|
|
||||||
connection(),
|
connection(),
|
||||||
login(),
|
login(),
|
||||||
password(),
|
password(),
|
||||||
@ -30,12 +66,11 @@ MySQL::MySQL():
|
|||||||
mysql_init(&connection);
|
mysql_init(&connection);
|
||||||
}
|
}
|
||||||
|
|
||||||
MySQL::~MySQL() {
|
DB::MySQL::~MySQL() {
|
||||||
mysql_close(&connection);
|
mysql_close(&connection);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void DB::MySQL::connect (const std::string& path) {
|
||||||
void MySQL::connect(const std::string& path) {
|
|
||||||
if (state != State::disconnected)
|
if (state != State::disconnected)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
@ -57,7 +92,7 @@ void MySQL::connect(const std::string& path) {
|
|||||||
state = State::connected;
|
state = State::connected;
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::setCredentials(const std::string& login, const std::string& password) {
|
void DB::MySQL::setCredentials (const std::string& login, const std::string& password) {
|
||||||
if (MySQL::login == login && MySQL::password == password)
|
if (MySQL::login == login && MySQL::password == password)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
@ -79,7 +114,7 @@ void MySQL::setCredentials(const std::string& login, const std::string& password
|
|||||||
throw std::runtime_error(std::string("Error changing credetials: ") + mysql_error(con));
|
throw std::runtime_error(std::string("Error changing credetials: ") + mysql_error(con));
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::setDatabase(const std::string& database) {
|
void DB::MySQL::setDatabase (const std::string& database) {
|
||||||
if (MySQL::database == database)
|
if (MySQL::database == database)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
@ -95,7 +130,7 @@ void MySQL::setDatabase(const std::string& database) {
|
|||||||
throw std::runtime_error(std::string("Error changing db: ") + mysql_error(con));
|
throw std::runtime_error(std::string("Error changing db: ") + mysql_error(con));
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::disconnect() {
|
void DB::MySQL::disconnect () {
|
||||||
if (state == State::disconnected)
|
if (state == State::disconnected)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
@ -104,7 +139,7 @@ void MySQL::disconnect() {
|
|||||||
mysql_init(con); //this is ridiculous!
|
mysql_init(con); //this is ridiculous!
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::executeFile(const std::filesystem::path& relativePath) {
|
void DB::MySQL::executeFile (const std::filesystem::path& relativePath) {
|
||||||
MYSQL* con = &connection;
|
MYSQL* con = &connection;
|
||||||
std::filesystem::path path = sharedPath() / relativePath;
|
std::filesystem::path path = sharedPath() / relativePath;
|
||||||
if (!std::filesystem::exists(path))
|
if (!std::filesystem::exists(path))
|
||||||
@ -114,9 +149,15 @@ void MySQL::executeFile(const std::filesystem::path& relativePath) {
|
|||||||
|
|
||||||
std::cout << "Executing file " << path << std::endl;
|
std::cout << "Executing file " << path << std::endl;
|
||||||
std::ifstream inputFile(path);
|
std::ifstream inputFile(path);
|
||||||
std::string query;
|
std::string block, comment;
|
||||||
while (std::getline(inputFile, query, ';')) {
|
while (getBlock(inputFile, block, comment)) {
|
||||||
int result = mysql_query(con, query.c_str());
|
if (!comment.empty())
|
||||||
|
std::cout << '\t' << comment << std::endl;
|
||||||
|
|
||||||
|
if (block.empty())
|
||||||
|
continue;
|
||||||
|
|
||||||
|
int result = mysql_query(con, block.c_str());
|
||||||
if (result != 0) {
|
if (result != 0) {
|
||||||
int errcode = mysql_errno(con);
|
int errcode = mysql_errno(con);
|
||||||
if (errcode == ER_EMPTY_QUERY)
|
if (errcode == ER_EMPTY_QUERY)
|
||||||
@ -128,9 +169,42 @@ void MySQL::executeFile(const std::filesystem::path& relativePath) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t MySQL::getVersion() {
|
bool DB::MySQL::getBlock(std::ifstream& file, std::string& block, std::string& name) {
|
||||||
|
if (file.eof())
|
||||||
|
return false;
|
||||||
|
|
||||||
|
block.clear();
|
||||||
|
name.clear();
|
||||||
|
|
||||||
|
if (file.peek() == '-') {
|
||||||
|
file.get();
|
||||||
|
if (file.peek() == '-') {
|
||||||
|
file.get();
|
||||||
|
std::getline(file, name);
|
||||||
|
} else {
|
||||||
|
file.unget();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
std::string line;
|
||||||
|
while (!file.eof()) {
|
||||||
|
if (file.peek() == '-')
|
||||||
|
return true;
|
||||||
|
|
||||||
|
if (!std::getline(file, line))
|
||||||
|
break;
|
||||||
|
|
||||||
|
if (!block.empty())
|
||||||
|
block.append(1, '\n');
|
||||||
|
|
||||||
|
block += line;
|
||||||
|
}
|
||||||
|
|
||||||
|
return !block.empty() || !name.empty();
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t DB::MySQL::getVersion () {
|
||||||
MYSQL* con = &connection;
|
MYSQL* con = &connection;
|
||||||
int result = mysql_query(con, "SELECT value FROM system WHERE `key` = 'version'");
|
int result = mysql_query(con, versionQuery);
|
||||||
|
|
||||||
if (result != 0) {
|
if (result != 0) {
|
||||||
unsigned int errcode = mysql_errno(con);
|
unsigned int errcode = mysql_errno(con);
|
||||||
@ -151,26 +225,284 @@ uint8_t MySQL::getVersion() {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::setVersion(uint8_t version) {
|
void DB::MySQL::setVersion (uint8_t version) {
|
||||||
std::string strVersion = std::to_string(version);
|
std::string strVersion = std::to_string(version);
|
||||||
Statement statement(&connection, updateQuery);
|
Statement statement(&connection, updateQuery);
|
||||||
statement.bind(strVersion.data(), MYSQL_TYPE_VAR_STRING);
|
statement.bind(strVersion.data(), MYSQL_TYPE_VAR_STRING);
|
||||||
statement.execute();
|
statement.execute();
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::migrate(uint8_t targetVersion) {
|
void DB::MySQL::migrate (uint8_t targetVersion) {
|
||||||
uint8_t currentVersion = getVersion();
|
uint8_t currentVersion = getVersion();
|
||||||
|
|
||||||
while (currentVersion < targetVersion) {
|
while (currentVersion < targetVersion) {
|
||||||
|
if (currentVersion == 255)
|
||||||
|
throw std::runtime_error("Maximum possible database version reached");
|
||||||
|
|
||||||
|
uint8_t nextVersion = currentVersion + 1;
|
||||||
std::string fileName = "migrations/m" + std::to_string(currentVersion) + ".sql";
|
std::string fileName = "migrations/m" + std::to_string(currentVersion) + ".sql";
|
||||||
std::cout << "Performing migration "
|
std::cout << "Performing migration "
|
||||||
<< std::to_string(currentVersion)
|
<< std::to_string(currentVersion)
|
||||||
<< " -> "
|
<< " -> "
|
||||||
<< std::to_string(++currentVersion)
|
<< std::to_string(nextVersion)
|
||||||
<< std::endl;
|
<< std::endl;
|
||||||
executeFile(fileName);
|
executeFile(fileName);
|
||||||
setVersion(currentVersion);
|
setVersion(nextVersion);
|
||||||
|
currentVersion = nextVersion;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "Database is now on actual version " << std::to_string(targetVersion) << std::endl;
|
std::cout << "Database is now on actual version " << std::to_string(targetVersion) << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
uint32_t DB::MySQL::registerAccount (const std::string& login, const std::string& hash) {
|
||||||
|
//TODO validate filed lengths!
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
MySQL::Transaction txn(con);
|
||||||
|
|
||||||
|
Statement addAcc(con, registerQuery);
|
||||||
|
|
||||||
|
std::string l = login; //I hate copying just to please this horible API
|
||||||
|
std::string h = hash;
|
||||||
|
addAcc.bind(l.data(), MYSQL_TYPE_STRING);
|
||||||
|
addAcc.bind(h.data(), MYSQL_TYPE_STRING);
|
||||||
|
try {
|
||||||
|
addAcc.execute();
|
||||||
|
} catch (const Duplicate& dup) {
|
||||||
|
throw DuplicateLogin(dup.what());
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t id = lastInsertedId();
|
||||||
|
static std::string defaultRole("default");
|
||||||
|
|
||||||
|
Statement addRole(con, assignRoleQuery);
|
||||||
|
addRole.bind(&id, MYSQL_TYPE_LONG, true);
|
||||||
|
addRole.bind(defaultRole.data(), MYSQL_TYPE_STRING);
|
||||||
|
addRole.execute();
|
||||||
|
|
||||||
|
txn.commit();
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string DB::MySQL::getAccountHash (const std::string& login) {
|
||||||
|
std::string l = login;
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
|
||||||
|
Statement getHash(con, selectHash);
|
||||||
|
getHash.bind(l.data(), MYSQL_TYPE_STRING);
|
||||||
|
getHash.execute();
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> result = getHash.fetchResult();
|
||||||
|
if (result.empty())
|
||||||
|
throw NoLogin("Couldn't find login " + l);
|
||||||
|
|
||||||
|
if (result[0].empty())
|
||||||
|
throw std::runtime_error("Error with the query \"selectHash\"");
|
||||||
|
|
||||||
|
return std::any_cast<const std::string&>(result[0][0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Session DB::MySQL::createSession (const std::string& login, const std::string& access, const std::string& renew) {
|
||||||
|
std::string l = login;
|
||||||
|
DB::Session res;
|
||||||
|
res.accessToken = access;
|
||||||
|
res.renewToken = renew;
|
||||||
|
static std::string testingDevice("Testing...");
|
||||||
|
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
|
||||||
|
Statement session(con, createSessionQuery);
|
||||||
|
session.bind(res.accessToken.data(), MYSQL_TYPE_STRING);
|
||||||
|
session.bind(res.renewToken.data(), MYSQL_TYPE_STRING);
|
||||||
|
session.bind(testingDevice.data(), MYSQL_TYPE_STRING);
|
||||||
|
session.bind(l.data(), MYSQL_TYPE_STRING);
|
||||||
|
session.execute();
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> result = session.fetchResult();
|
||||||
|
if (result.empty())
|
||||||
|
throw std::runtime_error("Error returning ids after insertion in sessions table");
|
||||||
|
|
||||||
|
res.id = std::any_cast<uint32_t>(result[0][0]);
|
||||||
|
res.owner = std::any_cast<uint32_t>(result[0][1]);
|
||||||
|
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t DB::MySQL::lastInsertedId () {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
int result = mysql_query(con, lastIdQuery);
|
||||||
|
|
||||||
|
if (result != 0)
|
||||||
|
throw std::runtime_error(std::string("Error executing last inserted id: ") + mysql_error(con));
|
||||||
|
|
||||||
|
std::unique_ptr<MYSQL_RES, ResDeleter> res(mysql_store_result(con));
|
||||||
|
if (!res)
|
||||||
|
throw std::runtime_error(std::string("Querying last inserted id returned no result: ") + mysql_error(con));
|
||||||
|
|
||||||
|
MYSQL_ROW row = mysql_fetch_row(res.get());
|
||||||
|
if (row)
|
||||||
|
return std::stoi(row[0]);
|
||||||
|
else
|
||||||
|
throw std::runtime_error(std::string("Querying last inserted id returned no rows"));
|
||||||
|
}
|
||||||
|
DB::Session DB::MySQL::findSession (const std::string& accessToken) {
|
||||||
|
std::string a = accessToken;
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
|
||||||
|
Statement session(con, selectSession);
|
||||||
|
session.bind(a.data(), MYSQL_TYPE_STRING);
|
||||||
|
session.execute();
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> result = session.fetchResult();
|
||||||
|
if (result.empty())
|
||||||
|
throw NoSession("Couldn't find session with token " + a);
|
||||||
|
|
||||||
|
return DB::Session(result[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<DB::Asset> DB::MySQL::listAssets (uint32_t owner) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
|
||||||
|
Statement st(con, selectAssets);
|
||||||
|
st.bind(&owner, MYSQL_TYPE_LONG, true);
|
||||||
|
st.execute();
|
||||||
|
std::vector<std::vector<std::any>> res = st.fetchResult();
|
||||||
|
|
||||||
|
std::size_t size = res.size();
|
||||||
|
std::vector<DB::Asset> result(size);
|
||||||
|
for (std::size_t i = 0; i < size; ++i)
|
||||||
|
result[i].parse(res[i]);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Asset DB::MySQL::addAsset(const Asset& asset) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
Asset result = asset;
|
||||||
|
|
||||||
|
Statement add(con, insertAsset);
|
||||||
|
add.bind(&result.owner, MYSQL_TYPE_LONG, true);
|
||||||
|
add.bind(&result.currency, MYSQL_TYPE_LONG, true);
|
||||||
|
add.bind(result.title.data(), MYSQL_TYPE_STRING);
|
||||||
|
add.bind(result.icon.data(), MYSQL_TYPE_STRING);
|
||||||
|
add.bind(&result.color, MYSQL_TYPE_LONG, true);
|
||||||
|
add.bind(&result.archived, MYSQL_TYPE_TINY);
|
||||||
|
add.execute();
|
||||||
|
|
||||||
|
result.id = lastInsertedId();
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
void DB::MySQL::updateAsset(const Asset& asset) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
Asset result = asset;
|
||||||
|
|
||||||
|
Statement update(con, updateAssetQuery);
|
||||||
|
update.bind(&result.owner, MYSQL_TYPE_LONG, true);
|
||||||
|
update.bind(&result.currency, MYSQL_TYPE_LONG, true);
|
||||||
|
update.bind(result.title.data(), MYSQL_TYPE_STRING);
|
||||||
|
update.bind(result.icon.data(), MYSQL_TYPE_STRING);
|
||||||
|
update.bind(&result.color, MYSQL_TYPE_LONG, true);
|
||||||
|
update.bind(&result.archived, MYSQL_TYPE_TINY);
|
||||||
|
update.bind(&result.id, MYSQL_TYPE_LONG, true);
|
||||||
|
update.execute();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool DB::MySQL::deleteAsset(uint32_t assetId, uint32_t actorId) {
|
||||||
|
Statement del(&connection, removeAsset);
|
||||||
|
del.bind(&assetId, MYSQL_TYPE_LONG, true);
|
||||||
|
del.bind(&actorId, MYSQL_TYPE_LONG, true);
|
||||||
|
del.execute();
|
||||||
|
|
||||||
|
if (del.affectedRows() == 0)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<DB::Currency> DB::MySQL::listUsedCurrencies(uint32_t owner) {
|
||||||
|
Statement list(&connection, selectUsedCurrencies);
|
||||||
|
list.bind(&owner, MYSQL_TYPE_LONG, true);
|
||||||
|
list.execute();
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> res = list.fetchResult();
|
||||||
|
|
||||||
|
std::size_t size = res.size();
|
||||||
|
std::vector<DB::Currency> result(size);
|
||||||
|
for (std::size_t i = 0; i < size; ++i)
|
||||||
|
result[i].parse(res[i]);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Transaction DB::MySQL::addTransaction(const DB::Transaction& transaction) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
DB::Transaction result = transaction;
|
||||||
|
|
||||||
|
std::string value = std::to_string(result.value);
|
||||||
|
|
||||||
|
Statement add(con, addTransactionQuery);
|
||||||
|
add.bind(&result.initiator, MYSQL_TYPE_LONG, true);
|
||||||
|
add.bind(&result.asset, MYSQL_TYPE_LONG, true);
|
||||||
|
add.bind(&result.parent, MYSQL_TYPE_LONG, true);
|
||||||
|
add.bind(value.data(), MYSQL_TYPE_STRING);
|
||||||
|
add.bind(&result.performed, MYSQL_TYPE_LONG, true);
|
||||||
|
add.execute();
|
||||||
|
|
||||||
|
result.id = lastInsertedId();
|
||||||
|
std::chrono::time_point currently = std::chrono::time_point_cast<std::chrono::seconds>(
|
||||||
|
std::chrono::system_clock::now()
|
||||||
|
);
|
||||||
|
result.modified = currently.time_since_epoch().count();
|
||||||
|
//todo actual value which could have changed after insertion
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
void DB::MySQL::updateTransaction(const DB::Transaction& transaction) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
DB::Transaction result = transaction;
|
||||||
|
|
||||||
|
std::string value = std::to_string(result.value);
|
||||||
|
|
||||||
|
Statement upd(con, updateTransactionQuery);
|
||||||
|
upd.bind(&result.initiator, MYSQL_TYPE_LONG, true);
|
||||||
|
upd.bind(&result.asset, MYSQL_TYPE_LONG, true);
|
||||||
|
upd.bind(&result.parent, MYSQL_TYPE_LONG, true);
|
||||||
|
upd.bind(value.data(), MYSQL_TYPE_STRING);
|
||||||
|
upd.bind(&result.performed, MYSQL_TYPE_LONG, true);
|
||||||
|
upd.bind(&result.id, MYSQL_TYPE_LONG, true);
|
||||||
|
upd.execute();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool DB::MySQL::deleteTransaction(uint32_t id, uint32_t actorId) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
|
||||||
|
Statement del(con, deleteTransactionQuery);
|
||||||
|
del.bind(&id, MYSQL_TYPE_LONG, true); //for actual transactions
|
||||||
|
del.bind(&id, MYSQL_TYPE_LONG, true); //for potential children
|
||||||
|
del.bind(&actorId, MYSQL_TYPE_LONG, true); //for preventing unauthorized removal, but it needs to be improved
|
||||||
|
del.execute(); //need to think of a parent with no children transactions...
|
||||||
|
|
||||||
|
if (del.affectedRows() == 0)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<DB::Transaction> DB::MySQL::listTransactions(uint32_t owner) {
|
||||||
|
MYSQL* con = &connection;
|
||||||
|
|
||||||
|
Statement get(con, selectAllTransactions);
|
||||||
|
get.bind(&owner, MYSQL_TYPE_LONG, true);
|
||||||
|
get.execute();
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> res = get.fetchResult();
|
||||||
|
std::size_t size = res.size();
|
||||||
|
|
||||||
|
std::vector<DB::Transaction> result(size);
|
||||||
|
for (std::size_t i = 0; i < size; ++i)
|
||||||
|
result[i].parse(res[i]);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
@ -1,37 +1,68 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
|
#include <optional>
|
||||||
|
|
||||||
#include <mysql.h>
|
#include <mysql.h>
|
||||||
|
|
||||||
#include "database/dbinterface.h"
|
#include "database/interface.h"
|
||||||
#include "utils/helpers.h"
|
#include "utils/helpers.h"
|
||||||
|
|
||||||
class MySQL : public DBInterface {
|
namespace DB {
|
||||||
|
class MySQL : public Interface {
|
||||||
class Statement;
|
class Statement;
|
||||||
|
class Transaction;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
MySQL();
|
MySQL ();
|
||||||
~MySQL() override;
|
~MySQL () override;
|
||||||
|
|
||||||
void connect(const std::string& path) override;
|
void connect (const std::string& path) override;
|
||||||
void disconnect() override;
|
void disconnect () override;
|
||||||
void setCredentials(const std::string& login, const std::string& password) override;
|
void setCredentials (const std::string& login, const std::string& password) override;
|
||||||
void setDatabase(const std::string& database) override;
|
void setDatabase (const std::string& database) override;
|
||||||
|
|
||||||
void migrate(uint8_t targetVersion) override;
|
void migrate (uint8_t targetVersion) override;
|
||||||
uint8_t getVersion() override;
|
uint8_t getVersion () override;
|
||||||
void setVersion(uint8_t version) override;
|
void setVersion (uint8_t version) override;
|
||||||
|
|
||||||
|
uint32_t registerAccount (const std::string& login, const std::string& hash) override;
|
||||||
|
std::string getAccountHash (const std::string& login) override;
|
||||||
|
|
||||||
|
Session createSession (const std::string& login, const std::string& access, const std::string& renew) override;
|
||||||
|
Session findSession (const std::string& accessToken) override;
|
||||||
|
|
||||||
|
std::vector<Asset> listAssets (uint32_t owner) override;
|
||||||
|
Asset addAsset (const Asset& asset) override;
|
||||||
|
void updateAsset (const Asset& asset) override;
|
||||||
|
bool deleteAsset(uint32_t assetId, uint32_t actorId) override;
|
||||||
|
|
||||||
|
std::vector<Currency> listUsedCurrencies(uint32_t owner) override;
|
||||||
|
|
||||||
|
DB::Transaction addTransaction(const DB::Transaction& transaction) override;
|
||||||
|
std::vector<DB::Transaction> listTransactions(uint32_t owner) override;
|
||||||
|
void updateTransaction(const DB::Transaction& transaction) override;
|
||||||
|
bool deleteTransaction(uint32_t id, uint32_t actorId) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void executeFile(const std::filesystem::path& relativePath);
|
void executeFile (const std::filesystem::path& relativePath);
|
||||||
|
bool getBlock (std::ifstream& file, std::string& block, std::string& name);
|
||||||
|
uint32_t lastInsertedId ();
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
MYSQL connection;
|
MYSQL connection;
|
||||||
std::string login;
|
std::string login;
|
||||||
std::string password;
|
std::string password;
|
||||||
std::string database;
|
std::string database;
|
||||||
|
|
||||||
|
struct ResDeleter {
|
||||||
|
void operator () (MYSQL_RES* res) {
|
||||||
|
mysql_free_result(res);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
};
|
||||||
|
}
|
||||||
|
@ -1,23 +1,24 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "statement.h"
|
#include "statement.h"
|
||||||
|
|
||||||
#include <cstring>
|
#include "mysqld_error.h"
|
||||||
|
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
static uint64_t TIME_LENGTH = sizeof(MYSQL_TIME);
|
static uint64_t TIME_LENGTH = sizeof(MYSQL_TIME);
|
||||||
|
|
||||||
MySQL::Statement::Statement(MYSQL* connection, const char* statement):
|
DB::MySQL::Statement::Statement(MYSQL* connection, const char* statement):
|
||||||
stmt(mysql_stmt_init(connection)),
|
stmt(mysql_stmt_init(connection)),
|
||||||
param(),
|
param()
|
||||||
lengths()
|
|
||||||
{
|
{
|
||||||
int result = mysql_stmt_prepare(stmt.get(), statement, strlen(statement));
|
int result = mysql_stmt_prepare(stmt.get(), statement, strlen(statement));
|
||||||
if (result != 0)
|
if (result != 0)
|
||||||
throw std::runtime_error(std::string("Error preparing statement: ") + mysql_stmt_error(stmt.get()));
|
throw std::runtime_error(std::string("Error preparing statement: ") + mysql_stmt_error(stmt.get()));
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::Statement::bind(void* value, enum_field_types type) {
|
void DB::MySQL::Statement::bind(void* value, enum_field_types type, bool usigned) {
|
||||||
MYSQL_BIND& result = param.emplace_back();
|
MYSQL_BIND& result = param.emplace_back();
|
||||||
std::memset(&result, 0, sizeof(result));
|
std::memset(&result, 0, sizeof(result));
|
||||||
|
|
||||||
@ -27,24 +28,135 @@ void MySQL::Statement::bind(void* value, enum_field_types type) {
|
|||||||
switch (type) {
|
switch (type) {
|
||||||
case MYSQL_TYPE_STRING:
|
case MYSQL_TYPE_STRING:
|
||||||
case MYSQL_TYPE_VAR_STRING:
|
case MYSQL_TYPE_VAR_STRING:
|
||||||
result.length = &lengths.emplace_back(strlen(static_cast<char*>(value)));
|
result.buffer_length = strlen(static_cast<char*>(value));
|
||||||
break;
|
break;
|
||||||
case MYSQL_TYPE_DATE:
|
case MYSQL_TYPE_DATE:
|
||||||
result.length = &TIME_LENGTH;
|
result.buffer_length = TIME_LENGTH;
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_LONG:
|
||||||
|
case MYSQL_TYPE_LONGLONG:
|
||||||
|
case MYSQL_TYPE_SHORT:
|
||||||
|
case MYSQL_TYPE_TINY:
|
||||||
|
result.is_unsigned = usigned;
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
lengths.pop_back();
|
|
||||||
throw std::runtime_error("Type: " + std::to_string(type) + " is not yet supported in bind");
|
throw std::runtime_error("Type: " + std::to_string(type) + " is not yet supported in bind");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void MySQL::Statement::execute() {
|
void DB::MySQL::Statement::execute() {
|
||||||
int result = mysql_stmt_bind_param(stmt.get(), param.data());
|
MYSQL_STMT* raw = stmt.get();
|
||||||
|
int result = mysql_stmt_bind_param(raw, param.data());
|
||||||
if (result != 0)
|
if (result != 0)
|
||||||
throw std::runtime_error(std::string("Error binding statement: ") + mysql_stmt_error(stmt.get()));
|
throw std::runtime_error(std::string("Error binding statement: ") + mysql_stmt_error(raw));
|
||||||
|
|
||||||
result = mysql_stmt_execute(stmt.get());
|
result = mysql_stmt_execute(raw);
|
||||||
if (result != 0)
|
if (result != 0) {
|
||||||
throw std::runtime_error(std::string("Error executing statement: ") + mysql_stmt_error(stmt.get()));
|
int errcode = mysql_stmt_errno(raw);
|
||||||
|
std::string text = mysql_stmt_error(raw);
|
||||||
|
switch (errcode) {
|
||||||
|
case ER_DUP_ENTRY:
|
||||||
|
throw Duplicate("Error executing statement: " + text);
|
||||||
|
default:
|
||||||
|
throw std::runtime_error("Error executing statement: " + text);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> DB::MySQL::Statement::fetchResult() {
|
||||||
|
MYSQL_STMT* raw = stmt.get();
|
||||||
|
if (mysql_stmt_store_result(raw) != 0)
|
||||||
|
throw std::runtime_error(std::string("Error fetching statement result: ") + mysql_stmt_error(raw)); //TODO not sure if it's valid here
|
||||||
|
|
||||||
|
MYSQL_RES* meta = mysql_stmt_result_metadata(raw);
|
||||||
|
if (meta == nullptr)
|
||||||
|
throw std::runtime_error(std::string("Error fetching statement result: ") + mysql_stmt_error(raw)); //TODO not sure if it's valid here
|
||||||
|
|
||||||
|
std::unique_ptr<MYSQL_RES, ResDeleter> mt(meta);
|
||||||
|
unsigned int numColumns = mysql_num_fields(meta);
|
||||||
|
MYSQL_BIND bind[numColumns];
|
||||||
|
std::memset(bind, 0, sizeof(bind));
|
||||||
|
|
||||||
|
std::vector<std::any> line(numColumns);
|
||||||
|
std::vector<long unsigned int> lengths(numColumns);
|
||||||
|
for (unsigned int i = 0; i < numColumns; ++i) {
|
||||||
|
MYSQL_FIELD *field = mysql_fetch_field_direct(meta, i);
|
||||||
|
|
||||||
|
switch (field->type) {
|
||||||
|
case MYSQL_TYPE_STRING:
|
||||||
|
case MYSQL_TYPE_VAR_STRING:
|
||||||
|
case MYSQL_TYPE_VARCHAR: {
|
||||||
|
line[i] = std::string();
|
||||||
|
std::string& str = std::any_cast<std::string&>(line[i]);
|
||||||
|
str.resize(field->length);
|
||||||
|
bind[i].buffer = str.data();
|
||||||
|
} break;
|
||||||
|
case MYSQL_TYPE_TINY:
|
||||||
|
line[i] = uint8_t{0};
|
||||||
|
bind[i].buffer = &std::any_cast<uint8_t&>(line[i]);
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_SHORT:
|
||||||
|
line[i] = uint16_t{0};
|
||||||
|
bind[i].buffer = &std::any_cast<uint16_t&>(line[i]);
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_LONG:
|
||||||
|
line[i] = uint32_t{0};
|
||||||
|
bind[i].buffer = &std::any_cast<uint32_t&>(line[i]);
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_LONGLONG:
|
||||||
|
line[i] = uint64_t{0};
|
||||||
|
bind[i].buffer = &std::any_cast<uint64_t&>(line[i]);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw std::runtime_error("Unsupported data fetching statement result " + std::to_string(field->type));
|
||||||
|
}
|
||||||
|
bind[i].buffer_type = field->type;
|
||||||
|
bind[i].buffer_length = field->length;
|
||||||
|
bind[i].length = &lengths[i];
|
||||||
|
if (field->flags & UNSIGNED_FLAG)
|
||||||
|
bind[i].is_unsigned = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (mysql_stmt_bind_result(raw, bind) != 0)
|
||||||
|
throw std::runtime_error(std::string("Error binding on fetching statement result: ") + mysql_stmt_error(raw));
|
||||||
|
|
||||||
|
std::vector<std::vector<std::any>> result;
|
||||||
|
int rc;
|
||||||
|
while ((rc = mysql_stmt_fetch(raw)) == 0) {
|
||||||
|
std::vector<std::any>& row = result.emplace_back(numColumns);
|
||||||
|
for (unsigned int i = 0; i < numColumns; ++i) {
|
||||||
|
switch (bind[i].buffer_type) {
|
||||||
|
case MYSQL_TYPE_STRING:
|
||||||
|
case MYSQL_TYPE_VAR_STRING:
|
||||||
|
case MYSQL_TYPE_VARCHAR: {
|
||||||
|
row[i] = std::string(std::any_cast<const std::string&>(line[i]).data(), lengths[i]);
|
||||||
|
} break;
|
||||||
|
case MYSQL_TYPE_TINY:
|
||||||
|
row[i] = std::any_cast<uint8_t>(line[i]);
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_SHORT:
|
||||||
|
row[i] = std::any_cast<uint16_t>(line[i]);
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_LONG:
|
||||||
|
row[i] = std::any_cast<uint32_t>(line[i]);
|
||||||
|
break;
|
||||||
|
case MYSQL_TYPE_LONGLONG:
|
||||||
|
row[i] = std::any_cast<uint64_t>(line[i]);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw std::runtime_error("Unsupported data fetching statement result " + std::to_string(bind[i].buffer_type));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (rc == 1)
|
||||||
|
throw std::runtime_error(std::string("Error occured fetching data ") + mysql_stmt_error(raw));
|
||||||
|
else if (rc == MYSQL_DATA_TRUNCATED)
|
||||||
|
throw std::runtime_error("Data has been truncated");
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned int DB::MySQL::Statement::affectedRows () {
|
||||||
|
return mysql_stmt_affected_rows(stmt.get());
|
||||||
}
|
}
|
||||||
|
@ -1,13 +1,16 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
#include <cstring>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
#include <tuple>
|
||||||
|
#include <any>
|
||||||
|
|
||||||
#include "mysql.h"
|
#include "mysql.h"
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
class MySQL::Statement {
|
class MySQL::Statement {
|
||||||
struct STMTDeleter {
|
struct STMTDeleter {
|
||||||
void operator () (MYSQL_STMT* stmt) {
|
void operator () (MYSQL_STMT* stmt) {
|
||||||
@ -17,11 +20,13 @@ class MySQL::Statement {
|
|||||||
public:
|
public:
|
||||||
Statement(MYSQL* connection, const char* statement);
|
Statement(MYSQL* connection, const char* statement);
|
||||||
|
|
||||||
void bind(void* value, enum_field_types type);
|
void bind(void* value, enum_field_types type, bool usigned = false);
|
||||||
void execute();
|
void execute();
|
||||||
|
unsigned int affectedRows();
|
||||||
|
std::vector<std::vector<std::any>> fetchResult();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::unique_ptr<MYSQL_STMT, STMTDeleter> stmt;
|
std::unique_ptr<MYSQL_STMT, STMTDeleter> stmt;
|
||||||
std::vector<MYSQL_BIND> param;
|
std::vector<MYSQL_BIND> param;
|
||||||
std::vector<uint64_t> lengths;
|
|
||||||
};
|
};
|
||||||
|
}
|
||||||
|
37
database/mysql/transaction.cpp
Normal file
37
database/mysql/transaction.cpp
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "transaction.h"
|
||||||
|
|
||||||
|
DB::MySQL::Transaction::Transaction(MYSQL* connection):
|
||||||
|
con(connection),
|
||||||
|
opened(false)
|
||||||
|
{
|
||||||
|
if (mysql_autocommit(con, 0) != 0)
|
||||||
|
throw std::runtime_error(std::string("Failed to start transaction") + mysql_error(con));
|
||||||
|
|
||||||
|
opened = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::MySQL::Transaction::~Transaction() {
|
||||||
|
if (opened)
|
||||||
|
abort();
|
||||||
|
}
|
||||||
|
|
||||||
|
void DB::MySQL::Transaction::commit() {
|
||||||
|
if (mysql_commit(con) != 0)
|
||||||
|
throw std::runtime_error(std::string("Failed to commit transaction") + mysql_error(con));
|
||||||
|
|
||||||
|
opened = false;
|
||||||
|
if (mysql_autocommit(con, 1) != 0)
|
||||||
|
throw std::runtime_error(std::string("Failed to return autocommit") + mysql_error(con));
|
||||||
|
}
|
||||||
|
|
||||||
|
void DB::MySQL::Transaction::abort() {
|
||||||
|
opened = false;
|
||||||
|
if (mysql_rollback(con) != 0)
|
||||||
|
throw std::runtime_error(std::string("Failed to rollback transaction") + mysql_error(con));
|
||||||
|
|
||||||
|
if (mysql_autocommit(con, 1) != 0)
|
||||||
|
throw std::runtime_error(std::string("Failed to return autocommit") + mysql_error(con));
|
||||||
|
}
|
21
database/mysql/transaction.h
Normal file
21
database/mysql/transaction.h
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "mysql.h"
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class MySQL::Transaction {
|
||||||
|
public:
|
||||||
|
Transaction(MYSQL* connection);
|
||||||
|
~Transaction();
|
||||||
|
|
||||||
|
void commit();
|
||||||
|
void abort();
|
||||||
|
|
||||||
|
private:
|
||||||
|
MYSQL* con;
|
||||||
|
bool opened;
|
||||||
|
};
|
||||||
|
}
|
57
database/pool.cpp
Normal file
57
database/pool.cpp
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "pool.h"
|
||||||
|
|
||||||
|
DB::Pool::Pool (Private):
|
||||||
|
std::enable_shared_from_this<Pool>(),
|
||||||
|
mutex(),
|
||||||
|
conditional(),
|
||||||
|
interfaces()
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Pool::~Pool () {
|
||||||
|
}
|
||||||
|
|
||||||
|
std::shared_ptr<DB::Pool> DB::Pool::create () {
|
||||||
|
return std::make_shared<Pool>(Private());
|
||||||
|
}
|
||||||
|
|
||||||
|
void DB::Pool::addInterfaces (
|
||||||
|
Interface::Type type,
|
||||||
|
std::size_t amount,
|
||||||
|
const std::string & login,
|
||||||
|
const std::string & password,
|
||||||
|
const std::string & database,
|
||||||
|
const std::string& path
|
||||||
|
) {
|
||||||
|
std::unique_lock lock(mutex);
|
||||||
|
for (std::size_t i = 0; i < amount; ++i) {
|
||||||
|
const std::unique_ptr<Interface>& ref = interfaces.emplace(Interface::create(type));
|
||||||
|
ref->setCredentials(login, password);
|
||||||
|
ref->setDatabase(database);
|
||||||
|
ref->connect(path);
|
||||||
|
}
|
||||||
|
|
||||||
|
lock.unlock();
|
||||||
|
conditional.notify_all();
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Resource DB::Pool::request () {
|
||||||
|
std::unique_lock lock(mutex);
|
||||||
|
while (interfaces.empty())
|
||||||
|
conditional.wait(lock);
|
||||||
|
|
||||||
|
std::unique_ptr<Interface> interface = std::move(interfaces.front());
|
||||||
|
interfaces.pop();
|
||||||
|
return Resource(std::move(interface), shared_from_this());
|
||||||
|
}
|
||||||
|
|
||||||
|
void DB::Pool::free (std::unique_ptr<Interface> interface) {
|
||||||
|
std::unique_lock lock(mutex);
|
||||||
|
|
||||||
|
interfaces.push(std::move(interface));
|
||||||
|
|
||||||
|
lock.unlock();
|
||||||
|
conditional.notify_one();
|
||||||
|
}
|
47
database/pool.h
Normal file
47
database/pool.h
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <memory>
|
||||||
|
#include <queue>
|
||||||
|
#include <mutex>
|
||||||
|
#include <condition_variable>
|
||||||
|
|
||||||
|
#include "interface.h"
|
||||||
|
#include "resource.h"
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Pool : public std::enable_shared_from_this<Pool> {
|
||||||
|
struct Private {};
|
||||||
|
friend class Resource;
|
||||||
|
|
||||||
|
void free(std::unique_ptr<Interface> interface);
|
||||||
|
|
||||||
|
public:
|
||||||
|
Pool(Private);
|
||||||
|
Pool(const Pool&) = delete;
|
||||||
|
Pool(Pool&&) = delete;
|
||||||
|
~Pool();
|
||||||
|
Pool& operator = (const Pool&) = delete;
|
||||||
|
Pool& operator = (Pool&&) = delete;
|
||||||
|
|
||||||
|
static std::shared_ptr<Pool> create();
|
||||||
|
Resource request();
|
||||||
|
void addInterfaces(
|
||||||
|
Interface::Type type,
|
||||||
|
std::size_t amount,
|
||||||
|
const std::string& login,
|
||||||
|
const std::string& password,
|
||||||
|
const std::string& database,
|
||||||
|
const std::string& path
|
||||||
|
);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::mutex mutex;
|
||||||
|
std::condition_variable conditional;
|
||||||
|
std::queue<std::unique_ptr<Interface>> interfaces;
|
||||||
|
|
||||||
|
};
|
||||||
|
}
|
38
database/resource.cpp
Normal file
38
database/resource.cpp
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "resource.h"
|
||||||
|
|
||||||
|
#include "pool.h"
|
||||||
|
|
||||||
|
DB::Resource::Resource (
|
||||||
|
std::unique_ptr<Interface> interface,
|
||||||
|
std::weak_ptr<Pool> parent
|
||||||
|
):
|
||||||
|
parent(parent),
|
||||||
|
interface(std::move(interface))
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Resource::Resource(Resource&& other):
|
||||||
|
parent(other.parent),
|
||||||
|
interface(std::move(other.interface))
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Resource::~Resource() {
|
||||||
|
if (!interface)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (std::shared_ptr<Pool> p = parent.lock())
|
||||||
|
p->free(std::move(interface));
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Resource& DB::Resource::operator = (Resource&& other) {
|
||||||
|
parent = other.parent;
|
||||||
|
interface = std::move(other.interface);
|
||||||
|
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
DB::Interface* DB::Resource::operator -> () {
|
||||||
|
return interface.get();
|
||||||
|
}
|
31
database/resource.h
Normal file
31
database/resource.h
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
|
#include "interface.h"
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Pool;
|
||||||
|
|
||||||
|
class Resource {
|
||||||
|
friend class Pool;
|
||||||
|
Resource(std::unique_ptr<Interface> interface, std::weak_ptr<Pool> parent);
|
||||||
|
|
||||||
|
public:
|
||||||
|
Resource(const Resource&) = delete;
|
||||||
|
Resource(Resource&& other);
|
||||||
|
~Resource();
|
||||||
|
|
||||||
|
Resource& operator = (const Resource&) = delete;
|
||||||
|
Resource& operator = (Resource&& other);
|
||||||
|
|
||||||
|
Interface* operator -> ();
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Pool> parent;
|
||||||
|
std::unique_ptr<Interface> interface;
|
||||||
|
};
|
||||||
|
}
|
18
database/schema/CMakeLists.txt
Normal file
18
database/schema/CMakeLists.txt
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
set(HEADERS
|
||||||
|
session.h
|
||||||
|
asset.h
|
||||||
|
currency.h
|
||||||
|
transaction.h
|
||||||
|
)
|
||||||
|
|
||||||
|
set(SOURCES
|
||||||
|
session.cpp
|
||||||
|
asset.cpp
|
||||||
|
currency.cpp
|
||||||
|
transaction.cpp
|
||||||
|
)
|
||||||
|
|
||||||
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
48
database/schema/asset.cpp
Normal file
48
database/schema/asset.cpp
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "asset.h"
|
||||||
|
|
||||||
|
DB::Asset::Asset ():
|
||||||
|
id(0),
|
||||||
|
owner(0),
|
||||||
|
currency(0),
|
||||||
|
title(),
|
||||||
|
icon(),
|
||||||
|
color(0),
|
||||||
|
archived(false)
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Asset::Asset (const std::vector<std::any>& vec):
|
||||||
|
id(std::any_cast<uint32_t>(vec[0])),
|
||||||
|
owner(std::any_cast<uint32_t>(vec[1])),
|
||||||
|
currency(std::any_cast<uint32_t>(vec[2])),
|
||||||
|
title(std::any_cast<const std::string&>(vec[3])),
|
||||||
|
icon(std::any_cast<const std::string&>(vec[4])),
|
||||||
|
color(std::any_cast<uint32_t>(vec[5])),
|
||||||
|
archived(std::any_cast<uint8_t>(vec[6]))
|
||||||
|
{}
|
||||||
|
|
||||||
|
void DB::Asset::parse (const std::vector<std::any>& vec) {
|
||||||
|
id = std::any_cast<uint32_t>(vec[0]);
|
||||||
|
owner = std::any_cast<uint32_t>(vec[1]);
|
||||||
|
currency = std::any_cast<uint32_t>(vec[2]);
|
||||||
|
title = std::any_cast<const std::string&>(vec[3]);
|
||||||
|
icon = std::any_cast<const std::string&>(vec[4]);
|
||||||
|
color = std::any_cast<uint32_t>(vec[5]);
|
||||||
|
archived = std::any_cast<uint8_t>(vec[6]);
|
||||||
|
}
|
||||||
|
|
||||||
|
nlohmann::json DB::Asset::toJSON () const {
|
||||||
|
nlohmann::json result = nlohmann::json::object();
|
||||||
|
|
||||||
|
result["id"] = id;
|
||||||
|
//result["owner"] = owner;
|
||||||
|
result["currency"] = currency;
|
||||||
|
result["title"] = title;
|
||||||
|
result["icon"] = icon;
|
||||||
|
result["color"] = color;
|
||||||
|
result["archived"] = archived;
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
33
database/schema/asset.h
Normal file
33
database/schema/asset.h
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <any>
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Asset {
|
||||||
|
public:
|
||||||
|
Asset ();
|
||||||
|
Asset (const std::vector<std::any>& vec);
|
||||||
|
|
||||||
|
void parse (const std::vector<std::any>& vec);
|
||||||
|
nlohmann::json toJSON () const;
|
||||||
|
|
||||||
|
public:
|
||||||
|
uint32_t id;
|
||||||
|
uint32_t owner;
|
||||||
|
uint32_t currency;
|
||||||
|
std::string title;
|
||||||
|
std::string icon;
|
||||||
|
uint32_t color;
|
||||||
|
// `balance` DECIMAL (20, 5) DEFAULT 0,
|
||||||
|
// `type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
bool archived;
|
||||||
|
};
|
||||||
|
}
|
40
database/schema/currency.cpp
Normal file
40
database/schema/currency.cpp
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "currency.h"
|
||||||
|
|
||||||
|
DB::Currency::Currency ():
|
||||||
|
id(0),
|
||||||
|
code(),
|
||||||
|
title(),
|
||||||
|
manual(false),
|
||||||
|
icon()
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Currency::Currency (const std::vector<std::any>& vec):
|
||||||
|
id(std::any_cast<uint32_t>(vec[0])),
|
||||||
|
code(std::any_cast<const std::string&>(vec[1])),
|
||||||
|
title(std::any_cast<const std::string&>(vec[2])),
|
||||||
|
manual(std::any_cast<uint8_t>(vec[3])),
|
||||||
|
icon(std::any_cast<const std::string&>(vec[4]))
|
||||||
|
{}
|
||||||
|
|
||||||
|
void DB::Currency::parse (const std::vector<std::any>& vec) {
|
||||||
|
id = std::any_cast<uint32_t>(vec[0]);
|
||||||
|
code = std::any_cast<const std::string&>(vec[1]);
|
||||||
|
title = std::any_cast<const std::string&>(vec[2]);
|
||||||
|
manual = std::any_cast<uint8_t>(vec[3]);
|
||||||
|
icon = std::any_cast<const std::string&>(vec[4]);
|
||||||
|
}
|
||||||
|
|
||||||
|
nlohmann::json DB::Currency::toJSON () const {
|
||||||
|
nlohmann::json result = nlohmann::json::object();
|
||||||
|
|
||||||
|
result["id"] = id;
|
||||||
|
result["code"] = code;
|
||||||
|
result["title"] = title;
|
||||||
|
result["manual"] = manual;
|
||||||
|
result["icon"] = icon;
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
35
database/schema/currency.h
Normal file
35
database/schema/currency.h
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <any>
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Currency {
|
||||||
|
public:
|
||||||
|
Currency ();
|
||||||
|
Currency (const std::vector<std::any>& vec);
|
||||||
|
|
||||||
|
void parse (const std::vector<std::any>& vec);
|
||||||
|
nlohmann::json toJSON () const;
|
||||||
|
|
||||||
|
public:
|
||||||
|
uint32_t id;
|
||||||
|
std::string code;
|
||||||
|
std::string title;
|
||||||
|
bool manual;
|
||||||
|
// `added` TIMESTAMP DEFAULT UTC_TIMESTAMP(),
|
||||||
|
// `type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
// `value` DECIMAL (20, 5) NOT NULL,
|
||||||
|
// `source` TEXT,
|
||||||
|
// `description` TEXT,
|
||||||
|
std::string icon;
|
||||||
|
|
||||||
|
};
|
||||||
|
}
|
18
database/schema/session.cpp
Normal file
18
database/schema/session.cpp
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "session.h"
|
||||||
|
|
||||||
|
DB::Session::Session ():
|
||||||
|
id(),
|
||||||
|
owner(),
|
||||||
|
accessToken(),
|
||||||
|
renewToken()
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Session::Session (const std::vector<std::any>& vec):
|
||||||
|
id(std::any_cast<unsigned int>(vec[0])),
|
||||||
|
owner(std::any_cast<unsigned int>(vec[1])),
|
||||||
|
accessToken(std::any_cast<const std::string&>(vec[2])),
|
||||||
|
renewToken(std::any_cast<const std::string&>(vec[3]))
|
||||||
|
{}
|
23
database/schema/session.h
Normal file
23
database/schema/session.h
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <any>
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Session {
|
||||||
|
public:
|
||||||
|
Session ();
|
||||||
|
Session (const std::vector<std::any>& vec);
|
||||||
|
|
||||||
|
public:
|
||||||
|
unsigned int id;
|
||||||
|
unsigned int owner;
|
||||||
|
std::string accessToken;
|
||||||
|
std::string renewToken;
|
||||||
|
};
|
||||||
|
}
|
50
database/schema/transaction.cpp
Normal file
50
database/schema/transaction.cpp
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "transaction.h"
|
||||||
|
|
||||||
|
DB::Transaction::Transaction():
|
||||||
|
id(0),
|
||||||
|
initiator(0),
|
||||||
|
asset(0),
|
||||||
|
parent(0),
|
||||||
|
value(0),
|
||||||
|
modified(0),
|
||||||
|
performed(0),
|
||||||
|
notes()
|
||||||
|
{}
|
||||||
|
|
||||||
|
DB::Transaction::Transaction(const std::vector<std::any>& vec):
|
||||||
|
id(std::any_cast<uint32_t>(vec[0])),
|
||||||
|
initiator(std::any_cast<uint32_t>(vec[1])),
|
||||||
|
asset(std::any_cast<uint32_t>(vec[2])),
|
||||||
|
parent(std::any_cast<uint32_t>(vec[3])),
|
||||||
|
value(std::any_cast<double>(vec[4])),
|
||||||
|
modified(std::any_cast<uint32_t>(vec[5])),
|
||||||
|
performed(std::any_cast<uint32_t>(vec[6])),
|
||||||
|
notes()
|
||||||
|
{}
|
||||||
|
|
||||||
|
void DB::Transaction::parse(const std::vector<std::any>& vec) {
|
||||||
|
id = std::any_cast<uint32_t>(vec[0]);
|
||||||
|
initiator = std::any_cast<uint32_t>(vec[1]);
|
||||||
|
asset = std::any_cast<uint32_t>(vec[2]);
|
||||||
|
parent = std::any_cast<uint32_t>(vec[3]);
|
||||||
|
value = std::any_cast<double>(vec[4]);
|
||||||
|
modified = std::any_cast<uint32_t>(vec[5]);
|
||||||
|
performed = std::any_cast<uint32_t>(vec[6]);
|
||||||
|
}
|
||||||
|
|
||||||
|
nlohmann::json DB::Transaction::toJSON() const {
|
||||||
|
nlohmann::json result = nlohmann::json::object();
|
||||||
|
|
||||||
|
result["id"] = id;
|
||||||
|
result["initiator"] = initiator;
|
||||||
|
result["asset"] = asset;
|
||||||
|
result["parent"] = parent;
|
||||||
|
result["value"] = value;
|
||||||
|
result["modified"] = modified;
|
||||||
|
result["performed"] = performed;
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
35
database/schema/transaction.h
Normal file
35
database/schema/transaction.h
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <any>
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
|
namespace DB {
|
||||||
|
class Transaction {
|
||||||
|
public:
|
||||||
|
Transaction ();
|
||||||
|
Transaction (const std::vector<std::any>& vec);
|
||||||
|
|
||||||
|
void parse (const std::vector<std::any>& vec);
|
||||||
|
nlohmann::json toJSON () const;
|
||||||
|
|
||||||
|
public:
|
||||||
|
uint32_t id;
|
||||||
|
uint32_t initiator;
|
||||||
|
// `type` INTEGER UNSIGNED NOT NULL,
|
||||||
|
uint32_t asset;
|
||||||
|
uint32_t parent;
|
||||||
|
double value;
|
||||||
|
// `state` INTEGER UNSIGNED DEFAULT 0,
|
||||||
|
uint32_t modified;
|
||||||
|
uint32_t performed;
|
||||||
|
// `party` INTEGER UNSIGNED,
|
||||||
|
std::string notes;
|
||||||
|
};
|
||||||
|
}
|
40
handler/CMakeLists.txt
Normal file
40
handler/CMakeLists.txt
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
set(HEADERS
|
||||||
|
handler.h
|
||||||
|
info.h
|
||||||
|
env.h
|
||||||
|
register.h
|
||||||
|
login.h
|
||||||
|
poll.h
|
||||||
|
assets.h
|
||||||
|
addasset.h
|
||||||
|
deleteasset.h
|
||||||
|
updateasset.h
|
||||||
|
currencies.h
|
||||||
|
addtransaction.h
|
||||||
|
transactions.h
|
||||||
|
deletetransaction.h
|
||||||
|
updatetransaction.h
|
||||||
|
)
|
||||||
|
|
||||||
|
set(SOURCES
|
||||||
|
handler.cpp
|
||||||
|
info.cpp
|
||||||
|
env.cpp
|
||||||
|
register.cpp
|
||||||
|
login.cpp
|
||||||
|
poll.cpp
|
||||||
|
assets.cpp
|
||||||
|
addasset.cpp
|
||||||
|
deleteasset.cpp
|
||||||
|
updateasset.cpp
|
||||||
|
currencies.cpp
|
||||||
|
addtransaction.cpp
|
||||||
|
transactions.cpp
|
||||||
|
deletetransaction.cpp
|
||||||
|
updatetransaction.cpp
|
||||||
|
)
|
||||||
|
|
||||||
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
78
handler/addasset.cpp
Normal file
78
handler/addasset.cpp
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "addasset.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::AddAsset::AddAsset (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("addAsset", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::AddAsset::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("currency");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
DB::Asset asset;
|
||||||
|
asset.currency = std::stoul(itr->second);
|
||||||
|
//TODO validate the currency
|
||||||
|
|
||||||
|
itr = form.find("title");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
asset.title = itr->second;
|
||||||
|
|
||||||
|
itr = form.find("icon");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
asset.icon = itr->second;
|
||||||
|
|
||||||
|
try {
|
||||||
|
itr = form.find("color");
|
||||||
|
if (itr != form.end())
|
||||||
|
asset.color = std::stoul(itr->second);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Insignificant error parsing color during asset addition: " << e.what() << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
|
||||||
|
asset.owner = session.owner;
|
||||||
|
asset = srv->getDatabase()->addAsset(asset);
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
session.assetAdded(asset);
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
20
handler/addasset.h
Normal file
20
handler/addasset.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//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 AddAsset : public Handler {
|
||||||
|
public:
|
||||||
|
AddAsset (const std::shared_ptr<Server>& server);
|
||||||
|
virtual void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
78
handler/addtransaction.cpp
Normal file
78
handler/addtransaction.cpp
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "addtransaction.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::AddTransaction::AddTransaction (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("addTransaction", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::AddTransaction::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("asset");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
DB::Transaction txn;
|
||||||
|
txn.asset = std::stoul(itr->second);
|
||||||
|
//TODO validate the asset
|
||||||
|
|
||||||
|
itr = form.find("value");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
txn.value = std::stod(itr->second);
|
||||||
|
|
||||||
|
itr = form.find("performed");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
txn.performed = std::stoul(itr->second);
|
||||||
|
|
||||||
|
itr = form.find("notes");
|
||||||
|
if (itr != form.end())
|
||||||
|
txn.notes = itr->second;
|
||||||
|
|
||||||
|
itr = form.find("parent");
|
||||||
|
if (itr != form.end())
|
||||||
|
txn.parent = std::stoul(itr->second);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
|
||||||
|
txn.initiator = session.owner;
|
||||||
|
txn = srv->getDatabase()->addTransaction(txn);
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
session.transactionAdded(txn);
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
20
handler/addtransaction.h
Normal file
20
handler/addtransaction.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//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 AddTransaction : public Handler {
|
||||||
|
public:
|
||||||
|
AddTransaction (const std::shared_ptr<Server>& server);
|
||||||
|
virtual void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
51
handler/assets.cpp
Normal file
51
handler/assets.cpp
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "assets.h"
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::Assets::Assets (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("assets", Request::Method::get),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Assets::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);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
std::vector<DB::Asset> assets = srv->getDatabase()->listAssets(session.owner);
|
||||||
|
|
||||||
|
nlohmann::json arr = nlohmann::json::array();
|
||||||
|
for (const DB::Asset& asset : assets)
|
||||||
|
arr.push_back(asset.toJSON());
|
||||||
|
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["assets"] = arr;
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
20
handler/assets.h
Normal file
20
handler/assets.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//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 Assets : public Handler::Handler {
|
||||||
|
public:
|
||||||
|
Assets (const std::shared_ptr<Server>& server);
|
||||||
|
void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
51
handler/currencies.cpp
Normal file
51
handler/currencies.cpp
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "currencies.h"
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::Currencies::Currencies (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("currencies", Request::Method::get),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Currencies::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);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
std::vector<DB::Currency> cur = srv->getDatabase()->listUsedCurrencies(session.owner);
|
||||||
|
|
||||||
|
nlohmann::json arr = nlohmann::json::array();
|
||||||
|
for (const DB::Currency& c : cur)
|
||||||
|
arr.push_back(c.toJSON());
|
||||||
|
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["currencies"] = arr;
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
21
handler/currencies.h
Normal file
21
handler/currencies.h
Normal file
@ -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 Currencies : public Handler {
|
||||||
|
public:
|
||||||
|
Currencies(const std::shared_ptr<Server>& server);
|
||||||
|
|
||||||
|
void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
59
handler/deleteasset.cpp
Normal file
59
handler/deleteasset.cpp
Normal file
@ -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 (const std::shared_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 " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
21
handler/deleteasset.h
Normal file
21
handler/deleteasset.h
Normal file
@ -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 (const std::shared_ptr<Server>& server);
|
||||||
|
|
||||||
|
virtual void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
59
handler/deletetransaction.cpp
Normal file
59
handler/deletetransaction.cpp
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "deletetransaction.h"
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::DeleteTransaction::DeleteTransaction (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("deleteTransaction", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::DeleteTransaction::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 txnId;
|
||||||
|
try {
|
||||||
|
txnId = 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()->deleteTransaction(txnId, session.owner);
|
||||||
|
if (!success)
|
||||||
|
return error(request, Response::Status::forbidden);
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
session.transactionRemoved(txnId);
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
21
handler/deletetransaction.h
Normal file
21
handler/deletetransaction.h
Normal file
@ -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 DeleteTransaction : public Handler {
|
||||||
|
public:
|
||||||
|
DeleteTransaction (const std::shared_ptr<Server>& server);
|
||||||
|
|
||||||
|
virtual void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
17
handler/env.cpp
Normal file
17
handler/env.cpp
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "env.h"
|
||||||
|
|
||||||
|
Handler::Env::Env():
|
||||||
|
Handler("env", Request::Method::get)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Env::handle(Request& request) {
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
request.printEnvironment(body);
|
||||||
|
|
||||||
|
Response& res = request.createResponse();
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
16
handler/env.h
Normal file
16
handler/env.h
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "handler.h"
|
||||||
|
|
||||||
|
namespace Handler {
|
||||||
|
|
||||||
|
class Env : public Handler {
|
||||||
|
public:
|
||||||
|
Env();
|
||||||
|
void handle(Request& request) override;
|
||||||
|
|
||||||
|
};
|
||||||
|
}
|
16
handler/handler.cpp
Normal file
16
handler/handler.cpp
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "handler.h"
|
||||||
|
|
||||||
|
Handler::Handler::Handler(const std::string& path, Request::Method method):
|
||||||
|
path(path),
|
||||||
|
method(method)
|
||||||
|
{}
|
||||||
|
|
||||||
|
Handler::Handler::~Handler() {}
|
||||||
|
|
||||||
|
void Handler::Handler::error (Request& request, Response::Status status) {
|
||||||
|
Response& res = request.createResponse(status);
|
||||||
|
res.send();
|
||||||
|
}
|
29
handler/handler.h
Normal file
29
handler/handler.h
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
|
#include "request/request.h"
|
||||||
|
#include "response/response.h"
|
||||||
|
|
||||||
|
namespace Handler {
|
||||||
|
|
||||||
|
class Handler {
|
||||||
|
protected:
|
||||||
|
Handler(const std::string& path, Request::Method method);
|
||||||
|
|
||||||
|
protected:
|
||||||
|
static void error (Request& request, Response::Status status);
|
||||||
|
|
||||||
|
public:
|
||||||
|
virtual ~Handler();
|
||||||
|
|
||||||
|
virtual void handle(Request& request) = 0;
|
||||||
|
|
||||||
|
const std::string path;
|
||||||
|
const Request::Method method;
|
||||||
|
};
|
||||||
|
}
|
18
handler/info.cpp
Normal file
18
handler/info.cpp
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "info.h"
|
||||||
|
|
||||||
|
Handler::Info::Info():
|
||||||
|
Handler("info", Request::Method::get)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Info::handle(Request& request) {
|
||||||
|
Response& res = request.createResponse();
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["type"] = PROJECT_NAME;
|
||||||
|
body["version"] = PROJECT_VERSION;
|
||||||
|
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
17
handler/info.h
Normal file
17
handler/info.h
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "handler.h"
|
||||||
|
#include "config.h"
|
||||||
|
|
||||||
|
namespace Handler {
|
||||||
|
|
||||||
|
class Info : public Handler {
|
||||||
|
public:
|
||||||
|
Info();
|
||||||
|
void handle(Request& request) override;
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
79
handler/login.cpp
Normal file
79
handler/login.cpp
Normal file
@ -0,0 +1,79 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "login.h"
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::Login::Login(const std::shared_ptr<Server>& server):
|
||||||
|
Handler("login", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Login::handle(Request& request) {
|
||||||
|
std::map form = request.getForm();
|
||||||
|
std::map<std::string, std::string>::const_iterator itr = form.find("login");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Result::noLogin, Response::Status::badRequest);
|
||||||
|
|
||||||
|
const std::string& login = itr->second;
|
||||||
|
if (login.empty())
|
||||||
|
return error(request, Result::emptyLogin, Response::Status::badRequest);
|
||||||
|
|
||||||
|
itr = form.find("password");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Result::noPassword, Response::Status::badRequest);
|
||||||
|
|
||||||
|
const std::string& password = itr->second;
|
||||||
|
if (password.empty())
|
||||||
|
return error(request, Result::emptyPassword, Response::Status::badRequest);
|
||||||
|
|
||||||
|
std::shared_ptr<Server> srv = server.lock();
|
||||||
|
if (!srv)
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
|
||||||
|
bool success = false;
|
||||||
|
try {
|
||||||
|
success = srv->validatePassword(login, password);
|
||||||
|
} catch (const DB::NoLogin& e) {
|
||||||
|
std::cerr << "Exception on logging in:\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Result::wrongCredentials, Response::Status::badRequest);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on logging in:\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on ogging in" << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
if (!success)
|
||||||
|
return error(request, Result::wrongCredentials, Response::Status::badRequest);
|
||||||
|
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["result"] = Result::success;
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->openSession(login);
|
||||||
|
body["accessToken"] = session.getAccessToken();
|
||||||
|
body["renewToken"] = session.getRenewToken();
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on opening a session:\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on opening a session" << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
|
||||||
|
Response& res = request.createResponse();
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Handler::Login::error(Request& request, Result result, Response::Status code) {
|
||||||
|
Response& res = request.createResponse(code);
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["result"] = result;
|
||||||
|
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
35
handler/login.h
Normal file
35
handler/login.h
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 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 Login : public Handler {
|
||||||
|
public:
|
||||||
|
Login(const std::shared_ptr<Server>& server);
|
||||||
|
void handle(Request& request) override;
|
||||||
|
|
||||||
|
enum class Result {
|
||||||
|
success,
|
||||||
|
noLogin,
|
||||||
|
emptyLogin,
|
||||||
|
noPassword,
|
||||||
|
emptyPassword,
|
||||||
|
wrongCredentials,
|
||||||
|
unknownError
|
||||||
|
};
|
||||||
|
|
||||||
|
private:
|
||||||
|
void error(Request& request, Result result, Response::Status code);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
51
handler/poll.cpp
Normal file
51
handler/poll.cpp
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "handler/poll.h"
|
||||||
|
|
||||||
|
#include "response/response.h"
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "request/redirect.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::Poll::Poll (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("poll", Request::Method::get),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Poll::handle (Request& request) {
|
||||||
|
std::string access = request.getAuthorizationToken();
|
||||||
|
if (access.empty())
|
||||||
|
return error(request, Result::tokenProblem, Response::Status::unauthorized);
|
||||||
|
|
||||||
|
if (access.size() != 32)
|
||||||
|
return error(request, Result::tokenProblem, Response::Status::badRequest);
|
||||||
|
|
||||||
|
std::shared_ptr<Server> srv = server.lock();
|
||||||
|
if (!srv)
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
throw Redirect(&session);
|
||||||
|
} catch (const Redirect& r) {
|
||||||
|
throw r;
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Result::tokenProblem, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Handler::Poll::error(Request& request, Result result, Response::Status status) {
|
||||||
|
Response& res = request.createResponse(status);
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["result"] = result;
|
||||||
|
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
35
handler/poll.h
Normal file
35
handler/poll.h
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
|
#include "handler.h"
|
||||||
|
#include "request/request.h"
|
||||||
|
#include "response/response.h"
|
||||||
|
|
||||||
|
class Server;
|
||||||
|
namespace Handler {
|
||||||
|
|
||||||
|
class Poll : public Handler {
|
||||||
|
public:
|
||||||
|
Poll (const std::shared_ptr<Server>& server);
|
||||||
|
void handle (Request& request) override;
|
||||||
|
|
||||||
|
enum class Result {
|
||||||
|
success,
|
||||||
|
tokenProblem,
|
||||||
|
replace,
|
||||||
|
timeout,
|
||||||
|
unknownError
|
||||||
|
};
|
||||||
|
|
||||||
|
static void error (Request& request, Result result, Response::Status status);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
68
handler/register.cpp
Normal file
68
handler/register.cpp
Normal file
@ -0,0 +1,68 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "register.h"
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::Register::Register(const std::shared_ptr<Server>& server):
|
||||||
|
Handler("register", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Register::handle(Request& request) {
|
||||||
|
std::map form = request.getForm();
|
||||||
|
std::map<std::string, std::string>::const_iterator itr = form.find("login");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Result::noLogin, Response::Status::badRequest);
|
||||||
|
|
||||||
|
const std::string& login = itr->second;
|
||||||
|
if (login.empty())
|
||||||
|
return error(request, Result::emptyLogin, Response::Status::badRequest);
|
||||||
|
|
||||||
|
//TODO login policies checkup
|
||||||
|
|
||||||
|
itr = form.find("password");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Result::noPassword, Response::Status::badRequest);
|
||||||
|
|
||||||
|
const std::string& password = itr->second;
|
||||||
|
if (password.empty())
|
||||||
|
return error(request, Result::emptyPassword, Response::Status::badRequest);
|
||||||
|
|
||||||
|
//TODO password policies checkup
|
||||||
|
|
||||||
|
std::shared_ptr<Server> srv = server.lock();
|
||||||
|
if (!srv)
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
|
||||||
|
try {
|
||||||
|
srv->registerAccount(login, password);
|
||||||
|
} catch (const DB::DuplicateLogin& e) {
|
||||||
|
std::cerr << "Exception on registration:\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Result::loginExists, Response::Status::conflict);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on registration:\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on registration" << std::endl;
|
||||||
|
return error(request, Result::unknownError, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
|
||||||
|
Response& res = request.createResponse();
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["result"] = Result::success;
|
||||||
|
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Handler::Register::error(Request& request, Result result, Response::Status code) {
|
||||||
|
Response& res = request.createResponse(code);
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["result"] = result;
|
||||||
|
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
36
handler/register.h
Normal file
36
handler/register.h
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 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 Register : public Handler {
|
||||||
|
public:
|
||||||
|
Register(const std::shared_ptr<Server>& server);
|
||||||
|
void handle(Request& request) override;
|
||||||
|
|
||||||
|
enum class Result {
|
||||||
|
success,
|
||||||
|
noLogin,
|
||||||
|
emptyLogin,
|
||||||
|
loginExists,
|
||||||
|
loginPolicyViolation,
|
||||||
|
noPassword,
|
||||||
|
emptyPassword,
|
||||||
|
passwordPolicyViolation,
|
||||||
|
unknownError
|
||||||
|
};
|
||||||
|
|
||||||
|
private:
|
||||||
|
void error(Request& request, Result result, Response::Status code);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
51
handler/transactions.cpp
Normal file
51
handler/transactions.cpp
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "transactions.h"
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::Transactions::Transactions (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("transactions", Request::Method::get),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::Transactions::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);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
std::vector<DB::Transaction> transactions = srv->getDatabase()->listTransactions(session.owner);
|
||||||
|
|
||||||
|
nlohmann::json arr = nlohmann::json::array();
|
||||||
|
for (const DB::Transaction& transaction : transactions)
|
||||||
|
arr.push_back(transaction.toJSON());
|
||||||
|
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["transactions"] = arr;
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
20
handler/transactions.h
Normal file
20
handler/transactions.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//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 Transactions : public Handler::Handler {
|
||||||
|
public:
|
||||||
|
Transactions (const std::shared_ptr<Server>& server);
|
||||||
|
void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
84
handler/updateasset.cpp
Normal file
84
handler/updateasset.cpp
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "updateasset.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::UpdateAsset::UpdateAsset (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("updateAsset", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::UpdateAsset::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);
|
||||||
|
|
||||||
|
DB::Asset asset;
|
||||||
|
asset.id = std::stoul(itr->second);
|
||||||
|
|
||||||
|
itr = form.find("currency");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
asset.currency = std::stoul(itr->second);
|
||||||
|
//TODO validate the currency
|
||||||
|
|
||||||
|
itr = form.find("title");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
asset.title = itr->second;
|
||||||
|
|
||||||
|
itr = form.find("icon");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
asset.icon = itr->second;
|
||||||
|
|
||||||
|
try {
|
||||||
|
itr = form.find("color");
|
||||||
|
if (itr != form.end())
|
||||||
|
asset.color = std::stoul(itr->second);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Insignificant error parsing color during asset addition: " << e.what() << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
|
||||||
|
asset.owner = session.owner;
|
||||||
|
srv->getDatabase()->updateAsset(asset);
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
session.assetChanged(asset);
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
20
handler/updateasset.h
Normal file
20
handler/updateasset.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//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 UpdateAsset : public Handler {
|
||||||
|
public:
|
||||||
|
UpdateAsset (const std::shared_ptr<Server>& server);
|
||||||
|
virtual void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
84
handler/updatetransaction.cpp
Normal file
84
handler/updatetransaction.cpp
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "updatetransaction.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include "server/server.h"
|
||||||
|
#include "server/session.h"
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
Handler::UpdateTransaction::UpdateTransaction (const std::shared_ptr<Server>& server):
|
||||||
|
Handler("updateTransaction", Request::Method::post),
|
||||||
|
server(server)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void Handler::UpdateTransaction::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);
|
||||||
|
|
||||||
|
DB::Transaction txn;
|
||||||
|
txn.id = std::stoul(itr->second);
|
||||||
|
|
||||||
|
itr = form.find("asset");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
txn.asset = std::stoul(itr->second);
|
||||||
|
//TODO validate the asset
|
||||||
|
|
||||||
|
itr = form.find("value");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
txn.value = std::stod(itr->second);
|
||||||
|
|
||||||
|
itr = form.find("performed");
|
||||||
|
if (itr == form.end())
|
||||||
|
return error(request, Response::Status::badRequest);
|
||||||
|
|
||||||
|
txn.performed = std::stoul(itr->second);
|
||||||
|
|
||||||
|
itr = form.find("notes");
|
||||||
|
if (itr != form.end())
|
||||||
|
txn.notes = itr->second;
|
||||||
|
|
||||||
|
itr = form.find("parent");
|
||||||
|
if (itr != form.end())
|
||||||
|
txn.parent = std::stoul(itr->second);
|
||||||
|
|
||||||
|
try {
|
||||||
|
Session& session = srv->getSession(access);
|
||||||
|
|
||||||
|
txn.initiator = session.owner;
|
||||||
|
srv->getDatabase()->updateTransaction(txn);
|
||||||
|
|
||||||
|
Response& res = request.createResponse(Response::Status::ok);
|
||||||
|
res.send();
|
||||||
|
|
||||||
|
session.transactionChanged(txn);
|
||||||
|
|
||||||
|
} catch (const DB::NoSession& e) {
|
||||||
|
return error(request, Response::Status::unauthorized);
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
std::cerr << "Exception on " << path << ":\n\t" << e.what() << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
} catch (...) {
|
||||||
|
std::cerr << "Unknown exception on " << path << std::endl;
|
||||||
|
return error(request, Response::Status::internalError);
|
||||||
|
}
|
||||||
|
}
|
20
handler/updatetransaction.h
Normal file
20
handler/updatetransaction.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//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 UpdateTransaction : public Handler {
|
||||||
|
public:
|
||||||
|
UpdateTransaction (const std::shared_ptr<Server>& server);
|
||||||
|
virtual void handle (Request& request) override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<Server> server;
|
||||||
|
};
|
||||||
|
}
|
9
main.cpp
9
main.cpp
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include <fcgiapp.h>
|
#include <fcgiapp.h>
|
||||||
|
|
||||||
@ -7,6 +7,7 @@
|
|||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
#include "config.h" //autogenereted by cmake in the root of bindir
|
#include "config.h" //autogenereted by cmake in the root of bindir
|
||||||
#include "utils/helpers.h"
|
#include "utils/helpers.h"
|
||||||
@ -34,6 +35,6 @@ int main(int argc, char** argv) {
|
|||||||
|
|
||||||
FCGX_Init();
|
FCGX_Init();
|
||||||
|
|
||||||
Server server;
|
auto server = std::make_shared<Server>();
|
||||||
server.run(sockfd);
|
server->run(sockfd);
|
||||||
}
|
}
|
||||||
|
@ -1,9 +1,16 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
set(HEADERS
|
set(HEADERS
|
||||||
request.h
|
request.h
|
||||||
|
redirect.h
|
||||||
|
|
||||||
|
redirectable.h
|
||||||
)
|
)
|
||||||
|
|
||||||
set(SOURCES
|
set(SOURCES
|
||||||
request.cpp
|
request.cpp
|
||||||
|
redirect.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
target_sources(pica PRIVATE ${SOURCES})
|
target_sources(pica PRIVATE ${SOURCES})
|
||||||
|
14
request/accepting.h
Normal file
14
request/accepting.h
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
|
#include "request/request.h"
|
||||||
|
|
||||||
|
class Accepting {
|
||||||
|
public:
|
||||||
|
virtual ~Accepting() {};
|
||||||
|
virtual void accept(std::unique_ptr<Request> request) = 0;
|
||||||
|
};
|
12
request/redirect.cpp
Normal file
12
request/redirect.cpp
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "redirect.h"
|
||||||
|
|
||||||
|
Redirect::Redirect(Accepting* destination):
|
||||||
|
destination(destination)
|
||||||
|
{}
|
||||||
|
|
||||||
|
const char* Redirect::what() const noexcept {
|
||||||
|
return "This is a redirect, should have beeh handled in router, but if you see it - something went terrebly wrong";
|
||||||
|
}
|
16
request/redirect.h
Normal file
16
request/redirect.h
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <exception>
|
||||||
|
|
||||||
|
#include "accepting.h"
|
||||||
|
|
||||||
|
class Redirect : std::exception {
|
||||||
|
public:
|
||||||
|
Redirect(Accepting* destination);
|
||||||
|
|
||||||
|
Accepting* destination;
|
||||||
|
const char* what() const noexcept override;
|
||||||
|
};
|
@ -1,13 +1,19 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "request.h"
|
#include "request.h"
|
||||||
|
|
||||||
constexpr static const char* GET("GET");
|
#include "response/response.h"
|
||||||
|
|
||||||
constexpr static const char* REQUEST_METHOD("REQUEST_METHOD");
|
constexpr static const char* REQUEST_METHOD("REQUEST_METHOD");
|
||||||
constexpr static const char* SCRIPT_FILENAME("SCRIPT_FILENAME");
|
constexpr static const char* SCRIPT_FILENAME("SCRIPT_FILENAME");
|
||||||
constexpr static const char* SERVER_NAME("SERVER_NAME");
|
constexpr static const char* SERVER_NAME("SERVER_NAME");
|
||||||
|
constexpr static const char* CONTENT_TYPE("CONTENT_TYPE");
|
||||||
|
constexpr static const char* CONTENT_LENGTH("CONTENT_LENGTH");
|
||||||
|
constexpr static const char* AUTHORIZATION("HTTP_AUTHORIZATION");
|
||||||
|
constexpr static const char* QUERY_STRING("QUERY_STRING");
|
||||||
|
|
||||||
|
constexpr static const char* urlEncoded("application/x-www-form-urlencoded");
|
||||||
|
|
||||||
// constexpr static const char* REQUEST_URI("REQUEST_URI");
|
// constexpr static const char* REQUEST_URI("REQUEST_URI");
|
||||||
//
|
//
|
||||||
@ -16,9 +22,20 @@ constexpr static const char* SERVER_NAME("SERVER_NAME");
|
|||||||
//
|
//
|
||||||
// constexpr static const char* SCRIPT_NAME("SCRIPT_NAME");
|
// constexpr static const char* SCRIPT_NAME("SCRIPT_NAME");
|
||||||
|
|
||||||
|
constexpr std::array<
|
||||||
|
std::pair<std::string_view, Request::Method>,
|
||||||
|
static_cast<uint8_t>(Request::Method::unknown)
|
||||||
|
> methods = {{
|
||||||
|
{"GET", Request::Method::get},
|
||||||
|
{"POST", Request::Method::post}
|
||||||
|
}};
|
||||||
|
|
||||||
Request::Request ():
|
Request::Request ():
|
||||||
state(State::initial),
|
state(State::initial),
|
||||||
raw()
|
raw(),
|
||||||
|
response(nullptr),
|
||||||
|
path(),
|
||||||
|
cachedMethod(Method::unknown)
|
||||||
{}
|
{}
|
||||||
|
|
||||||
Request::~Request() {
|
Request::~Request() {
|
||||||
@ -27,21 +44,44 @@ Request::~Request() {
|
|||||||
|
|
||||||
void Request::terminate() {
|
void Request::terminate() {
|
||||||
switch (state) {
|
switch (state) {
|
||||||
|
case State::terminated:
|
||||||
|
case State::initial:
|
||||||
|
break;
|
||||||
case State::accepted:
|
case State::accepted:
|
||||||
|
std::cout << "A termination of accepted request that was not responded to, it's probably an error" << std::endl;
|
||||||
|
FCGX_Finish_r(&raw);
|
||||||
|
break;
|
||||||
|
case State::responding:
|
||||||
|
std::cout << "A termination of responding request that was not actually sent, it's probably an error" << std::endl;
|
||||||
|
FCGX_Finish_r(&raw);
|
||||||
|
break;
|
||||||
case State::responded:
|
case State::responded:
|
||||||
FCGX_Finish_r(&raw);
|
FCGX_Finish_r(&raw);
|
||||||
break;
|
break;
|
||||||
default:
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
state = State::terminated;
|
||||||
|
}
|
||||||
|
std::string_view Request::methodName() const {
|
||||||
|
if (state == State::initial || state == State::terminated)
|
||||||
|
throw std::runtime_error("An attempt to read request method on not accepted request");
|
||||||
|
|
||||||
|
return FCGX_GetParam(REQUEST_METHOD, raw.envp);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Request::isGet() const {
|
Request::Method Request::method() const {
|
||||||
if (state != State::accepted)
|
if (cachedMethod != Method::unknown)
|
||||||
throw std::runtime_error("An attempt to read request type on a wrong request state");
|
return cachedMethod;
|
||||||
|
|
||||||
std::string_view method(FCGX_GetParam(REQUEST_METHOD, raw.envp));
|
std::string_view method = methodName();
|
||||||
return method == GET;
|
for (const auto& pair : methods) {
|
||||||
|
if (pair.first == method) {
|
||||||
|
cachedMethod = pair.second;
|
||||||
|
return pair.second;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return Request::Method::unknown;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Request::wait(int socketDescriptor) {
|
bool Request::wait(int socketDescriptor) {
|
||||||
@ -60,38 +100,91 @@ bool Request::wait(int socketDescriptor) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
OStream Request::getOutputStream() {
|
OStream Request::getOutputStream() {
|
||||||
if (state != State::accepted)
|
|
||||||
throw std::runtime_error("An attempt to request output stream on a wrong request state");
|
|
||||||
|
|
||||||
return OStream(raw.out);
|
return OStream(raw.out);
|
||||||
}
|
}
|
||||||
|
|
||||||
OStream Request::getErrorStream() {
|
OStream Request::getErrorStream() {
|
||||||
if (state != State::accepted)
|
|
||||||
throw std::runtime_error("An attempt to request error stream on a wrong request state");
|
|
||||||
|
|
||||||
return OStream(raw.err);
|
return OStream(raw.err);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string Request::getPath(const std::string& serverName) const {
|
bool Request::active() const {
|
||||||
|
return state != State::initial && state != State::terminated;
|
||||||
|
}
|
||||||
|
|
||||||
|
Response& Request::createResponse() {
|
||||||
|
if (state != State::accepted)
|
||||||
|
throw std::runtime_error("An attempt create response to the request in the wrong state");
|
||||||
|
|
||||||
|
response = std::unique_ptr<Response>(new Response(*this));
|
||||||
|
state = State::responding;
|
||||||
|
|
||||||
|
return *response.get();
|
||||||
|
}
|
||||||
|
|
||||||
|
Response& Request::createResponse(Response::Status status) {
|
||||||
|
if (state != State::accepted)
|
||||||
|
throw std::runtime_error("An attempt create response to the request in the wrong state");
|
||||||
|
|
||||||
|
response = std::unique_ptr<Response>(new Response(*this, status));
|
||||||
|
state = State::responding;
|
||||||
|
|
||||||
|
return *response.get();
|
||||||
|
}
|
||||||
|
|
||||||
|
uint16_t Request::responseCode() const {
|
||||||
|
if (state != State::responded)
|
||||||
|
throw std::runtime_error("An attempt create read response code on the wrong state");
|
||||||
|
|
||||||
|
return response->statusCode();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Request::responseIsComplete() {
|
||||||
|
switch (state) {
|
||||||
|
case State::initial:
|
||||||
|
throw std::runtime_error("An attempt to mark the request as complete, but it wasn't even accepted yet");
|
||||||
|
break;
|
||||||
|
case State::accepted:
|
||||||
|
throw std::runtime_error("An attempt to mark the request as complete, but it wasn't responded");
|
||||||
|
break;
|
||||||
|
case State::responding:
|
||||||
|
state = State::responded;
|
||||||
|
std::cout << responseCode() << '\t' << methodName() << '\t' << path << std::endl;
|
||||||
|
break;
|
||||||
|
case State::responded:
|
||||||
|
throw std::runtime_error("An attempt to mark the request as a complete for the second time");
|
||||||
|
break;
|
||||||
|
case State::terminated:
|
||||||
|
throw std::runtime_error("An attempt to mark the request as a complete on a terminated request");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Request::State Request::currentState() const {
|
||||||
|
return state;
|
||||||
|
}
|
||||||
|
|
||||||
|
void Request::readPath(const std::string& serverName) {
|
||||||
if (state != State::accepted)
|
if (state != State::accepted)
|
||||||
throw std::runtime_error("An attempt to request path on a wrong request state");
|
throw std::runtime_error("An attempt to request path on a wrong request state");
|
||||||
|
|
||||||
std::string path;
|
if (!path.empty())
|
||||||
|
std::cout << "Request already has path \"" + path + "\", but it's being read again, probably an error";
|
||||||
|
|
||||||
std::string_view scriptFileName(FCGX_GetParam(SCRIPT_FILENAME, raw.envp));
|
std::string_view scriptFileName(FCGX_GetParam(SCRIPT_FILENAME, raw.envp));
|
||||||
std::string::size_type snLocation = scriptFileName.find(serverName);
|
std::string::size_type snLocation = scriptFileName.find(serverName);
|
||||||
|
|
||||||
if (snLocation != std::string::npos) {
|
if (snLocation != std::string::npos) {
|
||||||
if (snLocation + serverName.size() < scriptFileName.size())
|
if (snLocation + serverName.size() < scriptFileName.size())
|
||||||
path = scriptFileName.substr(snLocation + serverName.size() + 1);
|
path = scriptFileName.substr(snLocation + serverName.size() + 1);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!path.empty()) {
|
if (!path.empty()) {
|
||||||
while (path.back() == '/')
|
while (path.back() == '/')
|
||||||
path.erase(path.end() - 1);
|
path.erase(path.end() - 1);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string Request::getPath() const {
|
||||||
return path;
|
return path;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -102,14 +195,17 @@ std::string Request::getServerName() const {
|
|||||||
return FCGX_GetParam(SERVER_NAME, raw.envp);;
|
return FCGX_GetParam(SERVER_NAME, raw.envp);;
|
||||||
}
|
}
|
||||||
|
|
||||||
void Request::printEnvironment(std::ostream& out) {
|
void Request::printEnvironment(std::ostream& out) const {
|
||||||
|
if (!active())
|
||||||
|
throw std::runtime_error("An attempt to print environment of a request in a wrong state");
|
||||||
|
|
||||||
char **envp = raw.envp;
|
char **envp = raw.envp;
|
||||||
for (int i = 0; envp[i] != nullptr; ++i) {
|
for (int i = 0; envp[i] != nullptr; ++i) {
|
||||||
out << envp[i] << "\n";
|
out << envp[i] << "\n";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void Request::printEnvironment(nlohmann::json& out) {
|
void Request::printEnvironment(nlohmann::json& out) const {
|
||||||
if (!out.is_object())
|
if (!out.is_object())
|
||||||
return;
|
return;
|
||||||
|
|
||||||
@ -121,3 +217,75 @@ void Request::printEnvironment(nlohmann::json& out) {
|
|||||||
out[std::string(value.substr(0, pos))] = std::string(value.substr(pos + 1, value.size()));
|
out[std::string(value.substr(0, pos))] = std::string(value.substr(pos + 1, value.size()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool Request::isFormUrlEncoded() const {
|
||||||
|
if (!active())
|
||||||
|
throw std::runtime_error("An attempt to read content type of a request in a wrong state");
|
||||||
|
|
||||||
|
std::string_view contentType(FCGX_GetParam(CONTENT_TYPE, raw.envp));
|
||||||
|
if (!contentType.empty() && contentType.find(urlEncoded) != std::string_view::npos) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned int Request::contentLength() const {
|
||||||
|
if (!active())
|
||||||
|
throw std::runtime_error("An attempt to read content length of a request in a wrong state");
|
||||||
|
|
||||||
|
char* cl = FCGX_GetParam(CONTENT_LENGTH, raw.envp);
|
||||||
|
if (cl != nullptr)
|
||||||
|
return atoi(cl);
|
||||||
|
else
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::map<std::string, std::string> Request::getForm() const {
|
||||||
|
if (!active())
|
||||||
|
throw std::runtime_error("An attempt to read form of a request in a wrong state");
|
||||||
|
|
||||||
|
switch (Request::method()) {
|
||||||
|
case Method::get:
|
||||||
|
return urlDecodeAndParse(FCGX_GetParam(QUERY_STRING, raw.envp));
|
||||||
|
break;
|
||||||
|
case Method::post:
|
||||||
|
return getFormPOST();
|
||||||
|
default:
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::map<std::string, std::string> Request::getFormPOST () const {
|
||||||
|
std::map<std::string, std::string> result;
|
||||||
|
std::string_view contentType(FCGX_GetParam(CONTENT_TYPE, raw.envp));
|
||||||
|
if (contentType.empty())
|
||||||
|
return result;
|
||||||
|
|
||||||
|
if (contentType.find(urlEncoded) != std::string_view::npos) {
|
||||||
|
unsigned int length = contentLength();
|
||||||
|
std::string postData(length, '\0');
|
||||||
|
FCGX_GetStr(&postData[0], length, raw.in);
|
||||||
|
result = urlDecodeAndParse(postData);
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string Request::getAuthorizationToken() const {
|
||||||
|
if (!active())
|
||||||
|
throw std::runtime_error("An attempt to read authorization token of a request in a wrong state");
|
||||||
|
|
||||||
|
const char* auth = FCGX_GetParam(AUTHORIZATION, raw.envp);
|
||||||
|
if (auth == nullptr)
|
||||||
|
return std::string();
|
||||||
|
|
||||||
|
std::string result(auth);
|
||||||
|
if (result.find("Bearer") != 0)
|
||||||
|
return std::string();
|
||||||
|
|
||||||
|
result.erase(0, 6);
|
||||||
|
trim(result);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
@ -1,24 +1,39 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
|
#include <string>
|
||||||
|
#include <array>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
#include <fcgiapp.h>
|
#include <fcgiapp.h>
|
||||||
|
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
#include "stream/ostream.h"
|
#include "stream/ostream.h"
|
||||||
|
#include "utils/formdecode.h"
|
||||||
|
#include "utils/helpers.h"
|
||||||
|
#include "response/response.h"
|
||||||
|
|
||||||
class Request {
|
class Request {
|
||||||
|
friend class Response;
|
||||||
public:
|
public:
|
||||||
enum class State {
|
enum class State {
|
||||||
initial,
|
initial,
|
||||||
accepted,
|
accepted,
|
||||||
responded
|
responding,
|
||||||
|
responded,
|
||||||
|
terminated
|
||||||
|
};
|
||||||
|
|
||||||
|
enum class Method {
|
||||||
|
get,
|
||||||
|
post,
|
||||||
|
unknown
|
||||||
};
|
};
|
||||||
|
|
||||||
Request();
|
Request();
|
||||||
@ -29,18 +44,37 @@ public:
|
|||||||
Request& operator = (Request&& other) = delete;
|
Request& operator = (Request&& other) = delete;
|
||||||
|
|
||||||
bool wait(int socketDescriptor);
|
bool wait(int socketDescriptor);
|
||||||
void terminate();
|
bool active() const;
|
||||||
bool isGet() const;
|
|
||||||
|
|
||||||
|
Response& createResponse();
|
||||||
|
Response& createResponse(Response::Status status);
|
||||||
|
|
||||||
|
uint16_t responseCode() const;
|
||||||
|
Method method() const;
|
||||||
|
std::string_view methodName() const;
|
||||||
|
State currentState() const;
|
||||||
|
bool isFormUrlEncoded() const;
|
||||||
|
unsigned int contentLength() const;
|
||||||
|
std::map<std::string, std::string> getForm() const;
|
||||||
|
|
||||||
|
void readPath(const std::string& serverName);
|
||||||
|
std::string getPath() const;
|
||||||
|
std::string getServerName() const;
|
||||||
|
std::string getAuthorizationToken() const;
|
||||||
|
void printEnvironment(std::ostream& out) const;
|
||||||
|
void printEnvironment(nlohmann::json& out) const;
|
||||||
|
|
||||||
|
private:
|
||||||
OStream getOutputStream();
|
OStream getOutputStream();
|
||||||
OStream getErrorStream();
|
OStream getErrorStream();
|
||||||
|
void responseIsComplete();
|
||||||
std::string getPath(const std::string& serverName) const;
|
void terminate();
|
||||||
std::string getServerName() const;
|
std::map<std::string, std::string> getFormPOST() const;
|
||||||
void printEnvironment(std::ostream& out);
|
|
||||||
void printEnvironment(nlohmann::json& out);
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
State state;
|
State state;
|
||||||
FCGX_Request raw;
|
FCGX_Request raw;
|
||||||
|
std::unique_ptr<Response> response;
|
||||||
|
std::string path;
|
||||||
|
mutable Method cachedMethod;
|
||||||
};
|
};
|
||||||
|
@ -1,3 +1,6 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
set(HEADERS
|
set(HEADERS
|
||||||
response.h
|
response.h
|
||||||
)
|
)
|
||||||
@ -6,4 +9,4 @@ set(SOURCES
|
|||||||
response.cpp
|
response.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
target_sources(pica PRIVATE ${SOURCES})
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
||||||
|
@ -1,45 +1,70 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "response.h"
|
#include "response.h"
|
||||||
|
|
||||||
constexpr std::array<std::string_view, static_cast<uint8_t>(Response::Status::__size)> statusCodes = {
|
#include "request/request.h"
|
||||||
|
|
||||||
|
constexpr std::array<uint16_t, static_cast<uint8_t>(Response::Status::__size)> statusCodes = {
|
||||||
|
200,
|
||||||
|
400,
|
||||||
|
401,
|
||||||
|
403,
|
||||||
|
404,
|
||||||
|
405,
|
||||||
|
409,
|
||||||
|
500
|
||||||
|
};
|
||||||
|
|
||||||
|
constexpr std::array<std::string_view, static_cast<uint8_t>(Response::Status::__size)> statuses = {
|
||||||
"Status: 200 OK",
|
"Status: 200 OK",
|
||||||
|
"Status: 400 Bad Request",
|
||||||
|
"Status: 401 Unauthorized",
|
||||||
|
"Status: 403 Forbidden",
|
||||||
"Status: 404 Not Found",
|
"Status: 404 Not Found",
|
||||||
"Status: 405 Method Not Allowed",
|
"Status: 405 Method Not Allowed",
|
||||||
|
"Status: 409 Conflict",
|
||||||
"Status: 500 Internal Error"
|
"Status: 500 Internal Error"
|
||||||
};
|
};
|
||||||
|
|
||||||
constexpr std::array<std::string_view, static_cast<uint8_t>(Response::ContentType::__size)> contentTypes = {
|
constexpr std::array<std::string_view, static_cast<uint8_t>(Response::ContentType::__size)> contentTypes = {
|
||||||
"Content-type: text/plain",
|
"Content-Type: text/plain",
|
||||||
"Content-type: application/json"
|
"Content-Type: application/json"
|
||||||
};
|
};
|
||||||
|
|
||||||
Response::Response():
|
Response::Response(Request& request):
|
||||||
|
request(request),
|
||||||
status(Status::ok),
|
status(Status::ok),
|
||||||
type(ContentType::text),
|
type(ContentType::text),
|
||||||
body()
|
body()
|
||||||
{}
|
{}
|
||||||
|
|
||||||
Response::Response(Status status):
|
Response::Response(Request& request, Status status):
|
||||||
|
request(request),
|
||||||
status(status),
|
status(status),
|
||||||
type(ContentType::text),
|
type(ContentType::text),
|
||||||
body()
|
body()
|
||||||
{}
|
{}
|
||||||
|
|
||||||
void Response::replyTo(Request& request) const {
|
void Response::send() const {
|
||||||
// OStream out = status == Status::ok ?
|
// OStream out = status == Status::ok ?
|
||||||
// request.getOutputStream() :
|
// request.getOutputStream() :
|
||||||
// request.getErrorStream();
|
// request.getErrorStream();
|
||||||
OStream out = request.getOutputStream();
|
OStream out = request.getOutputStream();
|
||||||
|
|
||||||
out << statusCodes[static_cast<uint8_t>(status)];
|
out << statuses[static_cast<uint8_t>(status)] << "\r\n";
|
||||||
if (!body.empty())
|
if (!body.empty())
|
||||||
out << '\n'
|
out << contentTypes[static_cast<uint8_t>(type)] << "\r\n"
|
||||||
<< contentTypes[static_cast<uint8_t>(type)]
|
<< "\r\n"
|
||||||
<< '\n'
|
|
||||||
<< '\n'
|
|
||||||
<< body;
|
<< body;
|
||||||
|
else
|
||||||
|
out << "\r\n";
|
||||||
|
|
||||||
|
request.responseIsComplete();
|
||||||
|
}
|
||||||
|
|
||||||
|
uint16_t Response::statusCode() const {
|
||||||
|
return statusCodes[static_cast<uint8_t>(status)];
|
||||||
}
|
}
|
||||||
|
|
||||||
void Response::setBody(const std::string& body) {
|
void Response::setBody(const std::string& body) {
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
@ -9,15 +9,21 @@
|
|||||||
|
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
#include "request/request.h"
|
|
||||||
#include "stream/ostream.h"
|
#include "stream/ostream.h"
|
||||||
|
|
||||||
|
class Request;
|
||||||
class Response {
|
class Response {
|
||||||
|
friend class Request;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
enum class Status {
|
enum class Status {
|
||||||
ok,
|
ok,
|
||||||
|
badRequest,
|
||||||
|
unauthorized,
|
||||||
|
forbidden,
|
||||||
notFound,
|
notFound,
|
||||||
methodNotAllowed,
|
methodNotAllowed,
|
||||||
|
conflict,
|
||||||
internalError,
|
internalError,
|
||||||
__size
|
__size
|
||||||
};
|
};
|
||||||
@ -27,14 +33,19 @@ public:
|
|||||||
json,
|
json,
|
||||||
__size
|
__size
|
||||||
};
|
};
|
||||||
Response();
|
|
||||||
Response(Status status);
|
|
||||||
|
|
||||||
void replyTo(Request& request) const;
|
uint16_t statusCode() const;
|
||||||
|
|
||||||
|
void send() const;
|
||||||
void setBody(const std::string& body);
|
void setBody(const std::string& body);
|
||||||
void setBody(const nlohmann::json& body);
|
void setBody(const nlohmann::json& body);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
Response(Request& request);
|
||||||
|
Response(Request& request, Status status);
|
||||||
|
|
||||||
|
private:
|
||||||
|
Request& request;
|
||||||
Status status;
|
Status status;
|
||||||
ContentType type;
|
ContentType type;
|
||||||
std::string body;
|
std::string body;
|
||||||
|
28
run.sh.in
Normal file
28
run.sh.in
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
start_service() {
|
||||||
|
if (systemctl is-active --quiet $1) then
|
||||||
|
echo "$1 is already running"
|
||||||
|
else
|
||||||
|
echo "$1 is not running, going to use sudo to start it"
|
||||||
|
if (sudo systemctl start $1) then
|
||||||
|
echo "$1 started"
|
||||||
|
else
|
||||||
|
exit
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
if [ ! -d "/run/pica" ]; then
|
||||||
|
echo "required unix socket was not found, going to use sudo to create it"
|
||||||
|
sudo mkdir /run/pica
|
||||||
|
sudo chown $USER:$USER /run/pica
|
||||||
|
fi
|
||||||
|
|
||||||
|
start_service "mariadb"
|
||||||
|
start_service "httpd"
|
||||||
|
|
||||||
|
$(dirname "$0")/@PROJECT_NAME@
|
@ -1,11 +1,16 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
set(HEADERS
|
set(HEADERS
|
||||||
server.h
|
server.h
|
||||||
router.h
|
router.h
|
||||||
|
session.h
|
||||||
)
|
)
|
||||||
|
|
||||||
set(SOURCES
|
set(SOURCES
|
||||||
server.cpp
|
server.cpp
|
||||||
router.cpp
|
router.cpp
|
||||||
|
session.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
target_sources(pica PRIVATE ${SOURCES})
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
||||||
|
@ -1,44 +1,77 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "router.h"
|
#include "router.h"
|
||||||
|
|
||||||
|
#include "request/redirect.h"
|
||||||
|
|
||||||
Router::Router():
|
Router::Router():
|
||||||
table()
|
get(),
|
||||||
{
|
post()
|
||||||
|
{}
|
||||||
|
|
||||||
}
|
void Router::addRoute(Handler handler) {
|
||||||
|
std::pair<std::map<std::string, Handler>::const_iterator, bool> result;
|
||||||
|
switch (handler->method) {
|
||||||
|
case Request::Method::get:
|
||||||
|
result = get.emplace(handler->path, std::move(handler));
|
||||||
|
break;
|
||||||
|
case Request::Method::post:
|
||||||
|
result = post.emplace(handler->path, std::move(handler));
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw std::runtime_error("An attempt to register handler with unsupported method type: " + std::to_string((int)handler->method));
|
||||||
|
}
|
||||||
|
|
||||||
void Router::addRoute(const std::string& path, const Handler& handler) {
|
|
||||||
auto result = table.emplace(path, handler);
|
|
||||||
if (!result.second)
|
if (!result.second)
|
||||||
std::cerr << "could'not add route " + path + " to the routing table";
|
throw std::runtime_error("could'not add route " + handler->path + " to the routing table");
|
||||||
}
|
}
|
||||||
|
|
||||||
void Router::route(const std::string& path, std::unique_ptr<Request> request, Server* server) {
|
void Router::route(std::unique_ptr<Request> request) {
|
||||||
auto itr = table.find(path);
|
std::map<std::string, Handler>::const_iterator itr, end;
|
||||||
if (itr == table.end())
|
switch (request->method()) {
|
||||||
return handleNotFound(path, std::move(request));
|
case Request::Method::get:
|
||||||
|
itr = get.find(request->getPath());
|
||||||
|
end = get.end();
|
||||||
|
break;
|
||||||
|
case Request::Method::post:
|
||||||
|
itr = post.find(request->getPath());
|
||||||
|
end = post.end();
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
return handleMethodNotAllowed(std::move(request));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (itr == end)
|
||||||
|
return handleNotFound(std::move(request));
|
||||||
|
|
||||||
try {
|
try {
|
||||||
bool result = itr->second(request.get(), server);
|
itr->second->handle(*request.get());
|
||||||
if (!result)
|
|
||||||
|
if (request->currentState() != Request::State::responded)
|
||||||
handleInternalError(std::runtime_error("handler failed to handle the request"), std::move(request));
|
handleInternalError(std::runtime_error("handler failed to handle the request"), std::move(request));
|
||||||
|
} catch (const Redirect& redirect) {
|
||||||
|
redirect.destination->accept(std::move(request));
|
||||||
} catch (const std::exception& e) {
|
} catch (const std::exception& e) {
|
||||||
handleInternalError(e, std::move(request));
|
handleInternalError(e, std::move(request));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void Router::handleNotFound(const std::string& path, std::unique_ptr<Request> request) {
|
void Router::handleNotFound(std::unique_ptr<Request> request) {
|
||||||
Response notFound(Response::Status::notFound);
|
Response& notFound = request->createResponse(Response::Status::notFound);
|
||||||
|
std::string path = request->getPath();
|
||||||
notFound.setBody(std::string("Path \"") + path + "\" was not found");
|
notFound.setBody(std::string("Path \"") + path + "\" was not found");
|
||||||
notFound.replyTo(*request.get());
|
notFound.send();
|
||||||
std::cerr << "Not found: " << path << std::endl;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void Router::handleInternalError(const std::exception& exception, std::unique_ptr<Request> request) {
|
void Router::handleInternalError(const std::exception& exception, std::unique_ptr<Request> request) {
|
||||||
Response error(Response::Status::internalError);
|
Response& error = request->createResponse(Response::Status::internalError);
|
||||||
error.setBody(std::string(exception.what()));
|
error.setBody(std::string(exception.what()));
|
||||||
error.replyTo(*request.get());
|
error.send();
|
||||||
std::cerr << "Internal error: " << exception.what() << std::endl;
|
}
|
||||||
|
|
||||||
|
void Router::handleMethodNotAllowed(std::unique_ptr<Request> request) {
|
||||||
|
Response& error = request->createResponse(Response::Status::methodNotAllowed);
|
||||||
|
error.setBody(std::string("Method not allowed"));
|
||||||
|
error.send();
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
@ -10,23 +10,25 @@
|
|||||||
|
|
||||||
#include "request/request.h"
|
#include "request/request.h"
|
||||||
#include "response/response.h"
|
#include "response/response.h"
|
||||||
|
#include "handler/handler.h"
|
||||||
|
|
||||||
class Server;
|
class Server;
|
||||||
|
|
||||||
class Router {
|
class Router {
|
||||||
|
using Handler = std::unique_ptr<Handler::Handler>;
|
||||||
public:
|
public:
|
||||||
using Handler = std::function<bool(Request*, Server*)>;
|
|
||||||
|
|
||||||
Router();
|
Router();
|
||||||
|
|
||||||
void addRoute(const std::string& path, const Handler& handler);
|
void addRoute(Handler handler);
|
||||||
void route(const std::string& path, std::unique_ptr<Request> request, Server* server);
|
void route(std::unique_ptr<Request> request);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void handleNotFound(const std::string& path, std::unique_ptr<Request> request);
|
void handleNotFound(std::unique_ptr<Request> request);
|
||||||
void handleInternalError(const std::exception& exception, std::unique_ptr<Request> request);
|
void handleInternalError(const std::exception& exception, std::unique_ptr<Request> request);
|
||||||
|
void handleMethodNotAllowed(std::unique_ptr<Request> request);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::map<std::string, Handler> table;
|
std::map<std::string, Handler> get;
|
||||||
|
std::map<std::string, Handler> post;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
@ -1,45 +1,96 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "server.h"
|
#include "server.h"
|
||||||
|
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
#include "database/exceptions.h"
|
||||||
|
|
||||||
|
#include "handler/info.h"
|
||||||
|
#include "handler/env.h"
|
||||||
|
#include "handler/register.h"
|
||||||
|
#include "handler/login.h"
|
||||||
|
#include "handler/poll.h"
|
||||||
|
#include "handler/assets.h"
|
||||||
|
#include "handler/addasset.h"
|
||||||
|
#include "handler/deleteasset.h"
|
||||||
|
#include "handler/currencies.h"
|
||||||
|
#include "handler/updateasset.h"
|
||||||
|
#include "handler/addtransaction.h"
|
||||||
|
#include "handler/transactions.h"
|
||||||
|
#include "handler/deletetransaction.h"
|
||||||
|
#include "handler/updatetransaction.h"
|
||||||
|
|
||||||
|
#include "taskmanager/route.h"
|
||||||
|
|
||||||
|
constexpr const char* pepper = "well, not much of a secret, huh?";
|
||||||
|
constexpr const char* dbLogin = "pica";
|
||||||
|
constexpr const char* dbPassword = "pica";
|
||||||
|
constexpr const char* dbName = "pica";
|
||||||
|
constexpr const char* dbPath = "/run/mysqld/mysqld.sock";
|
||||||
|
constexpr uint8_t dbConnectionsCount = 4;
|
||||||
|
constexpr uint32_t pollTimout = 10000;
|
||||||
|
|
||||||
|
constexpr const char* randomChars = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
|
||||||
|
constexpr uint8_t saltSize = 16;
|
||||||
|
constexpr uint8_t hashSize = 32;
|
||||||
|
constexpr uint8_t hashParallel = 1;
|
||||||
|
constexpr uint8_t hashIterations = 2;
|
||||||
|
constexpr uint32_t hashMemoryCost = 65536;
|
||||||
|
|
||||||
constexpr uint8_t currentDbVesion = 1;
|
constexpr uint8_t currentDbVesion = 1;
|
||||||
|
|
||||||
Server::Server():
|
Server::Server ():
|
||||||
|
std::enable_shared_from_this<Server>(),
|
||||||
terminating(false),
|
terminating(false),
|
||||||
requestCount(0),
|
requestCount(0),
|
||||||
serverName(std::nullopt),
|
serverName(std::nullopt),
|
||||||
router(),
|
router(std::make_shared<Router>()),
|
||||||
db()
|
pool(DB::Pool::create()),
|
||||||
|
taskManager(std::make_shared<TM::Manager>()),
|
||||||
|
scheduler(std::make_shared<TM::Scheduler>(taskManager)),
|
||||||
|
sessions()
|
||||||
{
|
{
|
||||||
std::cout << "Startig pica..." << std::endl;
|
std::cout << "Startig pica..." << std::endl;
|
||||||
|
|
||||||
db = DBInterface::create(DBInterface::Type::mysql);
|
|
||||||
std::cout << "Database type: MySQL" << std::endl;
|
std::cout << "Database type: MySQL" << std::endl;
|
||||||
|
pool->addInterfaces(
|
||||||
|
DB::Interface::Type::mysql,
|
||||||
|
dbConnectionsCount,
|
||||||
|
dbLogin,
|
||||||
|
dbPassword,
|
||||||
|
dbName,
|
||||||
|
dbPath
|
||||||
|
);
|
||||||
|
|
||||||
db->setCredentials("pica", "pica");
|
DB::Resource db = pool->request();
|
||||||
db->setDatabase("pica");
|
|
||||||
|
|
||||||
bool connected = false;
|
|
||||||
try {
|
|
||||||
db->connect("/run/mysqld/mysqld.sock");
|
|
||||||
connected = true;
|
|
||||||
std::cout << "Successfully connected to the database" << std::endl;
|
|
||||||
|
|
||||||
} catch (const std::runtime_error& e) {
|
|
||||||
std::cerr << "Couldn't connect to the database: " << e.what() << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (connected)
|
|
||||||
db->migrate(currentDbVesion);
|
db->migrate(currentDbVesion);
|
||||||
|
|
||||||
router.addRoute("info", Server::info);
|
|
||||||
router.addRoute("env", Server::printEnvironment);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Server::~Server() {}
|
Server::~Server () {}
|
||||||
|
|
||||||
|
void Server::run (int socketDescriptor) {
|
||||||
|
std::shared_ptr<Server> srv = shared_from_this();
|
||||||
|
|
||||||
|
router->addRoute(std::make_unique<Handler::Info>());
|
||||||
|
router->addRoute(std::make_unique<Handler::Env>());
|
||||||
|
router->addRoute(std::make_unique<Handler::Register>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::Login>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::Poll>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::Assets>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::AddAsset>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::DeleteAsset>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::Currencies>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::UpdateAsset>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::AddTransaction>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::Transactions>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::DeleteTransaction>(srv));
|
||||||
|
router->addRoute(std::make_unique<Handler::UpdateTransaction>(srv));
|
||||||
|
|
||||||
|
taskManager->start();
|
||||||
|
scheduler->start();
|
||||||
|
|
||||||
void Server::run(int socketDescriptor) {
|
|
||||||
while (!terminating) {
|
while (!terminating) {
|
||||||
std::unique_ptr<Request> request = std::make_unique<Request>();
|
std::unique_ptr<Request> request = std::make_unique<Request>();
|
||||||
bool result = request->wait(socketDescriptor);
|
bool result = request->wait(socketDescriptor);
|
||||||
@ -51,7 +102,7 @@ void Server::run(int socketDescriptor) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void Server::handleRequest(std::unique_ptr<Request> request) {
|
void Server::handleRequest (std::unique_ptr<Request> request) {
|
||||||
++requestCount;
|
++requestCount;
|
||||||
if (!serverName) {
|
if (!serverName) {
|
||||||
try {
|
try {
|
||||||
@ -59,49 +110,115 @@ void Server::handleRequest(std::unique_ptr<Request> request) {
|
|||||||
std::cout << "received server name " << serverName.value() << std::endl;
|
std::cout << "received server name " << serverName.value() << std::endl;
|
||||||
} catch (...) {
|
} catch (...) {
|
||||||
std::cerr << "failed to read server name" << std::endl;
|
std::cerr << "failed to read server name" << std::endl;
|
||||||
Response error(Response::Status::internalError);
|
Response& error = request->createResponse(Response::Status::internalError);
|
||||||
error.replyTo(*request.get());
|
error.send();
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!request->isGet()) {
|
request->readPath(serverName.value());
|
||||||
static const Response methodNotAllowed(Response::Status::methodNotAllowed);
|
|
||||||
methodNotAllowed.replyTo(*request.get());
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
auto route = std::make_unique<TM::Route>(router, std::move(request));
|
||||||
|
taskManager->schedule(std::move(route));
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string Server::generateRandomString (std::size_t length) {
|
||||||
|
std::random_device rd;
|
||||||
|
std::mt19937 gen(rd());
|
||||||
|
std::uniform_int_distribution<uint8_t> distribution(0, std::strlen(randomChars) - 1);
|
||||||
|
|
||||||
|
std::string result(length, 0);
|
||||||
|
for (size_t i = 0; i < length; ++i)
|
||||||
|
result[i] = randomChars[distribution(gen)];
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned int Server::registerAccount (const std::string& login, const std::string& password) {
|
||||||
|
std::size_t encSize = argon2_encodedlen(
|
||||||
|
hashIterations, hashMemoryCost,
|
||||||
|
hashParallel, saltSize, hashSize, Argon2_id
|
||||||
|
);
|
||||||
|
|
||||||
|
std::string hash(encSize, 0);
|
||||||
|
std::string salt = generateRandomString(saltSize);
|
||||||
|
std::string spiced = password + pepper;
|
||||||
|
|
||||||
|
int result = argon2id_hash_encoded(
|
||||||
|
hashIterations, hashMemoryCost, hashParallel,
|
||||||
|
spiced.data(), spiced.size(),
|
||||||
|
salt.data(), saltSize,
|
||||||
|
hashSize, hash.data(), encSize
|
||||||
|
);
|
||||||
|
|
||||||
|
if (result != ARGON2_OK)
|
||||||
|
throw std::runtime_error(std::string("Hashing failed: ") + argon2_error_message(result));
|
||||||
|
|
||||||
|
DB::Resource db = pool->request();
|
||||||
|
return db->registerAccount(login, hash);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool Server::validatePassword (const std::string& login, const std::string& password) {
|
||||||
|
DB::Resource db = pool->request();
|
||||||
|
std::string hash = db->getAccountHash(login);
|
||||||
|
|
||||||
|
std::string spiced = password + pepper;
|
||||||
|
int result = argon2id_verify(hash.data(), spiced.data(), spiced.size());
|
||||||
|
|
||||||
|
switch (result) {
|
||||||
|
case ARGON2_OK:
|
||||||
|
return true;
|
||||||
|
case ARGON2_VERIFY_MISMATCH:
|
||||||
|
return false;
|
||||||
|
default:
|
||||||
|
throw std::runtime_error(std::string("Failed to verify password: ") + argon2_error_message(result));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Session& Server::openSession (const std::string& login) {
|
||||||
|
std::string accessToken, renewToken;
|
||||||
|
DB::Session s;
|
||||||
|
s.id = 0;
|
||||||
|
int counter = 10;
|
||||||
|
do {
|
||||||
try {
|
try {
|
||||||
std::string path = request->getPath(serverName.value());
|
accessToken = generateRandomString(32);
|
||||||
router.route(path.data(), std::move(request), this);
|
renewToken = generateRandomString(32);
|
||||||
} catch (const std::exception e) {
|
DB::Resource db = pool->request();
|
||||||
Response error(Response::Status::internalError);
|
s = db->createSession(login, accessToken, renewToken);
|
||||||
error.setBody(std::string(e.what()));
|
break;
|
||||||
error.replyTo(*request.get());
|
} catch (const DB::Duplicate& e) {
|
||||||
|
std::cout << "Duplicate on creating session, trying again with different tokens";
|
||||||
}
|
}
|
||||||
|
} while (--counter != 0);
|
||||||
|
|
||||||
|
if (s.id == 0)
|
||||||
|
throw std::runtime_error("Couldn't create session, ran out of attempts");
|
||||||
|
|
||||||
|
std::unique_ptr<Session>& session = sessions[accessToken]
|
||||||
|
= std::make_unique<Session>(scheduler, s.id, s.owner, s.accessToken, s.renewToken, pollTimout);
|
||||||
|
return *session.get();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Server::printEnvironment(Request* request, Server* server) {
|
Session& Server::getSession (const std::string& accessToken) {
|
||||||
(void)server;
|
Sessions::const_iterator itr = sessions.find(accessToken);
|
||||||
nlohmann::json body = nlohmann::json::object();
|
if (itr != sessions.end())
|
||||||
request->printEnvironment(body);
|
return *(itr->second);
|
||||||
|
|
||||||
Response res;
|
DB::Resource db = pool->request();
|
||||||
res.setBody(body);
|
DB::Session s = db->findSession(accessToken);
|
||||||
res.replyTo(*request);
|
std::unique_ptr<Session>& session = sessions[accessToken] = std::make_unique<Session>(
|
||||||
|
scheduler,
|
||||||
return true;
|
s.id,
|
||||||
|
s.owner,
|
||||||
|
s.accessToken,
|
||||||
|
s.renewToken,
|
||||||
|
pollTimout
|
||||||
|
);
|
||||||
|
return *session.get();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Server::info(Request* request, Server* server) {
|
|
||||||
(void)server;
|
|
||||||
Response res;
|
|
||||||
nlohmann::json body = nlohmann::json::object();
|
|
||||||
body["type"] = "Pica";
|
|
||||||
body["version"] = "0.0.1";
|
|
||||||
|
|
||||||
res.setBody(body);
|
DB::Resource Server::getDatabase () {
|
||||||
res.replyTo(*request);
|
return pool->request();
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
@ -10,35 +10,50 @@
|
|||||||
#include <string_view>
|
#include <string_view>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
#include <fcgiapp.h>
|
#include <fcgiapp.h>
|
||||||
#include <fcgio.h>
|
#include <fcgio.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
|
||||||
#include <nlohmann/json.hpp>
|
#include <argon2.h>
|
||||||
|
|
||||||
#include "request/request.h"
|
#include "request/request.h"
|
||||||
#include "response/response.h"
|
#include "response/response.h"
|
||||||
#include "router.h"
|
#include "router.h"
|
||||||
#include "database/dbinterface.h"
|
#include "session.h"
|
||||||
|
#include "database/pool.h"
|
||||||
|
#include "utils/helpers.h"
|
||||||
|
#include "config.h"
|
||||||
|
#include "taskmanager/manager.h"
|
||||||
|
#include "taskmanager/scheduler.h"
|
||||||
|
|
||||||
class Server {
|
class Server : public std::enable_shared_from_this<Server> {
|
||||||
public:
|
public:
|
||||||
Server();
|
Server();
|
||||||
~Server();
|
~Server();
|
||||||
|
|
||||||
void run(int socketDescriptor);
|
void run(int socketDescriptor);
|
||||||
|
|
||||||
|
unsigned int registerAccount(const std::string& login, const std::string& password);
|
||||||
|
bool validatePassword(const std::string& login, const std::string& password);
|
||||||
|
Session& openSession(const std::string& login);
|
||||||
|
Session& getSession(const std::string& accessToken);
|
||||||
|
DB::Resource getDatabase();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void handleRequest(std::unique_ptr<Request> request);
|
void handleRequest(std::unique_ptr<Request> request);
|
||||||
|
static std::string generateRandomString(std::size_t length);
|
||||||
static bool info(Request* request, Server* server);
|
|
||||||
static bool printEnvironment(Request* request, Server* server);
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
using Sessions = std::map<std::string, std::unique_ptr<Session>>;
|
||||||
|
|
||||||
bool terminating;
|
bool terminating;
|
||||||
uint64_t requestCount;
|
uint64_t requestCount;
|
||||||
std::optional<std::string> serverName;
|
std::optional<std::string> serverName;
|
||||||
Router router;
|
std::shared_ptr<Router> router;
|
||||||
std::unique_ptr<DBInterface> db;
|
std::shared_ptr<DB::Pool> pool;
|
||||||
|
std::shared_ptr<TM::Manager> taskManager;
|
||||||
|
std::shared_ptr<TM::Scheduler> scheduler;
|
||||||
|
Sessions sessions;
|
||||||
};
|
};
|
||||||
|
218
server/session.cpp
Normal file
218
server/session.cpp
Normal file
@ -0,0 +1,218 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "session.h"
|
||||||
|
|
||||||
|
#include "handler/poll.h"
|
||||||
|
|
||||||
|
Session::Session(
|
||||||
|
std::weak_ptr<TM::Scheduler> scheduler,
|
||||||
|
unsigned int id,
|
||||||
|
unsigned int owner,
|
||||||
|
const std::string& access,
|
||||||
|
const std::string& renew,
|
||||||
|
unsigned int timeout
|
||||||
|
):
|
||||||
|
id(id),
|
||||||
|
owner(owner),
|
||||||
|
scheduler(scheduler),
|
||||||
|
access(access),
|
||||||
|
renew(renew),
|
||||||
|
polling(nullptr),
|
||||||
|
timeoutId(TM::Scheduler::none),
|
||||||
|
timeout(timeout),
|
||||||
|
mtx(),
|
||||||
|
cache({
|
||||||
|
{"system", {
|
||||||
|
{"invalidate", true}
|
||||||
|
}}
|
||||||
|
})
|
||||||
|
{}
|
||||||
|
|
||||||
|
Session::~Session () {
|
||||||
|
if (timeoutId != TM::Scheduler::none) {
|
||||||
|
if (std::shared_ptr<TM::Scheduler> sch = scheduler.lock())
|
||||||
|
sch->cancel(timeoutId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string Session::getAccessToken() const {
|
||||||
|
return access;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string Session::getRenewToken() const {
|
||||||
|
return renew;
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::accept(std::unique_ptr<Request> request) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::shared_ptr<TM::Scheduler> sch = scheduler.lock();
|
||||||
|
if (polling) {
|
||||||
|
if (timeoutId != TM::Scheduler::none) {
|
||||||
|
if (sch)
|
||||||
|
sch->cancel(timeoutId);
|
||||||
|
|
||||||
|
timeoutId = TM::Scheduler::none;
|
||||||
|
}
|
||||||
|
Handler::Poll::error(*polling.get(), Handler::Poll::Result::replace, Response::Status::ok);
|
||||||
|
polling.reset();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::map form = request->getForm();
|
||||||
|
auto clear = form.find("clearCache");
|
||||||
|
if (clear != form.end() && clear->second == "all")
|
||||||
|
cache.clear();
|
||||||
|
|
||||||
|
if (!cache.empty())
|
||||||
|
return sendUpdates(std::move(request));
|
||||||
|
|
||||||
|
if (!sch) {
|
||||||
|
std::cerr << "Was unable to schedule polling timeout, replying with an error" << std::endl;
|
||||||
|
Handler::Poll::error(*request.get(), Handler::Poll::Result::unknownError, Response::Status::internalError);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
timeoutId = sch->schedule(std::bind(&Session::onTimeout, this), timeout);
|
||||||
|
polling = std::move(request);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::sendUpdates (std::unique_ptr<Request> request) {
|
||||||
|
Response& res = request->createResponse(Response::Status::ok);
|
||||||
|
nlohmann::json body = nlohmann::json::object();
|
||||||
|
body["result"] = Handler::Poll::Result::success;
|
||||||
|
nlohmann::json& data = body["data"] = nlohmann::json::object();
|
||||||
|
|
||||||
|
for (const auto& category : cache) {
|
||||||
|
nlohmann::json& cat = data[category.first] = nlohmann::json::object();
|
||||||
|
for (const auto& entry : category.second)
|
||||||
|
cat[entry.first] = entry.second;
|
||||||
|
}
|
||||||
|
|
||||||
|
res.setBody(body);
|
||||||
|
res.send();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::onTimeout () {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
timeoutId = TM::Scheduler::none;
|
||||||
|
Handler::Poll::error(*polling.get(), Handler::Poll::Result::timeout, Response::Status::ok);
|
||||||
|
polling.reset();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::assetAdded (const DB::Asset& asset) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::map<std::string, nlohmann::json>& assets = cache["assets"];
|
||||||
|
auto addedItr = assets.find("added");
|
||||||
|
if (addedItr == assets.end())
|
||||||
|
addedItr = assets.emplace("added", nlohmann::json::array()).first;
|
||||||
|
|
||||||
|
addedItr->second.push_back(asset.toJSON());
|
||||||
|
checkUpdates();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::assetChanged (const DB::Asset& asset) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::map<std::string, nlohmann::json>& assets = cache["assets"];
|
||||||
|
auto itr = assets.find("changed");
|
||||||
|
if (itr == assets.end())
|
||||||
|
itr = assets.emplace("changed", nlohmann::json::array()).first;
|
||||||
|
|
||||||
|
removeByID(itr->second, asset.id);
|
||||||
|
itr->second.push_back(asset.toJSON());
|
||||||
|
|
||||||
|
checkUpdates();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::assetRemoved (unsigned int assetId) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::map<std::string, nlohmann::json>& assets = cache["assets"];
|
||||||
|
auto itr = assets.find("added");
|
||||||
|
if (itr != assets.end())
|
||||||
|
removeByID(itr->second, assetId);
|
||||||
|
else {
|
||||||
|
itr = assets.find("removed");
|
||||||
|
if (itr == assets.end())
|
||||||
|
itr = assets.emplace("removed", nlohmann::json::array()).first;
|
||||||
|
|
||||||
|
itr->second.push_back(assetId);
|
||||||
|
}
|
||||||
|
|
||||||
|
itr = assets.find("changed");
|
||||||
|
if (itr != assets.end())
|
||||||
|
removeByID(itr->second, assetId);
|
||||||
|
|
||||||
|
checkUpdates();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::transactionAdded(const DB::Transaction& txn) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::map<std::string, nlohmann::json>& txns = cache["transactions"];
|
||||||
|
auto itr = txns.find("changed");
|
||||||
|
if (itr == txns.end())
|
||||||
|
itr = txns.emplace("changed", nlohmann::json::array()).first;
|
||||||
|
|
||||||
|
removeByID(itr->second, txn.id);
|
||||||
|
itr->second.push_back(txn.toJSON());
|
||||||
|
|
||||||
|
checkUpdates();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::transactionChanged(const DB::Transaction& txn) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::map<std::string, nlohmann::json>& txns = cache["transactions"];
|
||||||
|
auto itr = txns.find("changed");
|
||||||
|
if (itr == txns.end())
|
||||||
|
itr = txns.emplace("changed", nlohmann::json::array()).first;
|
||||||
|
|
||||||
|
removeByID(itr->second, txn.id);
|
||||||
|
itr->second.push_back(txn.toJSON());
|
||||||
|
|
||||||
|
checkUpdates();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::transactionRemoved(unsigned int txnId) {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
std::map<std::string, nlohmann::json>& txns = cache["transactions"];
|
||||||
|
auto itr = txns.find("added");
|
||||||
|
if (itr != txns.end())
|
||||||
|
removeByID(itr->second, txnId);
|
||||||
|
else {
|
||||||
|
itr = txns.find("removed");
|
||||||
|
if (itr == txns.end())
|
||||||
|
itr = txns.emplace("removed", nlohmann::json::array()).first;
|
||||||
|
|
||||||
|
itr->second.push_back(txnId);
|
||||||
|
}
|
||||||
|
|
||||||
|
itr = txns.find("changed");
|
||||||
|
if (itr != txns.end())
|
||||||
|
removeByID(itr->second, txnId);
|
||||||
|
|
||||||
|
checkUpdates();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::removeByID(nlohmann::json& array, unsigned int id) {
|
||||||
|
array.erase(
|
||||||
|
std::remove_if(
|
||||||
|
array.begin(),
|
||||||
|
array.end(),
|
||||||
|
[id](const nlohmann::json& item) {
|
||||||
|
return item["id"].get<unsigned int>() == id;
|
||||||
|
}
|
||||||
|
),
|
||||||
|
array.end()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Session::checkUpdates () {
|
||||||
|
std::shared_ptr<TM::Scheduler> sch = scheduler.lock();
|
||||||
|
if (polling) {
|
||||||
|
if (timeoutId != TM::Scheduler::none) {
|
||||||
|
if (sch)
|
||||||
|
sch->cancel(timeoutId);
|
||||||
|
|
||||||
|
timeoutId = TM::Scheduler::none;
|
||||||
|
}
|
||||||
|
sendUpdates(std::move(polling));
|
||||||
|
}
|
||||||
|
}
|
66
server/session.h
Normal file
66
server/session.h
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <map>
|
||||||
|
#include <mutex>
|
||||||
|
|
||||||
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
|
#include "request/accepting.h"
|
||||||
|
#include "taskmanager/scheduler.h"
|
||||||
|
|
||||||
|
#include "database/schema/asset.h"
|
||||||
|
#include "database/schema/transaction.h"
|
||||||
|
|
||||||
|
class Session : public Accepting {
|
||||||
|
public:
|
||||||
|
Session (
|
||||||
|
std::weak_ptr<TM::Scheduler> scheduler,
|
||||||
|
unsigned int id,
|
||||||
|
unsigned int owner,
|
||||||
|
const std::string& access,
|
||||||
|
const std::string& renew,
|
||||||
|
unsigned int timeout
|
||||||
|
);
|
||||||
|
Session (const Session&) = delete;
|
||||||
|
Session (Session&& other) = delete;
|
||||||
|
~Session ();
|
||||||
|
Session& operator = (const Session&) = delete;
|
||||||
|
Session& operator = (Session&& other) = delete;
|
||||||
|
|
||||||
|
std::string getAccessToken () const;
|
||||||
|
std::string getRenewToken () const;
|
||||||
|
void accept (std::unique_ptr<Request> request) override;
|
||||||
|
|
||||||
|
const unsigned int id;
|
||||||
|
const unsigned int owner;
|
||||||
|
|
||||||
|
void assetAdded (const DB::Asset& asset);
|
||||||
|
void assetChanged (const DB::Asset& asset);
|
||||||
|
void assetRemoved (unsigned int assetId);
|
||||||
|
|
||||||
|
void transactionAdded(const DB::Transaction& txn);
|
||||||
|
void transactionChanged(const DB::Transaction& txn);
|
||||||
|
void transactionRemoved(unsigned int txnId);
|
||||||
|
|
||||||
|
private:
|
||||||
|
void onTimeout ();
|
||||||
|
void sendUpdates (std::unique_ptr<Request> request);
|
||||||
|
void checkUpdates ();
|
||||||
|
|
||||||
|
void static removeByID (nlohmann::json& array, unsigned int id);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::weak_ptr<TM::Scheduler> scheduler;
|
||||||
|
std::string access;
|
||||||
|
std::string renew;
|
||||||
|
std::unique_ptr<Request> polling;
|
||||||
|
TM::Record::ID timeoutId;
|
||||||
|
TM::Scheduler::Delay timeout;
|
||||||
|
std::mutex mtx;
|
||||||
|
|
||||||
|
std::map<std::string, std::map<std::string, nlohmann::json>> cache;
|
||||||
|
};
|
@ -1,3 +1,6 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
set(HEADERS
|
set(HEADERS
|
||||||
stream.h
|
stream.h
|
||||||
ostream.h
|
ostream.h
|
||||||
@ -8,4 +11,4 @@ set(SOURCES
|
|||||||
ostream.cpp
|
ostream.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
target_sources(pica PRIVATE ${SOURCES})
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "ostream.h"
|
#include "ostream.h"
|
||||||
|
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#include "stream.h"
|
#include "stream.h"
|
||||||
|
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
// SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
// SPDX-License-Identifier: GPL-3.0-or-later
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
22
taskmanager/CMakeLists.txt
Normal file
22
taskmanager/CMakeLists.txt
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
#SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
#SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
set(HEADERS
|
||||||
|
manager.h
|
||||||
|
job.h
|
||||||
|
route.h
|
||||||
|
scheduler.h
|
||||||
|
function.h
|
||||||
|
record.h
|
||||||
|
)
|
||||||
|
|
||||||
|
set(SOURCES
|
||||||
|
manager.cpp
|
||||||
|
job.cpp
|
||||||
|
route.cpp
|
||||||
|
scheduler.cpp
|
||||||
|
function.cpp
|
||||||
|
record.cpp
|
||||||
|
)
|
||||||
|
|
||||||
|
target_sources(${PROJECT_NAME} PRIVATE ${SOURCES})
|
12
taskmanager/function.cpp
Normal file
12
taskmanager/function.cpp
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "function.h"
|
||||||
|
|
||||||
|
TM::Function::Function (const std::function<void()>& fn):
|
||||||
|
fn(fn)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void TM::Function::execute () {
|
||||||
|
fn();
|
||||||
|
}
|
20
taskmanager/function.h
Normal file
20
taskmanager/function.h
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "functional"
|
||||||
|
|
||||||
|
#include "job.h"
|
||||||
|
|
||||||
|
namespace TM {
|
||||||
|
class Function : public Job {
|
||||||
|
public:
|
||||||
|
Function(const std::function<void()>& fn);
|
||||||
|
|
||||||
|
void execute () override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::function<void()> fn;
|
||||||
|
};
|
||||||
|
}
|
9
taskmanager/job.cpp
Normal file
9
taskmanager/job.cpp
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "job.h"
|
||||||
|
|
||||||
|
TM::Job::Job ()
|
||||||
|
{}
|
||||||
|
|
||||||
|
TM::Job::~Job () {}
|
19
taskmanager/job.h
Normal file
19
taskmanager/job.h
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
namespace TM {
|
||||||
|
class Job {
|
||||||
|
public:
|
||||||
|
Job();
|
||||||
|
Job(const Job& other) = delete;
|
||||||
|
Job(Job&& other) = delete;
|
||||||
|
virtual ~Job();
|
||||||
|
|
||||||
|
Job& operator = (const Job& other) = delete;
|
||||||
|
Job& operator = (Job&& other) = delete;
|
||||||
|
|
||||||
|
virtual void execute() = 0;
|
||||||
|
};
|
||||||
|
}
|
69
taskmanager/manager.cpp
Normal file
69
taskmanager/manager.cpp
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "manager.h"
|
||||||
|
|
||||||
|
TM::Manager::Manager ():
|
||||||
|
terminating(false),
|
||||||
|
threads(),
|
||||||
|
queue(),
|
||||||
|
mtx(),
|
||||||
|
cond()
|
||||||
|
{}
|
||||||
|
|
||||||
|
TM::Manager::~Manager () {
|
||||||
|
std::unique_lock lock(mtx);
|
||||||
|
if (threads.empty())
|
||||||
|
return;
|
||||||
|
|
||||||
|
lock.unlock();
|
||||||
|
stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
void TM::Manager::start () {
|
||||||
|
std::lock_guard lock(mtx);
|
||||||
|
|
||||||
|
std::size_t amount = std::thread::hardware_concurrency();
|
||||||
|
for (std::size_t i = 0; i < amount; ++i)
|
||||||
|
threads.emplace_back(std::thread(&Manager::loop, this));
|
||||||
|
}
|
||||||
|
|
||||||
|
void TM::Manager::stop () {
|
||||||
|
std::unique_lock lock(mtx);
|
||||||
|
|
||||||
|
terminating = true;
|
||||||
|
|
||||||
|
lock.unlock();
|
||||||
|
cond.notify_all();
|
||||||
|
for (std::thread& thread : threads)
|
||||||
|
thread.join();
|
||||||
|
|
||||||
|
lock.lock();
|
||||||
|
threads.clear();
|
||||||
|
terminating = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
void TM::Manager::loop () {
|
||||||
|
while (true) {
|
||||||
|
std::unique_lock lock(mtx);
|
||||||
|
while (!terminating && queue.empty())
|
||||||
|
cond.wait(lock);
|
||||||
|
|
||||||
|
if (terminating)
|
||||||
|
return;
|
||||||
|
|
||||||
|
std::unique_ptr<Job> job = std::move(queue.front());
|
||||||
|
queue.pop();
|
||||||
|
lock.unlock();
|
||||||
|
|
||||||
|
job->execute();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void TM::Manager::schedule (std::unique_ptr<Job> job) {
|
||||||
|
std::unique_lock lock(mtx);
|
||||||
|
queue.emplace(std::move(job));
|
||||||
|
|
||||||
|
lock.unlock();
|
||||||
|
cond.notify_one();
|
||||||
|
}
|
40
taskmanager/manager.h
Normal file
40
taskmanager/manager.h
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <queue>
|
||||||
|
#include <vector>
|
||||||
|
#include <thread>
|
||||||
|
#include <memory>
|
||||||
|
#include <mutex>
|
||||||
|
#include <condition_variable>
|
||||||
|
|
||||||
|
#include "job.h"
|
||||||
|
|
||||||
|
namespace TM {
|
||||||
|
class Manager {
|
||||||
|
public:
|
||||||
|
Manager();
|
||||||
|
Manager(const Manager&) = delete;
|
||||||
|
Manager(Manager&&) = delete;
|
||||||
|
~Manager();
|
||||||
|
|
||||||
|
Manager& operator = (const Manager&) = delete;
|
||||||
|
Manager& operator = (Manager&&) = delete;
|
||||||
|
|
||||||
|
void start();
|
||||||
|
void stop();
|
||||||
|
void schedule(std::unique_ptr<Job> job);
|
||||||
|
|
||||||
|
private:
|
||||||
|
void loop();
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool terminating;
|
||||||
|
std::vector<std::thread> threads;
|
||||||
|
std::queue<std::unique_ptr<Job>> queue;
|
||||||
|
std::mutex mtx;
|
||||||
|
std::condition_variable cond;
|
||||||
|
};
|
||||||
|
}
|
18
taskmanager/record.cpp
Normal file
18
taskmanager/record.cpp
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "record.h"
|
||||||
|
|
||||||
|
TM::Record::Record (ID id, const Task& task, Time time):
|
||||||
|
id(id),
|
||||||
|
task(task),
|
||||||
|
time(time)
|
||||||
|
{}
|
||||||
|
|
||||||
|
bool TM::Record::operator < (const Record& other) const {
|
||||||
|
return time < other.time;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool TM::Record::operator > (const Record& other) const {
|
||||||
|
return time > other.time;
|
||||||
|
}
|
26
taskmanager/record.h
Normal file
26
taskmanager/record.h
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <chrono>
|
||||||
|
#include <functional>
|
||||||
|
|
||||||
|
namespace TM {
|
||||||
|
class Record {
|
||||||
|
public:
|
||||||
|
using Time = std::chrono::time_point<std::chrono::steady_clock>;
|
||||||
|
using Task = std::function<void()>;
|
||||||
|
using ID = uint64_t;
|
||||||
|
|
||||||
|
Record(ID id, const Task& task, Time time);
|
||||||
|
|
||||||
|
ID id;
|
||||||
|
Task task;
|
||||||
|
Time time;
|
||||||
|
|
||||||
|
bool operator > (const Record& other) const;
|
||||||
|
bool operator < (const Record& other) const;
|
||||||
|
};
|
||||||
|
}
|
13
taskmanager/route.cpp
Normal file
13
taskmanager/route.cpp
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#include "route.h"
|
||||||
|
|
||||||
|
TM::Route::Route (std::shared_ptr<Router> router, std::unique_ptr<Request> request):
|
||||||
|
router(router),
|
||||||
|
request(std::move(request))
|
||||||
|
{}
|
||||||
|
|
||||||
|
void TM::Route::execute () {
|
||||||
|
router->route(std::move(request));
|
||||||
|
}
|
23
taskmanager/route.h
Normal file
23
taskmanager/route.h
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
//SPDX-FileCopyrightText: 2023 Yury Gubich <blue@macaw.me>
|
||||||
|
//SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
|
||||||
|
#include "job.h"
|
||||||
|
#include "server/router.h"
|
||||||
|
#include "request/request.h"
|
||||||
|
|
||||||
|
namespace TM {
|
||||||
|
class Route : public Job {
|
||||||
|
public:
|
||||||
|
Route(std::shared_ptr<Router> router, std::unique_ptr<Request> request);
|
||||||
|
|
||||||
|
void execute () override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::shared_ptr<Router> router;
|
||||||
|
std::unique_ptr<Request> request;
|
||||||
|
};
|
||||||
|
}
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user