angular-cn/docs/dgeni-package/services/ExportTreeVisitor.js

module.exports = function ExportTreeVisitor(ParseTreeVisitor, log) {

  function ExportTreeVisitorImpl() {
    ParseTreeVisitor.call(this);
  }
  ExportTreeVisitorImpl.prototype = {

    __proto__: ParseTreeVisitor.prototype,

    visitExportDeclaration: function(tree) {
      // We are entering an export declaration - create an object to track it
      this.currentExport = {
        comment: tree.commentBefore,
        location: tree.location
      };
      log.silly('enter', tree.type, tree.commentBefore ? 'has comment' : '');
      ParseTreeVisitor.prototype.visitExportDeclaration.call(this, tree);
      log.silly('exit', this.currentExport);

      // We are exiting the export declaration - store the export object
      this.exports.push(this.currentExport);
      this.currentExport = null;
    },

    visitVariableStatement: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);
        this.currentExport.name = "VARIABLE_STATEMENT";
      }
    },

    visitVariableDeclaration: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);
        this.currentExport.name = tree.lvalue;
        this.currentExport.variableDeclaration = tree;
      }
    },

    visitFunctionDeclaration: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);
        this.currentExport.name = tree.name.identifierToken.value;
        this.currentExport.functionKind = tree.functionKind;
        this.currentExport.parameters = tree.parameterList.parameters;
        this.currentExport.typeAnnotation = tree.typeAnnotation;
        this.currentExport.annotations = tree.annotations;
        this.currentExport.docType = 'function';

        log.silly(tree.type, tree.commentBefore ? 'has comment' : '');
      }
    },
    visitClassDeclaration: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);
        this.currentExport.name = tree.name.identifierToken.value;
        this.currentExport.superClass = tree.superClass;
        this.currentExport.annotations = tree.annotations;
        this.currentExport.elements = tree.elements;
        this.currentExport.docType = 'class';
      }
    },
    visitAsyncFunctionDeclaration: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);
      }
    },

    visitExportDefault: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);
        this.currentExport.name = 'DEFAULT';
        this.currentExport.defaultExport = tree;
        // Default exports are either classes, functions or expressions
        // So we let the super class continue down...
        ParseTreeVisitor.prototype.visitExportDefault.call(this, tree);
      }
    },

    visitNamedExport: function(tree) {
      if ( this.currentExport ) {
        this.updateExport(tree);

        this.currentExport.namedExport = tree;
        this.currentExport.name = 'NAMED_EXPORT';
        // TODO: work out this bit!!
        // We need to cope with any export specifiers in the named export
      }
    },

    // TODO - if the export is an expression, find the thing that is being
    // exported and use it and its comments for docs

    updateExport: function(tree) {
      this.currentExport.comment = this.currentExport.comment || tree.commentBefore;
      this.currentExport.docType = tree.type;
    },

    visit: function(tree) {
      this.exports = [];
      ParseTreeVisitor.prototype.visit.call(this, tree);
    }
  };

  return ExportTreeVisitorImpl;
};