Index: lib/application.js |
=================================================================== |
new file mode 100644 |
--- /dev/null |
+++ b/lib/application.js |
@@ -0,0 +1,234 @@ |
+/* |
+ * This Source Code is subject to the terms of the Mozilla Public License |
+ * version 2.0 (the "License"). You can obtain a copy of the License at |
+ * http://mozilla.org/MPL/2.0/. |
+ */ |
+ |
+Cu.import( "resource://gre/modules/FileUtils.jsm" ); |
+ |
+let {Storage} = require( "storage" ); |
+let {Instruction, Instruction_Set, Input_String, Input_File} = require( "instruction" ); |
+let {Long_Task} = require( "task" ); |
+let {Crawler} = require( "crawler" ); |
+let {Logger} = require( "logger" ); |
+ |
+/** |
+ * @fileOverview The purpose of this module is to isolate the parameters needed to run a crawl from the means by which |
+ * those parameters are collected by top-level interface code. There are two basic forms of interface: interactive and |
+ * automated. The interactive interface uses a XUL dialog and starts a crawl when the user presses a button. The |
+ * automated interface handles command line options and starts a crawl when the command line arrives. |
+ * |
+ * There are two kinds of arguments to the Application_Session constructor. The first kind are internal resources needed |
+ * to perform the crawl. Notable here is the 'window' argument, which designates where to open the tabs for each crawled |
+ * site. This comes easy in the interactive version, but a window must be created for the automated one. The second kind |
+ * of argument are the external parameters that the user manipulates, such as the specification for the instructions. |
+ * |
+ * Internal. |
+ * - window. The window in which to run the tabs. |
+ * - progress. A progress monitor. Only useful for interactive modes. |
+ * - output (transient). If required, output to a log window in YAML format. |
+ * External. |
+ * - instructions. Contains a list of all the sites to crawl. There are multiple ways of obtaining this set, all of |
+ * which start with a text stream which is then parsed and converted to an {Instruction_Set} object. Mandatory. |
+ * - Local input file. Parameter is a local file name. By convention local files are in YAML format. |
+ * - Text string. Parameter is the text itself plus the name of one of the formats. |
+ * - YAML format. Used for an interactive, multiline textbox. (Not implemented) |
+ * - JSON format. Used for accepting instructions from the command line. (Not implemented) |
+ * - output (persistent). The recorded results of the crawl. |
+ * - local file name. Optional for interactive use, since a log window might be adequate. Mandatory for automated |
+ * use, otherwise what's the point? |
+ * - format. Either YAML or JSON. Default to JSON. |
+ * - leave_open. Leave the tabs open after crawling them. Only useful for interaction, say, in order to examine the DOM |
+ * after a trial. Always false for automated use. Default is false for interactive use. |
+ * - number_of_tabs. The number of simultaneous tabs. Built-in default if not present. |
+ */ |
+ |
+/** |
+ * @param {Window} window |
+ * @param {boolean} leave_open |
+ * @param {number} n_tabs |
+ * @param {number} time_limit |
+ * @param {Function} progress |
+ * @constructor |
+ */ |
+var Application_Session = function( window, leave_open, n_tabs, time_limit, progress ) |
+{ |
+ this.runnable = true; |
+ this.instructions = []; |
+ this.outputs = []; |
+ this.window = window; |
+ this.leave_open = leave_open; |
+ this.n_tabs = n_tabs; |
+ this.time_limit = time_limit; |
+ this.progress = progress; |
+}; |
+exports.Application_Session = Application_Session; |
+ |
+Application_Session.prototype.run = function( finisher, catcher ) |
+{ |
+ this.finisher = finisher; |
+ this.catcher = catcher; |
+ if ( !this.runnable ) |
+ { |
+ this._run_catch( new Error( "Application_Session is not runnable" ) ); |
+ return; |
+ } |
+ this.runnable = false; |
+ |
+ this.current_crawler = new Crawler( |
+ this.instructions, this.outputs, this.window, |
+ this.time_limit, this.leave_open, this.n_tabs |
+ ); |
+ |
+ |
+ if ( this.progress ) |
+ { |
+ /* |
+ * Add an instance-specific notice member to the crawler's progress instance. This is cleaner than |
+ * bothering with a subclass of the progress-notification class. |
+ */ |
+ this.current_crawler.progress.notice = function( notice ) |
+ { |
+ notice( this ); |
+ }.bind( this.current_crawler.progress, this.progress ); |
+ } |
+ this.current_crawl = new Long_Task( this.current_crawler, this.instructions.size * 3 ); |
+ this.current_crawl.run( this._run_finally.bind( this ), this._run_catch.bind( this ) ); |
+}; |
+ |
+/** |
+ * |
+ * @param {*} ex |
+ * Value to treat as a thrown exception. Treated as an opaque type. |
+ * @private |
+ */ |
+Application_Session.prototype._run_catch = function( ex ) |
+{ |
+ if ( this.catcher ) this.catcher( ex ); |
+ this._run_finally(); |
+}; |
+ |
+Application_Session.prototype._run_finally = function() |
+{ |
+ if ( this.finisher ) this.finisher(); |
+}; |
+ |
+/** |
+ * Close the application session. |
+ */ |
+Application_Session.prototype.close = function() |
+{ |
+ if ( this.current_crawl ) |
+ { |
+ this.current_crawl.close(); |
+ this.current_crawl = null; |
+ } |
+ if ( this.current_crawler ) |
+ { |
+ this.current_crawler.close(); |
+ this.current_crawler = null; |
+ } |
+}; |
+ |
+/** |
+ * Set an input string to specify the instruction set. |
+ * |
+ * @param {string} s |
+ */ |
+Application_Session.prototype.set_input_string = function( s ) |
+{ |
+ this.instructions = new Instruction_Set.Parsed( new Input_String( s ) ); |
+}; |
+ |
+/** |
+ * Set an input file from which to read the instruction set. |
+ * |
+ * @param {string} file_path |
+ */ |
+Application_Session.prototype.set_input_file = function( file_path ) |
+{ |
+ var f = new FileUtils.File( file_path ); |
+ if ( !f.exists() ) |
+ { |
+ throw new Error( "Input file does not exist. path = " + f.path ); |
+ } |
+ if ( !f.isFile() ) |
+ { |
+ throw new Error( "Input file path does not name a file. path = " + f.path ); |
+ } |
+ if ( !f.isReadable() ) |
+ { |
+ throw new Error( "Input file is not readable. path = " + f.path ); |
+ } |
+ this.instructions = new Instruction_Set.Parsed( new Input_File( f ) ); |
+}; |
+ |
+/** |
+ * Add a storage-encoding pair to the output list for the current session. |
+ * |
+ * @param storage |
+ * @param {String} encode |
+ * Either "JSON" or "YAML". |
+ */ |
+Application_Session.prototype.add_output = function( storage, encode ) |
+{ |
+ this.outputs.push( { storage: storage, encode: encode } ); |
+}; |
+ |
+/** |
+ * Add a file to the output list. Throws if it detects obvious problems with the arguments specifying a valid output, |
+ * namely, if the directory path is not a writable directory or if the encoding is invalid. |
+ * |
+ * @param {string} directory_path |
+ * Directory path for the output file. |
+ * @param {string} base_name |
+ * Base name for the output file. |
+ * @param {boolean} append_timestamp |
+ * Whether to append a timestamp to the base name, or not. |
+ * @param {string} encode |
+ * Either "JSON" or "YAML". |
+ * @return {string} |
+ * The constructed file name. |
+ */ |
+Application_Session.prototype.add_output_file = function( directory_path, base_name, append_timestamp, encode ) |
+{ |
+ var file = Cc["@mozilla.org/file/local;1"].createInstance( Ci.nsILocalFile ); |
+ file.initWithPath( directory_path ); |
+ if ( !file.exists() ) |
+ { |
+ throw new Error( "Output directory path does not exist. path = " + directory_path ); |
+ } |
+ if ( !file.isDirectory() ) |
+ { |
+ throw new Error( "Output directory path does not name a directory. path = " + directory_path ); |
+ } |
+ if ( !file.isWritable() ) |
+ { |
+ throw new Error( "Output directory is not writable. path = " + directory_path ); |
+ } |
+ var file_name = ( base_name && base_name.length > 0 ) ? base_name : "crawl-results"; |
+ if ( append_timestamp ) |
+ { |
+ file_name += filename_timestamp(); |
+ } |
+ switch ( encode ) |
+ { |
+ case "JSON": |
+ file_name += ".json"; |
+ break; |
+ case "YAML": |
+ file_name += ".yaml"; |
+ break; |
+ default: |
+ throw new Error( "Invalid encoding = " + encode + ". Must be JSON or YAML." ); |
+ } |
+ file.append( file_name ); |
+ this.add_output( new Storage.Local_File( file ), encode ); |
+ return file.path; |
+}; |
+ |
+function filename_timestamp() |
+{ |
+ var s = Logger.timestamp(); |
+ return "_" + s.substr( 0, 10 ) + "_" + s.substr( 11, 2 ) + "-" + s.substr( 14, 2 ) + "-" + s.substr( 17, 2 ); |
+} |