Merge pull request #512 from 0xProject/feature/resolver

Dependencies resolver for deployer

1 files changed, 66 insertions, 0 deletions

diff --git a/packages/sol-resolver/src/resolvers/name_resolver.ts b/packages/sol-resolver/src/resolvers/name_resolver.ts
new file mode 100644
index 000000000..6849b7610
--- /dev/null
+++ b/packages/sol-resolver/src/resolvers/name_resolver.ts
@@ -0,0 +1,66 @@
+import * as fs from 'fs';
+import * as _ from 'lodash';
+import * as path from 'path';
+
+import { ContractSource } from '../types';
+
+import { EnumerableResolver } from './enumerable_resolver';
+
+export class NameResolver extends EnumerableResolver {
+    private _contractsDir: string;
+    constructor(contractsDir: string) {
+        super();
+        this._contractsDir = contractsDir;
+    }
+    public resolveIfExists(lookupContractName: string): ContractSource | undefined {
+        const SOLIDITY_FILE_EXTENSION = '.sol';
+        let contractSource: ContractSource | undefined;
+        const onFile = (filePath: string) => {
+            const contractName = path.basename(filePath, SOLIDITY_FILE_EXTENSION);
+            if (contractName === lookupContractName) {
+                const source = fs.readFileSync(filePath).toString();
+                contractSource = {
+                    source,
+                    path: filePath,
+                };
+                return true;
+            }
+            return undefined;
+        };
+        this._traverseContractsDir(this._contractsDir, onFile);
+        return contractSource;
+    }
+    public getAll(): ContractSource[] {
+        const SOLIDITY_FILE_EXTENSION = '.sol';
+        const contractSources: ContractSource[] = [];
+        const onFile = (filePath: string) => {
+            const contractName = path.basename(filePath, SOLIDITY_FILE_EXTENSION);
+            const source = fs.readFileSync(filePath).toString();
+            const contractSource = {
+                source,
+                path: filePath,
+            };
+            contractSources.push(contractSource);
+        };
+        this._traverseContractsDir(this._contractsDir, onFile);
+        return contractSources;
+    }
+    // tslint:disable-next-line:prefer-function-over-method
+    private _traverseContractsDir(dirPath: string, onFile: (filePath: string) => true | void): boolean {
+        let dirContents: string[] = [];
+        try {
+            dirContents = fs.readdirSync(dirPath);
+        } catch (err) {
+            throw new Error(`No directory found at ${dirPath}`);
+        }
+        for (const fileName of dirContents) {
+            const entryPath = path.join(dirPath, fileName);
+            const isDirectory = fs.lstatSync(entryPath).isDirectory();
+            const isComplete = isDirectory ? this._traverseContractsDir(entryPath, onFile) : onFile(entryPath);
+            if (isComplete) {
+                return isComplete;
+            }
+        }
+        return false;
+    }
+}