com.impetus.ankush.agent.action.impl.YAMLManipulator.java Source code

Introduction

Here is the source code for com.impetus.ankush.agent.action.impl.YAMLManipulator.java

Source

/*******************************************************************************
 * ===========================================================
 * 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.agent.action.impl;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

import org.apache.commons.io.FileUtils;
import org.yaml.snakeyaml.Yaml;

import com.impetus.ankush.agent.action.Manipulatable;

public class YAMLManipulator implements Manipulatable {

    /**
     * Delete conf value.
     * 
     * @param file
     *            the file
     * @param propertyName
     *            the property name
     * @return true, if successful
     */
    public boolean deleteConfValue(String file, String objPropertyName) {

        String propertyName = (String) objPropertyName;
        Yaml yaml = new Yaml();
        try {
            InputStream fis = new FileInputStream(file);
            Object javaObject = yaml.load(fis);
            Map<String, Object> map = (Map<String, Object>) javaObject;
            if (map.containsKey(propertyName)) {
                map.remove(propertyName);
            }
            fis.close();
            // save to file
            File confFile = new File(file);
            if (!confFile.exists()) {
                System.err.println("File " + file + " does not exists.");
            }
            String dumped = yaml.dumpAsMap(map);
            FileUtils.writeStringToFile(confFile, dumped, false);

            return true;

        } catch (FileNotFoundException e) {
            System.err.println(e.getMessage());
        } catch (IOException e) {
            System.err.println(e.getMessage());
        } catch (Exception e) {
            System.err.println(e.getMessage());
        }
        return false;
    }

    /**
     * Edits the conf value.
     * 
     * @param file
     *            the file
     * @param propertyName
     *            the property name
     * @param newPropertyValue
     *            the new property value
     * @return true, if successful
     */
    public boolean editConfValue(String file, String propertyName, String newPropertyValue) {

        boolean isInteger = checkNumber((String) newPropertyValue);
        boolean isBoolean = false;

        if (String.valueOf(newPropertyValue).equalsIgnoreCase("true")
                || String.valueOf(newPropertyValue).equalsIgnoreCase("false")) {
            isBoolean = true;
        }

        List<Object> newPropertyValueList = new ArrayList<Object>();
        if (String.valueOf(newPropertyValue).contains(",")) {
            List<Object> list = new ArrayList<Object>(Arrays.asList(String.valueOf(newPropertyValue).split(",")));
            for (Object val : list) {
                if (isInteger) {
                    newPropertyValueList.add(Integer.valueOf((String) val));
                } else if (isBoolean) {
                    newPropertyValueList.add(Boolean.valueOf((String) val));
                } else {
                    newPropertyValueList.add((String) val);
                }
            }
        } else {
            if (isInteger) {
                newPropertyValueList.add(Integer.valueOf((String) newPropertyValue));
            } else if (isBoolean) {
                newPropertyValueList.add(Boolean.valueOf((String) newPropertyValue));
            } else {
                newPropertyValueList.add((String) newPropertyValue);
            }
        }

        Yaml yaml = new Yaml();
        try {
            InputStream fis = new FileInputStream(file);
            Object javaObject = yaml.load(fis);
            Map<Object, Object> map = (Map<Object, Object>) javaObject;

            Object propertyValue = new Object();

            if (newPropertyValueList.size() > 1) {
                propertyValue = (List<Object>) newPropertyValueList;
            } else {
                propertyValue = newPropertyValueList.get(0);
            }

            map.put(propertyName, propertyValue);
            fis.close();
            // save to file
            File confFile = new File(file);
            if (!confFile.exists()) {
                System.err.println("File " + file + " does not exists.");
            }

            String dumped = yaml.dumpAsMap(map);
            FileUtils.writeStringToFile(confFile, dumped, false);

            return true;

        } catch (FileNotFoundException e) {
            System.err.println(e.getMessage());
        } catch (IOException e) {
            System.err.println(e.getMessage());
        } catch (Exception e) {
            System.err.println(e.getMessage());
        }
        return false;
    }

    private boolean checkNumber(String propertyValue) {
        try {
            double d;
            if (propertyValue.contains(",")) {
                String[] splittedArray = propertyValue.split(",");
                for (int i = 0; i < splittedArray.length; i++) {
                    d = Double.parseDouble(splittedArray[i]);
                }
            } else {
                d = Double.parseDouble(propertyValue);
            }
        } catch (NumberFormatException nfe) {
            return false;
        }
        return true;
    }

    /**
     * Write conf value.
     * 
     * @param file
     *            the file
     * @param propertyName
     *            the property name
     * @param propertyValue
     *            the property value
     * @return true, if successful
     */
    public boolean writeConfValue(String file, String propertyName, String propertyValue) {

        return editConfValue(file, propertyName, propertyValue);

    }

    /**
     * Read conf value.
     * 
     * @param file
     *            the file
     * @param propertyName
     *            the property name
     * @return the string
     */
    public String readConfValue(String file, String objPropertyName) {

        String propertyName = (String) objPropertyName;

        Yaml yaml = new Yaml();
        try {
            InputStream fis = new FileInputStream(file);
            Object javaObject = yaml.load(fis);
            Map<String, Object> map = (Map<String, Object>) javaObject;

            for (Map.Entry<String, Object> entry : map.entrySet()) {
                if (entry.getKey().equals(propertyName)) {
                    return (String) entry.getValue();
                }
            }
            fis.close();

        } catch (FileNotFoundException e) {
            System.err.println(e.getMessage());
        } catch (IOException e) {
            System.err.println(e.getMessage());
        } catch (Exception e) {
            System.err.println(e.getMessage());
        }
        return null;
    }

}