/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

package com.facebook.react.uimanager;

import com.facebook.react.common.MapBuilder;
import com.facebook.react.common.ReactConstants;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.annotation.Nullable;
import android.util.Log;

/**
 * Class that stores the mapping between native view name used in JS and the corresponding instance
 * of {@link ViewManager}.
 */
public final class ViewManagerRegistry {

  private final Map<String, ViewManager> mViewManagers;
  private final @Nullable UIManagerModule.ViewManagerResolver mViewManagerResolver;
  private static Map<String, String> sComponentNames = new HashMap<>();

  static {
    sComponentNames.put("Image", "RCTImageView");
    sComponentNames.put("ForwardRef(Image)", "RCTImageView");
    sComponentNames.put("Text", "RCTText");
    sComponentNames.put("TextInput", "AndroidTextInput");
    sComponentNames.put("TouchableHighlight", "RCTView");
    sComponentNames.put("WebView", "RCTWebView");
  }

  public ViewManagerRegistry(UIManagerModule.ViewManagerResolver viewManagerResolver) {
    mViewManagers = MapBuilder.newHashMap();
    mViewManagerResolver = viewManagerResolver;
  }

  public ViewManagerRegistry(List<ViewManager> viewManagerList) {
    Map<String, ViewManager> viewManagerMap = MapBuilder.newHashMap();
    for (ViewManager viewManager : viewManagerList) {
      viewManagerMap.put(viewManager.getName(), viewManager);
    }

    mViewManagers = viewManagerMap;
    mViewManagerResolver = null;
  }

  public ViewManagerRegistry(Map<String, ViewManager> viewManagerMap) {
    mViewManagers =
        viewManagerMap != null ? viewManagerMap : MapBuilder.<String, ViewManager>newHashMap();
    mViewManagerResolver = null;
  }

  public ViewManager get(String className) {
    String newClassName;
    if (sComponentNames.containsKey(className)) {
      newClassName = sComponentNames.get(className);
    } else {
      newClassName = className;
    }

    ViewManager viewManager = mViewManagers.get(newClassName);
    if (viewManager != null) {
      return viewManager;
    }
    if (mViewManagerResolver != null) {
      viewManager = mViewManagerResolver.getViewManager(newClassName);
      if (viewManager != null) {
        mViewManagers.put(newClassName, viewManager);
        return viewManager;
      }
    }
    Log.w(ReactConstants.TAG, "No ViewManager defined for class: " + newClassName);
    throw new IllegalViewOperationException("No ViewManager defined for class " + newClassName);
  }
}