package com.qliktrialreactnativestraighttable;

import android.content.Context;
import android.graphics.Canvas;
import android.graphics.Color;
import android.text.Layout;
import android.util.Log;
import android.view.Gravity;
import android.view.View;
import android.view.ViewGroup;
import android.widget.FrameLayout;
import android.widget.LinearLayout;
import android.widget.RelativeLayout;
import android.widget.ScrollView;
import android.widget.Space;
import android.widget.TextView;

import androidx.recyclerview.widget.LinearLayoutManager;
import androidx.recyclerview.widget.RecyclerView;

import java.util.ArrayList;
import java.util.List;

public class TableViewFactory {
  public HeaderViewFactory headerViewFactory = null;
  public CustomHorizontalScrollView scrollView = null;
  public MockVerticalScrollView verticalScrollBar = null;
  public MockHorizontalScrollView horizontalScrollView = null;
  public RootLayout rootLayout = null;
  public HeaderView headerView = null;
  public RowCountView rowCountView = null;
  public TotalsView totalsView = null;
  public CustomRecyclerView firstColumnRecyclerView = null;
  public CustomRecyclerView coupledRecyclerView = null;
  public List<GrabberView> grabbers = null;
  public List<TotalsCell> totalsCells = null;
  public HeaderCell firstColumnHeaderCell = null;
  public TotalsViewCell firstColumnTotalsCell = null;
  public ScreenGuideView screenGuideView = null;
  private List<DataColumn> dataColumns = null;
  public String totalsPosition;
  private final Context context;
  private final ColumnWidths columnWidths;
  private final DataProvider dataProvider;
  private final TableView tableView;
  private final DragBox dragBox;
  private final DragBox firstColumnDragBox;

  public TableViewFactory(TableView tableView, ColumnWidths columnWidths, DataProvider dataProvider, DragBox dragBox, DragBox firstColumnDragBox) {
    this.tableView = tableView;
    this.columnWidths = columnWidths;
    this.dataProvider = dataProvider;
    this.context = tableView.getContext();
    this.dragBox = dragBox;
    this.firstColumnDragBox = firstColumnDragBox;
  }

  public void createAll() {
    this.dataColumns = dataProvider.dataColumns;
    this.totalsCells = dataProvider.totalsCells;
    this.totalsPosition = dataProvider.totalsPosition;

    updateRowHeights();
    createHeaderFactory();
    createScrollView();
    createMockScrollBar();

    tableView.addView(scrollView);
    scrollView.addView(rootLayout);
    tableView.addView(verticalScrollBar);
    tableView.addView(horizontalScrollView);

    if(tableView.headerContentStyle.wrap) {
      if(totalsView != null) {
        totalsView.testTextWrap();
      }
      if(firstColumnTotalsCell != null) {
        firstColumnTotalsCell.testTextWrap();
      }
      headerView.testTextWrap();
    }

    updateFirstColumnHeaderHeight();
    updateTotalsViewHeight();
    tempListenForLayoutChanges();

    dataProvider.notifyDataSetChanged();

    tableView.post(new Runnable() {
      @Override
      public void run() {
        updateGrabbers();
        updateScrollbarBounds();
        updateRecyclerViewMargins();
        testTextWrap();
      }
    });
  }

  private void tempListenForLayoutChanges() {
    // add a temporary listener to do text wrapping
    if(coupledRecyclerView != null) {
      coupledRecyclerView.addOnLayoutChangeListener(new View.OnLayoutChangeListener() {
        @Override
        public void onLayoutChange(View v, int left, int top, int right, int bottom, int oldLeft, int oldTop, int oldRight, int oldBottom) {
          if(coupledRecyclerView.testTextWrap(false)) {
            coupledRecyclerView.removeOnLayoutChangeListener(this);
          }
        }
      });
    }
  }

