Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.export.jdbc.service;

import java.util.ArrayList;

import java.util.Collection;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.export.ExportAttributes;

import org.gvsig.export.spi.AttributeNamesTranslator;

/**

 *

 * @author osc

 */

public class ExportJDBCAttributeNamesTranslator implements AttributeNamesTranslator {

    private boolean removeSpacesInIdentifiers;

    private boolean translateIdentifiersToLowerCase;

    private boolean translateHyphens;

    public ExportJDBCAttributeNamesTranslator() {

    }

    private String buildTranslatedName(String source_name, Collection<String> current_names) {

        String target_name = source_name;

        if (this.removeSpacesInIdentifiers) {

            //target_name = target_name.replaceAll("\\s","");

            target_name = StringUtils.normalizeSpace(target_name).replace(" ", "_");

        }

        if (this.translateHyphens) {

            target_name = target_name.replace("-", "_");

        }

        if (this.translateIdentifiersToLowerCase) {

            target_name = target_name.toLowerCase();

        }

        if (!current_names.contains(target_name)) {

            return target_name;

        }

        if (!current_names.contains(target_name)) {

            return target_name;

        }

        for (int i = 0; i < 255; i++) {

            target_name = source_name + i;

            if (!current_names.contains(target_name)) {

                return target_name;

            }

        }

        return source_name.substring(0, 4) + "_" + (System.currentTimeMillis() % 1000000);

    }

    @Override

    public boolean isValidName(ExportAttributes attributes, int index, String name) {

        ExportAttributes.ExportAttribute selectedAttr = attributes.get(index);

        //

        String selfName = this.getNameSuggestion(attributes, index, name);

        if (!selfName.equals(name)) {

            return false;

        }

        // check with others values

        boolean valid = true;

        for (int i = 0; i < attributes.size(); i++) {

            if (i == index) {

                continue;

            }

            ExportAttributes.ExportAttribute attr = attributes.get(i);

            String newName = attr.getNewName();

            if (newName.equals(name)) {

                return false;

            }

            if (newName.equals(selfName)) {

                return false;

            }

            if (newName.isEmpty()) {

                return false;

            }

            if (this.removeSpacesInIdentifiers) {

                //if (newName.replaceAll("\\s","") != newName) {

                String s = StringUtils.normalizeSpace(newName).replace(" ", "_");

                if (!newName.equals(s)) {

                    return false;

                }

            }

            if (this.translateHyphens) {

                if (newName.contains("-")) {

                    return false;

                }

            }

            if (this.translateIdentifiersToLowerCase) {

                if (newName.toLowerCase() != newName) {

                    return false;

                }

            }

        }

        return valid;

    }

    @Override

    public String getNameSuggestion(ExportAttributes attributes, int index, String name) {

        ArrayList names = new ArrayList();

        boolean oneTime = true;

        for (ExportAttributes.ExportAttribute attribute : attributes) {

            if (attribute.getNewName().equals(name) && oneTime) {

                oneTime = false;

                continue;

            }

            names.add(attribute.getNewName());

        }

        String target_name = buildTranslatedName(name, names);

        return target_name;

    }

    public boolean getRemoveSpacesInIdentifiers() {

        return this.removeSpacesInIdentifiers;

    }

    public void setRemoveSpacesInIdentifiers(boolean selected) {

        this.removeSpacesInIdentifiers = selected;

    }

    public boolean getTranslateIdentifiersToLowerCase() {

        return this.translateIdentifiersToLowerCase;

    }

    public void setTranslateIdentifiersToLowerCase(boolean selected) {

        this.translateIdentifiersToLowerCase = selected;

    }

    public boolean getTranslateHyphens() {

        return this.translateHyphens;

    }

    public void setTranslateHyphens(boolean selected) {

        this.translateHyphens = selected;

    }

}