/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.android.material.internal;

import com.google.android.material.R;

import static androidx.annotation.RestrictTo.Scope.LIBRARY_GROUP;

import android.content.Context;
import android.content.res.ColorStateList;
import android.graphics.Color;
import android.graphics.Typeface;
import android.graphics.drawable.ColorDrawable;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.StateListDrawable;
import androidx.appcompat.view.menu.MenuItemImpl;
import androidx.appcompat.view.menu.MenuView;
import androidx.appcompat.widget.TooltipCompat;
import android.util.AttributeSet;
import android.util.TypedValue;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.view.ViewStub;
import android.view.accessibility.AccessibilityEvent;
import android.widget.CheckedTextView;
import android.widget.FrameLayout;
import androidx.annotation.Dimension;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.annotation.RestrictTo;
import androidx.core.content.res.ResourcesCompat;
import androidx.core.graphics.drawable.DrawableCompat;
import androidx.core.view.AccessibilityDelegateCompat;
import androidx.core.view.ViewCompat;
import androidx.core.view.accessibility.AccessibilityNodeInfoCompat;
import androidx.core.widget.TextViewCompat;

/** @hide */
@RestrictTo(LIBRARY_GROUP)
public class NavigationMenuItemView extends ForegroundLinearLayout implements MenuView.ItemView {

  private static final int[] CHECKED_STATE_SET = {android.R.attr.state_checked};

  private int iconSize;

  private boolean needsEmptyIcon;

  boolean checkable;

  boolean isBold = true;

  private final CheckedTextView textView;

  private FrameLayout actionArea;

  private MenuItemImpl itemData;

  private ColorStateList iconTintList;

  private boolean hasIconTintList;

  private Drawable emptyDrawable;

  private final AccessibilityDelegateCompat accessibilityDelegate =
      new AccessibilityDelegateCompat() {

        @Override
        public void onInitializeAccessibilityNodeInfo(
            View host, @NonNull AccessibilityNodeInfoCompat info) {
          super.onInitializeAccessibilityNodeInfo(host, info);
          info.setCheckable(checkable);
          info.setRoleDescription(
              getResources().getString(R.string.item_view_role_description));
        }
      };

  public NavigationMenuItemView(@NonNull Context context) {
    this(context, null);
  }

  public NavigationMenuItemView(@NonNull Context context, @Nullable AttributeSet attrs) {
    this(context, attrs, 0);
  }

  public NavigationMenuItemView(
      @NonNull Context context, @Nullable AttributeSet attrs, int defStyleAttr) {
    super(context, attrs, defStyleAttr);
    setOrientation(HORIZONTAL);
    LayoutInflater.from(context).inflate(R.layout.design_navigation_menu_item, this, true);
    setIconSize(context.getResources().getDimensionPixelSize(R.dimen.design_navigation_icon_size));
    textView = findViewById(R.id.design_menu_item_text);
    ViewCompat.setAccessibilityDelegate(textView, accessibilityDelegate);
  }

  @Override
  public void initialize(@NonNull MenuItemImpl itemData, int menuType) {
    this.itemData = itemData;
    if (itemData.getItemId() > 0) {
      setId(itemData.getItemId());
    }

    setVisibility(itemData.isVisible() ? VISIBLE : GONE);

    if (getBackground() == null) {
      setBackground(createDefaultBackground());
    }

    setCheckable(itemData.isCheckable());
    setChecked(itemData.isChecked());
    setEnabled(itemData.isEnabled());
    setTitle(itemData.getTitle());
    setIcon(itemData.getIcon());
    setActionView(itemData.getActionView());
    setContentDescription(itemData.getContentDescription());
    TooltipCompat.setTooltipText(this, itemData.getTooltipText());
    adjustAppearance();
  }

  public void initialize(@NonNull MenuItemImpl itemData, boolean isBold) {
    this.isBold = isBold;
    initialize(itemData, 0);
  }

  private boolean shouldExpandActionArea() {
    return itemData.getTitle() == null
        && itemData.getIcon() == null
        && itemData.getActionView() != null;
  }

  private void adjustAppearance() {
    if (shouldExpandActionArea()) {
      // Expand the actionView area
      textView.setVisibility(View.GONE);
      if (actionArea != null) {
        LayoutParams params = (LayoutParams) actionArea.getLayoutParams();
        params.width = LayoutParams.MATCH_PARENT;
        actionArea.setLayoutParams(params);
      }
    } else {
      textView.setVisibility(View.VISIBLE);
      if (actionArea != null) {
        LayoutParams params = (LayoutParams) actionArea.getLayoutParams();
        params.width = LayoutParams.WRAP_CONTENT;
        actionArea.setLayoutParams(params);
      }
    }
  }

