net.binis.codegen.compiler.CGMethodDeclaration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of code-generator Show documentation

Binis Code Generation

There is a newer version: 1.2.22

package net.binis.codegen.compiler;

/*-
 * #%L
 * code-generator
 * %%
 * Copyright (C) 2021 - 2024 Binis Belev
 * %%
 * 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.
 * #L%
 */

import com.sun.source.util.Trees;
import lombok.extern.slf4j.Slf4j;

import javax.lang.model.element.Element;

import static java.util.Objects.isNull;
import static java.util.Objects.nonNull;
import static net.binis.codegen.tools.Reflection.*;
import static net.binis.codegen.tools.Tools.withRes;

@Slf4j
public class CGMethodDeclaration extends CGDeclaration {

    protected CGList params;

    public static Class theClass() {
        return loadClass("com.sun.tools.javac.tree.JCTree$JCMethodDecl");
    }

    public static CGMethodDeclaration create(Trees trees, Element element) {
        return new CGMethodDeclaration(trees, element);
    }

    public CGMethodDeclaration(Trees trees, Element element) {
        super(trees, element);
    }

    public CGMethodDeclaration(Object instance) {
        super(instance);
    }

    public CGList getParameters() {
        if (isNull(params)) {
            params = new CGList<>(getFieldValue(instance, "params"), this::onModify, CGVariableDecl.class);
        }
        return params;
    }

    @Override
    protected void init() {
        cls = CGMethodDeclaration.theClass();
    }

    private void onModify(CGList list) {
        setFieldValue(instance, "annotations", list.getInstance());
        params = list;
    }

    public boolean isConstructor() {
        return "".equals(getName().toString());
    }

    public CGBlock getBody() {
        var body = getFieldValue(instance, "body");
        return nonNull(body) ? new CGBlock(body) : null;
    }

    public void setBody(CGBlock body) {
        setFieldValue(instance, "body", nonNull(body) ? body.getInstance() : null);
    }

    public CGType getReturnType() {
        return withRes(getFieldValue(instance, "restype"), CGType::new);
    }

}