@fruits-chain/react-native-xiaoshu/src/theme/create-style.ts

🌈 React Native UI library

import type { TokensType } from './interface'

export type ComponentVar = Record<string, any>

type Creator<T> = (v: ComponentVar, t?: TokensType) => T

type KeyType = [ComponentVar, Creator<any>]

const StyleMap: Map<KeyType, any> = new Map()

export const createStyle = <T>(
  componentVar: ComponentVar,
  creator: Creator<T>,
  tokens?: TokensType,
): T => {
  let myStyle: T

  for (let [key, value] of StyleMap) {
    if (key[1] === creator) {
      if (key[0] === componentVar) {
        myStyle = value
      } else {
        StyleMap.delete(key)
      }
    }
  }

  // @ts-ignore
  if (!myStyle) {
    myStyle = creator(componentVar, tokens)

    StyleMap.set([componentVar, creator], myStyle)
  }

  return myStyle
}