wappalyzer/drivers/chrome/js/driver.js

/**
 * Chrome driver
 */

(function() {
	if ( wappalyzer == null ) return;

	var w = wappalyzer;

	var
		debug = true,
		tab,
		tabCache = {}
		;

	w.driver = {
		/**
		 * Log messages to console
		 */
		log: function(args) {
			if ( debug ) {
				console.log(args.message);
			}
		},

		/**
		 * Initialize
		 */
		init: function(callback) {
			w.log('init');

			chrome.browserAction.setBadgeBackgroundColor({ color: [255, 102, 0, 255] });

			chrome.extension.onRequest.addListener(function(request, sender, sendResponse) {
				if ( typeof(request.msg != 'undefined' ) ) {
					w.log('request: ' + request.msg);

					switch ( request.msg ) {
						case 'analyze':
							tab = sender.tab;

							chrome.browserAction.setBadgeText({ tabId: tab.id, text: '' });

							tabCache[tab.id] = {
								count: 0,
								appsDetected: {}
								};

							w.analyze(tab.url, tab.url, { html: request.html, env: request.env });

							break;
						case 'get_apps':
							sendResponse({ tabCache: tabCache[request.tab.id] });

							break;
					}
				}
			});

			chrome.tabs.getAllInWindow(null, function(tabs) {
				tabs.map(function(tab) {
					if ( tab.url.match(/https?:\/\//) ) {
						chrome.tabs.executeScript(tab.id, { file: 'js/content.js' });
					}
				})
			});

			chrome.tabs.onRemoved.addListener(function(tabId) {
				w.log('remove tab');

				tabCache[tabId] = null;
			});
		},

		/**
		 * Display apps
		 */
		displayApps: function() {
			var count = w.detected[tab.url].length.toString();

			tabCache[tab.id] = {
				count: count,
				appsDetected: w.detected[tab.url]
				};

			chrome.browserAction.setBadgeText({ tabId: tab.id, text: count });
		},
	};

	w.init();
})();

/*
var wappalyzer = (function(){
	self = {
		debug:    false,
		tabCache: {},

		log: function(message) {
			if ( self.debug && message ) {
				console.log(message);
			}
		},

		init: function() {
			self.log('init');

			chrome.browserAction.setBadgeBackgroundColor({ color: [255, 102, 0, 255] });

			chrome.extension.onRequest.addListener(function(request, sender, sendResponse) {
				if ( typeof(request.msg != 'undefined' ) ) {
					self.log('request: ' + request.msg);

					switch ( request.msg ) {
						case 'analyze':
							wappalyzer.analyze(sender.tab.id, sender.tab.url, request.html, request.env);

							break;
						case 'get_apps':
							sendResponse({ tabCache: wappalyzer.tabCache[request.tab.id] });

							break;
					}
				}
			});

			chrome.tabs.getAllInWindow(null, function(tabs) {
				for ( i in tabs ) {
					if ( tabs[i].url.match(/https?:\/\//) ) {
				chrome.tabs.executeScript(tabs[i].id, { file: 'content.js' });
				}
				}
			});

			chrome.tabs.onRemoved.addListener(function(tabId) {
				self.log('remove tab');

				wappalyzer.tabCache[tabId] = null;
			});
		},

		analyze: function(tabId, url, html, environmentVars) {
			var app;
			chrome.browserAction.setBadgeText({ tabId: tabId, text: '' });

			wappalyzer.tabCache[tabId] = {
				count: 0,
				appsDetected: {}
			};

			if ( html && html.length > 50000 ) {
				html = html.substring(0, 25000) + html.substring(html.length - 25000, html.length);
			}

			for ( var appName in wappalyzer.apps ) {
				app = wappalyzer.apps[appName];
				// Scan html
				if ( html ) {

					if ( typeof(wappalyzer.tabCache[tabId].appsDetected[appName]) == 'undefined' ) {
						if ( typeof(app.html) != 'undefined' ) {
							var regex = app.html;

							if ( regex.test(html) ) {
								wappalyzer.register(tabId, appName);
							}
						}
					}
				}

				// Scan script tags
				if ( html && typeof app.script != 'undefined' ) {
					var
						regex = /<script[^>]+src=("|')([^"']+)\1/ig,
									match = []
										;

					while ( match = regex.exec(html) ) {
						if ( app.script.test(match[2]) ) {
							wappalyzer.register(tabId, appName);

							break;
						}
					}
				}

				// Scan meta tags
				if ( html && typeof app.meta != 'undefined' ) {
					var
						regex = /<meta[^>]+>/ig,
									match = []
										;

					while ( match = regex.exec(html) ) {
						for ( meta in app.meta ) {
							if ( new RegExp('name=["\']' + meta + '["\']', 'i').test(match) ) {
								var content = match.toString().match(/content=("|')([^"']+)("|')/i);

								if ( app.meta[meta].test(content[2]) ) {
									wappalyzer.register(tabId, appName);

									break;
								}
							}
						}
					}
				}

				// Scan url
				if ( typeof(wappalyzer.tabCache[tabId].appsDetected[appName]) == 'undefined' ) {
					if ( url && typeof(app.url) != 'undefined' ) {
						var regex = app.url;

						if ( regex.test(url) ) {
							wappalyzer.register(tabId, appName);
						}
					}
				}

				// Scan environment variables
				if ( typeof(wappalyzer.tabCache[tabId].appsDetected[appName]) == 'undefined' ) {
					if ( environmentVars && typeof app.env != 'undefined' ) {
						var regex = app.env;

						for ( var i in environmentVars ) {
							try {
								if ( regex.test(environmentVars[i]) ) {
									wappalyzer.register(tabId, appName);
								}
							}
							catch(e) { }
						}
					}
				}
			}

			html = null;
		},

		register: function(tabId, appName) {
			wappalyzer.tabCache[tabId].appsDetected[appName] = {
				cats: {},
				name: appName
			};

			for ( cat in wappalyzer.apps[appName].cats ) {
				wappalyzer.tabCache[tabId].appsDetected[appName].cats[cat] = wappalyzer.cats[wappalyzer.apps[appName].cats[cat]];
			}

			wappalyzer.tabCache[tabId].count = 0;

			for ( i in wappalyzer.tabCache[tabId].appsDetected ) {
				wappalyzer.tabCache[tabId].count ++;
			}

			chrome.browserAction.setBadgeText({ tabId: tabId, text: wappalyzer.tabCache[tabId].count.toString() });
		}
	};

	return self;
})();
*/