summaryrefslogtreecommitdiffstats
path: root/packages/gitbook-plugin-copy-code/src/index.js
blob: cfbe1b1ae1ea7140d7dea57c851e411bd5ad9582 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
const copy = require('copy-to-clipboard');
const GitBook = require('gitbook-core');
const { React } = GitBook;

/**
 * Get children as text
 * @param {React.Children} children
 * @return {String}
 */
function getChildrenToText(children) {
    return React.Children.map(children, child => {
        if (typeof child === 'string') {
            return child;
        } else {
            return child.props.children ?
                getChildrenToText(child.props.children) : '';
        }
    }).join('');
}

const CodeBlockWithCopy = React.createClass({
    propTypes: {
        children: React.PropTypes.node
    },

    onClick(event) {
        const { children } = this.props;

        event.preventDefault();
        event.stopPropagation();

        const text = getChildrenToText(children);
        copy(text);
    },

    render() {
        const { children } = this.props;

        return (
            <div className="CodeBlockWithCopy-Container">
                <GitBook.ImportCSS href="gitbook/copy-code/button.css" />

                {children}
                <span className="CodeBlockWithCopy-Button" onClick={this.onClick}>Copy</span>
            </div>
        );
    }
});

module.exports = GitBook.createPlugin({
    activate: (dispatch, getState, { Components }) => {
        dispatch(Components.registerComponent(CodeBlockWithCopy, { role: 'html:pre' }));
    }
});