/*
 * Copyright 2018 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.dialog;

import com.google.android.material.R;

import static com.google.android.material.internal.ThemeEnforcement.createThemedContext;

import android.content.Context;
import android.content.DialogInterface.OnCancelListener;
import android.content.DialogInterface.OnClickListener;
import android.content.DialogInterface.OnDismissListener;
import android.content.DialogInterface.OnKeyListener;
import android.content.DialogInterface.OnMultiChoiceClickListener;
import android.content.res.ColorStateList;
import android.content.res.TypedArray;
import android.database.Cursor;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.InsetDrawable;
import android.support.annotation.ArrayRes;
import android.support.annotation.AttrRes;
import android.support.annotation.DrawableRes;
import android.support.annotation.Nullable;
import android.support.annotation.Px;
import android.support.annotation.StringRes;
import android.support.annotation.StyleRes;
import com.google.android.material.internal.ThemeEnforcement;
import com.google.android.material.shape.MaterialShapeDrawable;
import android.support.v4.view.ViewCompat;
import android.support.v7.app.AlertDialog;
import android.util.TypedValue;
import android.view.View;
import android.view.Window;
import android.widget.AdapterView;
import android.widget.ListAdapter;

/**
 * An extension of {@link AlertDialog.Builder} for use with a Material theme (e.g.,
 * Theme.MaterialComponents).
 *
 * <p>This Builder must be used in order for AlertDialog objects to respond to color and shape
 * theming provided by Material themes.
 *
 * <p>The type of dialog returned is still an {@link AlertDialog}; there is no specific Material
 * implementation of {@link AlertDialog}.
 */
public class MaterialAlertDialogBuilder extends AlertDialog.Builder {

  private static final float DEFAULT_DIM_AMOUNT = 0.32f;
  @AttrRes private static final int DEF_STYLE_ATTR = R.attr.alertDialogStyle;
  @StyleRes private static final int DEF_STYLE_RES = R.style.AlertDialog_MaterialComponents;

  private Drawable background;
  @Px private int backgroundInsetStart;
  @Px private int backgroundInsetTop;
  @Px private int backgroundInsetEnd;
  @Px private int backgroundInsetBottom;

  public MaterialAlertDialogBuilder(Context context) {
    this(context, 0);
  }

  public MaterialAlertDialogBuilder(Context context, int themeResId) {
    super(createThemedContext(context, null, DEF_STYLE_ATTR, DEF_STYLE_RES), themeResId);
    // Ensure we are using the correctly themed context rather than the context that was passed in.
    context = getContext();

    TypedArray attributes =
        ThemeEnforcement.obtainStyledAttributes(
            context, null, R.styleable.MaterialAlertDialog, DEF_STYLE_ATTR, DEF_STYLE_RES);

    backgroundInsetStart =
        attributes.getDimensionPixelSize(
            R.styleable.MaterialAlertDialog_backgroundInsetStart,
            context
                .getResources()
                .getDimensionPixelSize(R.dimen.mtrl_alert_dialog_background_inset_start));
    backgroundInsetTop =
        attributes.getDimensionPixelSize(
            R.styleable.MaterialAlertDialog_backgroundInsetTop,
            context
                .getResources()
                .getDimensionPixelSize(R.dimen.mtrl_alert_dialog_background_inset_top));

    backgroundInsetEnd =
        attributes.getDimensionPixelSize(
            R.styleable.MaterialAlertDialog_backgroundInsetEnd,
            context
                .getResources()
                .getDimensionPixelSize(R.dimen.mtrl_alert_dialog_background_inset_end));
    backgroundInsetBottom =
        attributes.getDimensionPixelSize(
            R.styleable.MaterialAlertDialog_backgroundInsetBottom,
            context
                .getResources()
                .getDimensionPixelSize(R.dimen.mtrl_alert_dialog_background_inset_bottom));

    attributes.recycle();

    TypedValue typedValue = new TypedValue();
    context.getTheme().resolveAttribute(R.attr.colorSurface, typedValue, true);
    int surfaceColor = typedValue.data;

    MaterialShapeDrawable materialShapeDrawable =
        new MaterialShapeDrawable(context, null, DEF_STYLE_ATTR, DEF_STYLE_RES);
    materialShapeDrawable.setFillColor(ColorStateList.valueOf(surfaceColor));
    background = materialShapeDrawable;
  }

