directory_replace_placeholder.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : LGPL-3.0-only
****************************************/
 
 
#include <fstream>
#include <set>
 
#include "directory_replace_placeholder.h"


bool directory_replace_isBinaryAsset(const PPath & file){
        static const std::set<PString> setBinaryExtension{
                //images
                "png", "jpg", "jpeg", "gif", "bmp", "ico", "svg", "webp", "tiff", "tif",
                //fonts
                "woff", "woff2", "ttf", "otf", "eot",
                //archives
                "zip", "gz", "tar", "7z", "bz2", "xz",
                //documents
                "pdf",
                //audio/video
                "mp3", "mp4", "wav", "ogg", "avi", "mov",
                //compiled/binary artefacts
                "so", "dll", "dylib", "a", "o", "exe", "class", "pyc"
        };
        return setBinaryExtension.find(file.getExtension().toLower()) != setBinaryExtension.end();
}


bool directory_replace_copyBinaryFile(const PPath & inputFile, const PPath & outputFile){
        std::ifstream in(inputFile.c_str(), std::ios::binary);
        if(!in.is_open()){return false;}
        std::ofstream out(outputFile.c_str(), std::ios::binary);
        if(!out.is_open()){return false;}
        out << in.rdbuf();
        return static_cast<bool>(out);
}


bool directory_replace_placeholderFile(const PPath & outputDir, const PPath & inputConfigDir, const PMapString & mapReplacePlaceholder){
        bool b(true);
        PVecPath vecFileInputDir = inputConfigDir.getAllFileInDir();
        for(PVecPath::iterator it(vecFileInputDir.begin()); it != vecFileInputDir.end(); ++it){
                PPath relativeFile = it->replace(mapReplacePlaceholder);
                PPath fullFile = inputConfigDir / *it;
                PPath outputFile = outputDir / relativeFile;
                if(directory_replace_isBinaryAsset(fullFile)){
                        //Binary/image assets are copied verbatim, the file name/path may still use placeholders, but the content must not be touched
                        b &= directory_replace_copyBinaryFile(fullFile, outputFile);
                        continue;
                }
                PString fileContent = fullFile.loadFileContent();
                //TODO : this could be optimised with the dico_replace_var algorithm, but this is enough for now
                PString replacedContent = fileContent.replace(mapReplacePlaceholder);
                b &= outputFile.saveFileContent(replacedContent);
        }
        return b;
}


bool directory_replace_placeholder(const PPath & outputDir, const PPath & inputConfigDir, const PMapString & mapReplacePlaceholder){
        if(!outputDir.createDirectory()){
                std::cerr << "directory_replace_placeholder : cannot create dir '"<<outputDir<<"'" << std::endl;
                return false;
        }
        //Let's get content of inputConfigDir
        PVecPath vecDirInputDir = inputConfigDir.getAllDirectoryInDir();
        bool b(true);
        for(PVecPath::iterator itDir(vecDirInputDir.begin()); itDir != vecDirInputDir.end(); ++itDir){
                PPath relativeDir = itDir->replace(mapReplacePlaceholder);
                PPath fullInputDir = inputConfigDir / *itDir;
                PPath fullOutputDir = outputDir / relativeDir;
                b &= fullOutputDir.createDirectory();
                
                b &= directory_replace_placeholderFile(fullOutputDir, fullInputDir, mapReplacePlaceholder);
                b &= directory_replace_placeholder(fullOutputDir, fullInputDir, mapReplacePlaceholder);
        }
        b &= directory_replace_placeholderFile(outputDir, inputConfigDir, mapReplacePlaceholder);
        if(!b){
                std::cerr << "directory_replace_placeholder : cannot create output directories architecture" << std::endl;
                std::cerr << "\tinputConfigDir = '"<<inputConfigDir<<"'" << std::endl;
                std::cerr << "\toutputDir = '"<<outputDir<<"'" << std::endl;
                return false;
        }
        return b;
}