package cn.zjy.actionsheet;

import android.app.DialogFragment;
import android.app.Fragment;
import android.app.FragmentManager;
import android.content.Context;
import android.content.DialogInterface;
import android.graphics.Color;
import android.graphics.Typeface;
import android.graphics.drawable.ColorDrawable;
import android.os.Bundle;
import android.text.Spannable;
import android.text.SpannableStringBuilder;
import android.text.TextUtils;
import android.text.style.AbsoluteSizeSpan;
import android.text.style.ForegroundColorSpan;
import android.view.Gravity;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.view.Window;
import android.view.WindowManager;
import android.widget.Button;
import android.widget.FrameLayout;
import android.widget.LinearLayout;
import android.widget.TextView;

import cn.zjy.actionsheet.rn.R;

public class ActionSheet extends DialogFragment implements View.OnClickListener {
    public static final String TAG = "ActionSheet";

    public static final String TITLE = "title";
    public static final String TITLE_COLOR = "title_color";
    public static final String TITLE_TEXT_SIZE = "title_text_size";

    public static final String MESSAGE = "message";
    public static final String MESSAGE_COLOR = "message_color";
    public static final String MESSAGE_FONT_SIZE = "message_font_size";
    public static final String MESSAGE_HEIGHT = "message_height";


    public static final String CANCEL_BTN_TITLE = "cancel_btn_title";
    public static final String CANCEL_BTN_TITLE_COLOR = "cancel_btn_title_color";
    public static final String CANCEL_BTN_TEXT_SIZE = "cancel_btn_text_size";

    public static final String OTHER_BTN_TITLES = "other_btn_titles";
    public static final String OTHER_BTN_TITLE_COLORS = "other_btn_title_colors";
    public static final String OTHER_BTN_TEXT_SIZE = "other_btn_text_size";
    public static final String OTHER_BTN_SUB_TITLES = "other_btn_sub_titles";
    public static final String OTHER_BTN_SUB_TITLE_COLORS = "other_btn_sub_title_colors";
    public static final String OTHER_BTN_SUB_TEXT_SIZE = "other_btn_sub_text_size";

    public static final String CANCELABLE_ON_TOUCH_OUTSIDE = "cancelable_on_touch_outside";

    private static final float DEFAULT_TITLE_TEXT_SIZE = 18;
    private static final float DEFAULT_BTN_TEXT_SIZE = 20;
    private static final float DEFAULT_BTN_SUB_TEXT_SIZE = 15;

    private static final int DEFAULT_TITLE_TEXT_COLOR = Color.parseColor("#929292");
    private static final int DEFAULT_BTN_TEXT_COLOR = Color.BLACK;
    private static final int DEFAULT_BTN_SUB_TEXT_COLOR = Color.parseColor("#929292");

    private ActionSheetListener mActionSheetListener;
    private int mClickedBtnIdx = -1;

    public void show(FragmentManager fragmentManager) {
        Fragment fragment = fragmentManager.findFragmentByTag(TAG);
        if (fragment != null && fragment instanceof ActionSheet) {
            ((ActionSheet) fragment).dismiss();
        }

        show(fragmentManager, TAG);
    }

    public void setActionSheetListener(ActionSheetListener listener) {
        mActionSheetListener = listener;
    }

    @Override
    public void onDismiss(DialogInterface dialog) {
        super.onDismiss(dialog);
        if (mActionSheetListener != null) {
            mActionSheetListener.onDismiss(this, mClickedBtnIdx >= 0);
        }
    }

