src/resourceshrinker/java/com/android/build/shrinker/r8integration/R8ResourceShrinkerState.java

// Copyright (c) 2023, the R8 project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
package com.android.build.shrinker.r8integration;

import static com.android.build.shrinker.r8integration.LegacyResourceShrinker.getUtfReader;

import com.android.aapt.Resources;
import com.android.aapt.Resources.ConfigValue;
import com.android.aapt.Resources.Entry;
import com.android.aapt.Resources.FileReference;
import com.android.aapt.Resources.Item;
import com.android.aapt.Resources.Package;
import com.android.aapt.Resources.Primitive;
import com.android.aapt.Resources.ResourceTable;
import com.android.aapt.Resources.Value;
import com.android.aapt.Resources.XmlAttribute;
import com.android.aapt.Resources.XmlElement;
import com.android.aapt.Resources.XmlNode;
import com.android.build.shrinker.ResourceShrinkerImplKt;
import com.android.build.shrinker.ResourceShrinkerModel;
import com.android.build.shrinker.ResourceTableUtilKt;
import com.android.build.shrinker.ShrinkerDebugReporter;
import com.android.build.shrinker.graph.ProtoResourcesGraphBuilder;
import com.android.build.shrinker.r8integration.LegacyResourceShrinker.ShrinkerResult;
import com.android.build.shrinker.usages.ProtoAndroidManifestUsageRecorderKt;
import com.android.build.shrinker.usages.ToolsAttributeUsageRecorderKt;
import com.android.ide.common.resources.ResourcesUtil;
import com.android.ide.common.resources.usage.ResourceStore;
import com.android.ide.common.resources.usage.ResourceUsageModel;
import com.android.ide.common.resources.usage.ResourceUsageModel.Resource;
import com.android.resources.ResourceType;
import com.android.tools.r8.FeatureSplit;
import com.android.tools.r8.origin.Origin;
import com.android.tools.r8.origin.PathOrigin;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;
import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
import it.unimi.dsi.fastutil.ints.IntSet;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.IdentityHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collectors;

public class R8ResourceShrinkerState {

  private final Function<Exception, RuntimeException> errorHandler;
  private final R8ResourceShrinkerModel r8ResourceShrinkerModel;
  private final Map<String, Supplier<InputStream>> xmlFileProviders = new HashMap<>();
  private final Map<String, byte[]> changedXmlFiles = new HashMap<>();
  private final Set<String> duplicatedResFolderEntries = new HashSet<>();
  private final List<Supplier<InputStream>> keepRuleFileProviders = new ArrayList<>();

  private final List<Supplier<InputStream>> manifestProviders = new ArrayList<>();
  private final Map<String, Supplier<InputStream>> resfileProviders = new HashMap<>();
  private final Map<FeatureSplit, ResourceTable> resourceTables = new HashMap<>();
  private final ShrinkerDebugReporter shrinkerDebugReporter;
  private final boolean enableXmlInlining;
  private ClassReferenceCallback enqueuerCallback;
  private MethodReferenceCallback methodCallback;
  private Map<Integer, Set<String>> resourceIdToXmlFiles;
  private Set<String> packageNames;
  private final Set<String> seenNoneClassValues = new HashSet<>();
  private final Set<Integer> seenResourceIds = new HashSet<>();
  private final Map<Resource, String> reachabilityMap = new ConcurrentHashMap<>();

  private static final Set<String> SPECIAL_MANIFEST_ELEMENTS =
      ImmutableSet.of(
          "provider",
          "activity",
          "service",
          "receiver",
          "instrumentation",
          "process",
          "application");

  private static final Set<String> SPECIAL_APPLICATION_ATTRIBUTES =
      ImmutableSet.of("backupAgent", "appComponentFactory", "zygotePreloadName");

  @FunctionalInterface
  public interface ClassReferenceCallback {
    boolean tryClass(String possibleClass, Origin xmlFileOrigin);
  }

  @FunctionalInterface
  public interface MethodReferenceCallback {
    void tryMethod(String methodName, Origin xmlFileOrigin);
  }

