censql/bin/pipeCommands/write.js

A NodeJS command line client for SAP HANA

var isValidPath = require('is-valid-path');
var path = require("path");
var fs = require("fs");
var colors = require("colors");

module.exports = function(linesIn, command) {

	var parts = command.trim().split(" ");

	if(parts.length < 2){
		parts[1] = "";
	}

    var fileLoc = parts[1];

    /**
 	 * Check if a file location was supplied
 	 */
    if(fileLoc.length == 0){
    	return ["No file path supplied!".bold + " Example: '" + "SELECT 'HELLO', 'WORLD' FROM DUMMY | write myfile.txt".dim + "'"]
    }

    /**
 	 * Make sure path is a valid path
 	 */
 	if(!isValidPath(fileLoc)){
 		return ["Invalid file path! " + fileLoc.red.bold];
 	}

 	var dirLoc = path.dirname(fileLoc);

 	/**
 	 * Check dirLoc is a directory
 	 */
 	try{
 		fs.lstatSync(dirLoc).isDirectory()
 	}catch(e){
 		if(e.code == "ENOENT"){
 			return [("Location: '" + dirLoc + "' does not exist!").red.bold]
 		}
 	}

 	/**
 	 * Check fileloc is not a directory
 	 */
 	try{
 		if(!fs.lstatSync(fileLoc).isFile()){
 			return [("'" + fileLoc + "' is a directory!").red.bold];
 		}
 	}catch(e){
 		if(e.code != "ENOENT"){
 			return [e.message.red.bold];
 		}
 	}

 	var output = linesIn.join("\n") + "\n";

 	/**
 	 * Write to file
 	 */
 	try{
 		fs.writeFileSync(fileLoc, output)
 	}catch(e){
 		return [e.message.red.bold];
 	}

    return [("File saved to: ".bold + fileLoc).green];
}