import { ViewStyle, StyleSheet } from "react-native";
import { ComponentContext, TailwindContext } from "./context";
import { matchAtRule } from "./match-at-rule";
import { normaliseSelector } from "./shared/selector";
import { AtRuleRecord, StyleArray } from "./types/common";

export interface GetStylesOptions {
  className: string;
  hover: boolean;
  focus: boolean;
  active: boolean;
  tailwindContext: TailwindContext;
  componentInteraction: ComponentContext;
}

export function getRuntimeStyles<P>({
  className,
  hover,
  focus,
  active,
  componentInteraction,
  tailwindContext,
}: GetStylesOptions): [P[], AtRuleRecord[]] {
  const { styles, media } = tailwindContext;
  const tailwindStyles: P[] = [];
  const childStyles = [] as AtRuleRecord[];
  const transforms: ViewStyle["transform"] = [];

  for (const name of className.trim().split(/\s+/)) {
    const selector = normaliseSelector(name);

    const styleArray: StyleArray = [];

    if (styles[selector]) {
      styleArray.push(styles[selector]);
    }

    if (media[selector]) {
      styleArray.push(
        ...media[selector].map((atRules, index) => ({
          ...styles[`${selector}.${index}`],
          atRules,
        }))
      );
    }

    if (styleArray.length === 0) {
      continue;
    }

    for (let styleRecord of styleArray) {
      // react-native-web <=17 still uses style IDs, so styleRecord may be a number
      if (typeof styleRecord === "object" && "atRules" in styleRecord) {
        let isForChildren = false;

        const { atRules, ...style } = styleRecord;

        const atRulesResult = atRules.every(([rule, params]) => {
          if (rule === "selector" && params === "(> *:not(:first-child))") {
            isForChildren = true;
            return true;
          } else if (rule === "selector" && params === "(> *)") {
            isForChildren = true;
            return true;
          }

          return matchAtRule({
            rule,
            params,
            hover,
            active,
            focus,
            componentInteraction,
            tailwindContext,
          });
        });

        if (isForChildren) {
          childStyles.push(styleRecord);
          continue;
        } else if (atRulesResult) {
          styleRecord = style;
        } else {
          continue;
        }
      }

      // For RWN <=0.17 we need to convert the styleID into an object to handle
      // merging transform.
      //
      // This is a performance issue, but hopefully move people will move onto
      // RWN 0.18 and use CSS
      if (typeof styleRecord === "number") {
        styleRecord = StyleSheet.flatten(styleRecord);
      }

      const { transform, ...style } = styleRecord;
      tailwindStyles.push(style as P);

      if (transform) {
        transforms.push(...transform);
      }
    }
  }

  if (transforms.length > 0) {
    // eslint-disable-next-line @typescript-eslint/no-explicit-any
    tailwindStyles.push({ transform: transforms } as any);
  }

  if (childStyles.length > 0) {
    return [tailwindStyles, childStyles];
  }

  return [tailwindStyles, []];
}