  public R8ResourceShrinkerState(
      Function<Exception, RuntimeException> errorHandler,
      ShrinkerDebugReporter shrinkerDebugReporter,
      boolean enableXmlInlining) {
    r8ResourceShrinkerModel = new R8ResourceShrinkerModel(shrinkerDebugReporter, true);
    this.shrinkerDebugReporter = shrinkerDebugReporter;
    this.errorHandler = errorHandler;
    this.enableXmlInlining = enableXmlInlining;
  }

  public void trace(int id, String reachableFrom) {
    if (!seenResourceIds.add(id)) {
      return;
    }
    Resource resource = r8ResourceShrinkerModel.getResourceStore().getResource(id);
    if (resource == null) {
      return;
    }
    assert reachableFrom != null;

    // For deterministic output, sort the strings lexicographically.
    reachabilityMap.compute(
        resource, (r, v) -> v == null || v.compareTo(reachableFrom) > 0 ? reachableFrom : v);
    ResourceUsageModel.markReachable(resource);
    traceXmlForResourceId(id);
    if (resource.references != null) {
      for (Resource reference : resource.references) {
        if (!reference.isReachable()) {
          trace(reference.value, resource.toString());
        }
      }
    }
  }

  public boolean hasResourceId(int resourceId) {
    return getR8ResourceShrinkerModel().getResourceStore().getResource(resourceId) != null;
  }

  public void traceKeepXmlAndManifest() {
    // We start by building the root set of all keep/discard rules to find those pinned resources
    // before marking additional resources in the trace.
    // We then explicitly trace those resources to transitively get the full set of reachable
    // resources and code.
    try {
      updateModelWithKeepXmlReferences();
    } catch (IOException e) {
      throw errorHandler.apply(e);
    }
    // ProcessToolsAttribute returns the resources that becomes live
    r8ResourceShrinkerModel
        .getResourceStore()
        .processToolsAttributes()
        .forEach(resource -> trace(resource.value, "keep xml file"));
    for (Supplier<InputStream> manifestProvider : manifestProviders) {
      traceXml("AndroidManifest.xml", manifestProvider.get(), ids -> {});
    }
  }

  public void setEnqueuerCallback(ClassReferenceCallback enqueuerCallback) {
    assert this.enqueuerCallback == null;
    this.enqueuerCallback = enqueuerCallback;
  }

  public void setEnqueuerMethodCallback(MethodReferenceCallback methodCallback) {
    assert this.methodCallback == null;
    this.methodCallback = methodCallback;
  }

  private synchronized Set<String> getPackageNames() {
    // TODO(b/325888516): Consider only doing this for the package corresponding to the current
    // feature.
    if (packageNames == null) {
      packageNames = new HashSet<>();
      for (ResourceTable resourceTable : resourceTables.values()) {
        for (Package aPackage : resourceTable.getPackageList()) {
          packageNames.add(aPackage.getPackageName());
        }
      }
    }
    return packageNames;
  }

  public void addManifestProvider(Supplier<InputStream> manifestProvider) {
    this.manifestProviders.add(manifestProvider);
  }

  public void addXmlFileProvider(Supplier<InputStream> inputStreamSupplier, String location) {
    if (this.xmlFileProviders.containsKey(location)) {
      duplicatedResFolderEntries.add(location);
    } else {
      this.xmlFileProviders.put(location, inputStreamSupplier);
    }
  }

  public void addKeepRuleRileProvider(Supplier<InputStream> inputStreamSupplier) {
    this.keepRuleFileProviders.add(inputStreamSupplier);
  }

  public void addResFileProvider(Supplier<InputStream> inputStreamSupplier, String location) {
    if (this.resfileProviders.containsKey(location)) {
      duplicatedResFolderEntries.add(location);
    }
    this.resfileProviders.put(location, inputStreamSupplier);
  }

  public void addResourceTable(InputStream inputStream, FeatureSplit featureSplit) {
    this.resourceTables.put(
        featureSplit, r8ResourceShrinkerModel.instantiateFromResourceTable(inputStream, true));
  }

  public R8ResourceShrinkerModel getR8ResourceShrinkerModel() {
    return r8ResourceShrinkerModel;
  }

  private byte[] getXmlOrResFileBytes(String path) {
    assert !path.startsWith("res/");
    String pathWithRes = "res/" + path;

    if (xmlFileProviders.containsKey(pathWithRes)) {
      return getXmlBytes(pathWithRes);
    }
    Supplier<InputStream> inputStreamSupplier = resfileProviders.get(pathWithRes);
    if (inputStreamSupplier == null) {
      // Ill formed resource table with file references inside res/ that does not exist.
      return null;
    }
    try {
      return inputStreamSupplier.get().readAllBytes();
    } catch (IOException ex) {
      throw errorHandler.apply(ex);
    }
  }

