revert #3

Author: siddharthkp

README.md

@@ -1,3 +1,5 @@
+![beta](https://img.shields.io/badge/build-beta-yellow.svg?style=flat)
+
 #### atom-import-cost
 
  
@@ -35,3 +37,5 @@ Please feel free to help out with these :)
 :star: this repo
 
  
+
+[![Sponsor](https://app.codesponsor.io/embed/LhLT2c31ydJzdLUuSR9f8mCA/siddharthkp/atom-import-cost.svg)](https://app.codesponsor.io/link/LhLT2c31ydJzdLUuSR9f8mCA/siddharthkp/atom-import-cost)

lib/imports.js

@@ -0,0 +1,38 @@
+'use babel'
+
+import readPkgUp from 'read-pkg-up'
+import Walker from 'node-source-walk'
+
+const isValidModule = ({ value }) => {
+  const regex = new RegExp('^([a-z0-9-_]{1,})$')
+  return regex.test(value)
+}
+
+module.exports = src => {
+  const walker = new Walker()
+  const dependencies = []
+
+  if (src === '') return dependencies
+
+  try {
+    walker.walk(src, function(node) {
+      switch (node.type) {
+        case 'ImportDeclaration':
+          if (node.source && node.source.value) {
+            dependencies.push(node.source)
+          }
+          break
+        case 'ExportNamedDeclaration':
+        case 'ExportAllDeclaration':
+          if (node.source && node.source.value) {
+            dependencies.push(node.source)
+          }
+          break
+        default:
+          return
+      }
+    })
+  } catch (e) {}
+
+  return dependencies.filter(isValidModule)
+}

lib/main.js

@@ -1,81 +1,69 @@
 'use babel'
 
-import { importCost, cleanup }  from 'import-cost';
-import Utils                    from './utils'
-
-let emitters  = {};
-let markers   = {};
+import { Point } from 'atom'
+import imports from './imports'
+import getSize from './size'
 
 export default {
+  subscriptions: null,
+  decorations: [],
+
   activate() {
-    atom.workspace.onDidChangeActivePaneItem(() => this.calculateImportCost())
+    console.log('import cost activated')
+    this.attachLabels()
+
+    atom.workspace.onDidChangeActivePaneItem(() => this.attachLabels())
 
     atom.workspace.observeTextEditors(editor =>
-      editor.onDidSave(() => this.calculateImportCost())
+      editor.onDidSave(() => this.attachLabels())
     )
+  },
 
-    this.calculateImportCost()
+  removeLabels() {
+    this.decorations.map(d => d.destroy())
+    this.decorations = []
   },
 
-  calculateImportCost() {
-    // Get the active editor
+  attachLabels() {
+    this.removeLabels()
     const editor = atom.workspace.getActiveTextEditor()
     if (!editor) return
 
-    // Get the current file path and check whether it is a valid file extension
-    const filePath      = editor.getPath();
-    const fileExtension = Utils.extractFileExtension(filePath);
-    if (!fileExtension) return
+    const code = editor.getText()
 
-    // Get the content of the active editor
-    const content = editor.getText()
+    const dependencies = imports(code)
 
-    // Remove active importCost event listeners
-    if (emitters[filePath]) {
-      emitters[filePath].removeAllListeners()
-    }
+    dependencies.map(d => {
+      const name = d.value
 
-    // Calculate import cost of the active editor
-    emitters[filePath] = importCost(filePath, content, fileExtension);
-    emitters[filePath].on('calculated', packageInfo => this.renderImportCost(editor, packageInfo))
-  },
+      const line = d.loc.start.line - 1 // starts at 0 in atom
+      const length = editor.lineLengthForScreenRow(line)
 
-  clearImportCost(packageName) {
-    let marker = markers[packageName]
-    if (marker) {
-      marker.destroy()
-      delete markers[packageName]
-    }
-  },
+      const label = document.createElement('span')
+      label.className = 'atom-import-cost-label'
+      label.textContent = '...'
+
+      const position = new Point(line, length)
 
-  renderImportCost(editor, packageInfo) {
-    // Remove labels
-    this.clearImportCost(packageInfo.name)
+      const marker = editor.markScreenPosition(position)
+      const decoration = editor.decorateMarker(marker, {
+        type: 'overlay',
+        item: label
+      })
 
-    // Get the position where we are going to insert the label
-    const row = packageInfo.line - 1
-    const col = editor.lineLengthForScreenRow(row)
+      getSize(name).then(size => {
+        if (size === 'invalid') label.textContent = ''
+        else if (size) label.textContent = size
+        else label.textContent = '¯\\_(ツ)_/¯ could not fetch size, API is down!'
 
-    // Create a label and render it
-    markers[packageInfo.name] = Utils.createMarker(packageInfo.size, editor, row, col)
+        decoration.setProperties({ item: label })
+      })
+
+      this.decorations.push(decoration)
+    })
   },
 
   deactivate() {
-    cleanup()
-
-    for (var emitter in emitters) {
-      if (emitters.hasOwnProperty(emitter)) {
-         emitters[emitter].removeAllListeners()
-         delete emitters[emitter]
-      }
-    }
-
-    for (var marker in markers) {
-      if (markers.hasOwnProperty(marker)) {
-         markers[marker].destroy()
-         delete markers[marker]
-      }
-    }
-
+    this.subscriptions.dispose()
   }
 }

lib/size.js

@@ -0,0 +1,32 @@
+'use babel'
+
+import axios from 'axios'
+import bytes from 'bytes'
+import versions from './versions'
+
+const cache = {}
+
+module.exports = name => {
+  return new Promise(resolve => {
+    if (versions(name)) name += '@' + versions(name)
+
+    if (cache[name]) {
+      if (cache[name] === 'invalid') resolve('invalid')
+      else resolve(bytes(cache[name]))
+    } else {
+      axios
+        .get('https://cost-of-modules.herokuapp.com/package?name=' + name)
+        .then(result => {
+          const size = result.data.gzipSize
+          cache[name] = size
+          resolve(bytes(size))
+        })
+        .catch(error => {
+          if (error.response && error.response.status === 400) {
+            cache[name] = 'invalid'
+            resolve('invalid')
+          } else resolve('error')
+        })
+    }
+  })
+}

lib/utils.js

@@ -0,0 +1,16 @@
+'use babel'
+
+import readPkgUp from 'read-pkg-up'
+
+module.exports = name => {
+  let dependencies = {}
+
+  try {
+    const directory = atom.workspace.getActiveTextEditor().getDirectoryPath()
+    dependencies = readPkgUp.sync({ cwd: directory }).pkg.dependencies
+  } catch (error) {}
+
+  if (!dependencies) dependencies = {}
+
+  return dependencies[name] || null
+}