/* * Copyright (C) 2024 Brett Terpstra * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #include #include #include #include "blt/std/time.h" #include namespace db { void guild_fs_manager::message_create(blt::u64 channel_id, blt::u64 msg_id, std::string_view content, std::string_view username, std::string_view display_name, const std::vector& attachments) { } void guild_fs_manager::message_delete(blt::u64 channel_id, blt::u64 msg_id) { } void guild_fs_manager::message_update(blt::u64 channel_id, blt::u64 msg_id, std::string_view new_content) { } void guild_fs_manager::flush(std::string new_path) { complete = 3; check_for_guild_updates(); flusher.flush_channels(channels); complete--; path = std::move(new_path); } void guild_fs_manager::check_for_channel_updates() { bot.guild_get(dpp::snowflake(guildID), [this](const dpp::confirmation_callback_t& con) { auto guild = con.get(); f_server_name.seekg(0); }); } void guild_fs_manager::check_for_guild_updates() { } void guild_fs_manager::await_completions() { BLT_DEBUG("Awaiting all requests to complete"); while (complete > 0) {} complete = 0; BLT_DEBUG("Awaiting completed"); } void fs_manager::flush() { BLT_TIME_FUNC(now); if (current_day != now.tm_yday) { is_flushing = true; std::this_thread::sleep_for(std::chrono::milliseconds(1000)); current_day = now.tm_yday; for (auto& gm : guild_handlers) { gm.second->flush(create_path(gm.first)); gm.second->await_completions(); } is_flushing = false; } } fs_manager::fs_manager(std::string_view archive_path, dpp::cluster& bot): bot(bot), archive_path(archive_path) { BLT_INFO("Creating with path '%s'", this->archive_path.c_str()); BLT_TIME_FUNC(now); current_day = now.tm_yday; blt::string::replaceAll(this->archive_path, "\\", "/"); if (!blt::string::ends_with(archive_path, '/')) this->archive_path += "/"; BLT_INFO("Modified path '%s'", this->archive_path.c_str()); } std::string fs_manager::create_path(blt::u64 guildID) { BLT_TIME_FUNC(now); auto local_guild_path = archive_path + "raw/" + std::to_string(guildID) + "/"; auto local_day_path = local_guild_path + std::to_string(now.tm_year + 1900) + "-" + std::to_string(now.tm_mon + 1) + "-" + std::to_string(now.tm_mday) + "/"; std::filesystem::create_directories(local_day_path); return local_day_path; } void channel_flusher::flush_channels(HASHMAP>& map) { channels_to_flush.reserve(map.size()); for (auto& v : map) channels_to_flush.push_back(std::move(v.second)); map.clear(); } }