|
| 1 | + |
| 2 | +const WILDFIRE = require("wildfire-for-js"); |
| 3 | + |
| 4 | + |
| 5 | +exports.for = function (API) { |
| 6 | + |
| 7 | + API.WILDFIRE = WILDFIRE; |
| 8 | + |
| 9 | + |
| 10 | + function isEnabled () { |
| 11 | + return true; |
| 12 | + } |
| 13 | + |
| 14 | + |
| 15 | + var httpHeaderChannel = API.httpHeaderChannel = API.WILDFIRE.HttpHeaderChannel({ |
| 16 | + "enableTransport": false |
| 17 | + }); |
| 18 | + httpHeaderChannel.setNoReceiverCallback(function(id) { |
| 19 | + API.console.error("trying to log to unknown receiver: " + id); |
| 20 | + }); |
| 21 | + |
| 22 | + |
| 23 | + // TODO: Use a PINF directive to get list of receivers. |
| 24 | + var receivers = [ |
| 25 | + ["receivers/firephp", require("./firephp"), null /* api */, null /* exports */] |
| 26 | + ]; |
| 27 | + receivers.forEach(function (receiver) { |
| 28 | + receiver[2] = API.makeAPI(API, { |
| 29 | + name: receiver[0] |
| 30 | + }); |
| 31 | + receiver[3] = receiver[1].for(receiver[2]); |
| 32 | + }); |
| 33 | + |
| 34 | + |
| 35 | + |
| 36 | + var announceDispatcher = API.WILDFIRE.Dispatcher(); |
| 37 | + announceDispatcher.setProtocol('http://registry.pinf.org/cadorn.org/wildfire/@meta/protocol/announce/0.1.0'); |
| 38 | + announceDispatcher.setChannel(httpHeaderChannel); |
| 39 | + |
| 40 | + function getAnnounceMessageForRequest (request) { |
| 41 | + |
| 42 | + if (!getAnnounceMessageForRequest._forHostnames) { |
| 43 | + getAnnounceMessageForRequest._forHostnames = {}; |
| 44 | + } |
| 45 | + var cache = getAnnounceMessageForRequest._forHostnames; |
| 46 | + |
| 47 | + // TODO: Don't just use `request.hostname` for cache. |
| 48 | + // TODO: If config changes this announceMessage needs to be reset |
| 49 | + |
| 50 | + if (cache[request.hostname]) { |
| 51 | + return cache[request.hostname]; |
| 52 | + } |
| 53 | + |
| 54 | + cache[request.hostname] = new API.WILDFIRE.Message(); |
| 55 | + cache[request.hostname].setData(JSON.stringify({ |
| 56 | + // TODO: Populate crypto-based key |
| 57 | + "authkey": null, |
| 58 | + "receivers": httpHeaderChannel.receivers.map(function (receiver) { |
| 59 | + return receiver.getId(); |
| 60 | + }) |
| 61 | + })); |
| 62 | + |
| 63 | + return cache[request.hostname]; |
| 64 | + } |
| 65 | + |
| 66 | + API.REQUEST_OBSERVER.register(function (request) { |
| 67 | + if (!isEnabled()) return; |
| 68 | + |
| 69 | + if (!request.httpChannel.getRequestHeader("User-Agent").match(/\sFirePHP\/([\.|\d]*)\s?/)) { |
| 70 | + request.httpChannel.setRequestHeader("User-Agent", request.httpChannel.getRequestHeader("User-Agent") + " FirePHP/0.5", false); |
| 71 | + } |
| 72 | + |
| 73 | + // NOTE: Do NOT set this header. This only works with FirePHPCore 0.3.2 and most users are using |
| 74 | + // FirePHPCore 0.3.1. If someone does not want their user-agent modified they should use FirePHP 1.0 with DeveloperCompanion |
| 75 | + // and disable the 'extensions.devcomp.firePHPCoreCompatibility' option. |
| 76 | + // request.httpChannel.setRequestHeader("X-FirePHP-Version", "0.4", false); |
| 77 | + |
| 78 | + var announceMessage = getAnnounceMessageForRequest(request); |
| 79 | + if (announceMessage) { |
| 80 | + // dispatch announce message |
| 81 | + announceDispatcher.dispatch(announceMessage); |
| 82 | + |
| 83 | + // flush channel |
| 84 | + httpHeaderChannel.flush({ |
| 85 | + setMessagePart: function(name, value) { |
| 86 | + request.httpChannel.setRequestHeader(name, value, false); |
| 87 | + }, |
| 88 | + getMessagePart: function(name) { |
| 89 | + if (request.httpChannel.getRequestHeader) { |
| 90 | + return null; |
| 91 | + } |
| 92 | + // HACK: Do not use exception for flow control |
| 93 | + try { |
| 94 | + return request.httpChannel.getRequestHeader(name); |
| 95 | + } catch (err) { |
| 96 | + return null; |
| 97 | + } |
| 98 | + } |
| 99 | + }); |
| 100 | + } else { |
| 101 | + // TODO: Use a proper unique ID + counter. |
| 102 | + request.httpChannel.setRequestHeader("x-request-id", ""+(new Date().getTime()) + "" + Math.floor(Math.random()*1000+1), false); |
| 103 | + } |
| 104 | + |
| 105 | + // API.console.log("REQUEST", request); |
| 106 | + |
| 107 | + }); |
| 108 | + |
| 109 | + API.RESPONSE_OBSERVER.on("response", function (response) { |
| 110 | + if (!isEnabled()) return; |
| 111 | + |
| 112 | + // API.console.log("RESPONSE", response); |
| 113 | + |
| 114 | + try { |
| 115 | + |
| 116 | + httpHeaderChannel.parseReceived(response.headers, { |
| 117 | + "MozillaRequestObserverListener": { |
| 118 | + "httpChannel": response.httpChannel |
| 119 | + }, |
| 120 | + "id": response.request.id, |
| 121 | + "url": response.request.url, |
| 122 | + "hostname": response.request.hostname, |
| 123 | + "port": response.request.port, |
| 124 | + "method": response.request.method, |
| 125 | + "status": response.status, |
| 126 | + "contentType": response.contentType, |
| 127 | + "requestHeaders": response.request.headers |
| 128 | + }); |
| 129 | + |
| 130 | + } catch (err) { |
| 131 | + API.console.error(err); |
| 132 | + } |
| 133 | + |
| 134 | + }); |
| 135 | +} |
| 136 | + |
0 commit comments