/*
 * Copyright (C) 2020 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.timepicker;

import com.google.android.material.R;

import static android.text.TextUtils.isEmpty;
import static com.google.android.material.timepicker.TimePickerView.GENERIC_VIEW_ACCESSIBILITY_CLASS_NAME;

import android.content.Context;
import android.content.res.ColorStateList;
import android.content.res.Configuration;
import android.os.Build.VERSION;
import android.os.Build.VERSION_CODES;
import android.os.LocaleList;
import android.text.Editable;
import android.text.InputFilter;
import android.text.TextWatcher;
import android.util.AttributeSet;
import android.view.LayoutInflater;
import android.view.View;
import android.widget.Checkable;
import android.widget.EditText;
import android.widget.FrameLayout;
import android.widget.TextView;
import androidx.annotation.ColorInt;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.annotation.VisibleForTesting;
import androidx.core.view.AccessibilityDelegateCompat;
import androidx.core.view.ViewCompat;
import androidx.core.view.accessibility.AccessibilityNodeInfoCompat;
import com.google.android.material.chip.Chip;
import com.google.android.material.color.MaterialColors;
import com.google.android.material.internal.TextWatcherAdapter;
import com.google.android.material.internal.ViewUtils;
import com.google.android.material.textfield.TextInputLayout;
import java.util.Arrays;

/**
 * A {@link Chip} that can switch to a {@link TextInputLayout} when checked to modify it's content.
 * It keeps the helper text from the TextInput always visible.
 */
class ChipTextInputComboView extends FrameLayout implements Checkable {

  private final Chip chip;
  private final TextInputLayout textInputLayout;
  private final EditText editText;
  private TextWatcher watcher;
  private TextView label;
  private CharSequence chipText = "";

  private boolean hasError = false;
  private ColorStateList originalChipBackgroundColor;
  private ColorStateList originalChipTextColor;
  private ColorStateList originalEditTextColor;
  private ColorStateList originalEditTextCursorColor;
  private ColorStateList originalLabelColor;
  @ColorInt private int originalChipStrokeColor;

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

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

  public ChipTextInputComboView(
      @NonNull Context context, @Nullable AttributeSet attrs, int defStyleAttr) {
    super(context, attrs, defStyleAttr);
    LayoutInflater inflater = LayoutInflater.from(context);
    chip = (Chip) inflater.inflate(R.layout.material_time_chip, this, false);
    chip.setAccessibilityClassName(GENERIC_VIEW_ACCESSIBILITY_CLASS_NAME);
    textInputLayout = (TextInputLayout) inflater.inflate(R.layout.material_time_input, this, false);
    editText = textInputLayout.getEditText();
    editText.setVisibility(INVISIBLE);
    watcher = new TextFormatter();
    editText.addTextChangedListener(watcher);
    updateHintLocales();
    addView(chip);
    addView(textInputLayout);
    label = findViewById(R.id.material_label);
    editText.setId(View.generateViewId());
    label.setLabelFor(editText.getId());
    editText.setSaveEnabled(false);
    editText.setLongClickable(false);
  }

  private void updateHintLocales() {
    if (VERSION.SDK_INT >= VERSION_CODES.N) {
      Configuration configuration = getContext().getResources().getConfiguration();
      final LocaleList locales = configuration.getLocales();
      editText.setImeHintLocales(locales);
    }
  }

  @Override
  public boolean isChecked() {
    return chip.isChecked();
  }

  @Override
  public void setChecked(boolean checked) {
    chip.setChecked(checked);
    if (checked) {
      chip.setText("");
      chip.setImportantForAccessibility(View.IMPORTANT_FOR_ACCESSIBILITY_NO);
    } else {
      chip.setText(chipText);
      chip.setImportantForAccessibility(View.IMPORTANT_FOR_ACCESSIBILITY_YES);
    }
    editText.setVisibility(checked ? VISIBLE : INVISIBLE);
    if (isChecked()) {
      ViewUtils.requestFocusAndShowKeyboard(editText, /* useWindowInsetsController= */ false);
    }
  }

  @Override
  public void toggle() {
    chip.toggle();
  }

  public void setText(CharSequence text) {
    String formattedText = formatText(text);
    chipText = formattedText;
    chip.setText(formattedText);
    if (!isEmpty(formattedText)) {
      editText.removeTextChangedListener(watcher);

      editText.setText(formattedText);
      setAccessibilityDelegate(editText, chipText.toString(), label.getText());

      editText.addTextChangedListener(watcher);
    }
  }

