/*
 * Copyright (C) 2023 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.android.wm.shell.bubbles.bar;

import android.annotation.ColorInt;
import android.content.Context;
import android.content.res.ColorStateList;
import android.graphics.Color;
import android.util.AttributeSet;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.view.accessibility.AccessibilityNodeInfo;
import android.widget.ImageView;
import android.widget.LinearLayout;
import android.widget.TextView;

import androidx.core.widget.ImageViewCompat;

import com.android.wm.shell.R;
import com.android.wm.shell.bubbles.Bubble;
import com.android.wm.shell.shared.TypefaceUtils;

import java.util.ArrayList;

/**
 * Bubble bar expanded view menu
 */
public class BubbleBarMenuView extends LinearLayout {

    private ViewGroup mBubbleSectionView;
    private ViewGroup mActionsSectionView;
    private ImageView mBubbleIconView;
    private ImageView mBubbleDismissIconView;
    private TextView mBubbleTitleView;
    // The animation has three stages. Each stage transition lasts until the animation ends. In
    // stage 1, the title item content fades in. In stage 2, the background of the option items
    // fades in. In stage 3, the option item content fades in.
    private static final int SHOW_MENU_STAGES_COUNT = 3;

    public BubbleBarMenuView(Context context) {
        this(context, null /* attrs */);
    }

    public BubbleBarMenuView(Context context, AttributeSet attrs) {
        this(context, attrs, 0 /* defStyleAttr */);
    }

    public BubbleBarMenuView(Context context, AttributeSet attrs, int defStyleAttr) {
        this(context, attrs, defStyleAttr, 0 /* defStyleRes */);
    }

    public BubbleBarMenuView(Context context, AttributeSet attrs, int defStyleAttr,
            int defStyleRes) {
        super(context, attrs, defStyleAttr, defStyleRes);
    }

    @Override
    protected void onFinishInflate() {
        super.onFinishInflate();
        mBubbleSectionView = findViewById(R.id.bubble_bar_manage_menu_bubble_section);
        mActionsSectionView = findViewById(R.id.bubble_bar_manage_menu_actions_section);
        mBubbleIconView = findViewById(R.id.bubble_bar_manage_menu_bubble_icon);
        mBubbleTitleView = findViewById(R.id.bubble_bar_manage_menu_bubble_title);
        TypefaceUtils.setTypeface(mBubbleTitleView, TypefaceUtils.FontFamily.GSF_TITLE_MEDIUM);
        mBubbleDismissIconView = findViewById(R.id.bubble_bar_manage_menu_dismiss_icon);
        updateThemeColors();

        mBubbleSectionView.setAccessibilityDelegate(new AccessibilityDelegate() {
            @Override
            public void onInitializeAccessibilityNodeInfo(View host, AccessibilityNodeInfo info) {
                super.onInitializeAccessibilityNodeInfo(host, info);
                info.addAction(new AccessibilityNodeInfo.AccessibilityAction(
                        AccessibilityNodeInfo.ACTION_CLICK, getResources().getString(
                        R.string.bubble_accessibility_action_collapse_menu)));
            }
        });
    }

    private void updateThemeColors() {
        mActionsSectionView.getBackground().setTint(
                mContext.getColor(com.android.internal.R.color.materialColorSurfaceBright));
        ImageViewCompat.setImageTintList(mBubbleDismissIconView,
                ColorStateList.valueOf(
                        mContext.getColor(com.android.internal.R.color.materialColorOnSurface)));
    }

    /** Animates the menu from the specified start scale. */
    public void animateFromStartScale(float currentScale, float progress) {
        int menuItemElevation = getResources().getDimensionPixelSize(
                R.dimen.bubble_manage_menu_elevation);
        setScaleX(currentScale);
        setScaleY(currentScale);
        setAlphaForTitleViews(progress);
        mBubbleSectionView.setElevation(menuItemElevation * progress);
        float actionsBackgroundAlpha = Math.max(0,
                (progress - (float) 1 / SHOW_MENU_STAGES_COUNT) * (SHOW_MENU_STAGES_COUNT - 1));
        float actionItemsAlpha = Math.max(0,
                (progress - (float) 2 / SHOW_MENU_STAGES_COUNT) * SHOW_MENU_STAGES_COUNT);
        mActionsSectionView.setAlpha(actionsBackgroundAlpha);
        mActionsSectionView.setElevation(menuItemElevation * actionsBackgroundAlpha);
        setMenuItemViewsAlpha(actionItemsAlpha);
    }

    private void setAlphaForTitleViews(float alpha) {
        mBubbleIconView.setAlpha(alpha);
        mBubbleTitleView.setAlpha(alpha);
        mBubbleDismissIconView.setAlpha(alpha);
    }

    private void setMenuItemViewsAlpha(float alpha) {
        for (int i = mActionsSectionView.getChildCount() - 1; i >= 0; i--) {
            mActionsSectionView.getChildAt(i).setAlpha(alpha);
        }
    }

    /** Update menu details with bubble info */
    void updateInfo(Bubble bubble) {
        if (bubble.getIcon() != null) {
            mBubbleIconView.setImageIcon(bubble.getIcon());
        } else {
            mBubbleIconView.setImageBitmap(bubble.getBubbleIcon());
        }
        mBubbleTitleView.setText(bubble.getTitle());
    }

    /**
     * Update menu action items views
     * @param actions used to populate menu item views
     */
    void updateActions(ArrayList<MenuAction> actions) {
        mActionsSectionView.removeAllViews();
        LayoutInflater inflater = LayoutInflater.from(mContext);

        for (MenuAction action : actions) {
            BubbleBarMenuItemView itemView = (BubbleBarMenuItemView) inflater.inflate(
                    R.layout.bubble_bar_menu_item, mActionsSectionView, false);
            itemView.update(action.mIcon, action.mTitle, action.mTint);
            itemView.setOnClickListener(action.mOnClick);
            mActionsSectionView.addView(itemView);
        }
    }

    /** Sets on close menu listener */
    void setOnCloseListener(Runnable onClose) {
        mBubbleSectionView.setOnClickListener(view -> {
            onClose.run();
        });
    }

    /**
     * Overridden to proxy to section views alpha.
     * @implNote
     * If animate alpha on the parent (menu container) view, section view shadows get distorted.
     * To prevent distortion and artifacts alpha changes applied directly on the section views.
     */
    @Override
    public void setAlpha(float alpha) {
        mBubbleSectionView.setAlpha(alpha);
        mActionsSectionView.setAlpha(alpha);
    }

    /**
     * Overridden to proxy section view alpha value.
     * @implNote
     * The assumption is that both section views have the same alpha value
     */
    @Override
    public float getAlpha() {
        return mBubbleSectionView.getAlpha();
    }

    /** Return title menu item height. */
    public float getTitleItemHeight() {
        return mBubbleSectionView.getHeight();
    }

    /**
     * Menu action details used to create menu items
     */
    static class MenuAction {
        private DrawableProvider mIcon;
        private @ColorInt int mTint;
        private String mTitle;
        private OnClickListener mOnClick;

        MenuAction(DrawableProvider icon, String title, OnClickListener onClick) {
            this(icon, title, Color.TRANSPARENT, onClick);
        }

        MenuAction(DrawableProvider icon, String title, @ColorInt int tint,
                OnClickListener onClick) {
            this.mIcon = icon;
            this.mTitle = title;
            this.mTint = tint;
            this.mOnClick = onClick;
        }
    }
}