  private byte[] getXmlBytes(String pathWithRes) {
    if (changedXmlFiles.containsKey(pathWithRes)) {
      return changedXmlFiles.get(pathWithRes);
    }
    Supplier<InputStream> inputStreamSupplier = xmlFileProviders.get(pathWithRes);
    if (inputStreamSupplier == null) {
      return null;
    }
    try {
      return inputStreamSupplier.get().readAllBytes();
    } catch (IOException ex) {
      throw errorHandler.apply(ex);
    }
  }

  public void setupReferences() {
    for (ResourceTable resourceTable : resourceTables.values()) {
      new ProtoResourcesGraphBuilder(this::getXmlOrResFileBytes, unused -> resourceTable)
          .buildGraph(r8ResourceShrinkerModel);
    }
  }

  public ShrinkerResult shrinkModel() throws IOException {
    updateModelWithManifestReferences();
    updateModelWithKeepXmlReferences();
    ResourceStore resourceStore = r8ResourceShrinkerModel.getResourceStore();
    resourceStore.processToolsAttributes();
    ImmutableSet<String> resEntriesToKeep = getResEntriesToKeep(resourceStore);
    List<Integer> resourceIdsToRemove = getResourcesToRemove();

    Map<FeatureSplit, ResourceTable> shrunkenTables = new IdentityHashMap<>();
    resourceTables.forEach(
        (featureSplit, resourceTable) -> {
          shrunkenTables.put(
              featureSplit,
              ResourceTableUtilKt.nullOutEntriesWithIds(resourceTable, resourceIdsToRemove, true));
        });
    for (Map.Entry<Resource, String> resourceStringEntry : reachabilityMap.entrySet()) {
      shrinkerDebugReporter.debug(
          () ->
              resourceStringEntry.getKey().toString()
                  + " reachable from "
                  + resourceStringEntry.getValue());
    }
    return new ShrinkerResult(resEntriesToKeep, shrunkenTables, changedXmlFiles);
  }

  private ImmutableSet<String> getResEntriesToKeep(ResourceStore resourceStore) {
    ImmutableSet.Builder<String> resEntriesToKeep = new ImmutableSet.Builder<>();
    for (String path : Iterables.concat(xmlFileProviders.keySet(), resfileProviders.keySet())) {
      if (ResourceShrinkerImplKt.isJarPathReachable(resourceStore, path)) {
        resEntriesToKeep.add(path);
      }
    }
    return resEntriesToKeep.build();
  }

  private void traceXmlForResourceId(int id) {
    Set<String> xmlFiles = getResourceIdToXmlFiles().get(id);
    if (xmlFiles != null) {
      for (String xmlFile : xmlFiles) {
        InputStream inputStream = xmlFileProviders.get(xmlFile).get();
        Resource resource = r8ResourceShrinkerModel.getResourceStore().getResource(id);
        traceXml(xmlFile, inputStream, inlinedIds -> pruneModel(resource, inlinedIds));
        if (duplicatedResFolderEntries.contains(xmlFile)) {
          traceDuplicatedXmlFileIds(id, xmlFile);
        }
      }
    }
  }

  private void pruneModel(Resource resource, IntSet inlinedIds) {
    List<Resource> references = resource.references;
    if (references != null) {
      references.removeIf(r -> inlinedIds.contains(r.value));
    }
  }

  private void traceDuplicatedXmlFileIds(int currentId, String xmlFile) {
    for (Map.Entry<Integer, Set<String>> entry : getResourceIdToXmlFiles().entrySet()) {
      if (entry.getValue().contains(xmlFile)) {
        if (entry.getKey() != currentId) {
          trace(entry.getKey(), "Duplicated xmlfile " + xmlFile);
        }
      }
    }
  }

