robolectric-shadows/shadows-core/src/main/resources/org/robolectric/shadows/ShadowAssetManager.java.vm

#set($Integer = 0)
#set($apiLevel = $Integer.parseInt($apiLevel))

#if ($apiLevel >= 21)
#set($ptrClass = "long")
#set($ptrClassBoxed = "Long")
#else
#set($ptrClass = "int")
#set($ptrClassBoxed = "Integer")
#end

package org.robolectric.shadows;

import android.content.res.AssetFileDescriptor;
import android.content.res.AssetManager;
import android.content.res.Resources;
import android.content.res.XmlResourceParser;
import android.util.TypedValue;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.ArrayList;
import org.jetbrains.annotations.NotNull;
import org.robolectric.RuntimeEnvironment;
import org.robolectric.manifest.AndroidManifest;
import org.robolectric.annotation.Implementation;
import org.robolectric.annotation.Implements;
import org.robolectric.annotation.HiddenApi;
import org.robolectric.res.Attribute;
import org.robolectric.res.DrawableNode;
import org.robolectric.res.DrawableResourceLoader;
import org.robolectric.res.FsFile;
import org.robolectric.res.ResName;
import org.robolectric.res.ResType;
import org.robolectric.res.ResourceIndex;
import org.robolectric.res.ResourceLoader;
import org.robolectric.res.Style;
import org.robolectric.res.StyleData;
import org.robolectric.res.TypedResource;
import org.robolectric.res.builder.XmlFileBuilder;
import org.robolectric.util.Strings;

import static org.robolectric.Shadows.shadowOf;
import static org.robolectric.shadows.ShadowApplication.getInstance;

@SuppressWarnings({"UnusedDeclaration"})
@Implements(AssetManager.class)
public final class ShadowAssetManager {
  public static final int STYLE_NUM_ENTRIES = 6;
  public static final int STYLE_TYPE = 0;
  public static final int STYLE_DATA = 1;
  public static final int STYLE_ASSET_COOKIE = 2;
  public static final int STYLE_RESOURCE_ID = 3;
  public static final int STYLE_CHANGING_CONFIGURATIONS = 4;
  public static final int STYLE_DENSITY = 5;

  private String qualifiers = "";
  private Map<$ptrClassBoxed, Resources.Theme> themesById = new LinkedHashMap<>();
  private int nextInternalThemeId = 1000;
  private static final Map<$ptrClassBoxed, List<OverlayedStyle>> APPLIED_STYLES = new HashMap<>();
  private AndroidManifest appManifest;
  private ResourceLoader resourceLoader;

  static AssetManager bind(AssetManager assetManager, AndroidManifest androidManifest, ResourceLoader resourceLoader) {
    ShadowAssetManager shadowAssetManager = shadowOf(assetManager);
    if (shadowAssetManager.appManifest != null) throw new RuntimeException("ResourceLoader already set!");
    shadowAssetManager.appManifest = androidManifest;
    shadowAssetManager.resourceLoader = resourceLoader;
    return assetManager;
  }

  @HiddenApi @Implementation
  public CharSequence getResourceText(int ident) {
    TypedResource value = getAndResolve(ident, getQualifiers(), true);
    if (value == null) return null;
    return (CharSequence) value.getData();
  }

  @HiddenApi @Implementation
  public CharSequence getResourceBagText(int ident, int bagEntryId) {
    throw new UnsupportedOperationException(); // todo
  }

  @HiddenApi @Implementation
  public String[] getResourceStringArray(final int id) {
    CharSequence[] resourceTextArray = getResourceTextArray(id);
    if (resourceTextArray == null) return null;
    String[] strings = new String[resourceTextArray.length];
    for (int i = 0; i < strings.length; i++) {
      strings[i] = resourceTextArray[i].toString();
    }
    return strings;
  }

  @HiddenApi @Implementation
  public boolean getResourceValue(int ident, int density, TypedValue outValue, boolean resolveRefs) {
    TypedResource value = getAndResolve(ident, getQualifiers(), resolveRefs);
    if (value == null) return false;

    getConverter(value).fillTypedValue(value.getData(), outValue);
    return true;
  }