  private void setMockScrollLayouts() {
    FrameLayout.LayoutParams verticalFrameLayout = new FrameLayout.LayoutParams((int) PixelUtils.dpToPx(5), FrameLayout.LayoutParams.MATCH_PARENT);
    verticalFrameLayout.gravity = Gravity.RIGHT;
    int headerHeight = headerView.getMeasuredHeight() > 0 ? headerView.getMeasuredHeight() : tableView.headerHeight;
    int extraTop = 0;
    int extraBottom = 0;
    if(tableView.totalsPosition != null) {
      if (tableView.totalsPosition.equals("top")) {
        extraTop = tableView.totalsHeight;
      } else if(tableView.totalsPosition.equals("bottom")) {
        extraBottom = tableView.totalsHeight;
      }
    }
    verticalFrameLayout.topMargin = headerHeight + extraTop;
    verticalFrameLayout.bottomMargin = TableTheme.DefaultRowHeight + extraBottom;
    verticalScrollBar.setLayoutParams(verticalFrameLayout);

    FrameLayout.LayoutParams horizontalFrameLayout = new FrameLayout.LayoutParams(FrameLayout.LayoutParams.MATCH_PARENT, (int) PixelUtils.dpToPx(5));
    horizontalFrameLayout.gravity = Gravity.BOTTOM;
    horizontalFrameLayout.bottomMargin = TableTheme.DefaultRowHeight;
    horizontalFrameLayout.rightMargin = TableTheme.HorizontalScrollViewPadding;

    horizontalScrollView.setLayoutParams(horizontalFrameLayout);
  }

  protected void updateScrollbarBounds() {
    setMockScrollLayouts();
    this.verticalScrollBar.requestLayout();
    this.horizontalScrollView.requestLayout();
  }

  protected void updateRowHeights() {
    tableView.cellContentStyle.themedRowHeight = tableView.cellContentStyle.getLineHeight() + CellView.PADDING_X_2;
    tableView.rowHeight = tableView.cellContentStyle.themedRowHeight;
    tableView.headerHeight = tableView.headerContentStyle.getLineHeight() + CellView.PADDING_X_2;
    tableView.totalsHeight = tableView.cellContentStyle.getLineHeight() + CellView.PADDING_X_2;
  }

  protected void createMockScrollBar() {
    verticalScrollBar = new MockVerticalScrollView(context);
    horizontalScrollView = new MockHorizontalScrollView(context);

    scrollView.setScrollbars(horizontalScrollView, verticalScrollBar);
    coupledRecyclerView.setScrollbar(verticalScrollBar);
  }

  protected void createScrollView() {
    this.scrollView = new CustomHorizontalScrollView(context, tableView);
    LinearLayout.LayoutParams scrollLayoutParams = new LinearLayout.LayoutParams(ScrollView.LayoutParams.MATCH_PARENT, ScrollView.LayoutParams.MATCH_PARENT);
    scrollLayoutParams.bottomMargin = TableTheme.DefaultRowHeight;

    this.scrollView.setLayoutParams(scrollLayoutParams);
    this.scrollView.setFillViewport(true);

    createRootLayout();
  }

  protected void createRootLayout() {
    this.rootLayout = new RootLayout(context);
    this.rootLayout.setColumnWidths(columnWidths);
    this.rootLayout.setBackgroundColor(tableView.instanceBackgroundColor);
    this.rootLayout.setPadding(0,0, TableTheme.HorizontalScrollViewPadding, 0);
    this.rootLayout.addView(dragBox);
    this.rootLayout.setZ(PixelUtils.dpToPx(1));

    createHeaderView();
  }

  protected void createHeaderView() {
    headerView = headerViewFactory.getHeaderView();
    rootLayout.addView(headerView);

    createTotalsView();
  }

  protected void createTotalsView() {
    if(!tableView.isDataView) {
      totalsView = headerViewFactory.getTotalsView();
      if (totalsView != null) {
        rootLayout.addView(totalsView);
      }
    }

    createRecyclerViews();
  }

