Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.expressionevaluator.suggestions;

import java.util.HashMap;

import java.util.Map;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.text.similarity.LevenshteinDistance;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionUtils;

import org.gvsig.expressionevaluator.swing.SuggestionProvider;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.util.Factory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class WrongNameFieldSuggestionProviderFactory implements Factory {

  private final FeatureStore store;

  private FeatureType featureType;

  public WrongNameFieldSuggestionProviderFactory(FeatureStore store) {

    this.store = store;

    try {

      this.featureType = store.getDefaultFeatureType();

    } catch (DataException ex) {

      this.featureType = null;

    }            

  }

  @Override

  public String getName() {

    return "WrongNameFieldSuggestion";

  }

  @Override

  public Object create(Object... parameters) {

    SuggestionProvider suggestionProvider = new WrongNameFieldSuggestionProvider();

    return suggestionProvider;

  }

  private class WrongNameFieldSuggestionProvider implements SuggestionProvider {

    @Override

    public String getSuggestion(String expression) {

      if ( StringUtils.isBlank(expression)) {

        return null;

      }

      final Map<String,String> similarFields = new HashMap<>();

      try {

        Code code = ExpressionUtils.compile(expression);

        if (code == null) {

          return null;

        }

        code.accept((Object obj) -> {

          if (obj instanceof Code.Identifier) {

            String varname = ((Code.Identifier) obj).name().toLowerCase();

            if( varname.length()<3  || featureType.get(varname)!=null ) {

              return;

            }

            if( similarFields.containsKey(varname) ) {

              return;

            }

            String suggestedName = getSuggestedAttrName(varname, 3);

            if( suggestedName!=null ) {

                similarFields.put(varname, suggestedName);

            }

          }

        });

      } catch (Throwable th) {

      }

      if( similarFields.isEmpty() ) {

        return null;

      }

      // Perhaps you wanted to use a similar field.

      // Quiz?s quiso usar un campo parecido.

      I18nManager i18n = ToolsLocator.getI18nManager();

      StringBuilder suggestion = new StringBuilder();

      suggestion.append(i18n.getTranslation("_Perhaps_you_wanted_to_use_a_similar_field"));

      suggestion.append("\n");

      boolean needAComma = false;

      for(Map.Entry<String, String> entry : similarFields.entrySet() ) {

        if( needAComma ) {

          suggestion.append(", ");

        } else {

          needAComma = true;

        }

        suggestion.append(entry.getKey());

        suggestion.append(" --> ");

        suggestion.append(entry.getValue());

      }

      return suggestion.toString();

    }

  }

  private String getSuggestedAttrName(String nameSearched, int maxdistance) {

    LevenshteinDistance comparator = LevenshteinDistance.getDefaultInstance();

    nameSearched = nameSearched.toLowerCase();

    String found_name = null;

    int found_distance = Integer.MAX_VALUE;

    for (FeatureAttributeDescriptor attr : featureType) {

      String name = attr.getName();

      int distance = comparator.apply(name.toLowerCase(), nameSearched);

      if( distance<found_distance ) {

        found_name = name;

        found_distance = distance;

      }

    }

    if( found_distance<=maxdistance ) {

      return found_name;

    }

    return null;

  }

}