UNPKG

@storybook/addon-docs

Version:

Document component usage and properties in Markdown

68 lines (58 loc) 2.21 kB
import "core-js/modules/es.object.assign.js"; import "core-js/modules/es.string.ends-with.js"; import "core-js/modules/es.array.slice.js"; import "core-js/modules/es.string.starts-with.js"; import "core-js/modules/es.string.replace.js"; import "core-js/modules/es.regexp.exec.js"; import { generate } from 'escodegen'; import dedent from 'ts-dedent'; var BASIC_OPTIONS = { format: { indent: { style: ' ' }, semicolons: false } }; var COMPACT_OPTIONS = Object.assign({}, BASIC_OPTIONS, { format: { newline: '' } }); var PRETTY_OPTIONS = Object.assign({}, BASIC_OPTIONS); export function generateCode(ast) { var compact = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false; return generate(ast, compact ? COMPACT_OPTIONS : PRETTY_OPTIONS); } export function generateObjectCode(ast) { var compact = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false; return !compact ? generateCode(ast) : generateCompactObjectCode(ast); } function generateCompactObjectCode(ast) { var result = generateCode(ast, true); // Cannot get escodegen to add a space before the last } with the compact mode settings. // Fix it until a better solution is found. if (!result.endsWith(' }')) { result = "".concat(result.slice(0, -1), " }"); } return result; } export function generateArrayCode(ast) { var compact = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false; return !compact ? generateMultilineArrayCode(ast) : generateCompactArrayCode(ast); } function generateMultilineArrayCode(ast) { var result = generateCode(ast); // escodegen add extra spacing before the closing bracket of a multiple line array with a nested object. // Fix it until a better solution is found. if (result.endsWith(' }]')) { result = dedent(result); } return result; } function generateCompactArrayCode(ast) { var result = generateCode(ast, true); // escodegen add extra an extra before the opening bracket of a compact array that contains primitive values. // Fix it until a better solution is found. if (result.startsWith('[ ')) { result = result.replace('[ ', '['); } return result; }