fix benchmarks by exposing memoization utils

packages/slate-html-serializer/benchmark/index.js

@@ -2,7 +2,7 @@
 
 import fs from 'fs'
 import { basename, extname, resolve } from 'path'
-import { __clear } from '../../slate/lib/utils/memoize'
+import { resetMemoization } from 'slate'
 
 /**
  * Benchmarks.
@@ -20,7 +20,7 @@ categories.forEach(category => {
 
     if (category == 'models') {
       after(() => {
-        __clear()
+        resetMemoization()
       })
     }
 

packages/slate-plain-serializer/benchmark/index.js

@@ -2,7 +2,7 @@
 
 import fs from 'fs'
 import { basename, extname, resolve } from 'path'
-import { __clear } from '../../slate/lib/utils/memoize'
+import { resetMemoization } from 'slate'
 
 /**
  * Benchmarks.
@@ -20,7 +20,7 @@ categories.forEach(category => {
 
     if (category == 'models') {
       after(() => {
-        __clear()
+        resetMemoization()
       })
     }
 

packages/slate/benchmark/changes/insert-text-by-key-multiple.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function({ change, keys }) {
   for (const key of keys) {
@@ -16,7 +16,7 @@ export function before(value) {
     .getTexts()
     .toArray()
     .map(t => t.key)
-  __clear()
+  resetMemoization()
   return { change, keys }
 }
 

packages/slate/benchmark/changes/insert-text-by-key.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function({ change, text }) {
   change.insertTextByKey(text.key, 0, 'a')
@@ -11,7 +11,7 @@ export default function({ change, text }) {
 export function before(value) {
   const change = value.change()
   const text = value.document.getLastText()
-  __clear()
+  resetMemoization()
   return { change, text }
 }
 

packages/slate/benchmark/index.js

@@ -2,7 +2,7 @@
 
 import fs from 'fs'
 import { basename, extname, resolve } from 'path'
-import { __clear } from '../lib/utils/memoize'
+import { resetMemoization } from '..'
 
 /**
  * Benchmarks.
@@ -20,7 +20,7 @@ categories.forEach(category => {
 
     if (category == 'models') {
       after(() => {
-        __clear()
+        resetMemoization()
       })
     }
 

packages/slate/benchmark/models/get-leaves.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function(text) {
   text.getLeaves()
@@ -10,7 +10,7 @@ export default function(text) {
 
 export function before(value) {
   const text = value.document.getFirstText()
-  __clear()
+  resetMemoization()
   return text
 }
 

packages/slate/benchmark/models/get-path.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function({ value, text }) {
   value.document.getPath(text.key)
@@ -10,7 +10,7 @@ export default function({ value, text }) {
 
 export function before(value) {
   const text = value.document.getLastText()
-  __clear()
+  resetMemoization()
   return { value, text }
 }
 

packages/slate/benchmark/models/has-node-multiple.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function({ value, keys }) {
   keys.forEach(key => {
@@ -15,7 +15,7 @@ export function before(value) {
     .getTexts()
     .toArray()
     .map(t => t.key)
-  __clear()
+  resetMemoization()
   return { value, keys }
 }
 

packages/slate/benchmark/models/has-node.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function({ value, text }) {
   value.document.hasNode(text.key)
@@ -10,7 +10,7 @@ export default function({ value, text }) {
 
 export function before(value) {
   const text = value.document.getLastText()
-  __clear()
+  resetMemoization()
   return { value, text }
 }
 

packages/slate/benchmark/models/update-node.js

@@ -2,7 +2,7 @@
 /* eslint-disable react/jsx-key */
 
 import h from '../../test/helpers/h'
-import { __clear } from '../../lib/utils/memoize'
+import { resetMemoization } from '../..'
 
 export default function({ value, next }) {
   value.document.updateNode(next)
@@ -13,7 +13,7 @@ export function before(value) {
   const { size } = texts
   const text = texts.get(Math.round(size / 2))
   const next = text.insertText(0, 'some text')
-  __clear()
+  resetMemoization()
   return { value, next }
 }
 

packages/slate/src/index.js

@@ -17,6 +17,7 @@ import Stack from './models/stack'
 import Text from './models/text'
 import Value from './models/value'
 import { resetKeyGenerator, setKeyGenerator } from './utils/generate-key'
+import { resetMemoization, useMemoization } from './utils/memoize'
 
 /**
  * Export.
@@ -45,6 +46,8 @@ export {
   Value,
   resetKeyGenerator,
   setKeyGenerator,
+  resetMemoization,
+  useMemoization,
 }
 
 export default {
@@ -67,4 +70,6 @@ export default {
   Value,
   resetKeyGenerator,
   setKeyGenerator,
+  resetMemoization,
+  useMemoization,
 }

packages/slate/src/utils/generate-key.js

@@ -55,4 +55,5 @@ resetKeyGenerator()
  * @type {Object}
  */
 
-export { generateKey as default, setKeyGenerator, resetKeyGenerator }
+export default generateKey
+export { setKeyGenerator, resetKeyGenerator }

packages/slate/src/utils/memoize.js

@@ -172,7 +172,7 @@ function setIn(map, keys, value) {
  * @return {Void}
  */
 
-function __clear() {
+function resetMemoization() {
   CACHE_KEY++
 
   if (CACHE_KEY >= Number.MAX_SAFE_INTEGER) {
@@ -187,7 +187,7 @@ function __clear() {
  * @return {Void}
  */
 
-function __enable(enabled) {
+function useMemoization(enabled) {
   ENABLED = enabled
 }
 
@@ -197,4 +197,5 @@ function __enable(enabled) {
  * @type {Object}
  */
 
-export { memoize as default, __clear, __enable }
+export default memoize
+export { resetMemoization, useMemoization }