Add support for scoped packages resolving to NPMResoler

1 files changed, 11 insertions, 2 deletions

diff --git a/packages/sol-resolver/src/resolvers/npm_resolver.ts b/packages/sol-resolver/src/resolvers/npm_resolver.ts
index a2df0dcad..505362c88 100644
--- a/packages/sol-resolver/src/resolvers/npm_resolver.ts
+++ b/packages/sol-resolver/src/resolvers/npm_resolver.ts
@@ -1,4 +1,5 @@
 import * as fs from 'fs';
+import * as _ from 'lodash';
 import * as path from 'path';
 
 import { ContractSource } from '../types';
@@ -13,12 +14,20 @@ export class NPMResolver extends Resolver {
     }
     public resolveIfExists(importPath: string): ContractSource | undefined {
         if (!importPath.startsWith('/')) {
-            const [packageName, ...other] = importPath.split('/');
+            let packageName;
+            let packageScope;
+            let other;
+            if (importPath.startsWith('@')) {
+                [packageScope, packageName, ...other] = importPath.split('/');
+            } else {
+                [packageName, ...other] = importPath.split('/');
+            }
             const pathWithinPackage = path.join(...other);
             let currentPath = this._packagePath;
             const ROOT_PATH = '/';
             while (currentPath !== ROOT_PATH) {
-                const lookupPath = path.join(currentPath, 'node_modules', packageName, pathWithinPackage);
+                const packagePath = _.isUndefined(packageScope) ? packageName : path.join(packageScope, packageName);
+                const lookupPath = path.join(currentPath, 'node_modules', packagePath, pathWithinPackage);
                 if (fs.existsSync(lookupPath) && fs.lstatSync(lookupPath).isFile()) {
                     const fileContent = fs.readFileSync(lookupPath).toString();
                     return {