org.uiautomation.ios.server.application.LocatorWithL10N Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ios-server Show documentation
The newest version!
package org.uiautomation.ios.server.application;

import org.openqa.selenium.WebDriverException;

import java.util.HashSet;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Helper to access the localization feature of the app.
 * @see LocatorWithL10N#translate(String)
 * @see LocatorWithL10N#translateKey(String)
 */
class LocatorWithL10N {

  private static final String pattern = "l10n\\(('|\")(.*?)('|\")\\)";
  private final IOSRunningApplication running;

  LocatorWithL10N(IOSRunningApplication app) {
    this.running = app;
  }

  private Set extractKeysToL10N(String locator) {
    Pattern p = Pattern.compile(pattern);
    Matcher matcher = p.matcher(locator);
    Set keys = new HashSet();
    while (matcher.find()) {
      keys.add(getKey(matcher.group()));
    }
    return keys;
  }

  private String getKey(String function) {
    String res = function.replaceAll("l10n\\(", "");
    res = res.replaceAll("(\"|'|\\))", "");
    return res;
  }


  public String translate(String locator) {
    Set keys = extractKeysToL10N(locator);
    String res = locator;
    for (String key : keys) {
      String value = translateKey(key);
      String originalFunction = "l10n\\('" + key + "'\\)|l10n\\(\"" + key + "\"\\)";
      String l10ned = "'" + value + "'";
      res = res.replaceAll(originalFunction, l10ned);
    }
    return res;
  }

  public String translateKey(String key) {
    LanguageDictionary dict = running.getCurrentDictionary();
    String value = dict.getContentForKey(key);
    if (value == null) {
      throw new WebDriverException("One of the key requested for localization :" + key
                                   + " isn't available in the l10n files.Most likely the key "
                                   + "provided is wrong.You can use the inspector to find the "
                                   + "correct keys.");
    }
    return LanguageDictionary.getRegexPattern(value);
  }
}