    @Override
    public void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setStyle(DialogFragment.STYLE_NORMAL, 0);
    }

    @Override
    public void onStart() {
        super.onStart();

        Window window = getDialog().getWindow();
        if (window == null) return;
        window.setBackgroundDrawable(new ColorDrawable(Color.TRANSPARENT));
        window.setGravity(Gravity.BOTTOM | Gravity.LEFT | Gravity.RIGHT);
        WindowManager.LayoutParams params = window.getAttributes();
        params.windowAnimations = R.style.ActionSheetAnimation;
        params.width = WindowManager.LayoutParams.MATCH_PARENT;
        window.setAttributes(params);
    }

    @Override
    public View onCreateView(LayoutInflater inflater, ViewGroup container, Bundle savedInstanceState) {
        Window window = getDialog().getWindow();
        if (window != null) {
            getDialog().getWindow().requestFeature(Window.FEATURE_NO_TITLE);
        }
        return createView();
    }

    private View createView() {
        Bundle args = getArguments();
        getDialog().setCanceledOnTouchOutside(args.getBoolean(CANCELABLE_ON_TOUCH_OUTSIDE, true));

        Context context = getActivity();
        LinearLayout layout = new LinearLayout(context);
        FrameLayout.LayoutParams lp = new FrameLayout.LayoutParams(FrameLayout.LayoutParams.MATCH_PARENT,
                FrameLayout.LayoutParams.MATCH_PARENT);
        int padding = (int) context.getResources().getDimension(R.dimen.action_sheet_padding);
        layout.setLayoutParams(lp);
        layout.setOrientation(LinearLayout.VERTICAL);
        layout.setGravity(Gravity.CENTER_HORIZONTAL);
        layout.setPadding(padding, 0, padding, padding);

        addTitle(layout);

        // Add message
        addMessage(layout);

        addOtherBtns(layout);
        addCancelBtn(layout);

        return layout;
    }

    private void addTitle(LinearLayout layout) {
        Bundle args = getArguments();
        String title = args.getString(TITLE);
        if (TextUtils.isEmpty(title)) return;

        int titleColor = args.getInt(TITLE_COLOR, DEFAULT_TITLE_TEXT_COLOR);
        float textSize = args.getFloat(TITLE_TEXT_SIZE, DEFAULT_TITLE_TEXT_SIZE);

        int titleHeight = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_title_height);

        int bottomMargin = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_btn_gap);
        LinearLayout.LayoutParams lp = new LinearLayout.LayoutParams(LinearLayout.LayoutParams.MATCH_PARENT,
                titleHeight);
        lp.bottomMargin = bottomMargin;

        int background = R.drawable.action_sheet_other_btns_bg_top;
        String[] titles = args.getStringArray(OTHER_BTN_TITLES);
        if (titles == null || titles.length == 0) {
            background = R.drawable.action_sheet_other_btns_bg_single;
        }
        TextView tvTitle = new TextView(getActivity());
        tvTitle.setText(title);
        tvTitle.setTypeface(null, Typeface.BOLD);
        tvTitle.setTextSize(textSize);
        tvTitle.setTextColor(titleColor);
        tvTitle.setGravity(Gravity.CENTER);
        tvTitle.setBackgroundResource(background);
        layout.addView(tvTitle, lp);
    }

    private void addMessage(LinearLayout layout) {
        Bundle args = getArguments();
        String desc = args.getString(MESSAGE);
        if (TextUtils.isEmpty(desc)) return;

        int titleColor = args.getInt(MESSAGE_COLOR, DEFAULT_TITLE_TEXT_COLOR);
        float textSize = args.getFloat(MESSAGE_FONT_SIZE, DEFAULT_TITLE_TEXT_SIZE);

        int messageHeight = args.getInt(MESSAGE_HEIGHT);
        int bottomMargin = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_btn_gap);
        LinearLayout.LayoutParams lp = new LinearLayout.LayoutParams(LinearLayout.LayoutParams.MATCH_PARENT,
                messageHeight);
        lp.bottomMargin = bottomMargin;

        int background = R.drawable.action_sheet_other_btns_bg_middle;
        String[] titles = args.getStringArray(OTHER_BTN_TITLES);
        if (titles == null || titles.length == 0) {
            background = R.drawable.action_sheet_other_btns_bg_top;
        }
        TextView tvMessage = new TextView(getActivity());
        tvMessage.setPadding(50, 10, 50, 0);
        tvMessage.setText(desc);
        tvMessage.setTextSize(textSize);
        tvMessage.setTextColor(titleColor);
        tvMessage.setGravity(Gravity.LEFT);
        tvMessage.setBackgroundResource(background);
        layout.addView(tvMessage, lp);
    }

    private void addOtherBtns(LinearLayout layout) {
        Bundle args = getArguments();
        String[] titles = args.getStringArray(OTHER_BTN_TITLES);
        if (titles == null || titles.length == 0) return;

        int[] titleColors = args.getIntArray(OTHER_BTN_TITLE_COLORS);
        float textSize = args.getFloat(OTHER_BTN_TEXT_SIZE, DEFAULT_BTN_TEXT_SIZE);
        String[] subTitles = args.getStringArray(OTHER_BTN_SUB_TITLES);
        int[] subTitleColors = args.getIntArray(OTHER_BTN_SUB_TITLE_COLORS);
        float subTextSize = args.getFloat(OTHER_BTN_SUB_TEXT_SIZE, DEFAULT_BTN_SUB_TEXT_SIZE);
        int btnHeight = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_btn_height);
        int bottomMargin = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_btn_gap);
        LinearLayout.LayoutParams lp = new LinearLayout.LayoutParams(LinearLayout.LayoutParams.MATCH_PARENT,
                btnHeight);
        lp.bottomMargin = bottomMargin;

        int len = titles.length;
        for (int i = 0; i < len; i++) {
            int titleColor = titleColors != null && i < titleColors.length ? titleColors[i] : DEFAULT_BTN_TEXT_COLOR;
            int subTitleColor = subTitleColors != null && i < subTitleColors.length ? subTitleColors[i] : DEFAULT_BTN_SUB_TEXT_COLOR;

            Button btn = new Button(getActivity());
            btn.setAllCaps(false);
            btn.setTag(i);
            btn.setBackgroundResource(getOtherBtnBackground(i));
            btn.setOnClickListener(this);
            btn.setPadding(0, 0, 0, 0);
            if (subTitles != null && i < subTitles.length && !TextUtils.isEmpty(subTitles[i])) {
                int titleLength = titles[i].length() + 1;
                int subTitleLength = subTitles[i].length();

                SpannableStringBuilder builder = new SpannableStringBuilder(titles[i] + "\n" + subTitles[i]);
                ForegroundColorSpan titleColorSpan = new ForegroundColorSpan(titleColor);
                AbsoluteSizeSpan titleSizeSpan = new AbsoluteSizeSpan((int) textSize, true);
                builder.setSpan(titleColorSpan, 0, titleLength, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);
                builder.setSpan(titleSizeSpan, 0, titleLength, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);

                ForegroundColorSpan subTitleColorSpan = new ForegroundColorSpan(subTitleColor);
                AbsoluteSizeSpan subTitleSizeSpan = new AbsoluteSizeSpan((int) subTextSize, true);
                builder.setSpan(subTitleColorSpan, titleLength, titleLength + subTitleLength, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);
                builder.setSpan(subTitleSizeSpan, titleLength, titleLength + subTitleLength, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);

                btn.setText(builder);
            } else {
                btn.setText(titles[i]);
                btn.setTextSize(textSize);
                btn.setTextColor(titleColor);
            }
            layout.addView(btn, lp);
        }
    }

    private int getOtherBtnBackground(int index) {
        int background = R.drawable.action_sheet_other_btns_bg_single;
        Bundle args = getArguments();
        String[] btnTitles = args.getStringArray(OTHER_BTN_TITLES);
        boolean hasTitle = !TextUtils.isEmpty(args.getString(TITLE));
        if (btnTitles == null || btnTitles.length == 0)
            return background;

        int totalCount = btnTitles.length;
        if (hasTitle) {
            totalCount++;
            index++;
        }
        if (totalCount == 1) {
            background = R.drawable.action_sheet_other_btns_bg_single;
        } else {
            if (index == 0) {
                background = R.drawable.action_sheet_other_btns_bg_top;
            } else if (index == totalCount - 1) {
                background = R.drawable.action_sheet_other_btns_bg_bottom;
            } else {
                background = R.drawable.action_sheet_other_btns_bg_middle;
            }
        }

        return background;
    }

    private void addCancelBtn(LinearLayout layout) {
        Bundle args = getArguments();
        String cancelTitle = args.getString(CANCEL_BTN_TITLE);
        if (TextUtils.isEmpty(cancelTitle)) return;

        int cancelBtnColor = args.getInt(CANCEL_BTN_TITLE_COLOR, DEFAULT_BTN_TEXT_COLOR);
        float textSize = args.getFloat(CANCEL_BTN_TEXT_SIZE, DEFAULT_BTN_TEXT_SIZE);
        int btnHeight = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_btn_height);
        LinearLayout.LayoutParams lp = new LinearLayout.LayoutParams(LinearLayout.LayoutParams.MATCH_PARENT,
                btnHeight);
        lp.topMargin = (int) getActivity().getResources().getDimension(R.dimen.action_sheet_padding);
        String[] otherBtnTitles = args.getStringArray(OTHER_BTN_TITLES);
        int index = otherBtnTitles == null ? 0 : otherBtnTitles.length;

        Button btnCancel = new Button(getActivity());
        btnCancel.setText(cancelTitle);
        btnCancel.setAllCaps(false);
        btnCancel.setTextSize(textSize);
        btnCancel.setTextColor(cancelBtnColor);
        btnCancel.setOnClickListener(this);
        btnCancel.setBackgroundResource(R.drawable.action_sheet_cancel_btn_bg);
        btnCancel.setTag(index);
        layout.addView(btnCancel, lp);
    }

    @Override
    public void onClick(View v) {
        mClickedBtnIdx = (int) v.getTag();
        if (mActionSheetListener != null && mClickedBtnIdx >= 0) {
            mActionSheetListener.onButtonClicked(ActionSheet.this, mClickedBtnIdx);
        }
        dismiss();
    }

    public static class Builder {
        private String mTitle;
        private String mMessage;
        private int mMessageColor;
        private float mMessageTextSize = -1;
        private int mMessageHeight = 200;
        private int mTitleColor;
        private float mTitleTextSize = -1;
        private String mCancelBtnTitle;
        private int mCancelBtnTitleColor;
        private float mCancelBtnTextSize = -1;
        private String[] mOtherBtnTitles;
        private int[] mOtherBtnTitleColors;
        private float mOtherBtnTextSize = -1;
        private String[] mOtherBtnSubTitles;
        private int[] mOtherBtnSubTitleColors;
        private float mOtherBtnSubTextSize = -1;
        private boolean mCancelableOnTouchOutside;
        private ActionSheetListener mActionSheetListener;

        public Builder setTitle(String title, int titleColor) {
            mTitle = title;
            mTitleColor = titleColor;
            return this;
        }

        public Builder setMessage(String message, int messageColor, int messageHeight) {
            mMessage = message;
            mMessageColor = messageColor;
            if(messageHeight > 0) {
                mMessageHeight = messageHeight;
            }

            return this;
        }

        public Builder setTitleTextSize(float textSize) {
            mTitleTextSize = textSize;
            return this;
        }

        public Builder setCancelBtn(String title, int titleColor) {
            mCancelBtnTitle = title;
            mCancelBtnTitleColor = titleColor;
            return this;
        }

        public Builder setCancelBtnTextSize(float textSize) {
            mCancelBtnTextSize = textSize;
            return this;
        }

        public Builder setOtherBtn(String[] titles, int[] titleColors) {
            mOtherBtnTitles = titles;
            mOtherBtnTitleColors = titleColors;
            return this;
        }

        public Builder setOtherBtnTextSize(float textSize) {
            mOtherBtnTextSize = textSize;
            return this;
        }

        public Builder setOtherBtnSub(String[] titles, int[] titleColors) {
            mOtherBtnSubTitles = titles;
            mOtherBtnSubTitleColors = titleColors;
            return this;
        }

        public Builder setOtherBtnSubTextSize(float textSize) {
            mOtherBtnSubTextSize = textSize;
            return this;
        }

        public Builder setCancelableOnTouchOutside(boolean cancelable) {
            mCancelableOnTouchOutside = cancelable;
            return this;
        }

        public Builder setActionSheetListener(ActionSheetListener listener) {
            mActionSheetListener = listener;
            return this;
        }

        public ActionSheet build() {
            Bundle bundle = new Bundle();
            bundle.putString(TITLE, mTitle);
            bundle.putInt(TITLE_COLOR, mTitleColor);
            if (mTitleTextSize > 0) {
                bundle.putFloat(TITLE_TEXT_SIZE, mTitleTextSize);
            }
            bundle.putString(CANCEL_BTN_TITLE, mCancelBtnTitle);
            bundle.putInt(CANCEL_BTN_TITLE_COLOR, mCancelBtnTitleColor);
            if (mCancelBtnTextSize > 0) {
                bundle.putFloat(CANCEL_BTN_TEXT_SIZE, mCancelBtnTextSize);
            }
            bundle.putStringArray(OTHER_BTN_TITLES, mOtherBtnTitles);
            bundle.putIntArray(OTHER_BTN_TITLE_COLORS, mOtherBtnTitleColors);
            if (mOtherBtnTextSize > 0) {
                bundle.putFloat(OTHER_BTN_TEXT_SIZE, mOtherBtnTextSize);
            }
            bundle.putStringArray(OTHER_BTN_SUB_TITLES, mOtherBtnSubTitles);
            bundle.putIntArray(OTHER_BTN_SUB_TITLE_COLORS, mOtherBtnSubTitleColors);
            if (mOtherBtnSubTextSize > 0) {
                bundle.putFloat(OTHER_BTN_SUB_TEXT_SIZE, mOtherBtnSubTextSize);
            }
            bundle.putBoolean(CANCELABLE_ON_TOUCH_OUTSIDE, mCancelableOnTouchOutside);

            ActionSheet actionSheet = new ActionSheet();
            actionSheet.setActionSheetListener(mActionSheetListener);
            actionSheet.setArguments(bundle);

            /**
             * Message
             */
            bundle.putString(MESSAGE, mMessage);
            bundle.putInt(MESSAGE_COLOR, mMessageColor);
            bundle.putInt(MESSAGE_HEIGHT, mMessageHeight);

            if (mMessageTextSize > 0) {
                bundle.putFloat(MESSAGE_FONT_SIZE, mMessageTextSize);
            }


            return actionSheet;
        }
    }

    public interface ActionSheetListener {
        void onDismiss(ActionSheet actionSheet, boolean isByBtn);

        void onButtonClicked(ActionSheet actionSheet, int index);
    }

}