  private Converter getConverter(TypedResource value) {
    return Converter.getConverter(value.getResType());
  }

  @HiddenApi @Implementation
  public CharSequence[] getResourceTextArray(final int id) {
    ResName resName = resourceLoader.getResourceIndex().getResName(id);
    if (resName == null) throw new Resources.NotFoundException("unknown resource " + id);
    TypedResource value = getAndResolve(resName, getQualifiers(), true);
    if (value == null) return null;
    TypedResource[] items = getConverter(value).getItems(value);
    CharSequence[] charSequences = new CharSequence[items.length];
    for (int i = 0; i < items.length; i++) {
      TypedResource typedResource = resolve(items[i], getQualifiers(), resName);
      charSequences[i] = getConverter(typedResource).asCharSequence(typedResource);
    }
    return charSequences;
  }

  @HiddenApi @Implementation
  public boolean getThemeValue($ptrClass theme, int ident, TypedValue outValue, boolean resolveRefs) {
    ResourceIndex resourceIndex = resourceLoader.getResourceIndex();
    ResName resName = resourceIndex.getResName(ident);
    Resources.Theme theTheme = getThemeByInternalId(theme);
    // Load the style for the theme we represent. E.g. "@style/Theme.Robolectric"
    ResName themeStyleName = resourceIndex.getResName(shadowOf(theTheme).getStyleResourceId());
    if (themeStyleName == null) return false; // is this right?

    Style themeStyle = resolveStyle(resourceLoader, null, themeStyleName, getQualifiers());

    //// Load the theme attribute for the default style attributes. E.g., attr/buttonStyle
    //ResName defStyleName = resourceLoader.getResourceIndex().getResName(ident);
    //
    //// Load the style for the default style attribute. E.g. "@style/Widget.Robolectric.Button";
    //String defStyleNameValue = themeStyle.getAttrValue(defStyleName);
    //ResName defStyleResName = new ResName(defStyleName.packageName, "style", defStyleName.name);
    //Style style = resolveStyle(resourceLoader, defStyleResName, getQualifiers());
    if (themeStyle != null) {
      int styleResourceId = shadowOf(theTheme).getStyleResourceId();
      List<OverlayedStyle> overlayThemeStyles = getOverlayThemeStyles(styleResourceId);
      Attribute attrValue = ShadowResources.getOverlayedThemeValue(resName, themeStyle, overlayThemeStyles);
      while(resolveRefs && attrValue != null && attrValue.isStyleReference()) {
        ResName attrResName = new ResName(attrValue.contextPackageName, "attr", attrValue.value.substring(1));
        attrValue = ShadowResources.getOverlayedThemeValue(attrResName, themeStyle, overlayThemeStyles);
      }
      if (attrValue != null) {
        TypedResource attrDataValue = resourceLoader.getValue(resName, getQualifiers());
        Converter.convertAndFill(attrValue, outValue, resourceLoader, getQualifiers(), resolveRefs);
        return true;
      }
    }
    return false;
  }

  @HiddenApi @Implementation
  public void ensureStringBlocks() {
  }

  @Implementation
  public final InputStream open(String fileName) throws IOException {
    return appManifest.getAssetsDirectory().join(fileName).getInputStream();
  }

  @Implementation
  public final String[] list(String path) throws IOException {
    FsFile file = appManifest.getAssetsDirectory().join(path);
    if (file.isDirectory()) {
      return file.listFileNames();
    }
    return new String[0];
  }

  @HiddenApi @Implementation
  public final InputStream openNonAsset(int cookie, String fileName, int accessMode) throws IOException {
    final ResName resName = qualifyFromNonAssetFileName(fileName);
    final DrawableNode drawableNode = resourceLoader.getDrawableNode(resName, getQualifiers());

    if (drawableNode == null) {
      throw new IOException("Unable to find resource for " + fileName);
    }

    return new ByteArrayInputStream(drawableNode.getFsFile().getBytes());
  }

