lithium-sdk/lib/component-dependencies.js

### Lithium Developer Community Area

'use strict';

const dependencyTree = require('dependency-tree');
const fs = require('fs-extra');
const path = require('path');

class ComponentDependencies {
  constructor(srcPath, destPath, resolvedComponentPathBase) {
    this.srcPath = srcPath;
    this.destPath = destPath;
    this.packageName = resolvedComponentPathBase;
  }

  walk(currentDirPath, componentsRootDir, depsMap) {
    fs.readdirSync(currentDirPath).map(f => path.join(currentDirPath, f)).forEach(filePath => {
      const stat = fs.statSync(filePath);
      if (stat.isFile() && path.parse(filePath).ext === '.jsx') {
        depsMap = this.captureDeps(filePath, depsMap, componentsRootDir);
      } else if (stat.isDirectory()) {
        return this.walk(filePath, componentsRootDir, depsMap);
      }
    });

    return depsMap;
  }

  captureDeps(filepath, resultsObj, componentsRootDir) {
    const tree = dependencyTree({
      filename: filepath,
      directory: process.cwd(),
      filter: p => p.indexOf(componentsRootDir) === 0
    });

    return Object.assign(resultsObj, this.deepObjKeyTransform(tree, componentsRootDir));
  }

  deepObjKeyTransform(componentTree, componentsRootDir, deps, topLevelComp) {
    Object.keys(componentTree).forEach(componentPath => {

      // componentPath examples:
      //  - Users/adam.ayres/dev/angular-li/limuirs/src/components/LimuirsContext.jsx
      //  - Users/adam.ayres/dev/angular-li/limuirs/src/components/boards/BoardList.jsx
      //  - Users/adam.ayres/dev/angular-li/limuirs/src/components/common/Feedback/index.jsx
      //  - Users/adam.ayres/dev/angular-li/limuirs/src/components/common/Feedback/FeedbackStore/index.js
      //
      // Transform to:
      //  - limuirs/components/LimuirsContext
      //  - limuirs/components/boards/BoardList
      //  - limuirs/components/common/Feedback
      //  - limuirs/components/common/Feedback/FeedbackStore

      let fullComponentName = path.relative(path.join(process.cwd(), this.srcPath), componentPath);
      const stripFromEnd = ['index.js', 'index.jsx', '.jsx', '.js'];

      for (let i = 0; i < stripFromEnd.length; i++) {
        const itemToStrip = stripFromEnd[i];
        if (fullComponentName.endsWith(itemToStrip)) {
          fullComponentName = fullComponentName.substring(0, fullComponentName.length - itemToStrip.length);
          break;
        }
      }

      if (fullComponentName.endsWith('/')) {
        fullComponentName = fullComponentName.substring(0, fullComponentName.length - 1);
      }

      fullComponentName = this.packageName + fullComponentName;

      if (!deps) {
        deps = {
          [fullComponentName]: []
        };
        topLevelComp = fullComponentName;
      } else if (deps[topLevelComp].indexOf(fullComponentName) === -1) {
        deps[topLevelComp].push(fullComponentName);
      }

      this.deepObjKeyTransform(componentTree[componentPath],
          componentsRootDir, deps, topLevelComp);
    });
    return deps;
  }

  createDepFile() {
    const fullSrcPath = path.resolve(this.srcPath);

    function handleError(err, reject) {
      if (err) {
        console.log(err);
        reject(err);
      }
    }

    //if source path doesn't exist, don't bomb out
    if (fs.existsSync(fullSrcPath)) {
      const deps = this.walk(fullSrcPath, fullSrcPath, {});
      return new Promise((resolve, reject) => {
        return fs.ensureFile(this.destPath, (err) => {
          handleError(err, reject);
          return fs.writeFile(this.destPath, JSON.stringify(deps, null, 2), (err) => {
            handleError(err, reject);
            console.log(`Component dependencies have been written to file: ${this.destPath}`);
            resolve();
          });
        });
      });
    } else {
      console.log(`Component dependency path not found: ${this.srcPath}.  Exiting.`);
      return Promise.resolve();
    }
  }
}

module.exports = ComponentDependencies;