poomer-efsw/poomer-efsw.cpp
2025-04-14 22:22:12 -06:00

108 lines
3.6 KiB
C++

#include <iostream>
#include <thread>
#include <chrono>
#include <filesystem>
#include <string>
#include <atomic>
#include "../bella_engine_sdk/src/bella_sdk/bella_engine.h" // For rendering
#include "../bella_engine_sdk/src/dl_core/dl_main.inl" // Core functionality from the Diffuse Logic engine
#include "../oom/oom_license.h" // common misc code
#include "../oom/oom_file.h" // common misc code
//==============================================================================
// GLOBAL VARIABLES AND FUNCTIONS
//==============================================================================
// Global flag to indicate program termination
std::atomic<bool> STOP(false);
// Signal handler for clean shutdown
void sigend(int) {
std::cout << std::endl << "Bye bye" << std::endl;
STOP = true;
// Give a short time for cleanup
std::this_thread::sleep_for(std::chrono::milliseconds(100));
exit(0); // Force exit after cleanup
}
//==============================================================================
// MAIN FUNCTION
//==============================================================================
int DL_main(dl::Args& args) {
args.add("wd", "watchdir", "", "watch directory for changes");
args.add("tp", "thirdparty", "", "prints third party licenses");
args.add("li", "licenseinfo", "", "prints license info");
std::string watchPath = ".";
// If --help was requested, print help and exit
if (args.helpRequested()) {
std::cout << args.help("poomer-efsw © 2025 Harvey Fong","", "1.0") << std::endl;
return 0;
}
// Handle other command-line options
if (args.have("--licenseinfo")) {
std::cout << "poomer-efsw © 2025 Harvey Fong" << std::endl;
std::cout << oom::license::printLicense() << std::endl;
return 0;
}
if (args.have("--thirdparty")) {
std::cout << oom::license::printBellaSDK() << "\n====\n" << std::endl;
std::cout << oom::license::printEFSW() << "\n====\n" << std::endl;
return 0;
}
if (args.have("--watchdir")) {
watchPath = args.value("--watchdir").buf();
}
// Set up signal handler for clean shutdown
signal(SIGINT, sigend);
// Create the file watcher to catch when .bsz files or zip files are created
oom::file::Watcher oomWatcher({"bsz", "zip"}, {"download"});
// Start watching the current directory
if (oomWatcher.startWatching(watchPath)) {
std::cout << "File watcher started" << std::endl;
} else {
std::cout << "Failed to start file watcher" << std::endl;
return 1;
}
// Main processing loop
while(!STOP) {
// Check for files to process
std::filesystem::path filePath;
// Check if we have files to delete
while (oomWatcher.getNextFileToDelete(filePath)) {
std::cout << "Processing deletion: " << filePath.string() << std::endl;
// Handle file deletion
// Add your deletion logic here
}
// Check if we have files to render
while (oomWatcher.getNextFileToRender(filePath)) {
std::cout << "Processing render: " << filePath.string() << std::endl;
// Handle file rendering
// Add your rendering logic here
}
// Sleep to avoid busy waiting
std::this_thread::sleep_for(std::chrono::milliseconds(1000));
std::cout << "Running..." << std::endl;
}
// Clean shutdown
std::cout << "Shutting down file watcher..." << std::endl;
oomWatcher.stopWatching();
std::cout << "Shutdown complete" << std::endl;
return 0;
}