  @Override
  public AlertDialog create() {
    AlertDialog alertDialog = super.create();
    Window window = alertDialog.getWindow();
    /* {@link Window#getDecorView()} should be called before any changes are made to the Window
     * as it locks in attributes and affects layout. */
    View decorView = window.getDecorView();
    window.setDimAmount(DEFAULT_DIM_AMOUNT);

    int backgroundInsetLeft;
    int backgroundInsetRight;
    if (ViewCompat.getLayoutDirection(decorView) == ViewCompat.LAYOUT_DIRECTION_LTR) {
      backgroundInsetLeft = backgroundInsetStart;
      backgroundInsetRight = backgroundInsetEnd;
    } else {
      backgroundInsetLeft = backgroundInsetEnd;
      backgroundInsetRight = backgroundInsetStart;
    }
    Drawable insetDrawable =
        new InsetDrawable(
            background,
            backgroundInsetLeft,
            backgroundInsetTop,
            backgroundInsetRight,
            backgroundInsetBottom);
    window.setBackgroundDrawable(insetDrawable);
    decorView.setOnTouchListener(
        new InsetDialogOnTouchListener(
            alertDialog, backgroundInsetLeft, backgroundInsetTop));
    return alertDialog;
  }

  public Drawable getBackground() {
    return background;
  }

  public MaterialAlertDialogBuilder setBackground(Drawable background) {
    this.background = background;
    return this;
  }

  public MaterialAlertDialogBuilder setBackgroundInsetStart(@Px int backgroundInsetStart) {
    this.backgroundInsetStart = backgroundInsetStart;
    return this;
  }

  public MaterialAlertDialogBuilder setBackgroundInsetTop(@Px int backgroundInsetTop) {
    this.backgroundInsetTop = backgroundInsetTop;
    return this;
  }

  public MaterialAlertDialogBuilder setBackgroundInsetEnd(@Px int backgroundInsetEnd) {
    this.backgroundInsetEnd = backgroundInsetEnd;
    return this;
  }

  public MaterialAlertDialogBuilder setBackgroundInsetBottom(@Px int backgroundInsetBottom) {
    this.backgroundInsetBottom = backgroundInsetBottom;
    return this;
  }

  // The following methods are all pass-through methods used to specify the return type for the
  // builder chain.

  @Override
  public MaterialAlertDialogBuilder setTitle(@StringRes int titleId) {
    return (MaterialAlertDialogBuilder) super.setTitle(titleId);
  }

  @Override
  public MaterialAlertDialogBuilder setTitle(@Nullable CharSequence title) {
    return (MaterialAlertDialogBuilder) super.setTitle(title);
  }

  @Override
  public MaterialAlertDialogBuilder setCustomTitle(@Nullable View customTitleView) {
    return (MaterialAlertDialogBuilder) super.setCustomTitle(customTitleView);
  }

  @Override
  public MaterialAlertDialogBuilder setMessage(@StringRes int messageId) {
    return (MaterialAlertDialogBuilder) super.setMessage(messageId);
  }

  @Override
  public MaterialAlertDialogBuilder setMessage(@Nullable CharSequence message) {
    return (MaterialAlertDialogBuilder) super.setMessage(message);
  }

  @Override
  public MaterialAlertDialogBuilder setIcon(@DrawableRes int iconId) {
    return (MaterialAlertDialogBuilder) super.setIcon(iconId);
  }

  @Override
  public MaterialAlertDialogBuilder setIcon(@Nullable Drawable icon) {
    return (MaterialAlertDialogBuilder) super.setIcon(icon);
  }

  @Override
  public MaterialAlertDialogBuilder setIconAttribute(@AttrRes int attrId) {
    return (MaterialAlertDialogBuilder) super.setIconAttribute(attrId);
  }

