com.rolfje.anonimatron.anonymizer.CountryCodeAnonymizer Maven / Gradle / Ivy

Go to download
package com.rolfje.anonimatron.anonymizer;

import com.rolfje.anonimatron.synonyms.StringSynonym;
import com.rolfje.anonimatron.synonyms.Synonym;
import org.springframework.util.StringUtils;

import java.security.SecureRandom;
import java.text.SimpleDateFormat;
import java.util.Locale;
import java.util.MissingResourceException;

/**
 * Produces valid 2 or 3 character country codes.
 */
public class CountryCodeAnonymizer implements Anonymizer {

	private static final String TYPE = "COUNTRY_CODE";
	protected static final Locale[] AVAILABLE_LOCALES = SimpleDateFormat.getAvailableLocales();
	SecureRandom r = new SecureRandom();

	@Override
	public String getType() {
		return TYPE;
	}

	@Override
	public Synonym anonymize(Object from, int size, boolean shortlived) {

		if (size < 2) {
			throw new UnsupportedOperationException("Can not produce country codes of one character.");
		}

		String country = null;
		while (country == null || country.length() < 1) {
			Locale l = AVAILABLE_LOCALES[r.nextInt(AVAILABLE_LOCALES.length)];

			try {
				if (size > 2) {
					country = l.getISO3Country();
					if(StringUtils.isEmpty(country)) {
						continue;
					}
					country = padRight(country, size);
				}
				else if (size == 2) {
					country = l.getCountry();
				}
			} catch (MissingResourceException e) {
				// Locale.getISOCountries() has inconsistent behaviour for "AN", "BU" and "CS" country codes
				// See https://bugs.openjdk.java.net/browse/JDK-8071929
			}
		}

		return new StringSynonym(
				getType(),
				from.toString(),
				country,
				shortlived
		);
	}

	public static String padRight(String s, int n) {
		return String.format("%1$-" + n + "s", s);
	}
}