  private ResName qualifyFromNonAssetFileName(String fileName) {
    if (fileName.startsWith("jar:")) {
      // Must remove "jar:" prefix, or else qualifyFromFilePath fails on Windows
      return ResName.qualifyFromFilePath("android", fileName.replaceFirst("jar:", ""));
    } else {
      return ResName.qualifyFromFilePath(appManifest.getPackageName(), fileName);
    }
  }

  @HiddenApi @Implementation
  public final AssetFileDescriptor openNonAssetFd(int cookie, String fileName) throws IOException {
    throw new UnsupportedOperationException();
  }

  @Implementation
  public final XmlResourceParser openXmlResourceParser(int cookie, String fileName) throws IOException {
    return XmlFileBuilder.getXmlResourceParser(fileName, "fixme", null);
  }

  @HiddenApi @Implementation
  public int addAssetPath(String path) {
    return 1;
  }

  @HiddenApi @Implementation
  public boolean isUpToDate() {
    return true;
  }

  @HiddenApi @Implementation
  public void setLocale(String locale) {
  }

  @Implementation
  public String[] getLocales() {
    return new String[0]; // todo
  }

  @HiddenApi @Implementation
  public void setConfiguration(int mcc, int mnc, String locale,
                 int orientation, int touchscreen, int density, int keyboard,
                 int keyboardHidden, int navigation, int screenWidth, int screenHeight,
                 int smallestScreenWidthDp, int screenWidthDp, int screenHeightDp,
                 int screenLayout, int uiMode, int majorVersion) {
  }

  @HiddenApi @Implementation
  public int[] getArrayIntResource(int arrayRes) {
    ResName resName = resourceLoader.getResourceIndex().getResName(arrayRes);
    if (resName == null) throw new Resources.NotFoundException("unknown resource " + arrayRes);
    TypedResource value = getAndResolve(resName, getQualifiers(), true);
    if (value == null) return null;
    TypedResource[] items = getConverter(value).getItems(value);
    int[] ints = new int[items.length];
    for (int i = 0; i < items.length; i++) {
      TypedResource typedResource = resolve(items[i], getQualifiers(), resName);
      ints[i] = getConverter(typedResource).asInt(typedResource);
    }
    return ints;
  }

  @HiddenApi @Implementation
  synchronized public int createTheme() {
    return nextInternalThemeId++;
  }

  @HiddenApi @Implementation
  synchronized public void releaseTheme($ptrClass theme) {
    themesById.remove(theme);
  }

  @HiddenApi @Implementation
  public static void applyThemeStyle($ptrClass theme, int styleRes, boolean force) {
    if (!APPLIED_STYLES.containsKey(theme)) {
      APPLIED_STYLES.put(theme, new LinkedList<OverlayedStyle>());
    }

    ResourceLoader resourceLoader = getInstance().getResourceLoader();
    ShadowAssetManager assetManager = shadowOf(getInstance().getAssets());

    ResourceIndex resourceIndex = resourceLoader.getResourceIndex();
    ResName resName = resourceIndex.getResName(styleRes);

    Style style = resolveStyle(resourceLoader, null, resName, assetManager.getQualifiers());

    List<OverlayedStyle> overlayedStyleList = APPLIED_STYLES.get(theme);
    OverlayedStyle styleToAdd = new OverlayedStyle(style, force);
    for (int i = 0; i < overlayedStyleList.size(); ++i) {
      if (styleToAdd.equals(overlayedStyleList.get(i))) {
        overlayedStyleList.remove(i);
        break;
      }
    }
    overlayedStyleList.add(styleToAdd);
  }

  static List<OverlayedStyle> getOverlayThemeStyles($ptrClass themeResourceId) {
    return APPLIED_STYLES.get(themeResourceId);
  }

  static class OverlayedStyle {
    Style style;
    boolean force;

    public OverlayedStyle(Style style, boolean force) {
      this.style = style;
      this.force = force;
    }

    @Override
    public boolean equals(Object obj) {
      if (!(obj instanceof OverlayedStyle)) {
        return false;
      }
      OverlayedStyle overlayedStyle = (OverlayedStyle) obj;
      return style.equals(overlayedStyle.style);
    }