  @Override
  public MaterialAlertDialogBuilder setPositiveButton(
      @StringRes int textId, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setPositiveButton(textId, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setPositiveButton(
      CharSequence text, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setPositiveButton(text, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setPositiveButtonIcon(Drawable icon) {
    return (MaterialAlertDialogBuilder) super.setPositiveButtonIcon(icon);
  }

  @Override
  public MaterialAlertDialogBuilder setNegativeButton(
      @StringRes int textId, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setNegativeButton(textId, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setNegativeButton(
      CharSequence text, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setNegativeButton(text, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setNegativeButtonIcon(Drawable icon) {
    return (MaterialAlertDialogBuilder) super.setNegativeButtonIcon(icon);
  }

  @Override
  public MaterialAlertDialogBuilder setNeutralButton(
      @StringRes int textId, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setNeutralButton(textId, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setNeutralButton(
      CharSequence text, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setNeutralButton(text, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setNeutralButtonIcon(Drawable icon) {
    return (MaterialAlertDialogBuilder) super.setNeutralButtonIcon(icon);
  }

  @Override
  public MaterialAlertDialogBuilder setCancelable(boolean cancelable) {
    return (MaterialAlertDialogBuilder) super.setCancelable(cancelable);
  }

  @Override
  public MaterialAlertDialogBuilder setOnCancelListener(OnCancelListener onCancelListener) {
    return (MaterialAlertDialogBuilder) super.setOnCancelListener(onCancelListener);
  }

  @Override
  public MaterialAlertDialogBuilder setOnDismissListener(OnDismissListener onDismissListener) {
    return (MaterialAlertDialogBuilder) super.setOnDismissListener(onDismissListener);
  }

  @Override
  public MaterialAlertDialogBuilder setOnKeyListener(OnKeyListener onKeyListener) {
    return (MaterialAlertDialogBuilder) super.setOnKeyListener(onKeyListener);
  }

  @Override
  public MaterialAlertDialogBuilder setItems(
      @ArrayRes int itemsId, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setItems(itemsId, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setItems(CharSequence[] items, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setItems(items, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setAdapter(
      final ListAdapter adapter, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setAdapter(adapter, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setCursor(
      final Cursor cursor, final OnClickListener listener, String labelColumn) {
    return (MaterialAlertDialogBuilder) super.setCursor(cursor, listener, labelColumn);
  }

  @Override
  public MaterialAlertDialogBuilder setMultiChoiceItems(
      @ArrayRes int itemsId, boolean[] checkedItems, final OnMultiChoiceClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setMultiChoiceItems(itemsId, checkedItems, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setMultiChoiceItems(
      CharSequence[] items, boolean[] checkedItems, final OnMultiChoiceClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setMultiChoiceItems(items, checkedItems, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setMultiChoiceItems(
      Cursor cursor,
      String isCheckedColumn,
      String labelColumn,
      final OnMultiChoiceClickListener listener) {
    return (MaterialAlertDialogBuilder)
        super.setMultiChoiceItems(cursor, isCheckedColumn, labelColumn, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setSingleChoiceItems(
      @ArrayRes int itemsId, int checkedItem, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setSingleChoiceItems(itemsId, checkedItem, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setSingleChoiceItems(
      Cursor cursor, int checkedItem, String labelColumn, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder)
        super.setSingleChoiceItems(cursor, checkedItem, labelColumn, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setSingleChoiceItems(
      CharSequence[] items, int checkedItem, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setSingleChoiceItems(items, checkedItem, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setSingleChoiceItems(
      ListAdapter adapter, int checkedItem, final OnClickListener listener) {
    return (MaterialAlertDialogBuilder) super.setSingleChoiceItems(adapter, checkedItem, listener);
  }

  @Override
  public MaterialAlertDialogBuilder setOnItemSelectedListener(
      final AdapterView.OnItemSelectedListener listener) {
    return (MaterialAlertDialogBuilder) super.setOnItemSelectedListener(listener);
  }

  @Override
  public MaterialAlertDialogBuilder setView(int layoutResId) {
    return (MaterialAlertDialogBuilder) super.setView(layoutResId);
  }

  @Override
  public MaterialAlertDialogBuilder setView(View view) {
    return (MaterialAlertDialogBuilder) super.setView(view);
  }
}