/src/mpv5/db/common/QueryData.java

/*
 *  This file is part of YaBS.
 *
 *  YaBS is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  YaBS 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with YaBS.  If not, see <http://www.gnu.org/licenses/>.
 */
package mpv5.db.common;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import mpv5.logging.Log;
import mpv5.data.*;

/**
 * This class is used to put any kind of data before passing it to the <code>QueryHandler</code> for transmission.
 * 
 */
public class QueryData {

    @Override
    public String toString() {
        String[] a = getKeys();
        String c = "";

        for (int i = 0; i < a.length; i++) {
            c += a[i] + " = " + getValue(a[i]).getWrapper() + getValue(a[i]).toString() + getValue(a[i]).getWrapper() + ", ";
        }

        if (c.length() > 2) {
            c = c.substring(0, c.length() - 2);
        }
        return c;
    }

    /**
     * Generate a <code>SaveString<code/>
     * @param s
     * @return
     */
    public static SaveString getSaveStringFor(String s) {
        return new SaveString(s, true);
    }
    private HashMap<String, SaveString> list = new HashMap<String, SaveString>();

    /**
     * This is a legacy constructor, to speed up the conversion of old style data into new style data
     * what  : {set, values}
     * @param what
     */
    public QueryData(String[] what) {
        String[] keys = what[0].split(",");
        String[] vals = what[1].split(",");

        for (int i = 0; i < vals.length; i++) {
            add(keys[i], vals[i]);
        }
    }

    public QueryData() {
    }

    /**
     * Adds a key with a value
     * @param <T> 
     * @param key
     * @param value
     */
    public <T extends Number> void add(String key, T value) {
        String string = String.valueOf(value);
        list.put(key, new SaveString(string, false));
    }
    
    /**
     * 
     * @param criteria 
     */
    public void add(QueryCriteria2 criteria){
        List<QueryParameter> d = criteria.getFields();
        for (int i = 0; i < d.size(); i++) {
            QueryParameter o = d.get(i);
            add(o.getKey(), o.getValue());
        }
    }

    /**
     * Adds a key with a value
     * @param key
     * @param value
     */
    public void add(String key, boolean value) {
        if (value) {
            list.put(key, new SaveString("1", false));
        } else {
            list.put(key, new SaveString("0", false));
        }
    }

    /**
     * Adds a key with a value
     * @param key
     * @param value
     */
    public void add(String key, String value) {
        list.put(key, new SaveString(value, true));
    }

    /**
     *
     * @return True if there is at least one value
     */
    public boolean hasValues() {
        return !list.isEmpty();
    }

    /**
     * Mask backslashes with even more backslashes
     * @param string
     * @return
     */
    public String maskBackslashes(String string) {
        Log.Debug(QueryData.class, "Masking Backslashes!");
        return new SaveString(string.replaceAll("\\\\", "\\\\\\\\"), true).toString();
    }

    /**
     *
     * @return An array of keys
     */
    public String[] getKeys() {
        return list.keySet().toArray(new String[]{});
    }

    /**
     * returns the value of the given key
     * @param key
     * @return
     */
    public SaveString getValue(String key) {
        return list.get(key);
    }

    /**
     * Generates a comma separated String represantion of the current values<br/>
     * with String values wrapped in single quotes
     * @return
     */
    public String getValuesString() {
        String[] k = getKeys();
        String s = "";

        for (int i = 0; i < k.length; i++) {
            mpv5.db.common.SaveString v = getValue(k[i]);
            s += v.getWrapper() + v.toString() + v.getWrapper() + ",";
        }
        if (s.length() > 1) {
            return s.substring(0, s.length() - 1);
        } else {
            return s;
        }
    }

    /**
     * 
     * @return An array of all values, in getKeys() - order
     */
    public String[] getValues() {
        return getValuesString().split(",");
    }

    /**
     * Generates a comma separated String representation of the current keys
     * @return
     */
    public String getKeysString() {
        String[] k = getKeys();
        String s = "";

        for (int i = 0; i < k.length; i++) {
            s += k[i] + ",";
        }
        if (s.length() > 1) {
            return s.substring(0, s.length() - 1);
        } else {
            return s;
        }
    }

    /**
     * This method acts as bridge between  {@link PropertyStore} and {@link QueryData}
     * All data will be of type <code>String</code>
     * @param properties
     */
    public void parse(PropertyStore properties) {
        List<String[]> data = properties.getList();
        for (int i = 0; i < data.size(); i++) {
            String[] s = data.get(i);
            add(s[0], s[1]);
        }
    }

    /**
     * Adds a key with a value
     * @param columnName
     * @param value
     */
    public void add(String columnName, Object value) {
        if (value instanceof Number) {
            add(columnName, (Number) value);
        } else if (value instanceof Boolean) {
            add(columnName, (Boolean) value);
        } else {
            add(columnName, String.valueOf(value));
        }
    }
}