com.composum.sling.core.ResourceModel Maven / Gradle / Ivy

package com.composum.sling.core;

import com.composum.sling.core.util.PropertyUtil;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ValueMap;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.HashMap;

public class ResourceModel extends AbstractSlingBean {

    private transient ValueMap propertyMap;
    private transient ValueMap values;

    /**
     * initialize bean using the context an the resource given explicitly
     */
    public ResourceModel(BeanContext context, Resource resource) {
        super(context, resource);
    }

    /**
     * initialize bean using the context with the 'resource' attribute within
     */
    public ResourceModel(BeanContext context) {
        super(context);
    }

    /**
     * if this constructor is used, the bean must be initialized using the 'initialize' method!
     */
    public ResourceModel() {
    }

    public  T getProperty(String name, T defaultValue) {
        Class type = PropertyUtil.getType(defaultValue);
        T value = getProperty(name, type);
        return value != null ? value : defaultValue;
    }

    public  T getProperty(String name, Class type) {
        return getValues().get(name, type);
    }

    public ValueMap getValues() {
        if (values == null) {
            values = resource.adaptTo(ValueMap.class);
        }
        return values;
    }

    // generic property access via generic Map for direct use in templates

    public abstract class GenericMap extends HashMap implements ValueMap {

        public static final String UNDEFINED = "";

        @Override
        @Nullable
        public Object get(@Nonnull final Object key) {
            return get((String) key, Object.class);
        }

        @SuppressWarnings("unchecked")
        @Override
        public  T get(@Nonnull String name, @Nonnull Class type) {
            if (name.startsWith("_jcr_")) {
                name = "jcr:" + name.substring(5);
            }
            Object value = super.get(name);
            if (value == null) {
                value = getValue(name, type);
                super.put(name, value != null ? value : UNDEFINED);
            }
            return value != UNDEFINED ? (T) value : null;
        }

        @SuppressWarnings("unchecked")
        @Override
        @Nonnull
        public  Type get(@Nonnull String name, @Nonnull Type defaultValue) {
            Type value = (Type) get(name, defaultValue.getClass());
            return value != null ? value : defaultValue;
        }

        protected abstract  T getValue(String key, @Nonnull Class type);
    }

    public class GenericProperty extends GenericMap {

        @Override
        @Nullable
        public  T getValue(String key, @Nonnull Class type) {
            return getProperty(key, type);
        }
    }

    @Nonnull
    public ValueMap getProperty() {
        if (propertyMap == null) {
            propertyMap = new GenericProperty();
        }
        return propertyMap;
    }
}