  public void recycle() {
    if (actionArea != null) {
      actionArea.removeAllViews();
    }
    textView.setCompoundDrawables(null, null, null, null);
  }

  private void setActionView(@Nullable View actionView) {
    if (actionView != null) {
      if (actionArea == null) {
        actionArea =
            (FrameLayout)
                ((ViewStub) findViewById(R.id.design_menu_item_action_area_stub)).inflate();
      }
      // Make sure to remove the existing parent if the View is reused
      if (actionView.getParent() != null) {
        ((ViewGroup) actionView.getParent()).removeView(actionView);
      }
      actionArea.removeAllViews();
      actionArea.addView(actionView);
    }
  }

  @Nullable
  private StateListDrawable createDefaultBackground() {
    TypedValue value = new TypedValue();
    if (getContext()
        .getTheme()
        .resolveAttribute(androidx.appcompat.R.attr.colorControlHighlight, value, true)) {
      StateListDrawable drawable = new StateListDrawable();
      drawable.addState(CHECKED_STATE_SET, new ColorDrawable(value.data));
      drawable.addState(EMPTY_STATE_SET, new ColorDrawable(Color.TRANSPARENT));
      return drawable;
    }
    return null;
  }

  @Override
  public MenuItemImpl getItemData() {
    return itemData;
  }

  @Override
  public void setTitle(CharSequence title) {
    textView.setText(title);
  }

  @Override
  public void setCheckable(boolean checkable) {
    refreshDrawableState();
    if (this.checkable != checkable) {
      this.checkable = checkable;
      accessibilityDelegate.sendAccessibilityEvent(
          textView, AccessibilityEvent.TYPE_WINDOW_CONTENT_CHANGED);
    }
  }

  @Override
  public void setChecked(boolean checked) {
    refreshDrawableState();
    textView.setChecked(checked);
    // TODO(b/246765947): Use component tokens to control font weight
    textView.setTypeface(
        textView.getTypeface(), checked && isBold ? Typeface.BOLD : Typeface.NORMAL);
  }

  @Override
  public void setShortcut(boolean showShortcut, char shortcutKey) {}

  @Override
  public void setIcon(@Nullable Drawable icon) {
    if (icon != null) {
      if (hasIconTintList) {
        Drawable.ConstantState state = icon.getConstantState();
        icon = DrawableCompat.wrap(state == null ? icon : state.newDrawable()).mutate();
        icon.setTintList(iconTintList);
      }
      icon.setBounds(0, 0, iconSize, iconSize);
    } else if (needsEmptyIcon) {
      if (emptyDrawable == null) {
        emptyDrawable =
            ResourcesCompat.getDrawable(
                getResources(), R.drawable.navigation_empty_icon, getContext().getTheme());
        if (emptyDrawable != null) {
          emptyDrawable.setBounds(0, 0, iconSize, iconSize);
        }
      }
      icon = emptyDrawable;
    }
    textView.setCompoundDrawablesRelative(icon, null, null, null);
  }

  public void setIconSize(@Dimension int iconSize) {
    this.iconSize = iconSize;
  }

  @Override
  public boolean prefersCondensedTitle() {
    return false;
  }

  @Override
  public boolean showsIcon() {
    return true;
  }

  @Override
  protected int[] onCreateDrawableState(int extraSpace) {
    final int[] drawableState = super.onCreateDrawableState(extraSpace + 1);
    if (itemData != null && itemData.isCheckable() && itemData.isChecked()) {
      mergeDrawableStates(drawableState, CHECKED_STATE_SET);
    }
    return drawableState;
  }

  void setIconTintList(ColorStateList tintList) {
    iconTintList = tintList;
    hasIconTintList = iconTintList != null;
    if (itemData != null) {
      // Update the icon so that the tint takes effect
      setIcon(itemData.getIcon());
    }
  }

  public void setTextAppearance(int textAppearance) {
    TextViewCompat.setTextAppearance(textView, textAppearance);
  }

  public void setTextColor(ColorStateList colors) {
    textView.setTextColor(colors);
  }

  public void setNeedsEmptyIcon(boolean needsEmptyIcon) {
    this.needsEmptyIcon = needsEmptyIcon;
  }

  public void setHorizontalPadding(int padding) {
    setPadding(padding, getPaddingTop(), padding, getPaddingBottom());
  }

  public void setIconPadding(int padding) {
    textView.setCompoundDrawablePadding(padding);
  }

  public void setMaxLines(int maxLines) {
    textView.setMaxLines(maxLines);
  }
}