    @Override
    public int hashCode() {
      return style.hashCode();
    }
  }

  @HiddenApi @Implementation
  public static void copyTheme($ptrClass dest, long source) {
    throw new UnsupportedOperationException();
  }

  /////////////////////////

  synchronized public void setTheme($ptrClass internalThemeId, Resources.Theme theme) {
    themesById.put(internalThemeId, theme);
  }

  synchronized private Resources.Theme getThemeByInternalId($ptrClass internalThemeId) {
    return themesById.get(internalThemeId);
  }

  static Style resolveStyle(ResourceLoader resourceLoader, Style appTheme, @NotNull ResName themeStyleName, String qualifiers) {
    TypedResource themeStyleResource = resourceLoader.getValue(themeStyleName, qualifiers);
    if (themeStyleResource == null) return null;
    StyleData themeStyleData = (StyleData) themeStyleResource.getData();
    return new StyleResolver(resourceLoader, themeStyleData, appTheme, themeStyleName, qualifiers);
  }

  TypedResource getAndResolve(int resId, String qualifiers, boolean resolveRefs) {
    ResName resName = resourceLoader.getResourceIndex().getResName(resId);
    if (resName == null) throw new Resources.NotFoundException("unknown resource " + resId);
    return getAndResolve(resName, qualifiers, resolveRefs);
  }

  TypedResource getAndResolve(@NotNull ResName resName, String qualifiers, boolean resolveRefs) {
    TypedResource value = resourceLoader.getValue(resName, qualifiers);
    if (resolveRefs) {
      value = resolve(value, qualifiers, resName);
    }

    // todo: make the drawable loader put stuff into the normal spot...
    if (value == null && DrawableResourceLoader.isStillHandledHere(resName)) {
      DrawableNode drawableNode = resourceLoader.getDrawableNode(resName, qualifiers);
      return new TypedResource<FsFile>(drawableNode.getFsFile(), ResType.FILE);
    }

    // todo: gross. this is so resources.getString(R.layout.foo) works for ABS.
    if (value == null && "layout".equals(resName.type)) {
      throw new UnsupportedOperationException("ugh, this doesn't work still?");
    }

    return value;
  }

  TypedResource resolve(TypedResource value, String qualifiers, ResName contextResName) {
    while (true) {
      if (value == null) return null;

      Object data = value.getData();
      if (data instanceof String) {
        String s = (String) data;
        if (s.equals("@null")) {
          return null;
        } else if (s.startsWith("@")) {
          String refStr = s.substring(1).replace("+", "");
          contextResName = ResName.qualifyResName(refStr, contextResName);
          value = resourceLoader.getValue(contextResName, qualifiers);
          // back through...
        } else {
          return value;
        }
      } else {
        return value;
      }
    }
  }

  public FsFile getAssetsDirectory() {
    return appManifest.getAssetsDirectory();
  }

  public String getQualifiers() {
    return RuntimeEnvironment.getQualifiers();
  }

  public void setQualifiers(String qualifiers) {
    RuntimeEnvironment.setQualifiers(qualifiers);
  }

  static class StyleResolver implements Style {
    private final ResourceLoader resourceLoader;
    private final List<StyleData> styles = new ArrayList<>();
    private final Style theme;
    private final ResName myResName;
    private final String qualifiers;

    public StyleResolver(ResourceLoader resourceLoader, StyleData styleData,
        Style theme, ResName myResName, String qualifiers) {
      this.resourceLoader = resourceLoader;
      this.theme = theme;
      this.myResName = myResName;
      this.qualifiers = qualifiers;
      styles.add(styleData);
    }

