-
-
Notifications
You must be signed in to change notification settings - Fork 9.1k
/
extractProps.ts
80 lines (65 loc) · 2.27 KB
/
extractProps.ts
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/* eslint-disable no-underscore-dangle,react/forbid-foreign-prop-types */
import PropTypes from 'prop-types';
import { isForwardRef, isMemo } from 'react-is';
import { PropDef } from '@storybook/components';
import { PropDefGetter, PropsExtractor, propsFromDocgen, hasDocgen } from '../../lib/propsUtils';
export interface PropDefMap {
[p: string]: PropDef;
}
const propTypesMap = new Map();
Object.keys(PropTypes).forEach(typeName => {
// @ts-ignore
const type = PropTypes[typeName];
propTypesMap.set(type, typeName);
propTypesMap.set(type.isRequired, typeName);
});
const propsFromPropTypes: PropDefGetter = (type, section) => {
const props: PropDefMap = {};
if (type.propTypes) {
Object.keys(type.propTypes).forEach(property => {
const typeInfo = type.propTypes[property];
const required = typeInfo.isRequired === undefined;
const docgenInfo =
type.__docgenInfo && type.__docgenInfo[section] && type.__docgenInfo[section][property];
const description = docgenInfo ? docgenInfo.description : null;
let propType = propTypesMap.get(typeInfo) || 'other';
if (propType === 'other') {
if (docgenInfo && docgenInfo.type) {
propType = docgenInfo.type.name;
}
}
props[property] = { name: property, type: propType, required, description };
});
}
if (type.defaultProps) {
Object.keys(type.defaultProps).forEach(property => {
const value = type.defaultProps[property];
if (value === undefined) {
return;
}
if (!props[property]) {
props[property] = { name: property, type: 'any', required: false };
}
props[property].defaultValue = value;
});
}
return Object.values(props);
};
export const getPropDefs: PropDefGetter = (type, section) => {
let processedType = type;
if (!hasDocgen(type)) {
if (isForwardRef(type) || type.render) {
processedType = type.render().type;
}
if (isMemo(type)) {
// (typeof type.type === 'function')?
processedType = type.type().type;
}
}
return hasDocgen(processedType)
? propsFromDocgen(processedType, section)
: propsFromPropTypes(processedType, section);
};
export const extractProps: PropsExtractor = component => ({
rows: getPropDefs(component, 'props'),
});