Function locateImports

packages/query-codemods/src/utils/index.cjs:23–73

(
    identifiers,
    packageName = '@tanstack/react-query',
  )

      .find(jscodeshift.ImportSpecifier, {})

  const locateImports = (
    identifiers,
    packageName = '@tanstack/react-query',
  ) => {
    const findNamespaceImportIdentifier = () => {
      const specifier = root
        .find(jscodeshift.ImportDeclaration, {
          source: {
            value: packageName,
          },
        })
        .find(jscodeshift.ImportNamespaceSpecifier)
        .paths()

      return specifier.length > 0 ? specifier[0].value.local : null
    }

    /**
     * First, we search for the namespace import identifier because if we have any, we assume the consumer uses
     * namespace imports. In this case, we won't search for named imports at all.
     */
    const namespaceImportIdentifier = findNamespaceImportIdentifier()

    if (namespaceImportIdentifier) {
      const identifierMap = {}

      for (const identifier of identifiers) {
        identifierMap[identifier] = jscodeshift.identifier(identifier)
      }

      return {
        namespace: namespaceImportIdentifier,
        ...identifierMap,
      }
    }

    const importSpecifiers = findImportSpecifiers(packageName)
    const identifierMap = {}

    for (const identifier of identifiers) {
      identifierMap[identifier] = findImportIdentifierOf(
        importSpecifiers,
        identifier,
      )
    }

    return {
      namespace: null,
      ...identifierMap,
    }
  }

  const findAllMethodCalls = () =>
    root