Added ES7 decorators support.

Author: gajus

README.md

@@ -136,7 +136,7 @@ Refer to [`css-modulesify`](https://github.com/css-modules/css-modulesify).
  */
 ```
 
-Then you need to decorate your component using `CSSModules`, e.g.
+You need to decorate your component using `react-css-modules`, e.g.
 
 ```js
 import React from 'react';
@@ -157,6 +157,26 @@ export default CSSModules(Car, styles);
 
 Thats it!
 
+As the name implies, `react-css-modules` is compatible with the [ES7 decorators](https://github.com/wycats/javascript-decorators) syntax:
+
+```js
+import React from 'react';
+import styles from './car.css';
+import CSSModules from 'react-css-modules';
+
+@CSSModules(styles)
+export default class extends React.Component {
+    render () {
+        return <div className='car'>
+            <div className='front-door'>front-door</div>
+            <div className='back-door'>back-door</div>
+        </div>;
+    }
+}
+```
+
+[Awesome!](https://twitter.com/intent/retweet?tweet_id=636497036603428864)
+
 ### Options
 
 Options are supplied as the third parameter to the `CSSModules` function.

dist/index.js

@@ -18,14 +18,18 @@ var _linkClass = require('./linkClass');
 
 var _linkClass2 = _interopRequireDefault(_linkClass);
 
+var functionConstructor = undefined,
+    decoratorConstructor = undefined;
+
 /**
- * @param {ReactClass} Component
+ * When used as a function.
+ *
+ * @param {Function} Component
  * @param {Object} styles CSS modules class map.
  * @param {Object} options {@link https://github.com/gajus/react-css-modules#options}
- * @return {ReactClass}
+ * @return {Function}
  */
-
-exports['default'] = function (Component, styles) {
+functionConstructor = function (Component, styles) {
     var options = arguments.length <= 2 || arguments[2] === undefined ? {} : arguments[2];
 
     return (function (_Component) {
@@ -60,4 +64,25 @@ exports['default'] = function (Component, styles) {
     })(Component);
 };
 
+/**
+ * When used as a ES7 decorator.
+ *
+ * @param {Object} styles CSS modules class map.
+ * @param {Object} options {@link https://github.com/gajus/react-css-modules#options}
+ * @return {Function}
+ */
+decoratorConstructor = function (styles, options) {
+    return function (Component) {
+        return functionConstructor(Component, styles, options);
+    };
+};
+
+exports['default'] = function () {
+    if (typeof arguments[0] === 'function') {
+        return functionConstructor(arguments[0], arguments[1], arguments[2]);
+    } else {
+        return decoratorConstructor(arguments[0], arguments[1]);
+    }
+};
+
 module.exports = exports['default'];

src/index.js

@@ -1,12 +1,17 @@
 import linkClass from './linkClass';
 
+let functionConstructor,
+    decoratorConstructor;
+
 /**
- * @param {ReactClass} Component
+ * When used as a function.
+ *
+ * @param {Function} Component
  * @param {Object} styles CSS modules class map.
  * @param {Object} options {@link https://github.com/gajus/react-css-modules#options}
- * @return {ReactClass}
+ * @return {Function}
  */
-export default (Component, styles, options = {}) => {
+functionConstructor = (Component, styles, options = {}) => {
     return class extends Component {
         render () {
             if (options.allowMultiple !== false) {
@@ -23,5 +28,26 @@ export default (Component, styles, options = {}) => {
 
             return linkClass(super.render(), styles, options);
         }
+    };
+};
+
+/**
+ * When used as a ES7 decorator.
+ *
+ * @param {Object} styles CSS modules class map.
+ * @param {Object} options {@link https://github.com/gajus/react-css-modules#options}
+ * @return {Function}
+ */
+decoratorConstructor = (styles, options) => {
+    return (Component) => {
+        return functionConstructor(Component, styles, options);
+    };
+};
+
+export default (...args) => {
+    if (typeof args[0] === 'function') {
+        return functionConstructor(args[0], args[1], args[2]);
+    } else {
+        return decoratorConstructor(args[0], args[1]);
     }
 };