tools/artifact_gen/render.cc

// Copyright 2025 gRPC authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "render.h"

#include <thread>

#include "absl/flags/flag.h"
#include "absl/log/check.h"
#include "absl/log/log.h"
#include "absl/strings/match.h"
#include "inja/inja.hpp"
#include "nlohmann/json.hpp"
#include "utils.h"

ABSL_FLAG(std::string, templates_dir, "", "Directory containing templates");
ABSL_FLAG(std::string, output_dir, "", "Directory to write rendered templates");

namespace {
void RenderTemplate(const std::string& filename, nlohmann::json build_yaml) {
  inja::Environment env;
  std::string rendered = env.render(
      LoadString(absl::GetFlag(FLAGS_templates_dir) + "/" + filename),
      build_yaml);
  std::ofstream ofs(absl::GetFlag(FLAGS_output_dir) + "/" +
                    filename.substr(0, filename.size() - 5));
  ofs << rendered;
}
}  // namespace

void RenderAllTemplates(nlohmann::json build_yaml) {
  CHECK(!absl::GetFlag(FLAGS_templates_dir).empty());
  CHECK(!absl::GetFlag(FLAGS_output_dir).empty());
  std::vector<std::thread> threads;
  for (const auto& filename :
       AllFilesInDir(absl::GetFlag(FLAGS_templates_dir))) {
    if (!absl::EndsWith(filename, ".inja")) {
      continue;
    }
    threads.emplace_back(
        [filename, build_yaml]() { RenderTemplate(filename, build_yaml); });
  }
  for (auto& thread : threads) {
    thread.join();
  }
}