io.fabric8.kubernetes.schema.generator.GeneratorUtils Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2015 Red Hat, Inc.
 *
 * 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.
 */
// Ported from https://github.com/manusa/yakc/blob/9272d649bfe05cd536d417fec64dcf679877bd14/buildSrc/src/main/java/com/marcnuri/yakc/schema/GeneratorUtils.java
package io.fabric8.kubernetes.schema.generator;

import io.fabric8.kubernetes.schema.generator.schema.SchemaFlattener;
import io.fabric8.kubernetes.schema.generator.schema.SchemaUtils;
import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.parser.OpenAPIV3Parser;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardOpenOption;
import java.util.Map;
import java.util.Objects;
import java.util.function.Predicate;

import static io.fabric8.kubernetes.schema.generator.schema.CrdParser.crdToOpenApi;

public class GeneratorUtils {

  private final GeneratorSettings settings;

  public GeneratorUtils(GeneratorSettings settings) {
    this.settings = settings;
  }

  public static Predicate> filter(GeneratorSettings gs) {
    return e -> {
      for (String regex : gs.getSkipGenerationRegexes()) {
        if (e.getKey().matches(regex)) {
          return false;
        }
      }
      if (gs.getIncludeGenerationRegexes().isEmpty()) {
        return true;
      }
      for (String regex : gs.getIncludeGenerationRegexes()) {
        if (e.getKey().matches(regex)) {
          return true;
        }
      }
      return false;
    };
  }

  public String readTemplate(String name) {
    try (InputStream stream = Objects
        .requireNonNull(SchemaUtils.class.getResourceAsStream("/templates/" + name + ".mustache"))) {
      return IOUtils.toString(stream, StandardCharsets.UTF_8);
    } catch (IOException ex) {
      settings.getLogger().severe(ex.getMessage());
      throw new GeneratorException("Can't load template " + name);
    }
  }

  public final void writeFile(Path file, String fileContents) {
    try {
      Files.write(file, fileContents.getBytes(StandardCharsets.UTF_8),
          StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
    } catch (IOException ex) {
      settings.getLogger().severe(ex.getMessage());
      throw new GeneratorException("Can't write java generated class " + file.toString());
    }
  }

  public static void cleanSourceDirectory(File sourceDirectory) {
    try {
      FileUtils.deleteDirectory(sourceDirectory);
    } catch (IOException ex) {
      throw new GeneratorException("Can't clean existent generated sources");
    }
  }

  public static OpenAPI parse(File schema) {
    if (schema == null || !schema.exists()) {
      throw new IllegalArgumentException("Schema file not found: " + schema);
    }
    return parse(schema.toURI());
  }

  public static OpenAPI parse(URI uri) {
    Objects.requireNonNull(uri, "A valid URL is required for OpenAPI parsing");
    final OpenAPI openApi;
    if (uri.getPath().toLowerCase().matches(".+\\.ya?ml$")) {
      try {
        openApi = crdToOpenApi(uri);
      } catch (IOException e) {
        throw new IllegalArgumentException("Error parsing CRD file: " + uri, e);
      }
    } else {
      openApi = new OpenAPIV3Parser().read(uri.toString());
    }
    SchemaFlattener.flatten(openApi);
    return openApi;
  }
}