#ifndef CODE_GENERATOR_H
#define CODE_GENERATOR_H

#include "data_structures.h"
#include "json_parser.h"
#include <string>
#include <map>
#include <memory>

class CodeGenerator {
private:
    Config config;
    std::map<std::string, SetupFile> commonSetups;
    std::map<std::string, SetupFile> variantSetups;
    std::map<std::string, Variable> globalVariables;
    
    // Parse JSON helpers
    Instrument parseInstrument(std::shared_ptr<JsonValue> json);
    Property parseProperty(std::shared_ptr<JsonValue> json);
    Variable parseVariable(std::shared_ptr<JsonValue> json);
    Setup parseSetup(std::shared_ptr<JsonValue> json);
    SetupFile parseSetupFile(std::shared_ptr<JsonValue> json);
    Template parseTemplate(std::shared_ptr<JsonValue> json);
    TestExecution parseTestExecution(std::shared_ptr<JsonValue> json);
    
    // Code generation helpers
    std::string generateInstrumentDefinitions();
    std::string generateTestClass(const Template& tmpl, const TestExecution& testExec);
    std::string generateSetupMethod(const std::string& setupName, const std::string& instrumentName);
    std::string generateMainFile(const std::vector<std::string>& testNames);
    
    // Variable substitution
    std::string substituteVariables(const std::string& expr, const std::map<std::string, std::string>& localVars);
    std::string processVariableExpression(const std::string& expr);
    
    // Setup merging
    Setup mergeSetup(const std::string& setupName, const std::string& instrumentName);
    
public:
    CodeGenerator(const Config& cfg);
    
    bool loadConfigurations();
    bool generateCode();
};

#endif