/*
* Firetweet - Twitter client for Android
*
* Copyright (C) 2012-2014 Mariotaku Lee <mariotaku.lee@gmail.com>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* 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.getlantern.firetweet.util.imageloader;
import android.content.Context;
import android.net.Uri;
import android.os.Build;
import android.text.TextUtils;
import android.webkit.URLUtil;
import com.nostra13.universalimageloader.core.assist.ContentLengthInputStream;
import com.nostra13.universalimageloader.core.download.BaseImageDownloader;
import com.squareup.pollexor.Thumbor;
import com.squareup.pollexor.ThumborUrlBuilder;
import org.getlantern.firetweet.Constants;
import org.getlantern.firetweet.R;
import org.getlantern.firetweet.constant.SharedPreferenceConstants;
import org.getlantern.firetweet.model.ParcelableAccount;
import org.getlantern.firetweet.model.ParcelableAccount.ParcelableCredentials;
import org.getlantern.firetweet.model.ParcelableMedia;
import org.getlantern.firetweet.util.MediaPreviewUtils;
import org.getlantern.firetweet.util.SharedPreferencesWrapper;
import org.getlantern.firetweet.util.Utils;
import java.io.IOException;
import java.io.InputStream;
import java.util.Locale;
import twitter4j.TwitterException;
import twitter4j.auth.Authorization;
import twitter4j.http.HeaderMap;
import twitter4j.http.HttpClientWrapper;
import twitter4j.http.HttpResponse;
import static org.getlantern.firetweet.util.FiretweetLinkify.PATTERN_TWITTER_PROFILE_IMAGES;
import static org.getlantern.firetweet.util.Utils.getImageLoaderHttpClient;
import static org.getlantern.firetweet.util.Utils.getNormalTwitterProfileImage;
import static org.getlantern.firetweet.util.Utils.getRedirectedHttpResponse;
import static org.getlantern.firetweet.util.Utils.getTwitterAuthorization;
import static org.getlantern.firetweet.util.Utils.getTwitterProfileImageOfSize;
public class FiretweetImageDownloader extends BaseImageDownloader implements Constants {
private final Context mContext;
private final SharedPreferencesWrapper mPreferences;
private final boolean mUseThumbor;
private Thumbor mThumbor;
private HttpClientWrapper mClient;
private boolean mFastImageLoading;
private final boolean mFullImage;
private final String mTwitterProfileImageSize;
public FiretweetImageDownloader(final Context context, final boolean fullImage, final boolean useThumbor) {
super(context);
mContext = context;
mPreferences = SharedPreferencesWrapper.getInstance(context, SHARED_PREFERENCES_NAME,
Context.MODE_PRIVATE, SharedPreferenceConstants.class);
mFullImage = fullImage;
mTwitterProfileImageSize = context.getString(R.string.profile_image_size);
mUseThumbor = useThumbor;
reloadConnectivitySettings();
}
public void reloadConnectivitySettings() {
mClient = getImageLoaderHttpClient(mContext);
mFastImageLoading = mPreferences.getBoolean(KEY_FAST_IMAGE_LOADING);
if (mUseThumbor && mPreferences.getBoolean(KEY_THUMBOR_ENABLED)) {
final String address = mPreferences.getString(KEY_THUMBOR_ADDRESS, null);
final String securityKey = mPreferences.getString(KEY_THUMBOR_SECURITY_KEY, null);
if (URLUtil.isValidUrl(address)) {
if (TextUtils.isEmpty(securityKey)) {
mThumbor = Thumbor.create(address);
} else {
mThumbor = Thumbor.create(address, securityKey);
}
} else {
mThumbor = null;
}
} else {
mThumbor = null;
}
}
@Override
protected InputStream getStreamFromNetwork(final String uriString, final Object extras) throws IOException {
if (uriString == null) return null;
final ParcelableMedia media = MediaPreviewUtils.getAllAvailableImage(uriString, mFullImage, mFullImage
|| !mFastImageLoading ? mClient : null);
try {
final String mediaUrl = media != null ? media.media_url : uriString;
if (isTwitterProfileImage(uriString)) {
final String replaced = getTwitterProfileImageOfSize(mediaUrl, mTwitterProfileImageSize);
return getStreamFromNetworkInternal(replaced, extras);
} else
return getStreamFromNetworkInternal(mediaUrl, extras);
} catch (final TwitterException e) {
final int statusCode = e.getStatusCode();
if (statusCode != -1 && isTwitterProfileImage(uriString) && !uriString.contains("_normal.")) {
try {
return getStreamFromNetworkInternal(getNormalTwitterProfileImage(uriString), extras);
} catch (final TwitterException e2) {
}
}
throw new IOException(String.format(Locale.US, "Error downloading image %s, error code: %d", uriString,
statusCode));
}
}
private String getReplacedUri(final Uri uri, final String apiUrlFormat) {
if (uri == null) return null;
if (apiUrlFormat == null) return uri.toString();
if (isTwitterUri(uri)) {
final StringBuilder sb = new StringBuilder();
final String domain = uri.getHost().replaceAll("\\.?twitter.com", "");
final String path = uri.getPath();
sb.append(Utils.getApiUrl(apiUrlFormat, domain, path));
final String query = uri.getQuery();
if (!TextUtils.isEmpty(query)) {
sb.append("?");
sb.append(query);
}
final String fragment = uri.getFragment();
if (!TextUtils.isEmpty(fragment)) {
sb.append("#");
sb.append(fragment);
}
return sb.toString();
}
return uri.toString();
}
private ContentLengthInputStream getStreamFromNetworkInternal(final String uriString, final Object extras)
throws IOException, TwitterException {
final Uri uri = Uri.parse(uriString);
final Authorization auth;
final ParcelableCredentials account;
if (isTwitterAuthRequired(uri) && extras instanceof AccountExtra) {
final AccountExtra accountExtra = (AccountExtra) extras;
account = ParcelableAccount.getCredentials(mContext, accountExtra.account_id);
auth = getTwitterAuthorization(mContext, accountExtra.account_id);
} else {
account = null;
auth = null;
}
String modifiedUri = getReplacedUri(uri, account != null ? account.api_url_format : null);
if (mThumbor != null) {
modifiedUri = mThumbor.buildImage(modifiedUri).filter(ThumborUrlBuilder.quality(85)).toUrl();
}
final HeaderMap additionalHeaders = new HeaderMap();
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.JELLY_BEAN_MR1) {
additionalHeaders.addHeader("Accept", "image/webp, */*");
}
final HttpResponse resp = getRedirectedHttpResponse(mClient, modifiedUri, uriString, auth, additionalHeaders);
return new ContentLengthInputStream(resp.asStream(), (int) resp.getContentLength());
}
private boolean isTwitterAuthRequired(final Uri uri) {
if (uri == null) return false;
return "ton.twitter.com".equalsIgnoreCase(uri.getHost());
}
private boolean isTwitterProfileImage(final String uriString) {
if (TextUtils.isEmpty(uriString)) return false;
return PATTERN_TWITTER_PROFILE_IMAGES.matcher(uriString).matches();
}
private boolean isTwitterUri(final Uri uri) {
if (uri == null) return false;
return "ton.twitter.com".equalsIgnoreCase(uri.getHost());
}
}