  public void createTotalsViewIfNeeded() {
    if (totalsView != null || tableView.isDataView || rootLayout == null) {
      return;
    }
    if (dataProvider.totalsCells == null || dataProvider.dataColumns == null) {
      return;
    }
    if (tableView.headerContentStyle == null || tableView.cellContentStyle == null) {
      return;
    }
    if (tableView.totalsPosition == null || tableView.totalsPosition.equals("noTotals")) {
      return;
    }

    this.totalsCells = dataProvider.totalsCells;
    this.totalsPosition = dataProvider.totalsPosition;
    this.dataColumns = dataProvider.dataColumns;

    headerViewFactory = new HeaderViewFactory(
      dataProvider.dataColumns,
      dataProvider.totalsCells,
      dataProvider.totalsLabel,
      tableView.totalsPosition,
      tableView,
      tableView.headerContentStyle,
      tableView.getContext()
    );

    totalsView = headerViewFactory.getTotalsView();
    if (totalsView != null) {
      rootLayout.addView(totalsView);

      if (tableView.isFirstColumnFrozen && totalsCells != null && totalsCells.size() > 0) {
        firstColumnTotalsCell = HeaderViewFactory.buildFixedTotalsCell(
          tableView, dataColumns.get(0), totalsCells.get(0), headerViewFactory.topPosition);
        tableView.addView(firstColumnTotalsCell);

        if (grabbers != null) {
          for (GrabberView view : grabbers) {
            view.setFixedTotalsCell(firstColumnTotalsCell);
          }
        }
      }

      updateTotalsViewHeight();
      updateRecyclerViewMargins();
    }
  }

  protected void createRecyclerViews() {
    CustomLinearLayoutManger linearLayout = new CustomLinearLayoutManger(context);
    linearLayout.recyclerView = coupledRecyclerView;

    coupledRecyclerView = new CustomRecyclerView(context, false, dataProvider, tableView, linearLayout, dragBox, firstColumnDragBox);
    coupledRecyclerView.setBackgroundColor(tableView.instanceBackgroundColor);
    FrameLayout.LayoutParams recyclerViewLayoutParams = new FrameLayout.LayoutParams(ViewGroup.LayoutParams.MATCH_PARENT, ViewGroup.LayoutParams.MATCH_PARENT);
    coupledRecyclerView.setLayoutParams(recyclerViewLayoutParams);
    coupledRecyclerView.setZ(0);
    coupledRecyclerView.setElevation(0);
    coupledRecyclerView.setAdapter(dataProvider);
    rootLayout.addView(coupledRecyclerView, recyclerViewLayoutParams);

    CustomLinearLayoutManger firstColumnLinearLayout = new CustomLinearLayoutManger(context);
    firstColumnRecyclerView = new CustomRecyclerView(context, true, dataProvider, tableView, firstColumnLinearLayout, dragBox, firstColumnDragBox);
    firstColumnRecyclerView.setBackgroundColor(tableView.instanceBackgroundColor);
    firstColumnLinearLayout.recyclerView = firstColumnRecyclerView;
    firstColumnRecyclerView.setAdapter(dataProvider);
    FrameLayout.LayoutParams firstColumnViewLayoutParams = new FrameLayout.LayoutParams(ViewGroup.LayoutParams.WRAP_CONTENT, ViewGroup.LayoutParams.MATCH_PARENT);
    firstColumnRecyclerView.setLayoutParams(firstColumnViewLayoutParams);

    if(tableView.isFirstColumnFrozen) {
      firstColumnHeaderCell = HeaderViewFactory.buildFixedColumnCell(rootLayout, dataColumns.get(0), tableView, headerViewFactory.topPosition);
      dataProvider.setFirstColumnFrozen(true);
      coupledRecyclerView.setViewToScrollCouple(firstColumnRecyclerView);
      firstColumnRecyclerView.setViewToScrollCouple(coupledRecyclerView);
      firstColumnRecyclerView.setZ(1);
      if(totalsCells != null && totalsCells.size() > 0 && !tableView.isDataView) {
        firstColumnTotalsCell = HeaderViewFactory.buildFixedTotalsCell(tableView, dataColumns.get(0), totalsCells.get(0), headerViewFactory.topPosition);
        tableView.addView(firstColumnTotalsCell);
      }
      tableView.addView(firstColumnRecyclerView);
      tableView.addView(firstColumnHeaderCell);
      tableView.addView(firstColumnDragBox);
    }

    createRowCount();
  }

