v1/src/main/java/com/xtremelabs/robolectric/res/ResourceReferenceResolver.java

package com.xtremelabs.robolectric.res;

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

class ResourceReferenceResolver<T> {
    private Map<String, T> attributeNamesToValues = new HashMap<String, T>();
    private Map<String, List<String>> unresolvedReferences = new HashMap<String, List<String>>();
    private String referenceType;
    private String[] allowedReferenceTypesForRawValues;

    ResourceReferenceResolver(String referenceType) {
        this(referenceType, new String[] { referenceType });
    }

    ResourceReferenceResolver(String referenceType, String[] allowedReferenceTypesForRawValues) {
        this.referenceType = referenceType;
        this.allowedReferenceTypesForRawValues = allowedReferenceTypesForRawValues;
    }

    public T getValue(String resourceName) {
        return attributeNamesToValues.get(resourceName);
    }

    public void processResource(String name, String rawValue, ResourceValueConverter loader, boolean isSystem) {
        String valuePointer = referenceType + "/" + name;

        boolean isAllowedReferenceType = false;
        for (String allowedReferenceTypesForRawValue : allowedReferenceTypesForRawValues) {
            if (rawValue.startsWith("@")
                    && rawValue.contains(allowedReferenceTypesForRawValue + "/")) {
                isAllowedReferenceType = true;
                break;
            }
        }

        if (isAllowedReferenceType) {
            addAttributeReference(rawValue, valuePointer);
        } else {
            if (isSystem) {
                valuePointer = "android:" + valuePointer;
            }
            addAttribute(valuePointer, (T) loader.convertRawValue(rawValue));
        }
    }

    public void addAttribute(String valuePointer, T value) {
        attributeNamesToValues.put(valuePointer, value);
        resolveUnresolvedReferences(valuePointer, value);
    }

    private void resolveUnresolvedReferences(String attributeName, T value) {
        List<String> references = unresolvedReferences.remove(attributeName);
        if (references == null) {
            return;
        }
        for (String reference : references) {
            attributeNamesToValues.put(reference, value);
        }
    }

    private void addUnresolvedReference(String valuePointer, String attributeName) {
        List<String> references = unresolvedReferences.get(attributeName);
        if (references == null) {
            references = new ArrayList<String>();
            unresolvedReferences.put(attributeName, references);
        }
        references.add(valuePointer);
    }

    private void addAttributeReference(String rawValue, String valuePointer) {
        String attributeName = rawValue.substring(1);
        T value = attributeNamesToValues.get(attributeName);
        if (value == null) {
            addUnresolvedReference(valuePointer, attributeName);
        } else {
            attributeNamesToValues.put(valuePointer, value);
        }
    }
}