michael@0: var Cc = Components.classes;
michael@0: var Ci = Components.interfaces;
michael@0: var Cu = Components.utils;
michael@0: 
michael@0: // From prio.h
michael@0: const PR_RDWR        = 0x04;
michael@0: const PR_CREATE_FILE = 0x08;
michael@0: const PR_TRUNCATE    = 0x20;
michael@0: 
michael@0: Cu.import("resource://gre/modules/FileUtils.jsm");
michael@0: Cu.import("resource://gre/modules/NetUtil.jsm");
michael@0: 
michael@0: var gBasePath = "tests/dom/apps/tests/";
michael@0: var gMiniManifestTemplate = "file_packaged_app.template.webapp";
michael@0: var gAppTemplate = "file_packaged_app.template.html";
michael@0: var gAppName = "appname";
michael@0: var gDevName = "devname";
michael@0: var gDevUrl = "http://dev.url";
michael@0: 
michael@0: function handleRequest(request, response) {
michael@0:   var query = getQuery(request);
michael@0: 
michael@0:   response.setHeader("Access-Control-Allow-Origin", "*", false);
michael@0: 
michael@0:   var packageSize = ("packageSize" in query) ? query.packageSize : 0;
michael@0:   var appName = ("appName" in query) ? query.appName : gAppName;
michael@0:   var devName = ("devName" in query) ? query.devName : gDevName;
michael@0:   var devUrl = ("devUrl" in query) ? query.devUrl : gDevUrl;
michael@0: 
michael@0:   // If this is a version update, update state, prepare the manifest,
michael@0:   // the application package and return.
michael@0:   if ("setVersion" in query) {
michael@0:     var version = query.setVersion;
michael@0:     setState("version", version);
michael@0:     var packageVersion = ("dontUpdatePackage" in query) ? version - 1 : version;
michael@0:     var packageName = "test_packaged_app_" + packageVersion + ".zip";
michael@0: 
michael@0:     setState("packageName", packageName);
michael@0:     var packagePath = "/" + gBasePath + "file_packaged_app.sjs?getPackage=" +
michael@0:                       packageName;
michael@0:     setState("packagePath", packagePath);
michael@0: 
michael@0:     if (version == packageVersion) {
michael@0:       // Create the application package.
michael@0:       var zipWriter = Cc["@mozilla.org/zipwriter;1"]
michael@0:                         .createInstance(Ci.nsIZipWriter);
michael@0:       var zipFile = FileUtils.getFile("TmpD", [packageName]);
michael@0:       zipWriter.open(zipFile, PR_RDWR | PR_CREATE_FILE | PR_TRUNCATE);
michael@0: 
michael@0:       // We want to run some tests without the manifest included in the zip.
michael@0:       if (version != "0") {
michael@0:         var manifestTemplate = gBasePath + gMiniManifestTemplate;
michael@0:         var manifest = makeResource(manifestTemplate, version, packagePath,
michael@0:                                     packageSize, appName, devName, devUrl);
michael@0:         addZipEntry(zipWriter, manifest, "manifest.webapp");
michael@0:       }
michael@0: 
michael@0:       var appTemplate = gBasePath + gAppTemplate;
michael@0:       var app = makeResource(appTemplate, version, packagePath, packageSize,
michael@0:                              appName, devName, devUrl);
michael@0:       addZipEntry(zipWriter, app, "index.html");
michael@0: 
michael@0:       zipWriter.close();
michael@0:     }
michael@0: 
michael@0:     response.setHeader("Content-Type", "text/html", false);
michael@0:     response.write("OK");
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   // Get the version from server state
michael@0:   var version = Number(getState("version"));
michael@0:   var packageName = String(getState("packageName"));
michael@0:   var packagePath = String(getState("packagePath"));
michael@0: 
michael@0:   var etag = getEtag(request, version);
michael@0: 
michael@0:   if (etagMatches(request, etag)) {
michael@0:     dump("Etags Match. Sending 304\n");
michael@0:     response.setStatusLine(request.httpVersion, "304", "Not modified");
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   response.setHeader("Etag", etag, false);
michael@0: 
michael@0:   // Serve the application package corresponding to the requested app version.
michael@0:   if ("getPackage" in query) {
michael@0:     var resource = readFile(packageName, true);
michael@0:     response.setHeader("Content-Type",
michael@0:                        "Content-Type: application/java-archive", false);
michael@0:     response.write(resource);
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   // Serve the mini-manifest corresponding to the requested app version.
michael@0:   if ("getManifest" in query) {
michael@0:     var template = gBasePath + gMiniManifestTemplate;
michael@0:     if (!("noManifestContentType" in query)) {
michael@0:       response.setHeader("Content-Type",
michael@0:                          "application/x-web-app-manifest+json", false);
michael@0:     }
michael@0:     packagePath = "wrongPackagePath" in query ? "" : packagePath;
michael@0:     var manifest = makeResource(template, version, packagePath, packageSize,
michael@0:                                 appName, devName, devUrl);
michael@0:     response.write(manifest);
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   response.setHeader("Content-type", "text-html", false);
michael@0:   response.write("KO");
michael@0: }
michael@0: 
michael@0: function getQuery(request) {
michael@0:   var query = {};
michael@0:   request.queryString.split('&').forEach(function (val) {
michael@0:     var [name, value] = val.split('=');
michael@0:     query[name] = unescape(value);
michael@0:   });
michael@0:   return query;
michael@0: }
michael@0: 
michael@0: function getEtag(request, version) {
michael@0:   return request.queryString.replace(/&/g, '-').replace(/=/g, '-') +
michael@0:          '-' + version;
michael@0: }
michael@0: 
michael@0: function etagMatches(request, etag) {
michael@0:   return request.hasHeader("If-None-Match") &&
michael@0:          request.getHeader("If-None-Match") == etag;
michael@0: }
michael@0: 
michael@0: // File and resources helpers
michael@0: 
michael@0: function addZipEntry(zipWriter, entry, entryName) {
michael@0:   var stream = Cc["@mozilla.org/io/string-input-stream;1"]
michael@0:                .createInstance(Ci.nsIStringInputStream);
michael@0:   stream.setData(entry, entry.length);
michael@0:   zipWriter.addEntryStream(entryName, Date.now(),
michael@0:                            Ci.nsIZipWriter.COMPRESSION_BEST, stream, false);
michael@0: }
michael@0: 
michael@0: function readFile(path, fromTmp) {
michael@0:   var dir = fromTmp ? "TmpD" : "CurWorkD";
michael@0:   var file = Cc["@mozilla.org/file/directory_service;1"]
michael@0:              .getService(Ci.nsIProperties)
michael@0:              .get(dir, Ci.nsILocalFile);
michael@0:   var fstream = Cc["@mozilla.org/network/file-input-stream;1"]
michael@0:                 .createInstance(Ci.nsIFileInputStream);
michael@0:   var split = path.split("/");
michael@0:   for(var i = 0; i < split.length; ++i) {
michael@0:     file.append(split[i]);
michael@0:   }
michael@0:   fstream.init(file, -1, 0, 0);
michael@0:   var data = NetUtil.readInputStreamToString(fstream, fstream.available());
michael@0:   fstream.close();
michael@0:   return data;
michael@0: }
michael@0: 
michael@0: function makeResource(templatePath, version, packagePath, packageSize,
michael@0:                       appName, developerName, developerUrl) {
michael@0:   var res = readFile(templatePath, false)
michael@0:             .replace(/VERSIONTOKEN/g, version)
michael@0:             .replace(/PACKAGEPATHTOKEN/g, packagePath)
michael@0:             .replace(/PACKAGESIZETOKEN/g, packageSize)
michael@0:             .replace(/NAMETOKEN/g, appName)
michael@0:             .replace(/DEVELOPERTOKEN/g, developerName)
michael@0:             .replace(/DEVELOPERURLTOKEN/g, developerUrl);
michael@0:   return res;
michael@0: }