  private void traceXml(
      String xmlFile, InputStream inputStream, Consumer<IntSet> inlinedIdsConsumer) {
    try {
      XmlNode xmlNode;
      if (changedXmlFiles.containsKey(xmlFile)) {
        xmlNode = XmlNode.parseFrom(changedXmlFiles.get(xmlFile));
      } else {
        xmlNode = XmlNode.parseFrom(inputStream);
        XmlNode.Builder xmlNodeBuilder = xmlNode.toBuilder();
        boolean changed = tryInlineValues(xmlNodeBuilder, inlinedIdsConsumer);
        if (changed) {
          xmlNode = xmlNodeBuilder.build();
          changedXmlFiles.put(xmlFile, xmlNode.toByteArray());
        }
      }
      visitNode(xmlNode, xmlFile, null);
      // Ensure that we trace the transitive reachable ids, without us having to iterate all
      // resources for the reachable marker.
      ProtoAndroidManifestUsageRecorderKt.recordUsagesFromNode(xmlNode, r8ResourceShrinkerModel)
          .iterator()
          .forEachRemaining(resource -> trace(resource.value, xmlFile));
    } catch (IOException e) {
      errorHandler.apply(e);
    }
  }

  private boolean tryInlineValues(
      XmlNode.Builder xmlNodeBuilder, Consumer<IntSet> inlinedIdsConsumer) {
    if (!enableXmlInlining) {
      return false;
    }
    XmlElement.Builder elementBuilder = xmlNodeBuilder.getElementBuilder();
    IntSet inlinedIds = null;
    for (XmlAttribute.Builder attributeBuilder : elementBuilder.getAttributeBuilderList()) {
      if (attributeBuilder.hasCompiledItem()) {
        Item compiledItem = attributeBuilder.getCompiledItem();
        if (compiledItem.hasRef()) {
          int id = compiledItem.getRef().getId();
          SingleValueResource res = getR8ResourceShrinkerModel().getSingleValueResourceOrNull(id);
          // Try to inline if single
          if (res != null) {
            Primitive prim = res.getPrimitive();
            if (prim != null) {
              attributeBuilder.getCompiledItemBuilder().clearRef().setPrim(prim);
            } else {
              attributeBuilder.setValue(res.getSingleValueLiteral());
              attributeBuilder.clearCompiledItem();
            }
            if (inlinedIds == null) {
              inlinedIds = new IntOpenHashSet();
            }
            inlinedIds.add(id);
          }
        }
      }
    }
    boolean changedChildren = false;
    for (XmlNode.Builder builder : elementBuilder.getChildBuilderList()) {
      changedChildren = tryInlineValues(builder, inlinedIdsConsumer) || changedChildren;
    }
    if (inlinedIds != null) {
      inlinedIdsConsumer.accept(inlinedIds);
      return true;
    }
    return changedChildren;
  }

  private void tryEnqueuerOnString(String possibleClass, String xmlName) {
    // There are a lot of xml tags and attributes that are evaluated over and over, if it is
    // not a class, ignore it.
    if (seenNoneClassValues.contains(possibleClass)) {
      return;
    }
    if (!enqueuerCallback.tryClass(possibleClass, new PathOrigin(Paths.get(xmlName)))) {
      seenNoneClassValues.add(possibleClass);
    }
  }

  private void visitNode(XmlNode xmlNode, String xmlName, String manifestPackageName) {
    XmlElement element = xmlNode.getElement();
    String xmlElementName = element.getName();
    tryEnqueuerOnString(xmlElementName, xmlName);

    for (XmlAttribute xmlAttribute : element.getAttributeList()) {
      if (xmlAttribute.getName().equals("package") && xmlElementName.equals("manifest")) {
        // We are traversing a manifest, record the package name if we see it.
        manifestPackageName = xmlAttribute.getValue();
      }
      String value = xmlAttribute.getValue();
      tryEnqueuerOnString(value, xmlName);
      if (value.startsWith(".")) {
        // package specific names, e.g. context
        getPackageNames().forEach(s -> tryEnqueuerOnString(s + value, xmlName));
      }
      if (manifestPackageName != null) {
        // Manifest case
        traceManifestSpecificValues(xmlName, manifestPackageName, xmlAttribute, element);
      }
      if (xmlAttribute.getName().equals("onClick")
          && xmlAttribute.getNamespaceUri().equals("http://schemas.android.com/apk/res/android")) {
        methodCallback.tryMethod(xmlAttribute.getValue(), new PathOrigin(Paths.get(xmlName)));
      }
    }
    for (XmlNode node : element.getChildList()) {
      visitNode(node, xmlName, manifestPackageName);
    }
  }

