/*
* Copyright (C) 2010-2014 Geometer Plus <contact@geometerplus.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 2 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, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301, USA.
*/
package org.geometerplus.fbreader.network.opds;
import java.util.*;
import java.net.URLEncoder;
import java.io.InputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import org.geometerplus.zlibrary.core.network.ZLNetworkException;
import org.geometerplus.zlibrary.core.network.ZLNetworkRequest;
import org.geometerplus.zlibrary.core.util.MimeType;
import org.geometerplus.fbreader.network.*;
import org.geometerplus.fbreader.network.authentication.NetworkAuthenticationManager;
import org.geometerplus.fbreader.network.urlInfo.*;
import org.geometerplus.fbreader.network.tree.NetworkItemsLoader;
public abstract class OPDSNetworkLink extends AbstractNetworkLink {
private TreeMap<RelationAlias,String> myRelationAliases;
private final LinkedList<URLRewritingRule> myUrlRewritingRules = new LinkedList<URLRewritingRule>();
private final Map<String,String> myExtraData = new HashMap<String,String>();
private NetworkAuthenticationManager myAuthenticationManager;
OPDSNetworkLink(int id, String siteName, String title, String summary, String language,
UrlInfoCollection<UrlInfoWithDate> infos) {
super(id, siteName, title, summary, language, infos);
}
final void setRelationAliases(Map<RelationAlias,String> relationAliases) {
if (relationAliases != null && relationAliases.size() > 0) {
myRelationAliases = new TreeMap<RelationAlias,String>(relationAliases);
} else {
myRelationAliases = null;
}
}
final void setUrlRewritingRules(List<URLRewritingRule> rules) {
myUrlRewritingRules.clear();
myUrlRewritingRules.addAll(rules);
}
final void setExtraData(Map<String,String> extraData) {
myExtraData.clear();
myExtraData.putAll(extraData);
}
final void setAuthenticationManager(NetworkAuthenticationManager mgr) {
myAuthenticationManager = mgr;
}
/*
public AccountStatus getAccountStatus(boolean force) {
if (myAuthenticationManager == null) {
return AccountStatus.NotSupported;
}
if ("".equals(myAuthenticationManager.UserNameOption.getValue())) {
return AccountStatus.NoUserName;
}
}
SignedIn,
SignedOut,
NotChecked
*/
ZLNetworkRequest createNetworkData(String url, MimeType mime, final OPDSCatalogItem.State result) {
if (url == null) {
return null;
}
final NetworkLibrary library = NetworkLibrary.Instance();
final NetworkCatalogItem catalogItem = result.Loader.getTree().Item;
library.startLoading(catalogItem);
url = rewriteUrl(url, false);
return new ZLNetworkRequest(url, mime, null, false) {
@Override
public void handleStream(InputStream inputStream, int length) throws IOException, ZLNetworkException {
if (result.Loader.confirmInterruption()) {
return;
}
new OPDSXMLReader(
new OPDSFeedHandler(getURL(), result), false
).read(inputStream);
if (result.Loader.confirmInterruption() && result.LastLoadedId != null) {
// reset state to load current page from the beginning
result.LastLoadedId = null;
} else {
result.Loader.getTree().confirmAllItems();
}
}
@Override
public void doAfter(boolean success) {
library.stopLoading(catalogItem);
}
};
}
@Override
public OPDSCatalogItem.State createOperationData(NetworkItemsLoader loader) {
return new OPDSCatalogItem.State(this, loader);
}
public ZLNetworkRequest simpleSearchRequest(String pattern, NetworkOperationData data) {
final UrlInfo info = getUrlInfo(UrlInfo.Type.Search);
if (info == null || info.Url == null) {
return null;
}
try {
pattern = URLEncoder.encode(pattern, "utf-8");
} catch (UnsupportedEncodingException e) {
}
return createNetworkData(info.Url.replace("%s", pattern), info.Mime, (OPDSCatalogItem.State)data);
}
public ZLNetworkRequest resume(NetworkOperationData data) {
return createNetworkData(data.ResumeURI, MimeType.APP_ATOM_XML, (OPDSCatalogItem.State)data);
}
public NetworkCatalogItem libraryItem() {
final UrlInfoCollection<UrlInfo> urlMap = new UrlInfoCollection<UrlInfo>();
urlMap.addInfo(getUrlInfo(UrlInfo.Type.Catalog));
urlMap.addInfo(getUrlInfo(UrlInfo.Type.Image));
urlMap.addInfo(getUrlInfo(UrlInfo.Type.Thumbnail));
return new OPDSCatalogItem(
this,
getTitle(),
getSummary(),
urlMap,
OPDSCatalogItem.Accessibility.ALWAYS,
OPDSCatalogItem.FLAGS_DEFAULT | OPDSCatalogItem.FLAG_ADD_SEARCH_ITEM,
myExtraData
);
}
public NetworkAuthenticationManager authenticationManager() {
return myAuthenticationManager;
}
public String rewriteUrl(String url, boolean isUrlExternal) {
final int apply = isUrlExternal
? URLRewritingRule.APPLY_EXTERNAL : URLRewritingRule.APPLY_INTERNAL;
for (URLRewritingRule rule : myUrlRewritingRules) {
if ((rule.whereToApply() & apply) != 0) {
url = rule.apply(url);
}
}
return url;
}
// rel and type must be either null or interned String objects.
String relation(String rel, MimeType type) {
if (myRelationAliases == null) {
return rel;
}
RelationAlias alias = new RelationAlias(rel, type.Name);
String mapped = myRelationAliases.get(alias);
if (mapped != null) {
return mapped;
}
if (type != null) {
alias = new RelationAlias(rel, null);
mapped = myRelationAliases.get(alias);
if (mapped != null) {
return mapped;
}
}
return rel;
}
private BasketItem myBasketItem;
@Override
public BasketItem getBasketItem() {
final String url = getUrl(UrlInfo.Type.ListBooks);
if (url != null && myBasketItem == null) {
myBasketItem = new OPDSBasketItem(this);
}
return myBasketItem;
}
@Override
public String toString() {
return "OPDSNetworkLink: {super=" + super.toString()
+ "; authManager=" + (myAuthenticationManager != null ? myAuthenticationManager.getClass().getName() : null)
+ "; relationAliases=" + myRelationAliases
+ "; rewritingRules=" + myUrlRewritingRules
+ "}";
}
}