110 lines
3.6 KiB
C++
110 lines
3.6 KiB
C++
#include <algorithm>
|
|
#include <dlfcn.h>
|
|
#include <iostream>
|
|
#include "functions.hpp"
|
|
#include "filesystem/filesystem.hpp"
|
|
|
|
std::vector< RenameLibrary > getLibraries(const std::vector<std::pair<std::string,std::string>> &libraries) {
|
|
// TODO get from config file
|
|
std::vector< RenameLibrary > result{};
|
|
for ( auto &library : libraries ) {
|
|
void *libhndl = dlopen( library.first.c_str(), RTLD_NOW );
|
|
if ( !libhndl ) {
|
|
std::cerr << "Could not load library " << std::get<0>(library) << std::endl;
|
|
closeLibraries( result );
|
|
return {};
|
|
}
|
|
RenameLibrary rl;
|
|
rl.libhndl = libhndl;
|
|
rl.init = ( bool ( * )(const std::string &) )dlsym( libhndl, "init" );
|
|
if ( !rl.init ) {
|
|
result.push_back( rl );
|
|
goto dlsymerror;
|
|
}
|
|
rl.getOptions = ( std::vector< RenameObject >( * )(
|
|
const RenameObject & ) )dlsym( libhndl, "getOptions" );
|
|
if ( !rl.getOptions ) {
|
|
result.push_back( rl );
|
|
goto dlsymerror;
|
|
}
|
|
rl.renamePath =
|
|
( bool ( * )( const std::string &, const RenameObject & ) )dlsym(
|
|
libhndl, "renamePath" );
|
|
if ( !rl.renamePath ) {
|
|
result.push_back( rl );
|
|
goto dlsymerror;
|
|
}
|
|
rl.getCustomKeys = ( std::vector< std::string >( * )() )dlsym(
|
|
libhndl, "getCustomKeys" );
|
|
if ( !rl.getCustomKeys ) {
|
|
result.push_back( rl );
|
|
goto dlsymerror;
|
|
}
|
|
rl.getName = ( const std::string(*)() ) dlsym( libhndl, "getName" );
|
|
rl.config = library.second;
|
|
result.push_back( rl );
|
|
}
|
|
return result;
|
|
|
|
dlsymerror:
|
|
std::cerr << "dlsym: " << dlerror() << std::endl;
|
|
closeLibraries( result );
|
|
return {};
|
|
}
|
|
|
|
void closeLibraries( std::vector< RenameLibrary > &libraries ) {
|
|
for ( auto &library : libraries ) {
|
|
dlclose( library.libhndl );
|
|
}
|
|
}
|
|
|
|
void addFilesRecursive(const FileObject &parent, std::vector< FileObject > &results, const std::string &filename, const std::string &containing_directory, bool dir_only = false ) {
|
|
auto path = containing_directory + FSLib::dir_divisor + filename;
|
|
if(!dir_only || FSLib::isDirectory(path)) {
|
|
if(!parent.getName().empty()) {
|
|
results.emplace_back(parent.getName() + FSLib::dir_divisor + filename);
|
|
} else {
|
|
results.emplace_back(filename);
|
|
}
|
|
results.back().setDepth(parent.getDepth() + 1);
|
|
}
|
|
if( FSLib::isDirectory(path) ) {
|
|
results.back().setFileType(TYPE_DIRECTORY);
|
|
for(const auto &entry : FSLib::Directory(path)) {
|
|
addFilesRecursive(results.back(), results, entry, path, dir_only);
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector< FileObject > getFilesInSource( const std::string &source_dir ) {
|
|
std::vector< FileObject > result;
|
|
for(const auto &entry : FSLib::Directory(source_dir)) {
|
|
addFilesRecursive(FileObject(""), result, entry, source_dir);
|
|
}
|
|
std::sort(result.begin(), result.end());
|
|
return result;
|
|
}
|
|
|
|
std::vector< FileObject > getTargetDirectories( const std::string &target_dir ) {
|
|
std::vector< FileObject > result;
|
|
for(const auto &entry : FSLib::Directory(target_dir)) {
|
|
addFilesRecursive(FileObject(""), result, entry, target_dir, true);
|
|
}
|
|
std::sort(result.begin(), result.end());
|
|
return result;
|
|
}
|
|
|
|
std::string safeJson(std::string input) {
|
|
for(size_t i = 0; i < input.size(); i++) {
|
|
if(input[i] == '\\') {
|
|
input.insert(i, "\\");
|
|
i++;
|
|
}
|
|
if(input[i] == '"') {
|
|
input.insert(i, "\\");
|
|
i++;
|
|
}
|
|
}
|
|
return input;
|
|
}
|