feat: persistence

simoneb · Dec 31, 2020 · d4a1880 · d4a1880
1 parent d286ac5
commit d4a1880
Show file tree

Hide file tree

Showing 11 changed files with 197 additions and 23 deletions.
diff --git a/examples/.eslintrc b/examples/.eslintrc
@@ -12,6 +12,7 @@
     "Source": true,
     "ControlledDeclarative": true,
     "ControlledImperative": true,
-    "Uncontrolled": true
+    "Uncontrolled": true,
+    "Persistence": true
   }
 }
diff --git a/examples/components/Example.js b/examples/components/Example.js
@@ -2,6 +2,7 @@ const choices = [
   ControlledDeclarative,
   ControlledImperative,
   Uncontrolled,
+  Persistence,
 ].reduce((acc, c) => ({ ...acc, [c.name]: c }), {})
 
 function Example() {

diff --git a/examples/components/Persistence.js b/examples/components/Persistence.js
@@ -0,0 +1,60 @@
+const { useState, useEffect } = React
+
+function Persistence() {
+  const {
+    currentToken,
+    hasToken,
+    useUpdateToken,
+    changePageNumber,
+    changePageSize,
+    pageNumber,
+    pageSize,
+  } = useTokenPagination({
+    defaultPageNumber: 1,
+    defaultPageSize: 5,
+    persister: useTokenPagination.localPersister('persistence'),
+  })
+  const [data, setData] = useState()
+
+  useEffect(() => {
+    async function fetchData() {
+      const params = new URLSearchParams({ pageSize })
+
+      if (currentToken) {
+        params.append('pageToken', currentToken)
+      }
+
+      const res = await fetch(`/api?${params.toString()}`)
+      const data = await res.json()
+
+      setData(data)
+    }
+
+    fetchData()
+  }, [pageSize, currentToken])
+
+  useUpdateToken(data?.nextPage)
+
+  function previousPage() {
+    changePageNumber(n => n - 1)
+  }
+  function nextPage() {
+    changePageNumber(n => n + 1)
+  }
+  function handleChangePageSize(e) {
+    changePageSize(+e.target.value)
+  }
+
+  return (
+    <Output
+      data={data}
+      pageNumber={pageNumber}
+      pageSize={pageSize}
+      changePageSize={handleChangePageSize}
+      previousPage={previousPage}
+      nextPage={hasToken(pageNumber + 1) ? nextPage : undefined}
+    />
+  )
+}
+
+Persistence.propTypes = {}
diff --git a/examples/index.html b/examples/index.html
@@ -16,6 +16,7 @@
     ></script>
     <script type="text/babel" src="components/ControlledImperative.js"></script>
     <script type="text/babel" src="components/Uncontrolled.js"></script>
+    <script type="text/babel" src="components/Persistence.js"></script>
     <script type="text/babel" src="components/Example.js"></script>
   </head>
   <body>

diff --git a/jest.config.js b/jest.config.js
@@ -14,7 +14,7 @@ module.exports = {
   // cacheDirectory: "C:\\Users\\simone\\AppData\\Local\\Temp\\jest",
 
   // Automatically clear mock calls and instances between every test
-  clearMocks: true,
+  // clearMocks: true,
 
   // Indicates whether the coverage information should be collected while executing the test
   // collectCoverage: false,

diff --git a/rollup.config.js b/rollup.config.js
@@ -5,6 +5,7 @@ export default {
       dir: 'cjs',
       format: 'cjs',
       exports: 'default',
+      esModule: false,
     },
     {
       dir: 'es',

diff --git a/src/controlled.js b/src/controlled.js
@@ -1,10 +1,20 @@
 import { useCallback, useEffect, useState, useMemo } from 'react'
+import { NULL_PERSISTER } from './persisters'
 import { assertNumber } from './utils'
 
-export default function useControlledTokenPagination(pageNumber) {
+const DEFAULTS = {
+  persister: NULL_PERSISTER,
+}
+
+export default function useControlledTokenPagination(pageNumber, options) {
+  options = { ...DEFAULTS, ...options }
+
   assertNumber('pageNumber', pageNumber)
 
-  const [mapping, setMapping] = useState({})
+  const [mapping, setMapping] = useState(() => {
+    const { mapping } = options.persister.hydrate()
+    return mapping || {}
+  })
 
   const updateToken = useCallback(
     nextToken => {
@@ -23,6 +33,10 @@ export default function useControlledTokenPagination(pageNumber) {
     [updateToken]
   )
 
+  useEffect(() => {
+    options.persister.persist({ mapping })
+  }, [options.persister, mapping])
+
   return useMemo(
     () => ({
       currentToken: mapping[pageNumber],

diff --git a/src/index.js b/src/index.js
@@ -1,5 +1,6 @@
 import useControlledTokenPagination from './controlled'
 import useUncontrolledTokenPagination from './uncontrolled'
+import * as persisters from './persisters'
 
 const variants = {
   number: useControlledTokenPagination,
@@ -15,3 +16,5 @@ export default function useTokenPagination(options) {
 
   return variant(options)
 }
+
+Object.assign(useTokenPagination, persisters)
diff --git a/src/persisters.js b/src/persisters.js
@@ -0,0 +1,26 @@
+export const NULL_PERSISTER = {
+  hydrate() {
+    return {}
+  },
+  persist() {},
+}
+
+function StoragePersister(key, storage) {
+  return {
+    hydrate() {
+      try {
+        return JSON.parse(storage.getItem(key) || '{}')
+      } catch (err) {
+        return {}
+      }
+    },
+    persist(value) {
+      storage.setItem(key, JSON.stringify({ ...this.hydrate(), ...value }))
+    },
+  }
+}
+
+export const localPersister = key =>
+  new StoragePersister(key, window.localStorage)
+export const sessionPersister = key =>
+  new StoragePersister(key, window.sessionStorage)
diff --git a/src/persisters.test.js b/src/persisters.test.js
@@ -0,0 +1,54 @@
+import { localPersister, sessionPersister } from './persisters'
+
+describe('persisters', () => {
+  describe('local', persisterTests(localPersister, 'localStorage'))
+
+  describe('session', persisterTests(sessionPersister, 'sessionStorage'))
+})
+
+function persisterTests(persisterFactory, storageName) {
+  return function () {
+    let persister, storage
+
+    beforeEach(() => {
+      Object.defineProperty(window, storageName, {
+        value: { setItem: jest.fn(), getItem: jest.fn() },
+      })
+      storage = window[storageName]
+      persister = persisterFactory('key')
+    })
+
+    it('persists', () => {
+      persister.persist({ some: 'value' })
+
+      expect(storage.setItem).toHaveBeenCalledWith(
+        'key',
+        JSON.stringify({ some: 'value' })
+      )
+    })
+
+    it('hydrates', () => {
+      const stored = { some: 'value' }
+      storage.getItem.mockReturnValue(JSON.stringify(stored))
+
+      expect(persister.hydrate()).toEqual(stored)
+    })
+
+    it('merges with existing value when persisting', () => {
+      const stored = { some: 'value' }
+      storage.getItem.mockReturnValue(JSON.stringify(stored))
+
+      persister.persist({ another: 'value' })
+
+      expect(storage.setItem).toHaveBeenCalledWith(
+        'key',
+        JSON.stringify({ some: 'value', another: 'value' })
+      )
+    })
+
+    it('handles errors', () => {
+      storage.getItem.mockReturnValue('invalid json')
+      expect(persister.hydrate()).toEqual({})
+    })
+  }
+}
diff --git a/src/uncontrolled.js b/src/uncontrolled.js
@@ -1,54 +1,67 @@
-import { useCallback, useState, useMemo } from 'react'
+import { useCallback, useState, useMemo, useEffect } from 'react'
 import useControlledTokenPagination from './controlled'
+import { NULL_PERSISTER } from './persisters'
 import { assertNumber } from './utils'
 
 const DEFAULTS = {
   defaultPageNumber: 1,
   resetPageNumberOnPageSizeChange: true,
+  persister: NULL_PERSISTER,
 }
 
+const changerTypes = ['function', 'number']
+
 export default function useUncontrolledTokenPagination(options) {
   options = { ...DEFAULTS, ...options }
 
   assertNumber('defaultPageNumber', options.defaultPageNumber)
   assertNumber('defaultPageSize', options.defaultPageSize)
 
-  const [{ pageNumber, pageSize }, setPagination] = useState({
-    pageNumber: options.defaultPageNumber,
-    pageSize: options.defaultPageSize,
+  const [{ pageNumber, pageSize }, setPagination] = useState(() => {
+    const { pageNumber, pageSize } = options.persister.hydrate()
+
+    return {
+      pageNumber: pageNumber || options.defaultPageNumber,
+      pageSize: pageSize || options.defaultPageSize,
+    }
   })
 
   const change = useCallback(
     (property, changer) => {
       const pageNumberReset = options.resetPageNumberOnPageSizeChange
         ? { pageNumber: options.defaultPageNumber }
-        : {}
+        : null
 
       const changerType = typeof changer
 
-      if (!['function', 'number'].includes(changerType)) {
+      if (!changerTypes.includes(changerType)) {
         throw new Error(
-          `Unsupported value ${changer} of type ${changerType} for ${property}`
+          `Unsupported value ${changer} of type ${changerType} for ${property}. Supported values are ${changerTypes}`
         )
       }
 
-      const paginate = p => {
-        return {
-          ...p,
-          ...pageNumberReset,
-          [property]: changerType === 'number' ? changer : changer(p[property]),
-        }
-      }
-
-      setPagination(paginate)
+      setPagination(p => ({
+        ...p,
+        ...pageNumberReset,
+        [property]: changerType === 'function' ? changer(p[property]) : changer,
+      }))
     },
     [options.defaultPageNumber, options.resetPageNumberOnPageSizeChange]
   )
 
-  const changePageNumber = useCallback(c => change('pageNumber', c), [change])
-  const changePageSize = useCallback(c => change('pageSize', c), [change])
+  const changePageNumber = useCallback(
+    changer => change('pageNumber', changer),
+    [change]
+  )
+  const changePageSize = useCallback(changer => change('pageSize', changer), [
+    change,
+  ])
+
+  const controlled = useControlledTokenPagination(pageNumber, options)
 
-  const controlled = useControlledTokenPagination(pageNumber)
+  useEffect(() => {
+    options.persister.persist({ pageNumber, pageSize })
+  }, [options.persister, pageNumber, pageSize])
 
   return useMemo(
     () => ({