get-require-targets.js 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. /**
  2. * @author Toru Nagashima
  3. * @copyright 2016 Toru Nagashima. All rights reserved.
  4. * See LICENSE file in root directory for full license.
  5. */
  6. "use strict"
  7. //------------------------------------------------------------------------------
  8. // Requirements
  9. //------------------------------------------------------------------------------
  10. const path = require("path")
  11. const resolve = require("resolve")
  12. const getResolvePaths = require("./get-resolve-paths")
  13. const getTryExtensions = require("./get-try-extensions")
  14. const getValueIfString = require("./get-value-if-string")
  15. const ImportTarget = require("./import-target")
  16. const stripImportPathParams = require("./strip-import-path-params")
  17. //------------------------------------------------------------------------------
  18. // Helpers
  19. //------------------------------------------------------------------------------
  20. /**
  21. * Checks whether or not a given node is a callee.
  22. *
  23. * @param {ASTNode} node - A node to check.
  24. * @returns {boolean} `true` if the node is a callee.
  25. */
  26. function isCallee(node) {
  27. return node.parent.type === "CallExpression" && node.parent.callee === node
  28. }
  29. /**
  30. * Gets references of "require".
  31. *
  32. * @param {escope.Scope} scope - The global scope.
  33. * @returns {escope.Reference[]} References of "require".
  34. */
  35. function getReferencesOfRequire(scope) {
  36. const variable = scope.set.get("require")
  37. if (!variable) {
  38. // Not found.
  39. return []
  40. }
  41. return variable.references
  42. }
  43. //------------------------------------------------------------------------------
  44. // Public Interface
  45. //------------------------------------------------------------------------------
  46. /**
  47. * Gets a list of `require()` targets.
  48. *
  49. * Core modules of Node.js (e.g. `fs`, `http`) are excluded.
  50. *
  51. * @param {RuleContext} context - The rule context.
  52. * @param {boolean} includeCore - The flag to include core modules.
  53. * @returns {ImportTarget[]} A list of found target's information.
  54. */
  55. module.exports = function getRequireTargets(context, includeCore) {
  56. const retv = []
  57. const basedir = path.dirname(path.resolve(context.getFilename()))
  58. const paths = getResolvePaths(context)
  59. const references = getReferencesOfRequire(context.getScope())
  60. const extensions = getTryExtensions(context)
  61. const options = { basedir, paths, extensions }
  62. for (const reference of references) {
  63. const node = reference.identifier
  64. // Skips if it's not a call of `require`.
  65. if (!isCallee(node)) {
  66. continue
  67. }
  68. // Gets the target module.
  69. const targetNode = node.parent.arguments[0]
  70. const rawName = getValueIfString(targetNode)
  71. const name = rawName && stripImportPathParams(rawName)
  72. if (name && (includeCore || !resolve.isCore(name))) {
  73. retv.push(new ImportTarget(targetNode, name, options))
  74. }
  75. }
  76. return retv
  77. }