  protected void createRowCount() {
    rowCountView = new RowCountView(context, tableView);
    coupledRecyclerView.setRowCountView(rowCountView);
    tableView.addView(rowCountView);

    createGrabbers();
  }

  protected void createGrabbers() {
    if (grabbers == null) {
      grabbers = new ArrayList<>();
      List<DataColumn> dataColumns = dataProvider.getDataColumns();
      int dragWidth = (int) PixelUtils.dpToPx(40);
      int offset = dragWidth / 2;
      int startOffset = 0;

      for (int i = 0; i < dataColumns.size(); i++) {
        GrabberView grabberView = new GrabberView(i, context, scrollView, tableView);
        FrameLayout.LayoutParams layoutParams = new FrameLayout.LayoutParams(dragWidth, ViewGroup.LayoutParams.MATCH_PARENT);
        startOffset += (dataColumns.get(i).width) - offset;
        offset = 0;

        grabberView.setBackgroundColor(Color.TRANSPARENT);
        grabberView.setTranslationX(startOffset);
        grabbers.add(grabberView);
        if (tableView.isFirstColumnFrozen && i == 0) {
          layoutParams.bottomMargin = TableTheme.DefaultRowHeight;
          tableView.addView(grabberView, layoutParams);
          grabberView.setElevation(PixelUtils.dpToPx(19));
        } else {
          rootLayout.addView(grabberView, layoutParams);
        }
      }
    }

    setupGrabbers();
  }

  protected void createMoreGrabbers(int n) {
      List<DataColumn> dataColumns = dataProvider.getDataColumns();
      int dragWidth = (int) PixelUtils.dpToPx(40);
      int startOffset = (int) grabbers.get(grabbers.size() - 1).getTranslationX();

      for (int i = 0; i < n; i++) {
        DataColumn column = dataColumns.get(grabbers.size() + i);
        GrabberView grabberView = new GrabberView(column.columnIndex, context, scrollView, tableView);
        FrameLayout.LayoutParams layoutParams = new FrameLayout.LayoutParams(dragWidth, ViewGroup.LayoutParams.MATCH_PARENT);
        startOffset += column.width;

        grabberView.setLayoutParams(layoutParams);
        grabberView.setBackgroundColor(Color.TRANSPARENT);
        grabberView.setTranslationX(startOffset);
        grabbers.add(grabberView);
        grabberView.setGrabbers(grabbers);
        rootLayout.addView(grabberView);
      }

    setupGrabbers();
  }

  protected void setupGrabbers() {
    for (GrabberView view : grabbers) {
      view.setDataProvider(dataProvider);
      view.setHeaderView(headerView);
      view.setGrabbers(grabbers);
      view.setRecyclerView(coupledRecyclerView);
      view.rootLayout = this.rootLayout;
      view.setFirstColumnRecyclerView(firstColumnRecyclerView);
      view.setFirstColumnHeader(firstColumnHeaderCell);
      view.setFixedTotalsCell(firstColumnTotalsCell);
      view.updateLayout();
    }
  }

  void invalidateLayout() {
    if (this.coupledRecyclerView != null) {
      int width = dataColumns.get(0).width;

      if (this.firstColumnHeaderCell != null) {
        ViewGroup.LayoutParams params = firstColumnHeaderCell.getLayoutParams();
        params.width = width;
        firstColumnHeaderCell.setLayoutParams(params);
        firstColumnHeaderCell.requestLayout();
      }
      if(this.firstColumnTotalsCell != null) {
        ViewGroup.LayoutParams params = firstColumnTotalsCell.getLayoutParams();
        params.width = width;
        firstColumnTotalsCell.setLayoutParams(params);
        firstColumnTotalsCell.requestLayout();
      }
      if(this.totalsView != null) {
        this.totalsView.updateLayout();
        this.totalsView.requestLayout();
      }

      this.headerView.updateLayout();
      this.headerView.requestLayout();
      this.dataProvider.invalidateLayout();

      updateRecyclerViewMargins();

      updateScrollbarBounds();
      updateGrabbers();

      this.rootLayout.requestLayout();
      this.scrollView.requestLayout();
    }
  }