    @Override public Attribute getAttrValue(ResName resName) {
      resName.mustBe("attr");
      for (StyleData style : styles) {
        Attribute value = style.getAttrValue(resName);
        if (value != null) return value;
      }
      int initialSize = styles.size();
      while (hasParent(styles.get(styles.size() - 1))) {
        StyleData parent = getParent(styles.get(styles.size() - 1));
        if (parent != null) {
          styles.add(parent);
        } else {
          break;
        }
      }
      for (int i = initialSize; i < styles.size(); i++) {
        StyleData style = styles.get(i);
        Attribute value = style.getAttrValue(resName);
        if (value != null) return value;
      }
      if (theme != null) {
        Attribute value = theme.getAttrValue(resName);
        if (value != null) return value;
      }

      return null;
    }

    private static String getParentStyleName(StyleData style) {
      if (style == null) {
        return null;
      }
      String parent = style.getParent();
      if (parent == null || parent.isEmpty()) {
        parent = null;
        String name = style.getName();
        if (name.contains(".")) {
          parent = name.substring(0, name.lastIndexOf('.'));
          if (parent.isEmpty()) {
            return null;
          }
        }
      }
      return parent;
    }

    private static boolean hasParent(StyleData style) {
      if (style == null) return false;
      String parent = style.getParent();
      return parent != null && !parent.isEmpty();
    }

    private StyleData getParent(StyleData style) {
      String parent = getParentStyleName(style);

      if (parent == null) return null;

      if (parent.startsWith("@")) parent = parent.substring(1);

      ResName styleRef = ResName.qualifyResName(parent, style.getPackageName(), "style");

      styleRef = dereferenceResName(styleRef);

      TypedResource typedResource = resourceLoader.getValue(styleRef, qualifiers);

      if (typedResource == null) {
        StringBuilder builder = new StringBuilder("Could not find any resource ")
            .append(" from reference ").append(styleRef)
            .append(" from style ").append(style)
            .append(" with theme ").append(theme);
        throw new RuntimeException(builder.toString());
      }

      Object data = typedResource.getData();
      if (data instanceof StyleData) {
        return (StyleData) data;
      } else {
        StringBuilder builder = new StringBuilder(styleRef.toString())
            .append(" does not resolve to a Style.")
            .append(" got ").append(data).append(" instead. ")
            .append(" from style ").append(style)
            .append(" with theme ").append(theme);
        throw new RuntimeException(builder.toString());
      }
    }

    private ResName dereferenceResName(ResName res) {
      ResName styleRef = res;
      boolean dereferencing = true;
      while ("attr".equals(styleRef.type) && dereferencing) {
        dereferencing = false;
        for (StyleData parentStyle : styles) {
          Attribute value = parentStyle.getAttrValue(styleRef);
          if (value != null) {
            styleRef = dereferenceAttr(value);
            dereferencing = true;
            break;
          }
        }
        if (!dereferencing && theme != null) {
          Attribute value = theme.getAttrValue(styleRef);
          if (value != null) {
            styleRef = dereferenceAttr(value);
            dereferencing = true;
          }
        }
      }

      return styleRef;
    }

    private ResName dereferenceAttr(Attribute attr) {
      if (attr.isResourceReference()) {
        return attr.getResourceReference();
      } else if (attr.isStyleReference()) {
        return attr.getStyleReference();
      }
      throw new RuntimeException("Found a " + attr + " but can't cast it :(");
    }

    @Override
    public boolean equals(Object obj) {
      if (!(obj instanceof StyleResolver)) {
        return false;
      }
      StyleResolver other = (StyleResolver) obj;

      return ((theme == null && other.theme == null) || (theme != null && theme.equals(other.theme)))
          && ((myResName == null && other.myResName == null)
            || (myResName != null && myResName.equals(other.myResName)))
          && Strings.equals(qualifiers, other.qualifiers);
    }

    @Override
    public int hashCode() {
      int hashCode = 0;
      hashCode = 31 * hashCode + (theme != null ? theme.hashCode() : 0);
      hashCode = 31 * hashCode + (myResName != null ? myResName.hashCode() : 0);
      hashCode = 31 * hashCode + Strings.nullToEmpty(qualifiers).hashCode();
      return hashCode;
    }

    @Override
    public String toString() {
      return "StyleResolver{"
          + "name=" + myResName
          + ", of=" + styles.get(0)
          + "}";
    }

  }

}