PrepareSurfaceShader.hpp

Go to the documentation of this file.

#pragma once
 
#include <Strings/components/utils/io/io.h>
#include <Strings/components/utils/string/string.h>
#include <Strings/components/logger/logger.h>
#include <Strings/components/utils/list/list.h>
 
#include "../ToolCommand.hpp"
#include "../console.h"
 
#ifdef TOOL_LIB_EXPORTS
#define TOOL_LIB_API __declspec(dllexport) 
#else
#define TOOL_LIB_API __declspec(dllimport) 
#endif
 
class PrepareSurfaceShaderCommand : public ToolCommand
{
private:
    std::string placeholder_marker = "@SURFACE_SHADER@";
 
    bool combine_files(std::string template_path, std::string surface_path, std::string output_path)
    {
        if (!BlamStrings::Utils::IO::FileExists(template_path))
        {
            BlamStrings::Logger::LogEvent("failed to read shader template: '" + template_path + "'", BlamLogLevel::Warning);
            return false;
        }
 
        if (!BlamStrings::Utils::IO::FileExists(surface_path))
        {
            BlamStrings::Logger::LogEvent("failed to read surface shader: '" + surface_path + "'", BlamLogLevel::Warning);
            return false;
        }
 
        std::string template_text = BlamStrings::Utils::IO::GetFileContentsAsString(template_path);
        std::string surface_text = BlamStrings::Utils::IO::GetFileContentsAsString(surface_path);
 
        if (!BlamStrings::Utils::String::Contains(template_text, placeholder_marker))
        {
            BlamStrings::Logger::LogEvent("template file does not contain '" + placeholder_marker + "' marker", BlamLogLevel::Warning);
            return false;
        }
 
        std::string generation_header =
            "// =====================================================================\n"
            "// GENERATED FILE - DO NOT EDIT\n"
            "// Produced by 'tool combine-shaders' from a template and surface file.\n"
            "// Modify the template or surface source instead.\n"
            "//\n"
            "// Template: " + template_path + "\n"
            "// Surface:  " + surface_path + "\n"
            "// =====================================================================\n\n";
 
        std::string combined_text = generation_header + BlamStrings::Utils::String::Replace(template_text, placeholder_marker, surface_text);
 
        if (!BlamStrings::Utils::IO::CreateNewFile(output_path, combined_text))
        {
            BlamStrings::Logger::LogEvent("failed to write output file: '" + output_path + "'", BlamLogLevel::Warning);
            return false;
        }
 
        return true;
    }
 
public:
 
    PrepareSurfaceShaderCommand()
    {
        command = "prepare-surface-shader";
        syntax = "prepare-surface-shader <template_path> <surface_path> <output_path>";
        description = "combines a shader template with a surface shader";
    }
 
    int execute(std::vector<std::string> args)
    {
        if (args.size() < 3)
        {
            BlamStrings::Logger::LogEvent("combine-shaders requires three arguments: template, surface, output", BlamLogLevel::Warning);
            return -1;
        }
 
        std::string template_path = args[0];
        std::string surface_path = args[1];
        std::string output_path = args[2];
 
        if (!BlamStrings::Utils::IO::FileExists(template_path))
        {
            BlamStrings::Logger::LogEvent("template file '" + template_path + "' does not exist", BlamLogLevel::Warning);
            return -1;
        }
 
        if (!BlamStrings::Utils::IO::FileExists(surface_path))
        {
            BlamStrings::Logger::LogEvent("surface file '" + surface_path + "' does not exist", BlamLogLevel::Warning);
            return -1;
        }
 
        bool success = combine_files(template_path, surface_path, output_path);
 
        if (success)
        {
            BlamStrings::Logger::LogEvent("shader combination succeeded: '" + output_path + "'", TerminalColor::BrightGreen);
            return 0;
        }
 
        BlamStrings::Logger::LogEvent("shader combination failed", TerminalColor::BrightRed);
        return -1;
    }
};