io.helidon.build.maven.snakeyaml.codegen.CodeGenModel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snakeyaml-codegen-maven-plugin Show documentation

There is a newer version: 4.0.15

Show newest version

/*
 * Copyright (c) 2020, 2021 Oracle and/or its affiliates.
 *
 * 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.
 *
 */
package io.helidon.build.maven.snakeyaml.codegen;

import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Data model used from the code generation template.
 */
class CodeGenModel {

    private String generatedPackageName;
    private String generatedClassName;
    private Collection types;
    private Set imports;
    private String interfacePrefix;
    private String implementationPrefix;

    CodeGenModel(String generatedPackageName, String generatedClassName, Collection types,
            Set imports, String interfacePrefix, String implementationPrefix) {
        this.generatedClassName = generatedClassName;
        this.generatedPackageName = generatedPackageName;
        this.types = types;
        this.imports = imports;
        this.interfacePrefix = interfacePrefix;
        this.implementationPrefix = implementationPrefix;
    }

    String generatedPackageName() {
        return generatedPackageName;
    }

    String generatedClassName() {
        return generatedClassName;
    }

    Collection typesToAugment() {
        return types.stream().filter(t -> t.implementationType() != null).collect(Collectors.toList());
    }

    List javaImports() {
        return filteredImports("java.");
    }

    List implImports() {
        return filteredImports(implementationPrefix);
    }

    List openAPIImports() {
        return filteredImports(interfacePrefix);
    }

    private List filteredImports(String namePrefix) {
        return imports.stream().filter(i -> i.name().startsWith(namePrefix)).sorted().collect(Collectors.toList());
    }
}