org.umlg.java.metamodel.annotation.OJAnnotatedOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-metamodel Show documentation
There is a newer version: 2.0.15
package org.umlg.java.metamodel.annotation;

import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

import org.umlg.java.metamodel.OJInterface;
import org.umlg.java.metamodel.OJOperation;
import org.umlg.java.metamodel.OJParameter;
import org.umlg.java.metamodel.OJPathName;
import org.umlg.java.metamodel.utilities.JavaStringHelpers;
import org.umlg.java.metamodel.utilities.JavaUtil;

public class OJAnnotatedOperation extends OJOperation implements OJAnnotatedElement {
    Map f_annotations = new TreeMap<>();
    private OJAnnotatedField resultVariable;
    private boolean interfaceDefault;
    private boolean isSynchronized = false;

    public OJAnnotatedOperation(String string, String returnPathName) {
        this(string);
        setReturnType(new OJPathName(returnPathName));
    }


    public OJAnnotatedOperation(String string, OJPathName returnPathName) {
        this(string);
        setReturnType(returnPathName);
    }

    public OJAnnotatedOperation(String string) {
        super();
        setName(string);
    }

    public boolean isInterfaceDefault() {
        return interfaceDefault;
    }

    public void setInterfaceDefault(boolean interfaceDefault) {
        this.interfaceDefault = interfaceDefault;
    }

    @Override
    public OJAnnotatedOperation getCopy() {
        OJAnnotatedOperation oper = new OJAnnotatedOperation(getName());
        super.copyValues(oper);
        oper.removeAllFromParameters();
        for (OJParameter f : getParameters()) {
            oper.addToParameters((OJParameter) f.getCopy());
        }
        for (OJAnnotationValue v : getAnnotations()) {
            oper.addAnnotationIfNew(v.getCopy());
        }
        // TODO copy exception
        return oper;
    }

    public boolean addAnnotationIfNew(OJAnnotationValue value) {
        if (f_annotations.containsKey(value.getType())) {
            return false;
        } else {
            putAnnotation(value);
            return true;
        }
    }

    public OJAnnotationValue removeAnnotation(OJPathName type) {
        return f_annotations.remove(type);
    }

    public Collection getAnnotations() {
        return f_annotations.values();
    }

    public OJAnnotationValue putAnnotation(OJAnnotationValue value) {
        return f_annotations.put(value.getType(), value);
    }

    @Override
    public String toJavaString() {
        StringBuilder result = new StringBuilder();
        if (!getComment().equals("")) {
            addJavaDocComment(result);
        }
        if (this.getNeedsSuppress()) {
            result.append("@SuppressWarnings(\"unchecked\")\n");
        }
        if (getAnnotations().size() > 0) {
            result.append(JavaStringHelpers.indent(JavaUtil.collectionToJavaString(getAnnotations(), "\n"), 0));
            result.append("\n");
        }
        if (this.isSynchronized()) {
            result.append("synchronized ");
        }
        // signature
        if (this.isAbstract()) {
            result.append("abstract ");
        }
        if (this.isInterfaceDefault()) {
            result.append("default ");
        }
        result.append(visToJava(this) + " ");
        if (this.getGenericTypeParam() != null) {
            result.append("<" + this.getGenericTypeParam().getLast() + "> ");
        }
        result.append(getReturnType().getCollectionTypeName());
        result.append(" " + getName());
        // params
        result.append("(" + paramsToJava(this) + ")");
        if (!getThrows().isEmpty()) {
            result.append(" throws " + exceptionsToJava(this));
        }
        if ((getOwner() instanceof OJAnnotatedInterface && !isStatic() && !isInterfaceDefault())
                ||getOwner() instanceof OJInterface
                || this.isAbstract()) {
            result.append(";\n");
        } else {
            result.append(" {\n");
            StringBuilder bodyStr = new StringBuilder();
            if (resultVariable != null) {
                if (resultVariable.getType() == null) {
                    resultVariable.setType(getReturnType());
                }
                bodyStr.append(resultVariable.toJavaString());
                bodyStr.append("\n");
            }
            bodyStr.append(getBody().toJavaString());
            if (resultVariable != null) {
                bodyStr.append("\nreturn result;");
            }
            result.append(JavaStringHelpers.indent(bodyStr, 1));
            if (result.charAt(result.length() - 1) == '\n') {
                result.deleteCharAt(result.length() - 1);
            }
            // closing bracket
            result.append("\n}\n");
        }
        return result.toString();
    }

    public void renameAll(Set renamePathNames, String suffix) {
        super.renameAll(renamePathNames, suffix);
        for (OJAnnotationValue annotation : getAnnotations()) {
            annotation.renameAll(renamePathNames, suffix);
        }
        if (resultVariable != null) {
            resultVariable.renameAll(renamePathNames, suffix);
        }
    }

    public OJOperation getDeepCopy() {
        OJAnnotatedOperation result = new OJAnnotatedOperation(getName());
        copyValuesDeep(result);
        return result;
    }

    public OJAnnotationValue findAnnotation(OJPathName ojPathName) {
        return AnnotationHelper.getAnnotation(this, ojPathName);
    }

    public void initializeResultVariable(String initialValue) {
        if (initialValue == null) {
            resultVariable = null;
        } else {
            resultVariable = new OJAnnotatedField("result", getReturnType());
            resultVariable.setInitExp(initialValue);
        }
    }

    public OJAnnotatedField getResultVariable() {
        return this.resultVariable;
    }

    public OJParameter findParameter(String fieldname) {
        for (OJParameter p : getParameters()) {
            if (p.getName().equals(fieldname)) {
                return p;
            }
        }
        return null;
    }

    public void setSynchronized() {
        this.isSynchronized = true;
    }

    public boolean isSynchronized() {
        return this.isSynchronized;
    }
}