@steambrew/client/src/class-mapper.ts

A support library for creating plugins with Millennium.

import { Module, findAllModules } from './webpack';

export interface ClassModule {
  [name: string]: string;
}

export const classMapList: ClassModule[] = findAllModules((m: Module) => {
  if (typeof m == 'object' && !m.__esModule) {
    const keys = Object.keys(m);
    // special case some libraries
    if (keys.length == 1 && m.version) return false;
    // special case localization
    if (keys.length > 1000 && m.AboutSettings) return false;

    return keys.length > 0 && keys.every((k) => !Object.getOwnPropertyDescriptor(m, k)?.get && typeof m[k] == 'string');
  }
  return false;
});

export const classMap: ClassModule = Object.assign({}, ...classMapList.map(obj =>
  Object.fromEntries(Object.entries(obj).map(([key, value]) => [key, value]))
));

export function findClass(name: string): string | void {
  return classMapList.find((m) => m?.[name])?.[name];
}

const findClassHandler = {
  get: (target: any, prop: string) => {
    if (typeof prop === "string") {
      return target(prop);
    }
  }
};

export const Classes = new Proxy(findClass, findClassHandler);

export function findClassModule(filter: (module: any) => boolean): ClassModule | void {
  return classMapList.find((m) => filter(m));
}

export function unminifyClass(minifiedClass: string): string | void {
  for (let m of classMapList) {
    for (let className of Object.keys(m)) {
      if (m[className] == minifiedClass) return className;
    }
  }
}