import type { MarketState } from "../../../types/financials";
import { compareSortValues, type SortDirection } from "../../../utils/sort-values";
import type { BoardQuoteMap } from "../shared/use-quote-board";
import { REGION_ORDER, type IndexEntry } from "./indices";

export type WorldIndexTableRow =
  | { type: "header"; region: IndexEntry["region"] }
  | { type: "row"; entry: IndexEntry };

export type WorldIndexColumnId =
  | "status"
  | "symbol"
  | "name"
  | "price"
  | "change"
  | "changePercent"
  | "time";

export interface WorldIndexSortPreference {
  columnId: WorldIndexColumnId | null;
  direction: SortDirection;
}

export const DEFAULT_SORT_PREFERENCE: WorldIndexSortPreference = {
  columnId: null,
  direction: "asc",
};

const MARKET_STATE_SORT_ORDER: Partial<Record<MarketState, number>> = {
  REGULAR: 0,
  PREPRE: 1,
  PRE: 1,
  POST: 1,
  POSTPOST: 1,
  CLOSED: 2,
};

function getSortValue(
  columnId: WorldIndexColumnId,
  entry: IndexEntry,
  quotes: BoardQuoteMap,
): string | number | null {
  const quote = quotes.get(entry.symbol)?.quote;

  switch (columnId) {
    case "status":
      return quote?.marketState ? (MARKET_STATE_SORT_ORDER[quote.marketState] ?? 3) : 3;
    case "symbol":
      return entry.shortName;
    case "name":
      return entry.name;
    case "price":
      return quote?.price ?? null;
    case "change":
      return quote?.change ?? null;
    case "changePercent":
      return quote?.changePercent ?? null;
    case "time":
      return quote?.lastUpdated ?? null;
  }
}

function sortEntries(
  entries: IndexEntry[],
  sortPreference: WorldIndexSortPreference,
  quotes: BoardQuoteMap,
): IndexEntry[] {
  const sortColumnId = sortPreference.columnId;
  if (!sortColumnId) return entries;
  return [...entries].sort((left, right) => compareSortValues(
    getSortValue(sortColumnId, left, quotes),
    getSortValue(sortColumnId, right, quotes),
    sortPreference.direction,
  ));
}

export function buildFlatRows(
  indicesByRegion: Map<IndexEntry["region"], IndexEntry[]>,
  sortPreference: WorldIndexSortPreference,
  quotes: BoardQuoteMap,
): WorldIndexTableRow[] {
  const rows: WorldIndexTableRow[] = [];
  for (const region of REGION_ORDER) {
    const entries = sortEntries(indicesByRegion.get(region) ?? [], sortPreference, quotes);
    if (entries.length === 0) continue;
    rows.push({ type: "header", region });
    for (const entry of entries) {
      rows.push({ type: "row", entry });
    }
  }
  return rows;
}

export function nextSortPreference(
  current: WorldIndexSortPreference,
  columnId: string,
): WorldIndexSortPreference {
  const typedColumnId = columnId as WorldIndexColumnId;
  if (current.columnId !== typedColumnId) {
    return { columnId: typedColumnId, direction: "asc" };
  }
  if (current.direction === "asc") {
    return { columnId: typedColumnId, direction: "desc" };
  }
  return DEFAULT_SORT_PREFERENCE;
}