ReplaceText.java

/*******************************************************************************
 * ===========================================================
 * Ankush : Big Data Cluster Management Solution
 * ===========================================================
 * 
 * (C) Copyright 2014, by Impetus Technologies
 * 
 * This is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License (LGPL v3) as
 * published by the Free Software Foundation;
 * 
 * This software is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License 
 * along with this software; if not, write to the Free Software Foundation, 
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 ******************************************************************************/
/**
 * 
 */
package com.impetus.ankush.common.scripting.impl;

import org.apache.commons.lang3.StringUtils;

import com.impetus.ankush.common.scripting.AnkushTask;

// TODO: Auto-generated Javadoc
/**
 * The Class ReplaceText.
 * 
 * @author Akhil
 */
public class ReplaceText extends AnkushTask {

	/** The file path. */
	private String filePath = null;

	/** The Target Text. */
	private String targetText = "";

	/** The Replacement Text. */
	private String replacementText = "";

	/** The password. */
	private String password = null;

	/** The flag for creating BackUp File. */
	private boolean createBackUpFile = true;

	private boolean addSudoOption = false;
	
	
	public ReplaceText(String targetText, String replacementText,
			String filePath, boolean createBackUpFile) {
		this.targetText = StringUtils.replace(targetText, "/", "\\/");
		this.replacementText = StringUtils.replace(replacementText, "/", "\\/");
		this.filePath = filePath;
		this.createBackUpFile = createBackUpFile;
		this.password = null;
		this.addSudoOption = false;
	}
	
	public ReplaceText(String targetText, String replacementText,
			String filePath, boolean createBackUpFile, String password) {
		this.targetText = StringUtils.replace(targetText, "/", "\\/");
		this.replacementText = StringUtils.replace(replacementText, "/", "\\/");
		this.filePath = filePath;
		this.createBackUpFile = createBackUpFile;
		this.password = password;
		this.addSudoOption = false;
	}
	
	/**
	 * Instantiates a new ReplaceText object.
	 * 
	 * @param targetText
	 *            the targetText
	 * @param replacementText
	 *            the replacementText
	 * @param filePath
	 *            the file path
	 * @param createBackUpFile
	 *            the createBackUpFile
	 * @param password
	 *            the password
	 * @param addSudoOption
	 *            the addSudoOption
	 */
	public ReplaceText(String targetText, String replacementText,
			String filePath, boolean createBackUpFile, String password, boolean addSudoOption) {
		this.targetText = StringUtils.replace(targetText, "/", "\\/");
		this.replacementText = StringUtils.replace(replacementText, "/", "\\/");
		this.filePath = filePath;
		this.createBackUpFile = createBackUpFile;
		this.password = password;
		this.addSudoOption = addSudoOption;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see net.neoremind.sshxcute.task.CustomTask#getCommand()
	 */
	@Override
	public String getCommand() {

		final StringBuilder sb = new StringBuilder();
		
		String prependSudo = "";
		if(addSudoOption) {
			if (this.password != null) {
				prependSudo = "echo '" + this.password + "' | sudo -S ";
			} else {
				prependSudo = "sudo -S ";
			}	
		}
		sb.append(prependSudo);
		if (this.createBackUpFile) {
			sb.append("sed -i.bak s/'" + this.targetText + "'/'"
					+ this.replacementText + "'/g " + this.filePath);
		} else {
			sb.append("sed -i s/'" + this.targetText + "'/'"
					+ this.replacementText + "'/g " + this.filePath);
		}
		sb.append(";");
		return sb.toString();
	}
}