  private void traceManifestSpecificValues(
      String xmlName, String packageName, XmlAttribute xmlAttribute, XmlElement element) {
    if (!SPECIAL_MANIFEST_ELEMENTS.contains(element.getName())) {
      return;
    }
    // All elements can have package specific name attributes pointing at classes.
    if (xmlAttribute.getName().equals("name")) {
      tryEnqueuerOnString(getFullyQualifiedName(packageName, xmlAttribute), xmlName);
    }
    // Application elements have multiple special case attributes, where the value is potentially
    // a class name (unqualified).
    if (element.getName().equals("application")) {
      if (SPECIAL_APPLICATION_ATTRIBUTES.contains(xmlAttribute.getName())) {
        tryEnqueuerOnString(getFullyQualifiedName(packageName, xmlAttribute), xmlName);
      }
    }
  }

  private static String getFullyQualifiedName(String packageName, XmlAttribute xmlAttribute) {
    return packageName + "." + xmlAttribute.getValue();
  }

  public Map<Integer, Set<String>> getResourceIdToXmlFiles() {
    if (resourceIdToXmlFiles == null) {
      resourceIdToXmlFiles = new HashMap<>();
      for (ResourceTable resourceTable : resourceTables.values()) {
        for (Package packageEntry : resourceTable.getPackageList()) {
          for (Resources.Type type : packageEntry.getTypeList()) {
            for (Entry entry : type.getEntryList()) {
              for (ConfigValue configValue : entry.getConfigValueList()) {
                if (configValue.hasValue()) {
                  Value value = configValue.getValue();
                  if (value.hasItem()) {
                    Item item = value.getItem();
                    if (item.hasFile()) {
                      FileReference file = item.getFile();
                      if (file.getType() == FileReference.Type.PROTO_XML) {
                        int id = ResourceTableUtilKt.toIdentifier(packageEntry, type, entry);
                        resourceIdToXmlFiles
                            .computeIfAbsent(id, unused -> new HashSet<>())
                            .add(file.getPath());
                      }
                    }
                  }
                }
              }
            }
          }
        }
      }
    }
    return resourceIdToXmlFiles;
  }

  private List<Integer> getResourcesToRemove() {
    return r8ResourceShrinkerModel.getResourceStore().getResources().stream()
        .filter(r -> !r.isReachable() && !r.isPublic())
        .filter(r -> r.type != ResourceType.ID)
        .map(r -> r.value)
        .collect(Collectors.toList());
  }

  // Temporary to support updating the reachable entries from the manifest, we need to instead
  // trace these in the enqueuer.
  public void updateModelWithManifestReferences() throws IOException {
    for (Supplier<InputStream> manifestProvider : manifestProviders) {
      ProtoAndroidManifestUsageRecorderKt.recordUsagesFromNode(
          XmlNode.parseFrom(manifestProvider.get()), r8ResourceShrinkerModel);
    }
  }

  public void updateModelWithKeepXmlReferences() throws IOException {
    for (Supplier<InputStream> keepRuleFileProvider : keepRuleFileProviders) {
      ToolsAttributeUsageRecorderKt.processRawXml(
          getUtfReader(keepRuleFileProvider.get().readAllBytes()), r8ResourceShrinkerModel);
    }
  }

  public void enqueuerDone(boolean isFinalTreeshaking) {
    enqueuerCallback = null;
    methodCallback = null;
    seenResourceIds.clear();
    if (!isFinalTreeshaking) {
      // After final tree shaking we will need the reachability bits to decide what to write out
      // from the model.
      clearReachableBits();
    }
  }

  public void clearReachableBits() {
    for (Resource resource : r8ResourceShrinkerModel.getResourceStore().getResources()) {
      resource.setReachable(false);
    }
  }

  public interface SingleValueResource {
    String getSingleValueLiteral();

    Primitive getPrimitive();
  }

  public static class StringSingleValueResource implements SingleValueResource {
    private final String value;

    StringSingleValueResource(String value) {
      this.value = value;
    }

    @Override
    public String getSingleValueLiteral() {
      return value;
    }

    @Override
    public Primitive getPrimitive() {
      return null;
    }
  }

  public static class ColorSingleValueResource implements SingleValueResource {