  private void createHeaderFactory() {
    if(this.tableView.isDataView) {
      tableView.headerContentStyle.color = Color.BLACK;
    }
    this.headerViewFactory = new HeaderViewFactory(dataColumns, totalsCells, dataProvider.totalsLabel, tableView.totalsPosition, tableView, tableView.headerContentStyle, context);
  }

  private void updateRecyclerViewMargins() {
    int extraTop = tableView.totalsPosition.equals("top") ? tableView.totalsHeight : 0;
    int extraBottom = tableView.totalsPosition.equals("bottom") ? tableView.totalsHeight : 0;
    if(tableView.isDataView) {
      extraTop = extraBottom = 0;
    }
    int marginTop = tableView.headerHeight + extraTop;

    FrameLayout.LayoutParams recyclerViewLayoutParams = (FrameLayout.LayoutParams) coupledRecyclerView.getLayoutParams();
    recyclerViewLayoutParams.topMargin = marginTop;
    recyclerViewLayoutParams.bottomMargin = extraBottom;

    coupledRecyclerView.setLayoutParams(recyclerViewLayoutParams);

    if (this.firstColumnRecyclerView != null) {
      FrameLayout.LayoutParams firstColumnRecyclerViewLayoutParams = (FrameLayout.LayoutParams) firstColumnRecyclerView.getLayoutParams();
      firstColumnRecyclerViewLayoutParams.topMargin = marginTop;
      firstColumnRecyclerViewLayoutParams.bottomMargin = TableTheme.DefaultRowHeight + extraBottom;
      firstColumnRecyclerView.setLayoutParams(firstColumnRecyclerViewLayoutParams);
      this.firstColumnRecyclerView.requestLayout();
    }
    this.coupledRecyclerView.requestLayout();
  }

  public void updateGrabbers() {
    if (grabbers != null) {
      int maxLineHeight = headerView.getMaxLineCount();
      int headerHeight = maxLineHeight * tableView.headerContentStyle.lineHeight + CellView.PADDING_X_2;
      List<DataColumn> dataColumns = dataProvider.getDataColumns();
      int dragWidth = (int) PixelUtils.dpToPx(40);
      int offset = dragWidth / 2;
      int startOffset = 0;

      for (int i = 0; i < dataColumns.size(); i++) {
        GrabberView grabberView = grabbers.get(i);
        startOffset += (dataColumns.get(i).width) - offset;
        offset = 0;
        grabberView.setTranslationX(startOffset);
        grabberView.setHeaderHeight(headerHeight);
        grabberView.postInvalidate();
      }
    }
  }

  public void updateHeaderViewLineCount() {
    int maxLineCount = headerView.getMaxLineCount();
    int headerHeight = (maxLineCount * tableView.headerContentStyle.lineHeight) + (CellView.PADDING_X_2);
    tableView.headerHeight = headerHeight;

    updateTotalsViewHeight();
    updateFirstColumnsHeights();

    ViewGroup.LayoutParams params = headerView.getLayoutParams();
    params.height = headerHeight;
    headerView.setLayoutParams(params);
    FrameLayout.LayoutParams recyclerParams = (FrameLayout.LayoutParams) coupledRecyclerView.getLayoutParams();
    recyclerParams.topMargin = headerHeight;

    if(headerViewFactory.topPosition && totalsView != null) {
      recyclerParams.topMargin += tableView.totalsHeight;
    } else if(headerViewFactory.bottomPosition && totalsView != null) {
      recyclerParams.bottomMargin = tableView.totalsHeight;
    }
    coupledRecyclerView.setLayoutParams(recyclerParams);

    tableView.post(new Runnable() {
      @Override
      public void run() {
        if(firstColumnRecyclerView != null) {
          firstColumnRecyclerView.requestLayout();
        }
        if(firstColumnHeaderCell != null) {
          firstColumnHeaderCell.requestLayout();
        }

        if(totalsView != null) {
          totalsView.requestLayout();
        }

        coupledRecyclerView.requestLayout();
        rootLayout.requestLayout();
        headerView.requestLayout();
        tableView.requestLayout();
      }
    });
  }

