VulpesOS/gaia/tools/extensions/httpd/bootstrap.js

/* -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- /
/* vim: set shiftwidth=2 tabstop=2 autoindent cindent expandtab: */

'use strict';

function startup(data, reason) {
  const CC = Components.Constructor;
  const Cc = Components.classes;
  const Ci = Components.interfaces;
  const Cu = Components.utils;
  const Cm = Components.manager.QueryInterface(Ci.nsIComponentRegistrar);

  Cu.import('resource://gre/modules/XPCOMUtils.jsm');
  Cu.import('resource://gre/modules/Services.jsm');
  try {
    // Gecko < 21
    Cu.import('resource:///modules/devtools/dbg-client.jsm');
  } catch (e) {
    // Gecko >= 21
    Cu.import('resource://gre/modules/devtools/dbg-client.jsm');
  }

  // Make sure all applications are considered launchable
  Cu.import('resource://gre/modules/Webapps.jsm');
  const env = Cc['@mozilla.org/process/environment;1'].
              getService(Ci.nsIEnvironment);
  DOMApplicationRegistry.allAppsLaunchable = true;

  const LANG_EN = 'en-US';
  const GAIA_DOMAIN = Services.prefs.getCharPref("extensions.gaia.domain");
  const GAIA_APPDIRS = Services.prefs.getCharPref("extensions.gaia.appdirs");
  const GAIA_ALLAPPDIRS = Services.prefs.getCharPref("extensions.gaia.allappdirs");
  const GAIA_DIR = Services.prefs.getCharPref("extensions.gaia.dir");
  const GAIA_PORT = Services.prefs.getIntPref("extensions.gaia.port");
  const GAIA_OFFICIAL = Services.prefs.getBoolPref("extensions.gaia.official");
  const LOCALES_FILE =
    Services.prefs.getCharPref("extensions.gaia.locales_file");
  const LOCALE_BASEDIR =
    Services.prefs.getCharPref("extensions.gaia.locale_basedir");

  const LocalFile = CC('@mozilla.org/file/local;1',
                       'nsILocalFile',
                       'initWithPath');

  const ScriptableInputStream = CC('@mozilla.org/scriptableinputstream;1',
                                   'nsIScriptableInputStream',
                                   'init');

  let debugEnv = env.get('DEBUG');
  let DEBUG = debugEnv === '*' || debugEnv.includes('bootstrap');

  let l10nManager, utils;
  function debug(data) {
    if (!DEBUG)
      return;
    dump(data + '\n');
  }

  function isl10nFile(fname) {
    var matched;
    return fname.contains('.ini') || fname.contains('manifest.webapp');
  }

  function startupHttpd(baseDir, port) {
    const httpdURL = 'chrome://httpd.js/content/httpd.js';
    let httpd = {};
    env.set('GAIA_DIR', GAIA_DIR);
    Services.scriptloader.loadSubScript(httpdURL, httpd);
    let server = new httpd.nsHttpServer();
    server.registerDirectory('/', new LocalFile(baseDir));
    server.registerContentType('appcache', 'text/cache-manifest');
    server.start(port);

    // Add {appname}.domain.tld:port to the list of domains handle by httpd
    let identity = server.identity;
    let scheme = 'http';

    let host = GAIA_DOMAIN;
    identity.add(scheme, host, port);

    let directories = getDirectories(GAIA_ALLAPPDIRS.split(' '));
    directories.forEach(function appendDir(name) {
      // Some app names can cause a raise here, preventing other apps
      // from being added.
      try {
        identity.add(scheme, name + '.' + host, port);
      } catch (e) {
        dump(e);
      }
    });
    identity.add(scheme,  'theme.' + host, port);

    let commonjs = {
      GAIA_BUILD_DIR: 'file://' + GAIA_DIR.replace(/\\/g, '/') + '/build/'
    };
    // Loading |require| then we can use build scripts of gaia.
    Services.scriptloader.loadSubScript(commonjs.GAIA_BUILD_DIR +
      '/xpcshell-commonjs.js', commonjs);
    utils = commonjs.require('./utils');

    if (LOCALE_BASEDIR) {
      let appDirs = GAIA_APPDIRS.split(' ');
      let multilocale = commonjs.require('./multilocale');
      let sharedDir = utils.getFile(GAIA_DIR, 'shared');
      l10nManager = new multilocale.L10nManager(GAIA_DIR,
        LOCALES_FILE, LOCALE_BASEDIR);
      let filesInShared = utils.ls(sharedDir, true);
      let localesFile = utils.resolve(LOCALES_FILE, GAIA_DIR);

      // Finding out which file we need to handle.
      appDirs.forEach(function(appDir) {
        let appDirFile = utils.getFile(appDir);
        let filesInApp = utils.ls(appDirFile, true);

        [...filesInApp, ...filesInShared].forEach(function(file) {
          let fname = file.leafName;
          let relativePath = file.path.substr(GAIA_DIR.length);
          // localized ini and manifest files will be generated in runtime
          // so we need handle path if client request those files.
          // and we use "LocalizationHandler" to handle them.
          if (isl10nFile(fname)) {
            server.registerPathHandler(relativePath, LocalizationHandler);
          }
        });
      });

      // languages.json
      server.registerFile('/shared/resources/languages.json', localesFile);
      server.registerDirectory('/locales/',
        new LocalFile(l10nManager.localeBasedir));
    }

    let testAgentBoilerplateDir = utils.getFile(GAIA_DIR, 'dev_apps',
      'test-agent', 'common', 'test', 'boilerplate');
    let proxyFile = testAgentBoilerplateDir.clone();
    let sandboxFile = testAgentBoilerplateDir.clone();
    proxyFile.append('_proxy.html');
    sandboxFile.append('_sandbox.html');
    server.registerFile('/test/unit/_proxy.html', proxyFile);
    server.registerFile('/test/unit/_sandbox.html', sandboxFile);

    server.registerPathHandler('/marionette', MarionetteHandler);

    server.registerDirectory(
      '/common/', new LocalFile(baseDir + '/dev_apps/test-agent/common')
    );
    server.registerDirectory('/shared/', new LocalFile(baseDir + '/shared'));

    let brandingLocalFile = new LocalFile(
      baseDir + '/shared/resources/branding/' + (GAIA_OFFICIAL ? 'official' : 'unofficial')
    );

    server.registerDirectory('/shared/resources/branding/', brandingLocalFile);
  }

  function getDirectories(appDirs) {
    return appDirs.map(function (path) {
      let appsDir = Cc['@mozilla.org/file/local;1']
                      .createInstance(Ci.nsILocalFile);
      appsDir.initWithPath(path);
      return appsDir.leafName;
    });
  }

  function LongPoll(connid, transport) {
    this.transport = transport;
    this.connid = connid;
    this.buffer = [];

    this.outstanding = {
      push: null,
      noop: null,
      timeout: null
    };
  }

  LongPoll.prototype = {
    cast: function cast(msg) {
      let outstanding = this.outstanding;
      clearTimeout(outstanding.noop);
      this.buffer.push(msg);

      if (outstanding.push === null)
        return;

      debug('cast ' + JSON.stringify(msg) + ' ' + outstanding.push);
      this.drain_queue(outstanding.push);
      outstanding.push.finish();
      outstanding.push = null;

      outstanding.timeout = setTimeout(function close(uid) {
        Connections.close(uid);
      }, 5000, this.connid);
    },

    drain_queue: function drain_queue(response) {
      let queue = this.buffer;
      this.buffer = [];

      let json = { 'messages': [] };
      queue.forEach(function(chunk) {
        if (chunk.noop) {
          //we need to send something to keep the client polling
          json.messages.push({});
          return;
        }
        json.messages.push({ 'id': this.connid, 'response': chunk });
      }.bind(this));

      response.write(JSON.stringify(json));
    },

    send: function send(obj) {
      this.transport.send(obj);
    }
  };

  let transports = {};

  function getUID() {
   return Cc['@mozilla.org/uuid-generator;1'].getService(Ci.nsIUUIDGenerator)
                                             .generateUUID()
                                             .toString();
  }

  var Connections = {
    _connections: {},
    close: function connections_close(uid) {
      if (uid in this._connections) {
        this._connections[uid].transport.close();
        delete this._connections[uid];
      }
    },

    add: function connections_add(uid, connection) {
      this._connections[uid] = connection;
    },

    get: function connections_get(uid) {
      return this._connections[uid] || null;
    }
  };

  var LocalizationHandler = {
    // we only handle GET method because we always use GET method to get those
    // files.
    handle: function lh_handle(request, response) {
      if (request.method === 'GET' && l10nManager) {
        let locales = JSON.parse(JSON.stringify(l10nManager.locales));
        // no need to handle en-US.
        var enIndex = locales.indexOf('en-US');
        if (enIndex !== -1) {
          locales.splice(enIndex, 1);
        }
        let {getFile, getFileContent} = utils;
        let {modifyLocaleIni, serializeIni} = l10nManager;

        let file = getFile(GAIA_DIR, request.path);
        if (file.leafName.contains('.ini')) {
          response.setHeader('Content-Type', 'text/plain');
          let iniOriginal = getFileContent(file);
          let modifiedIni = serializeIni(modifyLocaleIni(iniOriginal, locales));
          response.write(modifiedIni);
        } else if (file.leafName.contains('manifest.webapp')) {
          let buildDirectoryFile = getFile(GAIA_DIR, 'build_stage',
            file.parent.leafName);
          response.setHeader('Content-Type', 'application/json');
          let webapp = {
            manifestFile: file,
            sourceDirectoryFile: file.parent,
            sourceDirectoryName: file.parent.leafName,
            buildDirectoryFile: buildDirectoryFile
          };
          let manifest = l10nManager.localizeManifest(webapp);
          response.write(JSON.stringify(manifest));
        }
      }
    }
  }

  var MarionetteHandler = {

    /**
     * Adding only test-agent for now.
     * In the future individual apps may need
     * access as well.
     */
    crosDomains: [
      'http://test-agent.gaiamobile.org',
      'http://test-agent.gaiamobile.org:8080',
      'http://test-agent.trunk.gaiamobile.org',
      'http://test-agent.trunk.gaiamobile.org:8080'
    ],

    // nsIHttpRequestHandler
    handle: function mh_handle(request, response) {
      try {
        this.handleCrosHeaders(request, response);

        switch (request.method) {
          case 'GET':
            this.onGet(request, response);
            break;
          case 'POST':
            this.onPost(request, response);
            break;
          case 'OPTIONS':
            this.onOptions(request, response);
            break;
          case 'PUT':
            this.onPut(request, response);
            break;
          case 'DELETE':
            this.onDelete(request, response);
            break;
          default:
            this.onDefault(request, response);
            break;
        }
      } catch (e) {
        response.setStatusLine('1.1', 500, 'Internal Server Error');
        response.write(e.toString() + e.stack == undefined ? '' : ' ' + e.stack);
      }
    },

    onOptions: function mh_options(request, response) {
      response.setStatusLine('1.1', 200, 'OK');
    },

    handleCrosHeaders: function mh_options(request, response) {
      if(!request.hasHeader('Origin'))
        return;

      let methods = 'GET, POST, PUT, DELETE, OPTIONS';
      let origin = request.getHeader('Origin');

      if (!origin || this.crosDomains.indexOf(origin) === -1) {
        return;
      }

      response.setHeader(
        'Access-Control-Allow-Origin',
        origin
      );

      response.setHeader(
        'Access-Control-Allow-Credentials',
        'true'
      );

      response.setHeader(
        'Access-Control-Max-Age',
        '2500'
      );

      response.setHeader(
        'Access-Control-Allow-Methods',
        methods
      );

      response.setHeader('Access-Control-Allow-Headers', 'Content-Type, *');
    },

    onGet: function mh_get(request, response) {
      debug('/marionette?' + request.queryString);
      if (request.queryString.length === 0) {
        response.setStatusLine('1.1', 404, 'Not Found');
        response.write(JSON.stringify({ 'error': 'connection not found' }));
        return;
      }
      response.setStatusLine('1.1', 200, 'OK');
      response.setHeader('Content-Type', 'application/json');

      let uid = request.queryString.split('=')[0];
      let connection = Connections.get(uid);
      if (!connection) {
        debug('Could not get connection for ' + request.queryString);
        return;
      }

      clearTimeout(connection.outstanding.timeout);

      if (connection.buffer.length !== 0) {
        connection.drain_queue(response);
        return;
      }

      response.processAsync();

      connection.outstanding.push = response;
      connection.outstanding.noop = setTimeout(function() {
        debug('sending noop');
        connection.cast({ 'noop': true });
      }, 25000);
    },

    onPost: function mh_post(request, response) {
      response.setStatusLine('1.1', 200, 'OK');
      response.setHeader('Content-Type', 'application/json');

      let uid = getUID();
      response.write(JSON.stringify({ 'id': uid }));

      request.bodyInputStream.asyncWait(function(stream) {
        let scriptable = new ScriptableInputStream(stream);
        let json = scriptable.read(scriptable.available());
        let { server: server, port: port } = JSON.parse(json);

        let hostport = server + ':' + port;
        if (hostport in transports) {
          Connections.close(transports[hostport]);
        }

        debug('Opening a transport socket to the debugger...');
        let transport = debuggerSocketConnect(server, parseInt(port));
        let connection = new LongPoll(uid, transport);
        Connections.add(uid, connection);
        transports[hostport] = uid;

        transport.hooks = {
          onPacket: function onPacket(pack) {
            debug('onpacket ' + JSON.stringify(pack));
            connection.cast(pack);
          },
          onClosed: function onClosed(status) {
            debug('onclosed ' + JSON.stringify(status));
          }
        };

        transport.ready();
      }, 0, 0, Services.tm.currentThread);
    },

    onPut: function mh_put(request, response) {
      response.setStatusLine('1.1', 201, 'Accepted');
      response.setHeader('Content-Type', 'text/plain', false);
      response.write('');

      let uid = request.queryString.split('=')[0];
      let connection = Connections.get(uid);
      if (!connection) {
        debug('Could not get connection for ' + request.queryString);
        return;
      }

      request.bodyInputStream.asyncWait(function(stream) {
        let scriptable = new ScriptableInputStream(stream);
        let json = scriptable.read(scriptable.available());
        connection.send(JSON.parse(json));
      }, 0, 0, Services.tm.currentThread);
    },

    onDelete: function mh_delete(request, response) {
      Connections.close(request.queryString.split('=')[0]);
    },

    onDefault: function mh_default(request, response) {
      response.setStatusLine('1.1', 405, 'Method Not Allowed');
      response.write('');
    }
  };

  function registerAppProtocol() {
    let classID = Components.ID('{e3bace70-f074-11e3-ac10-0800200c9a66}');
    let contract = '@mozilla.org/network/protocol;1?name=app';
    let instance = null;

    function AppProtocol() {}

    AppProtocol.prototype = {
      scheme: 'app',
      classID: classID,
      QueryInterface: XPCOMUtils.generateQI([Ci.nsIProtocolHandler]),
      newURI: function(aSpec, aOriginCharset, aBaseURI) {
        let uri = Cc["@mozilla.org/network/standard-url;1"]
                  .createInstance(Ci.nsIStandardURL);
        uri.init(Ci.nsIStandardURL.URLTYPE_STANDARD, -1, aSpec, aOriginCharset,
                 aBaseURI);
        return uri.QueryInterface(Ci.nsIURI);
      },
      newChannel2: function(aURI, aLoadInfo) {
        let url = aURI.QueryInterface(Ci.nsIURL);
        let appId = aURI.host;
        let fileSpec = url.filePath;
        let spec = 'http://' + appId + ':' + GAIA_PORT + fileSpec;
        let uri = Services.io.newURI(spec, null, null);
        let channel = Services.io.newChannelFromURIWithLoadInfo(uri, aLoadInfo);
        channel.QueryInterface(Ci.nsIChannel).originalURI = aURI;
        return channel;
      },
      newChannel: function(aURI) {
        return newChannel2(aURI, null);
      }
    };

    let newFactory = {
      createInstance: function(outer, iid) {
        if (outer) {
          throw Components.results.NS_ERROR_NO_AGGREGATION;
        }
        if (instance === null) {
          instance = new AppProtocol();
        }
        return instance;
      },
      lockFactory: function(aLock) {
         throw Components.results.NS_ERROR_NOT_IMPLEMENTED;
      },
      QueryInterface: XPCOMUtils.generateQI([Ci.nsIFactory])
    };
    Cm.registerFactory(classID, '', contract, newFactory);
  }

  registerAppProtocol();
  startupHttpd(GAIA_DIR, GAIA_PORT);
}

function shutdown(data, reason) {
}

function install(data, reason) {
}

function uninstall(data, reason) {
}