LimeAttributes.java

/*
 * Copyright (c) 2017 OBiBa. All rights reserved.
 *
 * This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.obiba.magma.datasource.limesurvey;

import java.util.Arrays;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Set;

import org.obiba.magma.Attribute;
import org.springframework.util.StringUtils;

import com.google.common.base.Strings;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

class LimeAttributes {

  static final String LIMESURVEY_NAMESPACE = "limesurvey";

  // Attributes that should not be part of the limesurvey namespace
  private static final Set<String> OPAL_ATTRIBUTES = ImmutableSet.of("label");

  private final Map<String, String> attributes;

  private LimeAttributes() {
    attributes = Maps.newHashMap();
  }

  public static LimeAttributes create() {
    return new LimeAttributes();
  }

  public LimeAttributes attribute(String key, String value) {
    attributes.put(key, value);
    return this;
  }

  public Iterable<Attribute> toMagmaAttributes(boolean keepOriginalLocalizable) {
    List<Attribute> attrs = Lists.newArrayList();
    for(Map.Entry<String, String> entry : attributes.entrySet()) {
      String attValue = entry.getValue();
      if (Strings.isNullOrEmpty(attValue)) continue;
      String cleaned = cleanAttributeValue(attValue);
      String[] key = entry.getKey().split(":");
      if(key.length > 1) {
        Locale locale = new Locale(key[1]);
        attrs.add(newAttribute(key[0]).withValue(locale, clean(cleaned)).build());
        if(keepOriginalLocalizable && !cleaned.equals(attValue)) {
          attrs.add(newAttribute("original" + StringUtils.capitalize(key[0])).withValue(locale, attValue).build());
        }
      } else {
        attrs.add(newAttribute(key[0]).withValue(cleaned).build());
      }
    }
    return attrs;
  }

  private String cleanAttributeValue(String value) {
    StringBuilder sb = new StringBuilder(value);
    for(String tag : Arrays.asList("script")) {
      int start;
      int end;
      do {
        start = sb.indexOf("<" + tag);
        end = sb.indexOf("</" + tag + ">", start) + tag.length() + 3;
        if(start != -1 && end != -1 && start < end) {
          sb.replace(start, end, " [script] ");
        }
      } while(start != -1 && end != -1 && start < end);
    }
    return clean(sb.toString());
  }

  private Attribute.Builder newAttribute(String key) {
    Attribute.Builder builder = Attribute.Builder.newAttribute(key);
    if(!OPAL_ATTRIBUTES.contains(key)) {
      builder.withNamespace(LIMESURVEY_NAMESPACE);
    }
    return builder;
  }

  private String clean(String label) {
    List<String> filter = Arrays.asList("div", "span", "font", "p");
    String copy = label;
    copy = deleteText(copy, "\n");
    copy = deleteText(copy, "\t");
    copy = deleteText(copy, "\r");
    copy = deleteText(copy, "<br />");
    for(String tag : filter) {
      copy = copy.replaceAll("</?" + tag + "[^>]*>", "");
    }
    return copy;
  }

  private String deleteText(String text, String del) {
    return text.replaceAll(del, "");
  }

}