    final Primitive colorPrim;

    private String toHexColor(int value, int truncate) {
      return "#" + Integer.toHexString(value).substring(truncate).toUpperCase();
    }

    ColorSingleValueResource(Primitive prim) {
      this.colorPrim = prim;
    }

    @Override
    public String getSingleValueLiteral() {
      int colorValue = getValue();
      if (colorPrim.hasColorRgb4Value()) {
        return toHexColor(colorValue, 4);
      } else if (colorPrim.hasColorRgb8Value()) {
        return toHexColor(colorValue, 2);
      } else if (colorPrim.hasColorArgb4Value()) {
        return toHexColor(colorValue, 3);
      } else if (colorPrim.hasColorArgb8Value()) {
        return toHexColor(colorValue, 0);
      } else {
        assert false;
        return null;
      }
    }

    public int getValue() {
      if (colorPrim.hasColorRgb4Value()) {
        return colorPrim.getColorRgb4Value();
      } else if (colorPrim.hasColorRgb8Value()) {
        return colorPrim.getColorRgb8Value();
      } else if (colorPrim.hasColorArgb4Value()) {
        return colorPrim.getColorArgb4Value();
      } else if (colorPrim.hasColorArgb8Value()) {
        return colorPrim.getColorArgb8Value();
      } else {
        assert false;
        return -1;
      }
    }

    @Override
    public Primitive getPrimitive() {
      return colorPrim;
    }
  }

  public static class R8ResourceShrinkerModel extends ResourceShrinkerModel {
    private final Map<Integer, SingleValueResource> resourcesWithSingleValue = new HashMap<>();

    public R8ResourceShrinkerModel(
        ShrinkerDebugReporter debugReporter, boolean supportMultipackages) {
      super(debugReporter, supportMultipackages);
    }

    public SingleValueResource getSingleValueResourceOrNull(int id) {
      return resourcesWithSingleValue.get(id);
    }

    // Similar to instantiation in ProtoResourceTableGatherer, but using an inputstream.
    ResourceTable instantiateFromResourceTable(InputStream inputStream, boolean includeStyleables) {
      try {
        ResourceTable resourceTable = ResourceTable.parseFrom(inputStream);
        instantiateFromResourceTable(resourceTable, includeStyleables);
        return resourceTable;
      } catch (IOException ex) {
        throw new RuntimeException(ex);
      }
    }

    void instantiateFromResourceTable(ResourceTable resourceTable, boolean includeStyleables) {
      ResourceTableUtilKt.entriesSequence(resourceTable)
          .iterator()
          .forEachRemaining(
              entryWrapper -> {
                ResourceType resourceType = ResourceType.fromClassName(entryWrapper.getType());
                Entry entry = entryWrapper.getEntry();
                int entryId = entryWrapper.getId();
                recordSingleValueResources(resourceType, entry, entryId);
                if (resourceType != null
                    && (resourceType != ResourceType.STYLEABLE || includeStyleables)) {
                  this.addResource(
                      resourceType,
                      entryWrapper.getPackageName(),
                      ResourcesUtil.resourceNameToFieldName(entry.getName()),
                      entryId);
                }
              });
    }

    private void recordSingleValueResources(ResourceType resourceType, Entry entry, int entryId) {
      if (!entry.hasOverlayableItem() && entry.getConfigValueList().size() == 1) {
        if (resourceType == ResourceType.STRING || resourceType == ResourceType.COLOR) {
          ConfigValue configValue = entry.getConfigValue(0);
          if (configValue.hasValue()) {
            Value value = configValue.getValue();
            if (value.hasItem()) {
              Item item = value.getItem();
              if (item.hasStr()) {
                resourcesWithSingleValue.put(
                    entryId, new StringSingleValueResource(item.getStr().getValue()));
              } else if (item.hasPrim()) {
                Primitive prim = item.getPrim();
                if (hasColor(prim)) {
                  resourcesWithSingleValue.put(entryId, new ColorSingleValueResource(prim));
                }
              }
            }
          }
        }
      }
    }

    private boolean hasColor(Primitive prim) {
      return prim.hasColorRgb4Value()
          || prim.hasColorRgb8Value()
          || prim.hasColorArgb4Value()
          || prim.hasColorArgb8Value();
    }
  }
}