import { z } from 'zod';
import { BaseConnector, html as H, type ConnectorMeta, type CrawlContext, type RawRecordInput, type RawRecordLike } from '@rareindex/connectors';
import { AssetAttributesSchema, NormalizedCatalogItemSchema, NormalizedListingSchema, NormalizedPriceObservationSchema, type NormalizedRecord } from '@rareindex/shared';
import { fetchRawHtml, sneakerCategory, styleCodeFromText } from '../../api/_luxury-lib/index.js';

/** Flight Club — search grid from __NEXT_DATA__ (lowest ask + retail, USD). */
const BASE = 'https://www.flightclub.com';
const PARSER_VERSION = '1.0.0';

export const ItemSchema = z.object({ id: z.string(), name: z.string(), brand: z.string().nullable(), image: z.string().nullable(), price: z.number().nullable(), retail: z.number().nullable(), slug: z.string() });
export const PagePayloadSchema = z.object({ kind: z.literal('search_page'), url: z.string(), seed: z.string(), page: z.number(), currency: z.string(), total: z.number().nullable(), items: z.array(ItemSchema) });
export type PagePayload = z.infer<typeof PagePayloadSchema>;

export function parseSearchPage(htmlText: string, url: string, seed: string, page: number): PagePayload | null {
  const nd = H.nextData(htmlText) as { props?: { pageProps?: { consumerSearchResult?: { data?: Array<Record<string, unknown>>; total?: number }; currency?: string; availableCurrencies?: unknown[] } } } | null;
  const r = nd?.props?.pageProps?.consumerSearchResult;
  if (!r?.data) return null;
  const items: z.infer<typeof ItemSchema>[] = [];
  for (const d of r.data) {
    const id = d.id !== undefined ? String(d.id) : null;
    const slug = typeof d.slug === 'string' ? d.slug : null;
    if (!id || !slug) continue;
    const priceObj = d.price as { localizedValue?: number } | undefined;
    const retailObj = d.retailPrice as { localizedValue?: number } | undefined;
    items.push({ id, name: String(d.name ?? '').trim(), brand: d.brandName ? String(d.brandName) : null, image: d.pictureUrl ? String(d.pictureUrl) : null, price: typeof priceObj?.localizedValue === 'number' && priceObj.localizedValue > 0 ? priceObj.localizedValue : null, retail: typeof retailObj?.localizedValue === 'number' && retailObj.localizedValue > 0 ? retailObj.localizedValue : null, slug });
  }
  return { kind: 'search_page', url, seed, page, currency: String(nd?.props?.pageProps?.currency ?? 'USD'), total: typeof r.total === 'number' ? r.total : null, items };
}

export class FlightClubConnector extends BaseConnector {
  readonly version = '1.0.0';
  readonly parserVersion = PARSER_VERSION;
  protected override minIntervalMs = 2000;

  async *crawl(ctx: CrawlContext): AsyncIterable<RawRecordInput> {
    const seeds = (ctx.options.seeds?.length ? ctx.options.seeds : (this.meta.config.seeds as string[] | undefined)) ?? [];
    const pages = ctx.options.mode === 'backfill' ? 10 : Number(this.meta.config.pagesPerSeed ?? 2);
    let count = 0;
    for (const seed of seeds) {
      for (let page = 1; page <= pages; page++) {
        if (ctx.signal?.aborted || this.reached(ctx, count)) return;
        const url = `${BASE}/${seed}${page > 1 ? `?page=${page}` : ''}`;
        await this.throttle();
        // __NEXT_DATA__ lives in a <script>; the shared engine's `html` format strips scripts, so ask Firecrawl for rawHtml.
        let res = await fetchRawHtml(ctx, url);
        let payload = res.success && res.html ? parseSearchPage(res.html, url, seed, page) : null;
        if (!payload) {
          res = await ctx.fetch(url, { engines: ['scrapfly'], expect: ['title', 'price', 'identifiers'], parse: (r) => {
            const first = r.html ? parseSearchPage(r.html, url, seed, page)?.items[0] : undefined;
            return first ? { title: first.name, price: first.price, identifiers: { slug: first.slug } } : null;
          } });
          payload = res.success && res.html ? parseSearchPage(res.html, url, seed, page) : null;
        }
        if (!payload || payload.items.length === 0) {
          if (page === 1) ctx.anomaly(payload ? 'empty_page' : 'page_fetch_failed', `${url}: ${res.error ?? res.httpStatus}`);
          break;
        }
        count++;
        yield { url, externalId: `search:${seed}:${page}`, kind: 'catalog_item', engine: res.engine, httpStatus: res.httpStatus, payload, fetchedAt: res.fetchedAt };
      }
    }
  }

  async normalize(raw: RawRecordLike): Promise<NormalizedRecord[]> {
    const p = PagePayloadSchema.parse(raw.payload);
    const out: NormalizedRecord[] = [];
    const currency = p.currency === 'USD' ? 'USD' : 'USD';
    for (const it of p.items) {
      const styleCode = styleCodeFromText(it.slug.split('-').slice(-2).join('-')) ?? styleCodeFromText(it.slug);
      const year = it.name.match(/\b(20\d{2}|19\d{2})\b\s*$/)?.[1];
      const attributes = AssetAttributesSchema.parse({
        categorySlug: sneakerCategory(it.brand, it.name),
        brand: it.brand,
        name: it.name.replace(/\s+\d{4}$/, '').trim(),
        year: year ? Number(year) : null,
        originalMsrp: it.retail,
        originalMsrpCurrency: it.retail ? currency : null,
        identifiers: { ...(styleCode ? { style_code: styleCode } : {}), flightclub_id: it.id },
        metadata: { slug: it.slug },
      });
      const sourceUrl = `${BASE}/${it.slug}`;
      const base = { connectorId: this.meta.id, sourceId: this.meta.sourceId, sourceUrl, rawTitle: it.name, imageUrls: it.image ? [it.image] : [], attributes, observedAt: raw.fetchedAt, parserVersion: PARSER_VERSION };
      const cond = { condition: 'new', conditionRaw: 'New (Flight Club consignment)', completeness: 'with_box' };
      out.push(NormalizedCatalogItemSchema.parse({ kind: 'catalog_item', ...base, externalId: `product:${it.id}`, confidence: 0.85 }));
      if (it.price) {
        out.push(NormalizedPriceObservationSchema.parse({ kind: 'price_observation', ...base, externalId: `product:${it.id}:low`, confidence: 0.7, condition: cond, priceKind: 'low', price: it.price, currency, observationDate: raw.fetchedAt }));
        out.push(NormalizedListingSchema.parse({ kind: 'listing', ...base, externalId: `product:${it.id}:lowest_ask`, confidence: 0.7, condition: cond, listingType: 'ask', price: it.price, currency, seller: 'Flight Club', location: 'US', availability: 'available' }));
      }
    }
    return out;
  }
}

export default function createConnector(meta: ConnectorMeta) {
  return new FlightClubConnector(meta);
}