// This file is part of the 64k demo project.
// It defines the ShaderComposer class for managing WGSL snippets.

#pragma once

#include <map>
#include <set>
#include <string>
#include <vector>

class ShaderComposer {
 public:
  static ShaderComposer& Get();

  void RegisterSnippet(const std::string& name, const std::string& code);

  using CompositionMap = std::map<std::string, std::string>;

  // Resolves #include directives recursively
  // substitutions: map "placeholder_name" -> "actual_snippet_name"
  std::string Compose(const std::vector<std::string>& dependencies,
                      const std::string& main_code,
                      const CompositionMap& substitutions = {});

  void VerifyIncludes() const;

 private:
  ShaderComposer() = default;

  void ResolveRecursive(const std::string& source, std::stringstream& ss,
                        std::set<std::string>& included,
                        const CompositionMap& substitutions);

  std::map<std::string, std::string> snippets_;
};