package com.borjabolufer.elevate.adapters;

import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.ImageView;
import android.widget.TextView;

import androidx.annotation.NonNull;
import androidx.recyclerview.widget.RecyclerView;

import com.borjabolufer.elevate.R;
import com.borjabolufer.elevate.ui.fragments.HomeFragment;

import java.util.List;

public class EjercicioAdapter extends RecyclerView.Adapter<RecyclerView.ViewHolder> {

    public interface OnItemClickListener {
        void onItemClick(HomeFragment.EjercicioItem item);
    }

    private static final int TYPE_HEADER = 0;
    private static final int TYPE_EJERCICIO = 1;

    private List<HomeFragment.EjercicioItem> ejercicios;
    private OnItemClickListener listener;

    public EjercicioAdapter(List<HomeFragment.EjercicioItem> ejercicios, OnItemClickListener listener) {
        this.ejercicios = ejercicios;
        this.listener = listener;
    }

    @Override
    public int getItemViewType(int position) {
        return ejercicios.get(position).isEsHeader() ? TYPE_HEADER : TYPE_EJERCICIO;
    }

    @NonNull
    @Override
    public RecyclerView.ViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
        if (viewType == TYPE_HEADER) {
            View view = LayoutInflater.from(parent.getContext())
                    .inflate(R.layout.item_ejercicio_header, parent, false);
            return new HeaderViewHolder(view);
        } else {
            View view = LayoutInflater.from(parent.getContext())
                    .inflate(R.layout.item_ejercicio, parent, false);
            return new EjercicioViewHolder(view);
        }
    }

    @Override
    public void onBindViewHolder(@NonNull RecyclerView.ViewHolder holder, int position) {
        HomeFragment.EjercicioItem item = ejercicios.get(position);
        if (holder instanceof HeaderViewHolder) {
            ((HeaderViewHolder) holder).bind(item);
        } else if (holder instanceof EjercicioViewHolder) {
            ((EjercicioViewHolder) holder).bind(item, listener);
        }
    }

    @Override
    public int getItemCount() {
        return ejercicios.size();
    }

    static class HeaderViewHolder extends RecyclerView.ViewHolder {
        TextView textHeader;

        HeaderViewHolder(@NonNull View itemView) {
            super(itemView);
            textHeader = itemView.findViewById(R.id.text_header);
        }

        void bind(HomeFragment.EjercicioItem item) {
            textHeader.setText(item.getNombre());
        }
    }

    static class EjercicioViewHolder extends RecyclerView.ViewHolder {
        TextView textNombre, textSeries, textRepeticiones;
        ImageView iconEjercicio;

        EjercicioViewHolder(@NonNull View itemView) {
            super(itemView);
            textNombre = itemView.findViewById(R.id.text_nombre_ejercicio);
            textSeries = itemView.findViewById(R.id.text_series);
            textRepeticiones = itemView.findViewById(R.id.text_repeticiones);
            iconEjercicio = itemView.findViewById(R.id.icon_ejercicio);
        }

        void bind(HomeFragment.EjercicioItem item, OnItemClickListener listener) {
            textNombre.setText(item.getNombre());

            if ("calentamiento".equals(item.getTipo()) || "enfriamiento".equals(item.getTipo())) {
                textSeries.setText("");
                textRepeticiones.setText("");
            } else {
                textSeries.setText(item.getSeries() > 0 ? item.getSeries() + " series" : "");
                textRepeticiones.setText(item.getRepeticiones() > 0 ? item.getRepeticiones() + " reps" : "");
            }

            int iconResource;
            int tintColor;
            switch (item.getTipo()) {
                case "calentamiento":
                    iconResource = R.drawable.ic_calentamiento;
                    tintColor = R.color.warning;
                    break;
                case "principal":
                    iconResource = R.drawable.ic_principal;
                    tintColor = R.color.primary;
                    break;
                case "enfriamiento":
                    iconResource = R.drawable.ic_enfriamiento;
                    tintColor = R.color.success;
                    break;
                default:
                    iconResource = R.drawable.ic_calentamiento;
                    tintColor = R.color.warning;
                    break;
            }

            iconEjercicio.setImageResource(iconResource);
            iconEjercicio.setColorFilter(itemView.getContext().getColor(tintColor));

            itemView.setOnClickListener(v -> {
                if (listener != null) {
                    listener.onItemClick(item);
                }
            });
        }
    }
}
