Add script to generate a file list

Author: kgryte

lib/node_modules/@stdlib/datasets/spam-assassin/scripts/file_list.js

@@ -0,0 +1,64 @@
+/*
+* Script to generate a file list.
+*/
+'use strict';
+
+// MAIN //
+
+var resolve = require( 'path' ).resolve;
+var writeFileSync = require( 'fs' ).writeFileSync;
+var readDirSync = require( '@stdlib/fs/read-dir' ).sync;
+
+
+// VARIABLES //
+
+var RE_EXT = /\.txt$/;
+var dirs = [
+	'easy-ham-1',
+	'easy-ham-2',
+	'hard-ham-1',
+	'spam-1',
+	'spam-2'
+];
+
+
+// MAIN //
+
+/**
+* Main execution sequence.
+*
+* @private
+*/
+function main() {
+	var files;
+	var fpath;
+	var dpath;
+	var opts;
+	var out;
+	var i;
+	var j;
+
+	out = [];
+	for ( i = 0; i < dirs.length; i++ ) {
+		dpath = resolve( __dirname, '..', 'data', dirs[ i ] );
+		files = readDirSync( dpath );
+		if ( files instanceof Error ) {
+			throw files;
+		}
+		for ( j = 0; j < files.length; j++ ) {
+			if (
+				files[ j ][ 0 ] !== '.' &&
+				RE_EXT.test( files[ j ] )
+			) {
+				out.push( dirs[ i ]+'/'+files[ j ] );
+			}
+		}
+	}
+	opts = {
+		'encoding': 'utf8'
+	};
+	fpath = resolve( __dirname, '..', 'data', 'file_list.json' );
+	writeFileSync( fpath, JSON.stringify( out, null, 2 ), opts );
+} // end FUNCTION main()
+
+main();