Add a useSnackbar hook.

README.md

@@ -23,7 +23,32 @@ import { SnackbarProvider } from 'material-ui-snackbar-provider'
 </SnackbarProvider>
 ```
 
-You can then display messages with the `withSnackbar` HOC and the injected `snackbar` prop in your components.
+You can then display messages with the `useSnackbar` hook.
+
+```js
+import { useSnackbar } from 'material-ui-snackbar-provider'
+
+export default function MyComponent (props) {
+  const snackbar = useSnackbar()
+
+  const handleSomething = () => {
+    snackbar.showMessage(
+      'Something happened!',
+      'Undo', () => handleUndo()
+    )
+  }
+
+  const handleUndo = () => {
+    // *snip*
+  }
+
+  return (
+    // *snip*
+  )
+}
+```
+
+If you're not using React ^16.8.0 and our you can't use hooks (e.g. in a class component), you can use the `withSnackbar` HOC and the injected `snackbar` prop instead.
 
 ```js
 import { withSnackbar } from 'material-ui-snackbar-provider'

src/index.js

@@ -1,2 +1,3 @@
 export { default as SnackbarProvider } from './SnackbarProvider'
 export { default as withSnackbar } from './withSnackbar'
+export { default as useSnackbar } from './useSnackbar'

src/useSnackbar.js

@@ -0,0 +1,6 @@
+import React from 'react'
+import SnackbarContext from './SnackbarContext'
+
+export default function useSnackbar () {
+  return React.useContext(SnackbarContext)
+}

src/useSnackbar.test.js

@@ -0,0 +1,35 @@
+/* eslint-env jest */
+import React from 'react'
+import { mount } from 'enzyme'
+import SnackbarContext from './SnackbarContext'
+import useSnackbar from './useSnackbar'
+
+describe('useSnackbar', () => {
+  it('injects the snackbar object', () => {
+    const dummySnackbarContext = {
+      showMessage: () => {}
+    }
+
+    function DummySnackbarProvider ({ children }) {
+      return (
+        <SnackbarContext.Provider value={dummySnackbarContext}>
+          {children}
+        </SnackbarContext.Provider>
+      )
+    }
+
+    let snackbar
+    const Component = function () {
+      snackbar = useSnackbar()
+      return null
+    }
+
+    mount(
+      <DummySnackbarProvider>
+        <Component />
+      </DummySnackbarProvider>
+    ).find(Component)
+
+    expect(snackbar).toBe(dummySnackbarContext)
+  })
+})