From 9915d15aeaa20edf5044e7298a175ad4d9a2284c Mon Sep 17 00:00:00 2001 From: Adam Borowski Date: Thu, 20 Jul 2017 17:26:12 +0200 Subject: [PATCH] Update CSSTransition.js (#126) support for removing multiple class names (for use with react-jss) --- src/CSSTransition.js | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/CSSTransition.js b/src/CSSTransition.js index 1803c758..540320b9 100644 --- a/src/CSSTransition.js +++ b/src/CSSTransition.js @@ -1,13 +1,14 @@ import * as PropTypes from 'prop-types'; import addOneClass from 'dom-helpers/class/addClass'; -import removeClass from 'dom-helpers/class/removeClass'; +import removeOneClass from 'dom-helpers/class/removeClass'; import React from 'react'; import Transition from './Transition'; import { classNamesShape } from './utils/PropTypes'; -const addClass = (node, classes) => classes.split(' ').forEach(c => addOneClass(node, c)); +const addClass = (node, classes) => classes && classes.split(' ').forEach(c => addOneClass(node, c)); +const removeClass = (node, classes) => classes && classes.split(' ').forEach(c => removeOneClass(node, c)); const propTypes = { ...Transition.propTypes,