94 lines
2.4 KiB
TypeScript
94 lines
2.4 KiB
TypeScript
import { basename, extname } from 'path'
|
|
|
|
const camelCase = (value: string): string => {
|
|
const val = value.replace(/[-_\s.]+(.)?/g, (_match, chr) =>
|
|
chr ? chr.toUpperCase() : ''
|
|
)
|
|
return val.substr(0, 1).toUpperCase() + val.substr(1)
|
|
}
|
|
|
|
const isValidIdentifier = (value: string): boolean =>
|
|
/^[a-zA-ZÀ-ÿ][0-9a-zA-ZÀ-ÿ]+$/.test(value)
|
|
|
|
type Node = {
|
|
type: string
|
|
declaration: {
|
|
type: string
|
|
body: any
|
|
id?: any
|
|
}
|
|
}
|
|
|
|
export default function transformer(file, api, options) {
|
|
const j = api.jscodeshift
|
|
const root = j(file.source)
|
|
|
|
let hasModifications: boolean
|
|
|
|
const returnsJSX = (node): boolean =>
|
|
node.type === 'JSXElement' ||
|
|
(node.type === 'BlockStatement' &&
|
|
j(node)
|
|
.find(j.ReturnStatement)
|
|
.some((path) => path.value.argument?.type === 'JSXElement'))
|
|
|
|
const hasRootAsParent = (path): boolean => {
|
|
const program = path.parentPath.parentPath.parentPath.parentPath.parentPath
|
|
return !program || program?.value?.type === 'Program'
|
|
}
|
|
|
|
const nameFunctionComponent = (path): void => {
|
|
const node: Node = path.value
|
|
|
|
if (!node.declaration) {
|
|
return
|
|
}
|
|
|
|
const isArrowFunction =
|
|
node.declaration.type === 'ArrowFunctionExpression' &&
|
|
returnsJSX(node.declaration.body)
|
|
const isAnonymousFunction =
|
|
node.declaration.type === 'FunctionDeclaration' && !node.declaration.id
|
|
|
|
if (!(isArrowFunction || isAnonymousFunction)) {
|
|
return
|
|
}
|
|
|
|
const fileName = basename(file.path, extname(file.path))
|
|
let name = camelCase(fileName)
|
|
|
|
// If the generated name looks off, don't add a name
|
|
if (!isValidIdentifier(name)) {
|
|
return
|
|
}
|
|
|
|
// Add `Component` to the end of the name if an identifier with the
|
|
// same name already exists
|
|
while (root.find(j.Identifier, { name }).some(hasRootAsParent)) {
|
|
// If the name is still duplicated then don't add a name
|
|
if (name.endsWith('Component')) {
|
|
return
|
|
}
|
|
name += 'Component'
|
|
}
|
|
|
|
hasModifications = true
|
|
|
|
if (isArrowFunction) {
|
|
path.insertBefore(
|
|
j.variableDeclaration('const', [
|
|
j.variableDeclarator(j.identifier(name), node.declaration),
|
|
])
|
|
)
|
|
|
|
node.declaration = j.identifier(name)
|
|
} else {
|
|
// Anonymous Function
|
|
node.declaration.id = j.identifier(name)
|
|
}
|
|
}
|
|
|
|
root.find(j.ExportDefaultDeclaration).forEach(nameFunctionComponent)
|
|
|
|
return hasModifications ? root.toSource(options) : null
|
|
}
|