io.github.jeddict.orm.generator.compiler.ConstructorSnippet.java Source code

Introduction

Here is the source code for io.github.jeddict.orm.generator.compiler.ConstructorSnippet.java

Source

/**
 * Copyright 2013-2018 the original author or authors from the Jeddict project (https://jeddict.github.io/).
 *
 * 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.github.jeddict.orm.generator.compiler;

import io.github.jeddict.jpa.spec.extend.AccessModifierType;
import io.github.jeddict.jpa.spec.extend.Constructor;
import io.github.jeddict.orm.generator.compiler.def.VariableDefSnippet;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.CLOSE_BRACES;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.CLOSE_PARANTHESES;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.COMMA;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.NEW_LINE;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.OPEN_BRACES;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.OPEN_PARANTHESES;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.SEMICOLON;
import static io.github.jeddict.orm.generator.util.ORMConverterUtil.SPACE;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import static org.apache.commons.lang.StringUtils.isNotBlank;

public class ConstructorSnippet implements Snippet {

    private final String className;
    private final Constructor constructor;
    private final List<VariableDefSnippet> parentVariableSnippets;
    private final List<VariableDefSnippet> localVariableSnippets;
    private final List<VariableDefSnippet> allVariableSnippets;

    public ConstructorSnippet(String className, Constructor constructor,
            List<VariableDefSnippet> parentVariableSnippets, List<VariableDefSnippet> localVariableSnippets) {
        this.className = className;
        this.constructor = constructor;
        this.parentVariableSnippets = parentVariableSnippets;
        this.localVariableSnippets = localVariableSnippets;
        this.allVariableSnippets = new ArrayList<>(parentVariableSnippets);
        this.allVariableSnippets.addAll(localVariableSnippets);
    }

    /**
     * @return the className
     */
    public String getClassName() {
        return className;
    }

    @Override
    public String getSnippet() throws InvalidDataException {
        StringBuilder builder = new StringBuilder();
        if (constructor.getAccessModifier() != AccessModifierType.DEFAULT) {
            builder.append(constructor.getAccessModifier().getValue()).append(SPACE);
        }

        builder.append(className).append(OPEN_PARANTHESES);
        if (!allVariableSnippets.isEmpty()) {
            for (VariableDefSnippet variableSnippet : allVariableSnippets) {
                builder.append(variableSnippet.getConstraintType()).append(SPACE).append(variableSnippet.getName())
                        .append(COMMA);
            }
            builder.setLength(builder.length() - 1);
        }
        builder.append(CLOSE_PARANTHESES);

        StringBuilder varAssign = new StringBuilder();
        if (!parentVariableSnippets.isEmpty()) {
            varAssign.append("super").append(OPEN_PARANTHESES);
            for (VariableDefSnippet parentVariableSnippet : parentVariableSnippets) {
                varAssign.append(parentVariableSnippet.getName()).append(COMMA).append(SPACE);
            }
            varAssign.setLength(varAssign.length() - 2);
            varAssign.append(CLOSE_PARANTHESES).append(SEMICOLON).append(NEW_LINE);
        }

        if (!localVariableSnippets.isEmpty()) {
            for (VariableDefSnippet localVariableSnippet : localVariableSnippets) {
                varAssign.append(String.format("this.%s=%s;", localVariableSnippet.getName(),
                        localVariableSnippet.getName())).append(NEW_LINE);
            }
        }

        builder.append(OPEN_BRACES).append(NEW_LINE);
        if (isNotBlank(constructor.getPreCode())) {
            builder.append(constructor.getPreCode()).append(NEW_LINE);
        }
        builder.append(varAssign);
        if (isNotBlank(constructor.getPostCode())) {
            builder.append(constructor.getPostCode()).append(NEW_LINE);
        }
        builder.append(CLOSE_BRACES);

        return builder.toString();
    }

    @Override
    public Collection<String> getImportSnippets() throws InvalidDataException {
        Collection<String> imports = new HashSet<>();
        for (VariableDefSnippet variableSnippet : allVariableSnippets) {
            imports.addAll(variableSnippet.getTypeImportSnippets());
        }
        return imports;
    }

}