package com.twitter;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import com.twitter.Extractor.Entity;
/**
* A class for adding HTML links to hashtag, username and list references in
* Tweet text.
*/
public class Autolink {
/** Default CSS class for auto-linked list URLs */
public static final String DEFAULT_LIST_CLASS = "tweet-url list-slug";
/** Default CSS class for auto-linked username URLs */
public static final String DEFAULT_USERNAME_CLASS = "tweet-url username";
/** Default CSS class for auto-linked hashtag URLs */
public static final String DEFAULT_HASHTAG_CLASS = "tweet-url hashtag";
/** Default CSS class for auto-linked cashtag URLs */
public static final String DEFAULT_CASHTAG_CLASS = "tweet-url cashtag";
/**
* Default href for username links (the username without the @ will be
* appended)
*/
public static final String DEFAULT_USERNAME_URL_BASE = "https://twitter.com/";
/**
* Default href for list links (the username/list without the @ will be
* appended)
*/
public static final String DEFAULT_LIST_URL_BASE = "https://twitter.com/";
/**
* Default href for hashtag links (the hashtag without the # will be
* appended)
*/
public static final String DEFAULT_HASHTAG_URL_BASE = "https://twitter.com/#!/base_dual_pane?q=%23";
/**
* Default href for cashtag links (the cashtag without the $ will be
* appended)
*/
public static final String DEFAULT_CASHTAG_URL_BASE = "https://twitter.com/#!/base_dual_pane?q=%24";
/** Default attribute for invisible span tag */
public static final String DEFAULT_INVISIBLE_TAG_ATTRS = "style='position:absolute;left:-9999px;'";
protected String urlClass = null;;
protected String listClass;
protected String usernameClass;
protected String hashtagClass;
protected String cashtagClass;
protected String usernameUrlBase;
protected String listUrlBase;
protected String hashtagUrlBase;
protected String cashtagUrlBase;
protected String invisibleTagAttrs;
protected boolean noFollow = true;
protected boolean usernameIncludeSymbol = false;
protected String symbolTag = null;
protected String textWithSymbolTag = null;
protected String urlTarget = null;
protected LinkAttributeModifier linkAttributeModifier = null;
protected LinkTextModifier linkTextModifier = null;
private final Extractor extractor = new Extractor();
/**
* Fullwidth ellipsis: '...'
*/
private static final String FULLWIDTH_ELLIPSIS = "\u2026";
public Autolink() {
urlClass = null;
listClass = DEFAULT_LIST_CLASS;
usernameClass = DEFAULT_USERNAME_CLASS;
hashtagClass = DEFAULT_HASHTAG_CLASS;
cashtagClass = DEFAULT_CASHTAG_CLASS;
usernameUrlBase = DEFAULT_USERNAME_URL_BASE;
listUrlBase = DEFAULT_LIST_URL_BASE;
hashtagUrlBase = DEFAULT_HASHTAG_URL_BASE;
cashtagUrlBase = DEFAULT_CASHTAG_URL_BASE;
invisibleTagAttrs = DEFAULT_INVISIBLE_TAG_ATTRS;
extractor.setExtractURLWithoutProtocol(false);
}
/**
* Auto-link hashtags, URLs, usernames and lists.
*
* @param text of the Tweet to auto-link
* @return text with auto-link HTML added
*/
public String autoLink(String text) {
text = escapeBrackets(text);
// extract entities
final List<Entity> entities = extractor.extractEntitiesWithIndices(text);
return autoLinkEntities(text, entities);
}
/**
* Auto-link $cashtag references in the provided Tweet text. The $cashtag
* links will have the cashtagClass CSS class added.
*
* @param text of the Tweet to auto-link
* @return text with auto-link HTML added
*/
public String autoLinkCashtags(final String text) {
return autoLinkEntities(text, extractor.extractCashtagsWithIndices(text));
}
public String autoLinkEntities(final String text, final List<Entity> entities) {
final StringBuilder builder = new StringBuilder(text.length() * 2);
int beginIndex = 0;
for (final Entity entity : entities) {
builder.append(text.subSequence(beginIndex, entity.start));
switch (entity.type) {
case URL:
linkToURL(entity, text, builder);
break;
case HASHTAG:
linkToHashtag(entity, text, builder);
break;
case MENTION:
linkToMentionAndList(entity, text, builder);
break;
case CASHTAG:
linkToCashtag(entity, text, builder);
break;
}
beginIndex = entity.end;
}
builder.append(text.subSequence(beginIndex, text.length()));
return builder.toString();
}
/**
* Auto-link #hashtag references in the provided Tweet text. The #hashtag
* links will have the hashtagClass CSS class added.
*
* @param text of the Tweet to auto-link
* @return text with auto-link HTML added
*/
public String autoLinkHashtags(final String text) {
return autoLinkEntities(text, extractor.extractHashtagsWithIndices(text));
}
/**
* Auto-link URLs in the Tweet text provided.
* <p/>
* This only auto-links URLs with protocol.
*
* @param text of the Tweet to auto-link
* @return text with auto-link HTML added
*/
public String autoLinkURLs(final String text) {
return autoLinkEntities(text, extractor.extractURLsWithIndices(text));
}
/**
* Auto-link the @username and @username/list references in the provided
* text. Links to @username references will have the usernameClass CSS
* classes added. Links to @username/list references will have the listClass
* CSS class added.
*
* @param text of the Tweet to auto-link
* @return text with auto-link HTML added
*/
public String autoLinkUsernamesAndLists(final String text) {
return autoLinkEntities(text, extractor.extractMentionsOrListsWithIndices(text));
}
public String escapeBrackets(final String text) {
final int len = text.length();
if (len == 0) return text;
final StringBuilder sb = new StringBuilder(len + 16);
for (int i = 0; i < len; ++i) {
final char c = text.charAt(i);
if (c == '>') {
sb.append(">");
} else if (c == '<') {
sb.append("<");
} else {
sb.append(c);
}
}
return sb.toString();
}
/**
* @return CSS class for auto-linked cashtag URLs
*/
public String getCashtagClass() {
return cashtagClass;
}
/**
* @return the href value for cashtag links (to which the cashtag will be
* appended)
*/
public String getCashtagUrlBase() {
return cashtagUrlBase;
}
/**
* @return CSS class for auto-linked hashtag URLs
*/
public String getHashtagClass() {
return hashtagClass;
}
/**
* @return the href value for hashtag links (to which the hashtag will be
* appended)
*/
public String getHashtagUrlBase() {
return hashtagUrlBase;
}
/**
* @return CSS class for auto-linked list URLs
*/
public String getListClass() {
return listClass;
}
/**
* @return the href value for list links (to which the username/list will be
* appended)
*/
public String getListUrlBase() {
return listUrlBase;
}
/**
* @return CSS class for auto-linked URLs
*/
public String getUrlClass() {
return urlClass;
}
/**
* @return CSS class for auto-linked username URLs
*/
public String getUsernameClass() {
return usernameClass;
}
/**
* @return the href value for username links (to which the username will be
* appended)
*/
public String getUsernameUrlBase() {
return usernameUrlBase;
}
/**
* @return if the current URL links will include rel="nofollow" (true by
* default)
*/
public boolean isNoFollow() {
return noFollow;
}
public void linkToCashtag(final Entity entity, final String text, final StringBuilder builder) {
final CharSequence cashtag = entity.getValue();
final Map<String, String> attrs = new LinkedHashMap<String, String>();
attrs.put("href", cashtagUrlBase + cashtag);
attrs.put("title", "$" + cashtag);
attrs.put("class", cashtagClass);
linkToTextWithSymbol(entity, "$", cashtag, attrs, builder);
}
public void linkToHashtag(final Entity entity, final String text, final StringBuilder builder) {
// Get the original hash char from text as it could be a full-width
// char.
final CharSequence hashChar = text.subSequence(entity.getStart(), entity.getStart() + 1);
final CharSequence hashtag = entity.getValue();
final Map<String, String> attrs = new LinkedHashMap<String, String>();
attrs.put("href", hashtagUrlBase + hashtag);
attrs.put("title", "#" + hashtag);
attrs.put("class", hashtagClass);
linkToTextWithSymbol(entity, hashChar, hashtag, attrs, builder);
}
public void linkToMentionAndList(final Entity entity, final String text, final StringBuilder builder) {
String mention = entity.getValue();
// Get the original at char from text as it could be a full-width char.
final CharSequence atChar = text.subSequence(entity.getStart(), entity.getStart() + 1);
final Map<String, String> attrs = new LinkedHashMap<String, String>();
if (entity.listSlug != null) {
mention += entity.listSlug;
attrs.put("class", listClass);
attrs.put("href", listUrlBase + mention);
} else {
attrs.put("class", usernameClass);
attrs.put("href", usernameUrlBase + mention);
}
linkToTextWithSymbol(entity, atChar, mention, attrs, builder);
}
public void linkToText(final Entity entity, CharSequence text, final Map<String, String> attributes,
final StringBuilder builder) {
if (noFollow) {
attributes.put("rel", "nofollow");
}
if (linkAttributeModifier != null) {
linkAttributeModifier.modify(entity, attributes);
}
if (linkTextModifier != null) {
text = linkTextModifier.modify(entity, text);
}
// append <a> tag
builder.append("<a");
for (final Map.Entry<String, String> entry : attributes.entrySet()) {
builder.append(" ").append(escapeHTML(entry.getKey())).append("=\"").append(escapeHTML(entry.getValue()))
.append("\"");
}
builder.append(">").append(text).append("</a>");
}
public void linkToTextWithSymbol(final Entity entity, final CharSequence symbol, CharSequence text,
final Map<String, String> attributes, final StringBuilder builder) {
final CharSequence taggedSymbol = symbolTag == null || symbolTag.length() == 0 ? symbol : String.format(
"<%s>%s</%s>", symbolTag, symbol, symbolTag);
text = escapeHTML(text);
final CharSequence taggedText = textWithSymbolTag == null || textWithSymbolTag.length() == 0 ? text : String
.format("<%s>%s</%s>", textWithSymbolTag, text, textWithSymbolTag);
final boolean includeSymbol = usernameIncludeSymbol || !Regex.AT_SIGNS.matcher(symbol).matches();
if (includeSymbol) {
linkToText(entity, taggedSymbol.toString() + taggedText, attributes, builder);
} else {
builder.append(taggedSymbol);
linkToText(entity, taggedText, attributes, builder);
}
}
public void linkToURL(final Entity entity, final String text, final StringBuilder builder) {
final CharSequence url = entity.getValue();
CharSequence linkText = escapeHTML(url);
if (entity.displayURL != null && entity.expandedURL != null) {
final String displayURLSansEllipses = entity.displayURL.replace(FULLWIDTH_ELLIPSIS, "");
final int diplayURLIndexInExpandedURL = entity.expandedURL.indexOf(displayURLSansEllipses);
if (diplayURLIndexInExpandedURL != -1) {
final String beforeDisplayURL = entity.expandedURL.substring(0, diplayURLIndexInExpandedURL);
final String afterDisplayURL = entity.expandedURL.substring(diplayURLIndexInExpandedURL
+ displayURLSansEllipses.length());
final String precedingEllipsis = entity.displayURL.startsWith(FULLWIDTH_ELLIPSIS) ? FULLWIDTH_ELLIPSIS
: "";
final String followingEllipsis = entity.displayURL.endsWith(FULLWIDTH_ELLIPSIS) ? FULLWIDTH_ELLIPSIS
: "";
final String invisibleSpan = "<span " + invisibleTagAttrs + ">";
final StringBuilder sb = new StringBuilder("<span class='tco-ellipsis'>");
sb.append(precedingEllipsis);
sb.append(invisibleSpan).append(" </span></span>");
sb.append(invisibleSpan).append(escapeHTML(beforeDisplayURL)).append("</span>");
sb.append("<span class='js-display-url'>").append(escapeHTML(displayURLSansEllipses)).append("</span>");
sb.append(invisibleSpan).append(escapeHTML(afterDisplayURL)).append("</span>");
sb.append("<span class='tco-ellipsis'>").append(invisibleSpan).append(" </span>")
.append(followingEllipsis).append("</span>");
linkText = sb;
} else {
linkText = entity.displayURL;
}
}
final Map<String, String> attrs = new LinkedHashMap<String, String>();
attrs.put("href", url.toString());
if (urlClass != null) {
attrs.put("class", urlClass);
}
if (urlClass != null && urlClass.length() != 0) {
attrs.put("class", urlClass);
}
if (urlTarget != null && urlTarget.length() != 0) {
attrs.put("target", urlTarget);
}
linkToText(entity, linkText, attrs, builder);
}
/**
* Set the CSS class for auto-linked cashtag URLs
*
* @param cashtagClass new CSS value.
*/
public void setCashtagClass(final String cashtagClass) {
this.cashtagClass = cashtagClass;
}
/**
* Set the href base for cashtag links.
*
* @param cashtagUrlBase new href base value
*/
public void setCashtagUrlBase(final String cashtagUrlBase) {
this.cashtagUrlBase = cashtagUrlBase;
}
/**
* Set the CSS class for auto-linked hashtag URLs
*
* @param hashtagClass new CSS value.
*/
public void setHashtagClass(final String hashtagClass) {
this.hashtagClass = hashtagClass;
}
/**
* Set the href base for hashtag links.
*
* @param hashtagUrlBase new href base value
*/
public void setHashtagUrlBase(final String hashtagUrlBase) {
this.hashtagUrlBase = hashtagUrlBase;
}
/**
* Set a modifier to modify attributes of a link based on an entity
*
* @param modifier LinkAttributeModifier instance
*/
public void setLinkAttributeModifier(final LinkAttributeModifier modifier) {
linkAttributeModifier = modifier;
}
/**
* Set a modifier to modify text of a link based on an entity
*
* @param modifier LinkTextModifier instance
*/
public void setLinkTextModifier(final LinkTextModifier modifier) {
linkTextModifier = modifier;
}
/**
* Set the CSS class for auto-linked list URLs
*
* @param listClass new CSS value.
*/
public void setListClass(final String listClass) {
this.listClass = listClass;
}
/**
* Set the href base for list links.
*
* @param listUrlBase new href base value
*/
public void setListUrlBase(final String listUrlBase) {
this.listUrlBase = listUrlBase;
}
/**
* Set if the current URL links will include rel="nofollow" (true by
* default)
*
* @param noFollow new noFollow value
*/
public void setNoFollow(final boolean noFollow) {
this.noFollow = noFollow;
}
/**
* Set HTML tag to be applied around #/@/# symbols in
* hashtags/usernames/lists/cashtag
*
* @param tag HTML tag without bracket. e.g., "b" or "s"
*/
public void setSymbolTag(final String tag) {
symbolTag = tag;
}
/**
* Set HTML tag to be applied around text part of
* hashtags/usernames/lists/cashtag
*
* @param tag HTML tag without bracket. e.g., "b" or "s"
*/
public void setTextWithSymbolTag(final String tag) {
textWithSymbolTag = tag;
}
/**
* Set the CSS class for auto-linked URLs
*
* @param urlClass new CSS value.
*/
public void setUrlClass(final String urlClass) {
this.urlClass = urlClass;
}
/**
* Set the value of the target attribute in auto-linked URLs
*
* @param target target value e.g., "_blank"
*/
public void setUrlTarget(final String target) {
urlTarget = target;
}
/**
* Set the CSS class for auto-linked username URLs
*
* @param usernameClass new CSS value.
*/
public void setUsernameClass(final String usernameClass) {
this.usernameClass = usernameClass;
}
/**
* Set if the at mark '@' should be included in the link (false by default)
*
* @param noFollow new noFollow value
*/
public void setUsernameIncludeSymbol(final boolean usernameIncludeSymbol) {
this.usernameIncludeSymbol = usernameIncludeSymbol;
}
/**
* Set the href base for username links.
*
* @param usernameUrlBase new href base value
*/
public void setUsernameUrlBase(final String usernameUrlBase) {
this.usernameUrlBase = usernameUrlBase;
}
private static CharSequence escapeHTML(final CharSequence text) {
final int length = text.length();
final StringBuilder builder = new StringBuilder(length * 2);
for (int i = 0; i < length; i++) {
final char c = text.charAt(i);
switch (c) {
case '&':
builder.append("&");
break;
case '>':
builder.append(">");
break;
case '<':
builder.append("<");
break;
case '"':
builder.append(""");
break;
case '\'':
builder.append("'");
break;
default:
builder.append(c);
break;
}
}
return builder;
}
public static interface LinkAttributeModifier {
public void modify(Entity entity, Map<String, String> attributes);
}
public static interface LinkTextModifier {
public CharSequence modify(Entity entity, CharSequence text);
}
}