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 org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.namestranslator.BaseNamesTranslator;

import org.gvsig.tools.namestranslator.NamesTranslator;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

/**

 *

 * @author osc

 */

public class ExportJDBCAttributeNamesTranslator extends BaseNamesTranslator {

    private ExportJDBCParameters parameters;

    public ExportJDBCAttributeNamesTranslator(ExportJDBCParameters parameters) {

        this.parameters = parameters;

    }

    public ExportJDBCAttributeNamesTranslator() {

    }

    public void setParameters(ExportJDBCParameters parameters) {

        this.parameters = parameters;

    }

    public static void registerPersistence() {

        PersistenceManager manager = ToolsLocator.getPersistenceManager();

        if (manager.getDefinition("ExportJDBCAttributeNamesTranslator") == null) {

            DynStruct definition = manager.addDefinition(ExportJDBCAttributeNamesTranslator.class,

                    "ExportJDBCAttributeNamesTranslator", "ExportJDBCAttributeNamesTranslator persistence definition", null, null);

            definition.extend(manager.getDefinition("BaseNamesTranslator"));

        }

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        super.saveToState(state);

    }

    @Override

    public void loadFromState(PersistentState state) throws PersistenceException {

        super.loadFromState(state);

    }

    @Override

    public NamesTranslator clone() throws CloneNotSupportedException {

        ExportJDBCAttributeNamesTranslator clone = (ExportJDBCAttributeNamesTranslator) super.clone();

        return clone;

    }

    protected String fixSyntax(String name) {

        String target_name = name;

        if (StringUtils.isBlank(target_name)) {

            target_name = DEFAULT_FIELD_NAME;

        }

        if (this.parameters.getRemoveSpacesInIdentifiers()) {

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

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

        }

        if (this.parameters.getTranslateHyphens()) {

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

        }

        if (this.parameters.getTranslateIdentifiersToLowerCase()) {

            target_name = target_name.toLowerCase();

        }

        return target_name;

    }

    @Override

    public String getSuggestion(String name) {

        String target_name = fixSyntax(name);

        if (isValid(target_name)) {

            return target_name;

        }

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

            String new_target_name = target_name + i;

            if (isValid(new_target_name)) {

                return new_target_name;

            }

        }

        return target_name.substring(0, 5) + (System.currentTimeMillis() % 1000000);

    }

    @Override

    public boolean isValid(String name) {

        if (!super.isValid(name)) {

            return false;

        }

        if (name.isEmpty()) {

            name = DEFAULT_FIELD_NAME;

        }

        if (this.parameters.getRemoveSpacesInIdentifiers()) {

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

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

            if (!name.equals(s)) {

                return false;

            }

        }

        if (this.parameters.getTranslateHyphens()) {

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

                return false;

            }

        }

        if (this.parameters.getTranslateIdentifiersToLowerCase()) {

            if (!StringUtils.equals(name.toLowerCase(), name)) {

                return false;

            }

        }

        return true;

    }

    @Override

    public int setTranslation(String sourceName, String translatedName) {

        String target_name = fixSyntax(translatedName);

        return super.setTranslation(sourceName, target_name);

    }

}