fix: polyfill flatMap

Author: gera2ld

src/plugins/prism.ts

@@ -1,4 +1,5 @@
 import { JSItem, CSSItem } from '../types';
+import { arrayFrom, flatMap } from '../util';
 
 export const styles: CSSItem[] = [
   {
@@ -32,9 +33,9 @@ export const scripts: JSItem[] = [
   },
 ];
 
-export function transform(nodes, mm): void {
+export function transform(nodes: HTMLElement[], mm): void {
   const { Prism } = window as any;
-  const langs = nodes.flatMap(node => Array.from(node.querySelectorAll('code[class*=language-]')))
+  const langs = flatMap(nodes, node => arrayFrom(node.querySelectorAll('code[class*=language-]')))
     .map(code => {
       const lang = code.className.match(/(?:^|\s)language-(\S+)|$/)[1];
       if (Prism.languages[lang]) {

src/util/base.ts

@@ -0,0 +1,56 @@
+const uniqId = Math.random().toString(36).slice(2, 8);
+let globalIndex = 0;
+export function getId(): string {
+  globalIndex += 1;
+  return `mm-${uniqId}-${globalIndex}`;
+}
+
+export function walkTree<T>(tree: T, callback: (item: T, next: () => void, parent?: T) => void, key = 'c'): void {
+  const walk = (item: T, parent?: T): void => callback(item, () => {
+    item[key]?.forEach((child: T) => {
+      walk(child, item);
+    });
+  }, parent);
+  walk(tree);
+}
+
+export function arrayFrom<T>(arrayLike: ArrayLike<T>): T[] {
+  if (Array.from) return Array.from(arrayLike);
+  const array = [];
+  for (let i = 0; i < arrayLike.length; i += 1) {
+    array.push(arrayLike[i]);
+  }
+  return array;
+}
+
+export function flatMap<T, U>(arrayLike: ArrayLike<T>, callback: (item?: T, index?: number, thisObj?: ArrayLike<T>) => U | U[]): U[] {
+  if ((arrayLike as Array<T>).flatMap) return (arrayLike as Array<T>).flatMap(callback);
+  const array = [];
+  for (let i = 0; i < arrayLike.length; i += 1) {
+    const result = callback(arrayLike[i], i, arrayLike);
+    if (Array.isArray(result)) array.push(...result);
+    else array.push(result);
+  }
+  return array;
+}
+
+export function addClass(className: string, ...rest: string[]): string {
+  const classList = (className || '').split(' ').filter(Boolean);
+  rest.forEach(item => {
+    if (item && classList.indexOf(item) < 0) classList.push(item);
+  });
+  return classList.join(' ');
+}
+
+export function childSelector<T extends Element>(filter?: string | ((el: T) => boolean)): () => T[] {
+  if (typeof filter === 'string') {
+    const tagName = filter;
+    filter = (el: T): boolean => el.tagName === tagName;
+  }
+  const filterFn = filter;
+  return function selector(): T[] {
+    let nodes = arrayFrom((this as HTMLElement).childNodes as NodeListOf<T>);
+    if (filterFn) nodes = nodes.filter(node => filterFn(node));
+    return nodes;
+  };
+}

src/util/index.ts

@@ -1,47 +1,3 @@
+export * from './base';
 export * from './html';
 export * from './loader';
-
-const uniqId = Math.random().toString(36).slice(2, 8);
-let globalIndex = 0;
-export function getId(): string {
-  globalIndex += 1;
-  return `mm-${uniqId}-${globalIndex}`;
-}
-
-export function walkTree<T>(tree: T, callback: (item: T, next: () => void, parent?: T) => void, key = 'c'): void {
-  const walk = (item: T, parent?: T): void => callback(item, () => {
-    item[key]?.forEach((child: T) => {
-      walk(child, item);
-    });
-  }, parent);
-  walk(tree);
-}
-
-export function arrayFrom<T>(arrayLike: ArrayLike<T>): T[] {
-  const array = [];
-  for (let i = 0; i < arrayLike.length; i += 1) {
-    array.push(arrayLike[i]);
-  }
-  return array;
-}
-
-export function addClass(className: string, ...rest: string[]): string {
-  const classList = (className || '').split(' ').filter(Boolean);
-  rest.forEach(item => {
-    if (item && classList.indexOf(item) < 0) classList.push(item);
-  });
-  return classList.join(' ');
-}
-
-export function childSelector<T extends Element>(filter?: string | ((el: T) => boolean)): () => T[] {
-  if (typeof filter === 'string') {
-    const tagName = filter;
-    filter = (el: T): boolean => el.tagName === tagName;
-  }
-  const filterFn = filter;
-  return function selector(): T[] {
-    let nodes = arrayFrom((this as HTMLElement).childNodes as NodeListOf<T>);
-    if (filterFn) nodes = nodes.filter(node => filterFn(node));
-    return nodes;
-  };
-}

src/util/loader.ts

@@ -1,5 +1,6 @@
 import { JSItem, JSScriptItem, CSSItem, IMarkmapPlugin } from '../types';
 import { escapeScript, wrapHtml } from './html';
+import { flatMap } from './base';
 
 export function buildCode(fn: Function, ...args: any[]): string {
   const params = args.map(arg => {
@@ -121,8 +122,8 @@ export function persistCSS(items: CSSItem[]): string[] {
 }
 
 export function persistPlugins(plugins: IMarkmapPlugin[], context?: any) {
-  const js = plugins.flatMap(plugin => persistJS(plugin.scripts, context)).join('');
-  const css = plugins.flatMap(plugin => persistCSS(plugin.styles)).join('');
+  const js = flatMap(plugins, plugin => persistJS(plugin.scripts, context)).join('');
+  const css = flatMap(plugins, plugin => persistCSS(plugin.styles)).join('');
   const processors = plugins.map(({ transform }) => transform);
   return {
     js,