de.hilling.lang.metamodel.ClassModel Maven / Gradle / Ivy

Go to download
package de.hilling.lang.metamodel;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.processing.ProcessingEnvironment;

/**
 * Properties model for class.
 */
public class ClassModel {
    private final ProcessingEnvironment env;
    private final Map attributes = new HashMap<>();
    private final List               names      = new ArrayList<>();

    ClassModel(ProcessingEnvironment env) {
        this.env = env;
    }

    AttributeInfo getInfo(String attributeName) {
        return attributes.computeIfAbsent(attributeName, this::addAttribute);
    }

    ProcessingEnvironment getEnvironment() {
        return env;
    }

    /**
     * @return map with all attributes for current class.
     */
    Map attributes() {
        return attributes;
    }

    /**
     * @return attribute names in declaration order.
     */
    List names() {
        return names;
    }

    private AttributeInfo addAttribute(String k) {
        names.add(k);
        return new AttributeInfo();
    }
}