fix #12

package.json

@@ -51,6 +51,12 @@
                     "command": "namespaceResolver.sort",
                     "alt": "namespaceResolver.sort",
                     "group": "0_namespace_resolver@3"
+                },
+                {
+                    "when": "resourceLangId == php",
+                    "command": "namespaceResolver.importall",
+                    "alt": "namespaceResolver.importall",
+                    "group": "0_namespace_resolver@4"
                 }
             ]
         },
@@ -107,6 +113,10 @@
             {
                 "title": "Sort Imports",
                 "command": "namespaceResolver.sort"
+            },
+            {
+                "title": "Import All Classes",
+                "command": "namespaceResolver.importall"
             }
         ],
         "keybindings": [
@@ -124,6 +134,11 @@
                 "command": "namespaceResolver.sort",
                 "key": "ctrl+alt+s",
                 "when": "editorTextFocus"
+            },
+            {
+                "command": "namespaceResolver.importall",
+                "key": "ctrl+alt+a",
+                "when": "editorTextFocus"
             }
         ]
     },

src/Resolver.js

@@ -27,6 +27,75 @@ class Resolver {
         this.importClass(selection, fqcn, replaceClassAfterImport);
     }
 
+    async importAll() {
+        const text = this.activeEditor().document.getText();
+        let matches = [];
+        let regex = /.*/;
+
+        let phpClasses = this.getPhpClasses(text);
+
+        for (let phpClass of phpClasses) {
+            await this.importCommand(phpClass);
+        }
+    }
+
+    getPhpClasses(text) {
+        let phpClasses = this.getExtended(text);
+        phpClasses = phpClasses.concat(this.getFromFunctionParameters(text));
+        phpClasses = phpClasses.concat(this.getInitializedWithNew(text));
+        phpClasses = phpClasses.concat(this.getFromStaticCalls(text));
+
+        // get unique class names only
+        phpClasses = phpClasses.filter((v, i, a) => a.indexOf(v) === i);
+        return phpClasses;
+    }
+
+    getExtended(text) {
+        let regex = /extends ([A-Z][A-Za-z0-9\-\_]*)/gm;
+        let matches = [];
+        let phpClasses = [];
+        while (matches = regex.exec(text)) {
+            phpClasses.push(matches[1]);
+        }
+        return phpClasses;
+    }
+
+    getFromFunctionParameters(text) {
+        let regex = /function [\S]+\((.*)\)/gm;
+        let matches = [];
+        let phpClasses = [];
+        while (matches = regex.exec(text)) {
+            let parameters = matches[1].split(', ');
+            for (let s of parameters) {
+                let phpClassName = s.substr(0, s.indexOf(' '));
+                if (phpClassName && /[A-Z]/.test(phpClassName[0])) { //starts with capital letter
+                    phpClasses.push(phpClassName);
+                }
+            }
+        }
+        return phpClasses;
+    }
+
+    getInitializedWithNew(text) {
+        let regex = /new ([A-Z][A-Za-z0-9\-\_]*)/gm;
+        let matches = [];
+        let phpClasses = [];
+        while (matches = regex.exec(text)) {
+            phpClasses.push(matches[1]);
+        }
+        return phpClasses;
+    }
+
+    getFromStaticCalls(text) {
+        let regex = /([A-Z][A-Za-z0-9\-\_]*)::/gm;
+        let matches = [];
+        let phpClasses = [];
+        while (matches = regex.exec(text)) {
+            phpClasses.push(matches[1]);
+        }
+        return phpClasses;
+    }
+
     importClass(selection, fqcn, replaceClassAfterImport = false) {
         let useStatements, declarationLines;
 
@@ -360,6 +429,10 @@ class Resolver {
     }
 
     resolving(selection) {
+        if ((typeof selection) == 'string') {
+            return selection;
+        }
+
         let wordRange = this.activeEditor().document.getWordRangeAtPosition(selection.active);
 
         if (wordRange === undefined) {

src/extension.js

@@ -28,6 +28,10 @@ function activate(context) {
         vscode.commands.registerCommand('namespaceResolver.sort', () => resolver.sortCommand())
     );
 
+    context.subscriptions.push(
+        vscode.commands.registerCommand('namespaceResolver.importall', () => resolver.importAll())
+    );
+
     context.subscriptions.push(vscode.workspace.onWillSaveTextDocument((event) => {
         if (
             event.document.languageId === 'php' &&