Skip to content
This repository was archived by the owner on Jun 16, 2023. It is now read-only.
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
const func = require('lib').func;
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
import { func } from 'lib';
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
const myFunc = require('lib').func;
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
import { func as myFunc } from 'lib';
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
import {defineTest} from 'jscodeshift/dist/testUtils'

const tests = [
'basic-case',
'different-names',
]

describe('require-lib-func-to-import-func-from-lib', () => {
tests.forEach(test => {
defineTest(
__dirname,
'require-lib-func-to-import-func-from-lib',
null,
`require-lib-func-to-import-func-from-lib/${test}`
)
})
})
75 changes: 75 additions & 0 deletions transforms/require-lib-func-to-import-func-from-lib.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
/**
* Transform
*
* const Lib = require('lib');
*
* to
*
* import Lib from 'lib';
*
* Only on global context
*/

// on https://astexplorer.net: Press ctrl+space for code completion

import Logger from './utils/logger'
import {isTopNode} from './utils/filters'

function transformer(file, api, options) {
const j = api.jscodeshift
const logger = new Logger(file, options)

// ------------------------------------------------------------------ SEARCH
const nodes = j(file.source)
.find(j.VariableDeclaration, {
declarations: [{
init: {
type: 'MemberExpression',
object: {
callee: {
name: 'require'
}
}
}
}]
}).filter(isTopNode)

logger.log(`${nodes.length} nodes will be transformed`)

// ----------------------------------------------------------------- REPLACE
return nodes.replaceWith(path => {
const rest = []
const imports = []
for (const declaration of path.node.declarations) {
if (j.MemberExpression.check(declaration.init)
&& declaration.init.object.callee.name === 'require'){
const importSpecifier = j.importSpecifier(declaration.init.property, declaration.id)

const sourcePath = declaration.init.object.arguments.shift()

if (declaration.init.object.arguments.length) {
logger.error(`${logger.lines(declaration)} too many arguments.`
+ 'Aborting transformation')
return file.source
}
if (!j.Literal.check(sourcePath)) {
logger.error(`${logger.lines(declaration)} bad argument.`
+ 'Expecting a string literal, got '
+ j(sourcePath).toSource()
+ '`. Aborting transformation')
return file.source
}
imports.push(j.importDeclaration([importSpecifier], sourcePath))
} else {
rest.push(declaration)
}
}
if (rest.length > 0) {
logger.warn(`${logger.lines(path.node)} introduced leftover`)
return [...imports, j.variableDeclaration(path.node.kind, rest)]
}
return imports
}).toSource()
}

export default transformer
3 changes: 1 addition & 2 deletions transforms/variable-require-to-import-default.js
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,7 @@ function transformer(file, api, options) {
const rest = []
const imports = []
for (const declaration of path.node.declarations) {
if (declaration.init !== null
&& declaration.init.type === 'CallExpression'
if (j.CallExpression.check(declaration.init)
&& declaration.init.callee.name === 'require'){
const importSpecifier = j.importDefaultSpecifier(declaration.id)

Expand Down