module.exports = AMDFormatter;

var CommonJSFormatter = require("./common");
var util              = require("../util");
var b                 = require("ast-types").builders;
var _                 = require("lodash");

function AMDFormatter(file) {
  this.file = file;
  this.ids  = {};
}

util.inherits(AMDFormatter, CommonJSFormatter);

AMDFormatter.prototype.transform = function (ast) {
  var program = ast.program;
  var body    = program.body;

  // build an array of module names

  var names = [b.literal("exports")];
  _.each(this.ids, function (id, name) {
    names.push(b.literal(name));
  });
  names = b.arrayExpression(names);

  // build up define container

  var params = _.values(this.ids);
  params.unshift(b.identifier("exports"));

  var container = b.functionExpression(null, params, b.blockStatement(body));
  var call = b.callExpression(b.identifier("define"), [names, container]);

  program.body = [b.expressionStatement(call)];
};

AMDFormatter.prototype._push = function (node) {
  var id  = node.source.value;
  var ids = this.ids;

  if (ids[id]) {
    return ids[id];
  } else {
    return this.ids[id] = b.identifier(this.file.generateUid(id));
  }
};

AMDFormatter.prototype.import = function (node) {
  this._push(node);
};

AMDFormatter.prototype.importSpecifier = function (specifier, node, nodes) {
  var key = util.getSpecifierName(specifier);
  var id  = specifier.id;

  // import foo from "foo";
  if (specifier.default) {
    id = b.identifier("default");
  }

  var ref;

  if (specifier.type === "ImportBatchSpecifier") {
    // import * as bar from "foo";
    ref = this._push(node);
  } else {
    // import foo from "foo";
    ref = b.memberExpression(this._push(node), id, false);
  }

  nodes.push(b.variableDeclaration("var", [
    b.variableDeclarator(key, ref)
  ]));
};

AMDFormatter.prototype.exportSpecifier = function (specifier, node, nodes) {
  var self = this;
  return this._exportSpecifier(function () {
    return self._push(node);
  }, specifier, node, nodes);
};