  @VisibleForTesting
  CharSequence getChipText() {
    return chipText;
  }

  private String formatText(CharSequence text) {
    return TimeModel.formatText(getResources(), text);
  }

  @Override
  public void setOnClickListener(@Nullable OnClickListener l) {
    chip.setOnClickListener(l);
  }

  @Override
  public void setTag(int key, Object tag) {
    chip.setTag(key, tag);
  }

  public void setHelperText(CharSequence helperText) {
    label.setText(helperText);
  }

  public void setCursorVisible(boolean visible) {
    editText.setCursorVisible(visible);
  }

  public void addInputFilter(InputFilter filter) {
    InputFilter[] current = editText.getFilters();
    InputFilter[] arr = Arrays.copyOf(current, current.length + 1);
    arr[current.length] = filter;
    editText.setFilters(arr);
  }

  public TextInputLayout getTextInput() {
    return textInputLayout;
  }

  public void setChipDelegate(AccessibilityDelegateCompat clickActionDelegate) {
    ViewCompat.setAccessibilityDelegate(chip, clickActionDelegate);
  }

  public void setError(boolean hasError) {
    if (this.hasError == hasError) {
      return;
    }
    this.hasError = hasError;

    if (hasError) {
      applyErrorColors();
    } else {
      clearErrorColors();
    }
  }

  private void applyErrorColors() {
    originalChipBackgroundColor = chip.getChipBackgroundColor();
    originalChipTextColor = chip.getTextColors();
    originalEditTextColor = editText.getTextColors();
    originalLabelColor = label.getTextColors();
    originalChipStrokeColor = textInputLayout.getBoxStrokeColor();

    // TODO(b/394610420): tokens and ColorStateList with error state
    ColorStateList colorErrorContainer =
        MaterialColors.getColorStateListOrNull(getContext(), R.attr.colorErrorContainer);
    ColorStateList colorOnErrorContainer =
        MaterialColors.getColorStateListOrNull(getContext(), R.attr.colorOnErrorContainer);
    if (colorErrorContainer != null && colorOnErrorContainer != null) {
      chip.setChipBackgroundColor(colorErrorContainer);
      chip.setTextColor(colorOnErrorContainer);
      editText.setTextColor(colorOnErrorContainer);
      textInputLayout.setBoxStrokeColor(colorOnErrorContainer.getDefaultColor());
      label.setTextColor(colorOnErrorContainer);
      if (VERSION.SDK_INT >= VERSION_CODES.Q) {
        originalEditTextCursorColor = textInputLayout.getCursorColor();
        textInputLayout.setCursorColor(colorOnErrorContainer);
      }
    }
  }

  private void clearErrorColors() {
    chip.setChipBackgroundColor(originalChipBackgroundColor);
    chip.setTextColor(originalChipTextColor);
    editText.setTextColor(originalEditTextColor);
    textInputLayout.setBoxStrokeColor(originalChipStrokeColor);
    label.setTextColor(originalLabelColor);
    if (VERSION.SDK_INT >= VERSION_CODES.Q) {
      textInputLayout.setCursorColor(originalEditTextCursorColor);
    }
  }

  public boolean hasError() {
    return hasError;
  }

  private class TextFormatter extends TextWatcherAdapter {

    private static final String DEFAULT_TEXT = "00";

    @Override
    public void afterTextChanged(Editable editable) {
      if (isEmpty(editable)) {
        chipText = formatText(DEFAULT_TEXT);
        return;
      }
      String formattedText = formatText(editable);
      chipText = isEmpty(formattedText) ? formatText(DEFAULT_TEXT) : formattedText;
      setAccessibilityDelegate(editText, editable.toString(), label.getText());
    }
  }

  private void setAccessibilityDelegate(
      @NonNull View view, CharSequence text, @Nullable CharSequence hint) {
    ViewCompat.setAccessibilityDelegate(
        view,
        new AccessibilityDelegateCompat() {
          @Override
          public void onInitializeAccessibilityNodeInfo(
              @NonNull View host, @NonNull AccessibilityNodeInfoCompat info) {
            super.onInitializeAccessibilityNodeInfo(host, info);
            info.setText(text);
            info.setHintText(hint);
            info.setMaxTextLength(2);
          }
        });
  }

  @Override
  protected void onConfigurationChanged(Configuration newConfig) {
    super.onConfigurationChanged(newConfig);
    updateHintLocales();
  }
}