[RFC] feat(handleRef): add an util for handling passed refs

CHANGELOG.md

@@ -35,6 +35,7 @@ This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.htm
 - Set default `chatBehavior` which uses Enter/Esc keys @sophieH29 ([#443](https://github.com/stardust-ui/react/pull/443))
 - Add `iconPosition` property to `Input` component @mnajdova ([#442](https://github.com/stardust-ui/react/pull/442))
 - Add `color`, `inverted` and `renderContent` props and `content` slot to `Segment` component @Bugaa92 ([#389](https://github.com/stardust-ui/react/pull/389))
+- Export `Ref` component and add `handleRef` util @layershifter ([#459](https://github.com/stardust-ui/react/pull/459))
 
 ### Documentation
 - Add all missing component descriptions and improve those existing @levithomason ([#400](https://github.com/stardust-ui/react/pull/400))

docs/src/examples/components/Ref/Types/RefExampleRef.tsx

@@ -0,0 +1,63 @@
+import React from 'react'
+import { Button, Grid, Ref, Segment } from '@stardust-ui/react'
+
+class RefExampleRef extends React.Component {
+  state = { isMounted: false }
+
+  createdRef = React.createRef<HTMLButtonElement>()
+  functionalRef = null
+
+  handleRef = node => (this.functionalRef = node)
+
+  componentDidMount() {
+    this.setState({ isMounted: true })
+  }
+
+  render() {
+    const { isMounted } = this.state
+
+    return (
+      <Grid columns={2}>
+        <Segment>
+          <Ref innerRef={this.handleRef}>
+            <Button primary>With functional ref</Button>
+          </Ref>
+          <Ref innerRef={this.createdRef}>
+            <Button>
+              With <code>createRef()</code>
+            </Button>
+          </Ref>
+        </Segment>
+
+        {isMounted && (
+          <code style={{ margin: 10 }}>
+            <pre>
+              {JSON.stringify(
+                {
+                  nodeName: this.functionalRef.nodeName,
+                  nodeType: this.functionalRef.nodeType,
+                  textContent: this.functionalRef.textContent,
+                },
+                null,
+                2,
+              )}
+            </pre>
+            <pre>
+              {JSON.stringify(
+                {
+                  nodeName: this.createdRef.current.nodeName,
+                  nodeType: this.createdRef.current.nodeType,
+                  textContent: this.createdRef.current.textContent,
+                },
+                null,
+                2,
+              )}
+            </pre>
+          </code>
+        )}
+      </Grid>
+    )
+  }
+}
+
+export default RefExampleRef

docs/src/examples/components/Ref/Types/index.tsx

@@ -0,0 +1,21 @@
+import * as React from 'react'
+
+import ComponentExample from 'docs/src/components/ComponentDoc/ComponentExample'
+import ExampleSection from 'docs/src/components/ComponentDoc/ExampleSection'
+
+const RefTypesExamples = () => (
+  <ExampleSection title="Types">
+    <ComponentExample
+      title="Ref"
+      description={
+        <span>
+          A component exposes the <code>innerRef</code> prop that always returns the DOM node of
+          both functional and class component children.
+        </span>
+      }
+      examplePath="components/Ref/Types/RefExampleRef"
+    />
+  </ExampleSection>
+)
+
+export default RefTypesExamples

docs/src/examples/components/Ref/index.tsx

@@ -0,0 +1,10 @@
+import * as React from 'react'
+import Types from './Types'
+
+const RefExamples: React.SFC = () => (
+  <div>
+    <Types />
+  </div>
+)
+
+export default RefExamples

src/components/Ref/Ref.tsx

@@ -1,19 +1,20 @@
 import * as PropTypes from 'prop-types'
-import * as _ from 'lodash'
-import { Children, Component } from 'react'
+import * as React from 'react'
 import { findDOMNode } from 'react-dom'
-import { ReactChildren } from 'utils'
+
+import { ReactChildren } from '../../../types/utils'
+import { handleRef } from '../../lib'
 
 export interface RefProps {
   children?: ReactChildren
-  innerRef?: (ref: HTMLElement) => void
+  innerRef?: React.Ref<any>
 }
 
 /**
  * This component exposes a callback prop that always returns the DOM node of both functional and class component
  * children.
  */
-export default class Ref extends Component<RefProps> {
+export default class Ref extends React.Component<RefProps> {
   static propTypes = {
     /**
      *  Used to set content when using childrenApi - internal only
@@ -22,18 +23,22 @@ export default class Ref extends Component<RefProps> {
     children: PropTypes.element,
 
     /**
-     * Called when componentDidMount.
+     * Called when a child component will be mounted or updated.
      *
      * @param {HTMLElement} node - Referred node.
      */
-    innerRef: PropTypes.func,
+    innerRef: PropTypes.oneOfType([PropTypes.func, PropTypes.object]),
   }
 
   componentDidMount() {
-    _.invoke(this.props, 'innerRef', findDOMNode(this))
+    handleRef(this.props, 'innerRef', findDOMNode(this))
+  }
+
+  componentWillUnmount() {
+    handleRef(this.props, 'innerRef', null)
   }
 
   render() {
-    return this.props.children && Children.only(this.props.children)
+    return this.props.children && React.Children.only(this.props.children)
   }
 }

src/index.ts

@@ -86,6 +86,7 @@ export {
   RadioGroupItemProps,
 } from './components/RadioGroup/RadioGroupItem'
 
+export { default as Ref, RefProps } from './components/Ref/Ref'
 export { default as Segment, SegmentProps } from './components/Segment/Segment'
 
 export { default as Status, StatusPropsWithDefaults, StatusProps } from './components/Status/Status'

src/lib/handleRef.ts

@@ -0,0 +1,36 @@
+import * as React from 'react'
+
+/** A type that ensures that passed prop is defined in props and matches the type. */
+type HandleRefProps<TProp extends string, N> = { [K in TProp]?: React.Ref<N> }
+
+/**
+ * The function that correctly handles passing refs.
+ *
+ * @param props All components props
+ * @param propName The name of a ref prop
+ * @param node A node that should be passed by ref
+ */
+const handleRef = <P extends string, N>(props: HandleRefProps<P, N>, propName: P, node: N) => {
+  const ref: React.Ref<N> = props[propName]
+
+  if (process.env.NODE_ENV !== 'production') {
+    if (typeof ref === 'string') {
+      throw new Error(
+        'We do not support refs as string, this is a legacy API and will be likely to be removed in one of the future releases of React.',
+      )
+    }
+  }
+
+  if (typeof ref === 'function') {
+    ref(node)
+    return
+  }
+
+  if (typeof ref === 'object') {
+    // @ts-ignore The `current` property is defined as readonly, however it's a valid way because
+    // `ref` is a mutable object
+    ref.current = node
+  }
+}
+
+export default handleRef

src/lib/index.ts

@@ -15,6 +15,8 @@ export { default as getElementType } from './getElementType'
 export { default as getUnhandledProps } from './getUnhandledProps'
 export { default as mergeThemes } from './mergeThemes'
 export { default as renderComponent, RenderResultConfig } from './renderComponent'
+
+export { default as handleRef } from './handleRef'
 export {
   htmlImageProps,
   htmlInputAttrs,

test/specs/components/Ref/Ref-test.tsx

@@ -1,8 +1,8 @@
-import * as React from 'react'
 import { shallow, mount } from 'enzyme'
+import * as React from 'react'
 
-import { CompositeClass, CompositeFunction, DOMClass, DOMFunction } from './fixtures'
 import Ref from 'src/components/Ref/Ref'
+import { CompositeClass, CompositeFunction, DOMClass, DOMFunction } from './fixtures'
 
 const testInnerRef = Component => {
   const innerRef = jest.fn()
@@ -42,5 +42,20 @@ describe('Ref', () => {
     it('returns node from a class component', () => {
       testInnerRef(CompositeClass)
     })
+
+    it('returns "null" after unmount', () => {
+      const innerRef = jest.fn()
+      const wrapper = mount(
+        <Ref innerRef={innerRef}>
+          <CompositeClass />
+        </Ref>,
+      )
+
+      innerRef.mockClear()
+      wrapper.unmount()
+
+      expect(innerRef).toHaveBeenCalledTimes(1)
+      expect(innerRef).toHaveBeenCalledWith(null)
+    })
   })
 })

test/specs/lib/handleRef-test.ts

@@ -0,0 +1,30 @@
+import * as React from 'react'
+import handleRef from 'src/lib/handleRef'
+
+describe('handleRef', () => {
+  it('throws an error when "ref" is string', () => {
+    const node = document.createElement('div')
+
+    expect(() => {
+      handleRef({ ref: 'ref' }, 'ref', node)
+    }).toThrowError()
+  })
+
+  it('calls with node when "ref" is function', () => {
+    const ref = jest.fn()
+    const node = document.createElement('div')
+
+    handleRef({ ref }, 'ref', node)
+
+    expect(ref).toBeCalledWith(node)
+  })
+
+  it('assigns to "current" when "ref" is object', () => {
+    const ref = React.createRef<HTMLDivElement>()
+    const node = document.createElement('div')
+
+    handleRef({ ref }, 'ref', node)
+
+    expect(ref.current).toBe(node)
+  })
+})