  public void updateFirstColumnsHeights() {
    if(firstColumnRecyclerView != null && tableView != null) {
      FrameLayout.LayoutParams dd = (FrameLayout.LayoutParams) firstColumnRecyclerView.getLayoutParams();
      if(dd != null) {
        dd.topMargin = tableView.headerHeight;
        if(headerViewFactory.topPosition && totalsView != null) {
          dd.topMargin += tableView.totalsHeight;
        } else if(headerViewFactory.bottomPosition && totalsView != null) {
          dd.bottomMargin = TableTheme.DefaultRowHeight + tableView.totalsHeight;
        }
        firstColumnRecyclerView.setLayoutParams(dd);
        updateFirstColumnHeaderHeight();
      }
    }
  }

  public void updateFirstColumnHeaderHeight() {
    if(firstColumnHeaderCell != null) {
      ViewGroup.LayoutParams params = firstColumnHeaderCell.getLayoutParams();
      params.height = tableView.headerHeight;
      firstColumnHeaderCell.setLayoutParams(params);
      if(tableView.headerContentStyle.wrap) {
        firstColumnHeaderCell.setMaxLines(tableView.headerHeight / tableView.headerContentStyle.lineHeight);
      }
    }
  }

  public void updateRecyclerViewLineCount(DataColumn column) {
    tableView.post(new Runnable() {
      @Override
      public void run() {
        coupledRecyclerView.updateLineHeight(column);
        coupledRecyclerView.requestLayout();
        rootLayout.requestLayout();
      }
    });
  }

  public void updateTotalsViewHeight() {
    if(totalsView != null) {
      int maxLineCount = totalsView.getMaxLineCount();
      int totalsViewHeight = (maxLineCount * tableView.cellContentStyle.lineHeight) + (CellView.PADDING_X_2);
      tableView.totalsHeight = totalsViewHeight;

      FrameLayout.LayoutParams pp = (FrameLayout.LayoutParams)totalsView.getLayoutParams();
      if(headerViewFactory.topPosition) {
        pp.topMargin = tableView.headerHeight;
      }
      pp.height = totalsViewHeight;
      totalsView.setLayoutParams(pp);

      if(firstColumnTotalsCell != null) {
        FrameLayout.LayoutParams fp = (FrameLayout.LayoutParams) firstColumnTotalsCell.getLayoutParams();
        fp.height = totalsViewHeight;
        if(headerViewFactory.topPosition) {
          fp.topMargin = tableView.headerHeight;
        } else {
          fp.bottomMargin = TableTheme.DefaultRowHeight;
        }

        firstColumnTotalsCell.setLayoutParams(fp);
        if(tableView.headerContentStyle.wrap) {
          firstColumnTotalsCell.setMaxLines(totalsViewHeight / tableView.cellContentStyle.lineHeight);
        }
      }

      tableView.post(new Runnable() {
        @Override
        public void run() {
          totalsView.requestLayout();
          if(firstColumnTotalsCell != null) {
            firstColumnTotalsCell.requestLayout();
          }
        }
      });
    }
  }

  private  void testTextWrap() {
    if(firstColumnRecyclerView != null) {
      firstColumnRecyclerView.testTextWrap(true);
    }
    if(coupledRecyclerView != null) {
      coupledRecyclerView.testTextWrap(true);
    }
  }
}