use of com.biglybt.core.subs.SubscriptionException in project BiglyBT by BiglySoftware.
the class SubscriptionDownloader method download.
protected void download() throws SubscriptionException {
log("Downloading");
Map map = JSONUtils.decodeJSON(subs.getJSON());
Long engine_id = (Long) map.get("engine_id");
String search_term = (String) map.get("search_term");
String networks = (String) map.get("networks");
Map filters = (Map) map.get("filters");
Engine engine = manager.getEngine(subs, map, false);
if (engine == null) {
throw (new SubscriptionException("Download failed, search engine " + engine_id + " not found"));
}
List sps = new ArrayList();
if (search_term != null) {
sps.add(new SearchParameter("s", search_term));
log(" Using search term '" + search_term + "' for engine " + engine.getString());
}
if (networks != null && networks.length() > 0) {
sps.add(new SearchParameter("n", networks));
}
/*
if ( mature != null ){
sps.add( new SearchParameter( "m", mature.toString()));
}
*/
SearchParameter[] parameters = (SearchParameter[]) sps.toArray(new SearchParameter[sps.size()]);
SubscriptionHistoryImpl history = (SubscriptionHistoryImpl) subs.getHistory();
try {
Map context = new HashMap();
context.put(Engine.SC_SOURCE, "subscription");
Result[] results = engine.search(parameters, context, -1, -1, null, null);
log(" Got " + results.length + " results");
SubscriptionResultFilterImpl result_filter = new SubscriptionResultFilterImpl(subs, filters);
results = result_filter.filter(results);
log(" Post-filter: " + results.length + " results");
SubscriptionResultImpl[] s_results = new SubscriptionResultImpl[results.length];
for (int i = 0; i < results.length; i++) {
SubscriptionResultImpl s_result = new SubscriptionResultImpl(history, results[i]);
s_results[i] = s_result;
}
SubscriptionResultImpl[] all_results = history.reconcileResults(engine, s_results);
checkAutoDownload(all_results);
history.setLastError(null, false);
} catch (Throwable e) {
log(" Download failed", e);
history.setLastError(Debug.getNestedExceptionMessage(e), e instanceof SearchLoginException);
throw (new SubscriptionException("Search failed", e));
}
}
use of com.biglybt.core.subs.SubscriptionException in project BiglyBT by BiglySoftware.
the class MetaSearchListener method handleSubscriptionAuth.
protected boolean handleSubscriptionAuth(final Subscription subs, final Map result) {
if (subs.getHistory().isAuthFail()) {
try {
Engine engine = subs.getEngine();
if (engine instanceof WebEngine) {
final WebEngine webEngine = (WebEngine) engine;
synchronized (active_subs_auth) {
if (active_subs_auth.contains(subs)) {
return (false);
}
active_subs_auth.add(subs);
}
Utils.execSWTThread(new Runnable() {
@Override
public void run() {
new ExternalLoginWindow(new ExternalLoginListener() {
private String previous_cookies;
private boolean result_sent;
@Override
public void canceled(ExternalLoginWindow window) {
try {
encodeResults(subs, result);
sendBrowserMessage("metasearch", "readSubscriptionResultsCompleted", result);
} finally {
completed();
}
}
@Override
public void cookiesFound(ExternalLoginWindow window, String cookies) {
if (handleCookies(cookies, false)) {
window.close();
}
}
@Override
public void done(ExternalLoginWindow window, String cookies) {
try {
if (!handleCookies(cookies, true)) {
encodeResults(subs, result);
sendBrowserMessage("metasearch", "readSubscriptionResultsCompleted", result);
}
} finally {
completed();
}
}
private void completed() {
synchronized (active_subs_auth) {
active_subs_auth.remove(subs);
}
}
private boolean handleCookies(String cookies, boolean force_if_ready) {
if (result_sent) {
return (false);
}
String[] required = webEngine.getRequiredCookies();
boolean skip = required.length == 0 && !force_if_ready;
if (CookieParser.cookiesContain(required, cookies)) {
webEngine.setCookies(cookies);
if (previous_cookies == null || !previous_cookies.equals(cookies)) {
previous_cookies = cookies;
if (!skip) {
// search operation will report outcome
result_sent = true;
try {
subs.getManager().getScheduler().download(subs, false, new SubscriptionDownloadListener() {
@Override
public void complete(Subscription subs) {
result.put("id", subs.getID());
encodeResults(subs, result);
sendBrowserMessage("metasearch", "readSubscriptionResultsCompleted", result);
}
@Override
public void failed(Subscription subs, SubscriptionException error) {
Debug.out(error);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(error));
sendBrowserMessage("metasearch", "readSubscriptionResultsFailed", result);
}
});
} catch (Throwable error) {
Debug.out(error);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(error));
sendBrowserMessage("metasearch", "readSubscriptionResultsFailed", result);
}
}
}
}
return (result_sent);
}
}, webEngine.getName(), webEngine.getLoginPageUrl(), false, webEngine.getAuthMethod(), subs.isMine());
}
});
return (true);
} else {
return (false);
}
} catch (Throwable e) {
Debug.printStackTrace(e);
return (false);
}
} else {
return (false);
}
}
use of com.biglybt.core.subs.SubscriptionException in project BiglyBT by BiglySoftware.
the class MetaSearchListener method handleMessage.
@Override
public void handleMessage(BrowserMessage message) {
String opid = message.getOperationId();
MetaSearchManager metaSearchManager = MetaSearchManagerFactory.getSingleton();
metaSearchManager.log("BrowserListener: received " + message);
if (OP_SEARCH.equals(opid)) {
Map decodedMap = message.getDecodedMap();
search(decodedMap, null);
} else if (OP_ENGINE_PREFERRED.equals(opid)) {
final Map decodedMap = message.getDecodedMap();
long engine_id = ((Long) decodedMap.get("engine_id")).longValue();
Engine engine = getEngineFromId(engine_id);
if (engine != null) {
metaSearchManager.getMetaSearch().enginePreferred(engine);
}
} else if (OP_ENGINE_LOGIN.equals(opid)) {
final Map decodedMap = message.getDecodedMap();
long engine_id = ((Long) decodedMap.get("engine_id")).longValue();
final Long sid = (Long) decodedMap.get("sid");
final Engine engine = getEngineFromId(engine_id);
if (engine != null && engine instanceof WebEngine) {
final WebEngine webEngine = (WebEngine) engine;
Utils.execSWTThread(new Runnable() {
@Override
public void run() {
new ExternalLoginWindow(new ExternalLoginListener() {
private String previous_cookies;
private boolean search_done;
@Override
public void canceled(ExternalLoginWindow window) {
/* gouss doesn't wan't anything on cancel
if ( !outcome_informed ){
outcome_informed = true;
Map params = getParams( webEngine );
params.put( "error", "operation cancelled" );
sendBrowserMessage("metasearch", "engineFailed", params );
}
*/
}
@Override
public void cookiesFound(ExternalLoginWindow window, String cookies) {
if (handleCookies(cookies, false)) {
window.close();
}
}
@Override
public void done(ExternalLoginWindow window, String cookies) {
handleCookies(cookies, true);
/*
if ( !outcome_informed ){
outcome_informed = true;
Map params = getParams( webEngine );
sendBrowserMessage("metasearch", "engineCompleted", params );
}
*/
}
private boolean handleCookies(String cookies, boolean force_if_ready) {
if (search_done) {
return (false);
}
String[] required = webEngine.getRequiredCookies();
boolean skip_search = required.length == 0 && !force_if_ready;
if (CookieParser.cookiesContain(required, cookies)) {
webEngine.setCookies(cookies);
if (previous_cookies == null || !previous_cookies.equals(cookies)) {
previous_cookies = cookies;
if (!skip_search) {
// search operation will report outcome
search_done = true;
search(decodedMap, webEngine);
}
}
}
return (search_done);
}
}, webEngine.getName(), webEngine.getLoginPageUrl(), false, webEngine.getAuthMethod(), engine.isMine());
}
});
} else {
Map<String, Object> params = new HashMap<>();
if (sid != null) {
params.put("sid", sid);
}
params.put("error", "engine not found or not a web engine");
sendBrowserMessage("metasearch", "engineFailed", params);
}
} else if (OP_GET_LOGIN_COOKIES.equals(opid)) {
final Map decodedMap = message.getDecodedMap();
final String url = ((String) decodedMap.get("url")).replaceAll("%s", "");
Utils.execSWTThread(new Runnable() {
@Override
public void run() {
new ExternalLoginWindow(new ExternalLoginListener() {
@Override
public void canceled(ExternalLoginWindow window) {
sendBrowserMessage("metasearch", "setCookiesFailed", new HashMap());
}
@Override
public void cookiesFound(ExternalLoginWindow window, String cookies) {
}
@Override
public void done(ExternalLoginWindow window, String cookies) {
String[] cookieNames = CookieParser.getCookiesNames(cookies);
Map<String, Object> params = new HashMap<>();
params.put("cookieNames", cookieNames);
params.put("currentCookie", cookies);
params.put("cookieMethod", window.proxyCaptureModeRequired() ? WebEngine.AM_PROXY : WebEngine.AM_TRANSPARENT);
sendBrowserMessage("metasearch", "setCookies", params);
}
}, url, url, true, WebEngine.AM_PROXY, true);
}
});
} else if (OP_GET_ENGINES.equals(opid)) {
String subscriptionId = null;
try {
final Map decodedMap = message.getDecodedMap();
subscriptionId = ((String) decodedMap.get("subs_id"));
} catch (Exception e) {
// No parameters
}
Engine[] engines = null;
if (subscriptionId != null) {
engines = new Engine[0];
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(subscriptionId);
if (subs != null) {
try {
Engine engine = subs.getEngine();
if (engine != null) {
engines = new Engine[] { engine };
}
} catch (Throwable e) {
Debug.out(e);
}
}
}
if (engines == null) {
engines = metaSearchManager.getMetaSearch().getEngines(true, true);
}
List params = new ArrayList();
for (int i = 0; i < engines.length; i++) {
Engine engine = engines[i];
if (!engine.isActive() || engine.getSource() == Engine.ENGINE_SOURCE_UNKNOWN) {
if (subscriptionId == null) {
continue;
}
}
Map<String, Object> engineMap = new HashMap<>();
engineMap.put("id", new Long(engine.getId()));
engineMap.put("name", engine.getName());
engineMap.put("favicon", engine.getIcon());
engineMap.put("dl_link_css", engine.getDownloadLinkCSS());
engineMap.put("selected", Engine.SEL_STATE_STRINGS[engine.getSelectionState()]);
engineMap.put("type", Engine.ENGINE_SOURCE_STRS[engine.getSource()]);
engineMap.put("shareable", Boolean.valueOf(engine.isShareable()));
params.add(engineMap);
}
sendBrowserMessage("metasearch", "enginesUsed", params);
} else if (OP_GET_ALL_ENGINES.equals(opid)) {
Engine[] engines = metaSearchManager.getMetaSearch().getEngines(false, true);
List params = new ArrayList();
for (int i = 0; i < engines.length; i++) {
Engine engine = engines[i];
if (engine.getSource() == Engine.ENGINE_SOURCE_UNKNOWN) {
continue;
}
Map<String, Object> engineMap = new HashMap<>();
engineMap.put("id", new Long(engine.getId()));
engineMap.put("name", engine.getName());
engineMap.put("favicon", engine.getIcon());
engineMap.put("dl_link_css", engine.getDownloadLinkCSS());
engineMap.put("selected", Engine.SEL_STATE_STRINGS[engine.getSelectionState()]);
engineMap.put("type", Engine.ENGINE_SOURCE_STRS[engine.getSource()]);
engineMap.put("shareable", Boolean.valueOf(engine.isShareable()));
params.add(engineMap);
}
sendBrowserMessage("metasearch", "engineList", params);
} else if (OP_SET_SELECTED_ENGINES.equals(opid)) {
Map decodedMap = message.getDecodedMap();
List template_ids = (List) decodedMap.get("template_ids");
long[] ids = new long[template_ids.size()];
for (int i = 0; i < ids.length; i++) {
Map m = (Map) template_ids.get(i);
ids[i] = ((Long) m.get("id")).longValue();
}
boolean auto = ((Boolean) decodedMap.get("auto")).booleanValue();
// there's some code that attempts to switch to 'auto=true' on first use as
// when 3110 defaults to false and the decision was made to switch this
// disable the behaviour if we are customised
Boolean is_default = (Boolean) decodedMap.get("set_default");
boolean skip = false;
if (is_default != null && is_default.booleanValue()) {
if (CustomizationManagerFactory.getSingleton().getActiveCustomization() != null) {
skip = true;
}
}
try {
if (!skip) {
metaSearchManager.setSelectedEngines(ids, auto);
}
Map<String, Object> params = new HashMap<>();
sendBrowserMessage("metasearch", "setSelectedCompleted", params);
} catch (Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("error", Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "setSelectedFailed", params);
}
} else if (OP_CHANGE_ENGINE_SELECTION.equals(opid)) {
Map decodedMap = message.getDecodedMap();
MetaSearch ms = metaSearchManager.getMetaSearch();
Engine[] engines = ms.getEngines(false, true);
Set selected = new HashSet();
for (int i = 0; i < engines.length; i++) {
Engine e = engines[i];
if (e.getSelectionState() == Engine.SEL_STATE_MANUAL_SELECTED) {
selected.add(new Long(e.getId()));
}
}
List l_engines = (List) decodedMap.get("engines");
for (int i = 0; i < l_engines.size(); i++) {
Map map = (Map) l_engines.get(i);
long id = ((Long) map.get("id")).longValue();
String str = (String) map.get("selected");
if (str.equalsIgnoreCase(Engine.SEL_STATE_STRINGS[Engine.SEL_STATE_MANUAL_SELECTED])) {
selected.add(new Long(id));
} else if (str.equalsIgnoreCase(Engine.SEL_STATE_STRINGS[Engine.SEL_STATE_DESELECTED])) {
selected.remove(new Long(id));
}
}
long[] ids = new long[selected.size()];
Iterator it = selected.iterator();
int pos = 0;
while (it.hasNext()) {
long id = ((Long) it.next()).longValue();
ids[pos++] = id;
}
try {
metaSearchManager.setSelectedEngines(ids, metaSearchManager.isAutoMode());
Map<String, Object> params = new HashMap<>();
sendBrowserMessage("metasearch", "changeEngineSelectionCompleted", params);
} catch (Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("error", Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "changeEngineSelectionFailed", params);
}
} else if (OP_GET_AUTO_MODE.equals(opid)) {
boolean mode = metaSearchManager.isAutoMode();
Map<String, Object> params = new HashMap<>();
params.put("auto", Boolean.valueOf(mode));
boolean custom = CustomizationManagerFactory.getSingleton().getActiveCustomization() != null;
params.put("is_custom", Boolean.valueOf(custom));
sendBrowserMessage("metasearch", "getAutoModeResult", params);
} else if (OP_SAVE_TEMPLATE.equals(opid)) {
Map decodedMap = message.getDecodedMap();
String type_str = (String) decodedMap.get("type");
String name = (String) decodedMap.get("name");
Long l_id = (Long) decodedMap.get("id");
long id = l_id == null ? -1 : l_id.longValue();
String json = (String) decodedMap.get("value");
String cookies = (String) decodedMap.get("current_cookie");
try {
Engine engine = metaSearchManager.addEngine(id, type_str.equals("json") ? Engine.ENGINE_TYPE_JSON : Engine.ENGINE_TYPE_REGEX, name, json);
engine.setMine(true);
if (cookies != null && engine instanceof WebEngine) {
WebEngine we = (WebEngine) engine;
we.setCookies(cookies);
}
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(engine.getId()));
sendBrowserMessage("metasearch", "saveTemplateCompleted", params);
} catch (Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "saveTemplateFailed", params);
}
} else if (OP_LOAD_TEMPLATE.equals(opid)) {
Map decodedMap = message.getDecodedMap();
long id = ((Long) decodedMap.get("id")).longValue();
Engine engine = metaSearchManager.getMetaSearch().getEngine(id);
if (engine == null) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", "Template not found");
sendBrowserMessage("metasearch", "loadTemplateFailed", params);
} else {
try {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(engine.getId()));
params.put("name", engine.getName());
int type = engine.getType();
params.put("type", type < Engine.ENGINE_TYPE_STRS.length ? Engine.ENGINE_TYPE_STRS[type] : type);
params.put("value", JSONObject.escape(engine.exportToJSONString()));
params.put("shareable", Boolean.valueOf(engine.isShareable()));
params.put("uid", engine.getUID());
params.put("supports_direct_download", Boolean.valueOf(engine.supportsField(Engine.FIELD_TORRENTLINK) || engine.supportsField(Engine.FIELD_DOWNLOADBTNLINK)));
params.put("auto_dl_supported", Boolean.valueOf(engine.getAutoDownloadSupported() == Engine.AUTO_DL_SUPPORTED_YES));
sendBrowserMessage("metasearch", "loadTemplateCompleted", params);
} catch (Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "loadTemplateFailed", params);
}
}
} else if (OP_DELETE_TEMPLATE.equals(opid)) {
Map decodedMap = message.getDecodedMap();
long id = ((Long) decodedMap.get("id")).longValue();
Engine engine = metaSearchManager.getMetaSearch().getEngine(id);
if (engine == null) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", "Template not found");
sendBrowserMessage("metasearch", "deleteTemplateFailed", params);
} else if (engine.getSource() != Engine.ENGINE_SOURCE_LOCAL) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", "Template is not local");
sendBrowserMessage("metasearch", "deleteTemplateFailed", params);
} else {
engine.delete();
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
sendBrowserMessage("metasearch", "deleteTemplateCompleted", params);
}
} else if (OP_TEST_TEMPLATE.equals(opid)) {
Map decodedMap = message.getDecodedMap();
final long id = ((Long) decodedMap.get("id")).longValue();
long match_count = ((Long) decodedMap.get("max_matches")).longValue();
String searchText = (String) decodedMap.get("searchText");
String headers = (String) decodedMap.get("headers");
final Long sid = (Long) decodedMap.get("sid");
Engine engine = metaSearchManager.getMetaSearch().getEngine(id);
if (engine == null) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", "Template not found");
if (sid != null)
params.put("sid", sid);
sendBrowserMessage("metasearch", "testTemplateFailed", params);
} else {
SearchParameter parameter = new SearchParameter("s", searchText);
SearchParameter[] parameters = new SearchParameter[] { parameter };
try {
engine.search(parameters, new HashMap(), (int) match_count, (int) match_count, headers, new ResultListener() {
private String content;
private List matches = new ArrayList();
@Override
public void contentReceived(Engine engine, String _content) {
content = _content;
}
@Override
public void matchFound(Engine engine, String[] fields) {
matches.add(fields);
}
@Override
public void resultsReceived(Engine engine, Result[] results) {
}
@Override
public void resultsComplete(Engine engine) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
if (sid != null)
params.put("sid", sid);
params.put("content", JSONObject.escape(content));
JSONArray l_matches = new JSONArray();
params.put("matches", l_matches);
for (int i = 0; i < matches.size(); i++) {
String[] match = (String[]) matches.get(i);
JSONArray l_match = new JSONArray();
l_matches.add(l_match);
Collections.addAll(l_match, match);
}
sendBrowserMessage("metasearch", "testTemplateCompleted", params);
}
@Override
public void engineFailed(Engine engine, Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", Debug.getNestedExceptionMessage(e));
if (sid != null)
params.put("sid", sid);
sendBrowserMessage("metasearch", "testTemplateFailed", params);
}
@Override
public void engineRequiresLogin(Engine engine, Throwable e) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", Debug.getNestedExceptionMessage(e));
if (sid != null)
params.put("sid", sid);
sendBrowserMessage("metasearch", "testTemplateRequiresLogin", params);
}
});
} catch (SearchException e) {
// listener handles
}
}
} else if (OP_EXPORT_TEMPLATE.equals(opid)) {
Map decodedMap = message.getDecodedMap();
final long id = ((Long) decodedMap.get("id")).longValue();
final Engine engine = metaSearchManager.getMetaSearch().getEngine(id);
if (engine == null) {
Map<String, Object> params = new HashMap<>();
params.put("error", "template '" + id + "' not found");
sendBrowserMessage("metasearch", "exportTemplateFailed", params);
} else {
final Shell shell = Utils.findAnyShell();
shell.getDisplay().asyncExec(new AERunnable() {
@Override
public void runSupport() {
FileDialog dialog = new FileDialog(shell, SWT.SYSTEM_MODAL | SWT.SAVE);
dialog.setFilterPath(TorrentOpener.getFilterPathData());
dialog.setText(MessageText.getString("metasearch.export.select.template.file"));
dialog.setFilterExtensions(VuzeFileHandler.getVuzeFileFilterExtensions());
dialog.setFilterNames(VuzeFileHandler.getVuzeFileFilterExtensions());
String path = TorrentOpener.setFilterPathData(dialog.open());
if (path != null) {
if (!VuzeFileHandler.isAcceptedVuzeFileName(path)) {
path = VuzeFileHandler.getVuzeFileName(path);
}
try {
engine.exportToVuzeFile(new File(path));
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
sendBrowserMessage("metasearch", "exportTemplateCompleted", params);
} catch (Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", "save failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "exportTemplateFailed", params);
}
} else {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(id));
params.put("error", "operation cancelled");
sendBrowserMessage("metasearch", "exportTemplateFailed", params);
}
}
});
}
} else if (OP_IMPORT_TEMPLATE.equals(opid)) {
final Shell shell = Utils.findAnyShell();
shell.getDisplay().asyncExec(new AERunnable() {
@Override
public void runSupport() {
FileDialog dialog = new FileDialog(shell, SWT.SYSTEM_MODAL | SWT.OPEN);
dialog.setFilterPath(TorrentOpener.getFilterPathData());
dialog.setText(MessageText.getString("metasearch.import.select.template.file"));
dialog.setFilterExtensions(VuzeFileHandler.getVuzeFileFilterExtensions());
dialog.setFilterNames(VuzeFileHandler.getVuzeFileFilterExtensions());
String path = TorrentOpener.setFilterPathData(dialog.open());
if (path != null) {
VuzeFileHandler vfh = VuzeFileHandler.getSingleton();
VuzeFile vf = vfh.loadAndHandleVuzeFile(path, VuzeFileComponent.COMP_TYPE_METASEARCH_TEMPLATE);
if (vf == null) {
Map<String, Object> params = new HashMap<>();
params.put("error", "invalid .biglybt file");
sendBrowserMessage("metasearch", "importTemplateFailed", params);
} else {
VuzeFileComponent[] comps = vf.getComponents();
for (int i = 0; i < comps.length; i++) {
VuzeFileComponent comp = comps[i];
if (comp.getType() == VuzeFileComponent.COMP_TYPE_METASEARCH_TEMPLATE) {
Engine engine = (Engine) comp.getData(Engine.VUZE_FILE_COMPONENT_ENGINE_KEY);
if (engine != null) {
Map<String, Object> params = new HashMap<>();
params.put("id", new Long(engine.getId()));
sendBrowserMessage("metasearch", "importTemplateCompleted", params);
return;
}
}
}
Map<String, Object> params = new HashMap<>();
params.put("error", "invalid search template file");
sendBrowserMessage("metasearch", "importTemplateFailed", params);
}
} else {
Map<String, Object> params = new HashMap<>();
// don't change this message as the UI uses it!
params.put("error", "operation cancelled");
sendBrowserMessage("metasearch", "importTemplateFailed", params);
}
}
});
} else if (OP_OPEN_SEARCH_RESULTS.equals(opid)) {
Map decodedMap = message.getDecodedMap();
openCloseSearchDetailsListener.openSearchResults(decodedMap);
} else if (OP_CLOSE_SEARCH_RESULTS.equals(opid)) {
Map decodedMap = message.getDecodedMap();
openCloseSearchDetailsListener.closeSearchResults(decodedMap);
} else if (OP_LOAD_TORRENT.equals(opid)) {
Map decodedMap = message.getDecodedMap();
String torrentUrl = (String) decodedMap.get("torrent_url");
String referer_str = (String) decodedMap.get("referer_url");
try {
Map headers = UrlUtils.getBrowserHeaders(referer_str);
String subscriptionId = ((String) decodedMap.get("subs_id"));
String subscriptionResultId = ((String) decodedMap.get("subs_rid"));
if (subscriptionId != null) {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(subscriptionId);
if (subs != null) {
try {
Engine engine = subs.getEngine();
if (engine != null && engine instanceof WebEngine) {
WebEngine webEngine = (WebEngine) engine;
if (webEngine.isNeedsAuth()) {
headers.put("Cookie", webEngine.getCookies());
}
}
} catch (Throwable e) {
Debug.out(e);
}
if (subscriptionResultId != null) {
subs.addPotentialAssociation(subscriptionResultId, torrentUrl);
}
}
} else {
try {
long engineID = ((Long) decodedMap.get("engine_id")).longValue();
Engine engine = metaSearchManager.getMetaSearch().getEngine(engineID);
if (engine != null) {
engine.addPotentialAssociation(torrentUrl);
}
if (engine != null && engine instanceof WebEngine) {
WebEngine webEngine = (WebEngine) engine;
if (webEngine.isNeedsAuth()) {
headers.put("Cookie", webEngine.getCookies());
}
}
} catch (Throwable e) {
Debug.printStackTrace(e);
}
}
try {
String hash = (String) decodedMap.get("hash");
if (hash != null) {
if (!torrentUrl.toLowerCase(Locale.US).startsWith("magnet:")) {
headers.put("X-Alternative-URI-1", UrlUtils.getMagnetURI(Base32.decode(hash)));
}
}
} catch (Throwable e) {
}
PluginInitializer.getDefaultInterface().getDownloadManager().addDownload(new URL(torrentUrl), headers);
Map<String, Object> params = new HashMap<>();
params.put("torrent_url", torrentUrl);
params.put("referer_url", referer_str);
sendBrowserMessage("metasearch", "loadTorrentCompleted", params);
} catch (Exception e) {
Map<String, Object> params = new HashMap<>();
params.put("torrent_url", torrentUrl);
params.put("referer_url", referer_str);
params.put("error", e.getMessage());
sendBrowserMessage("metasearch", "loadTorrentFailed", params);
}
} else if (OP_HAS_LOAD_TORRENT.equals(opid)) {
Map<String, Object> params = new HashMap<>();
params.put("result", "1");
sendBrowserMessage("metasearch", "hasLoadTorrent", params);
} else if (OP_CREATE_SUBSCRIPTION.equals(opid)) {
Map decodedMap = message.getDecodedMap();
Long tid = (Long) decodedMap.get("tid");
String name = (String) decodedMap.get("name");
Boolean isPublic = (Boolean) decodedMap.get("is_public");
Map options = (Map) decodedMap.get("options");
Boolean isEnabled = (Boolean) options.get("is_enabled");
Boolean autoDownload = (Boolean) options.get("auto_dl");
Map<String, Object> result = new HashMap<>();
if (tid != null)
result.put("tid", tid);
try {
JSONObject payload = new JSONObject();
// change this you need to change update too below
payload.put("engine_id", decodedMap.get("engine_id"));
payload.put("search_term", decodedMap.get("search_term"));
payload.put("filters", decodedMap.get("filters"));
payload.put("schedule", decodedMap.get("schedule"));
payload.put("options", decodedMap.get("options"));
Subscription subs = SubscriptionManagerFactory.getSingleton().create(name, isPublic.booleanValue(), payload.toString());
subs.getHistory().setDetails(isEnabled == null ? true : isEnabled.booleanValue(), autoDownload == null ? false : autoDownload.booleanValue());
result.put("id", subs.getID());
subs.requestAttention();
sendBrowserMessage("metasearch", "createSubscriptionCompleted", result);
} catch (Throwable e) {
Debug.out(e);
result.put("error", "create failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "createSubscriptionFailed", result);
}
} else if (OP_READ_SUBSCRIPTION.equals(opid)) {
Map decodedMap = message.getDecodedMap();
final Long tid = (Long) decodedMap.get("tid");
final String sid = (String) decodedMap.get("id");
Map<String, Object> result = new HashMap<>();
if (tid != null)
result.put("tid", tid);
try {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
result.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "readSubscriptionFailed", result);
} else {
boolean shareable = subs.isShareable();
// override public flag if not shareable
result.put("id", subs.getID());
result.put("name", subs.getName());
result.put("is_public", Boolean.valueOf(shareable && subs.isPublic()));
result.put("is_author", Boolean.valueOf(subs.isMine()));
result.put("is_shareable", Boolean.valueOf(shareable));
result.put("auto_dl_supported", Boolean.valueOf(subs.isAutoDownloadSupported()));
SubscriptionHistory history = subs.getHistory();
Map<String, Object> options = new HashMap<>();
result.put("options", options);
options.put("is_enabled", Boolean.valueOf(history.isEnabled()));
options.put("auto_dl", Boolean.valueOf(history.isAutoDownload()));
Map<String, Object> info = new HashMap<>();
result.put("info", info);
info.put("last_scan", new Long(history.getLastScanTime()));
info.put("last_new", new Long(history.getLastNewResultTime()));
info.put("num_unread", new Long(history.getNumUnread()));
info.put("num_read", new Long(history.getNumRead()));
String json = subs.getJSON();
Map map = JSONUtils.decodeJSON(json);
result.put("engine_id", map.get("engine_id"));
result.put("search_term", map.get("search_term"));
result.put("filters", map.get("filters"));
result.put("schedule", map.get("schedule"));
sendBrowserMessage("metasearch", "readSubscriptionCompleted", result);
}
} catch (Throwable e) {
Debug.out(e);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "readSubscriptionFailed", result);
}
} else if (OP_UPDATE_SUBSCRIPTION.equals(opid)) {
Map decodedMap = message.getDecodedMap();
final Long tid = (Long) decodedMap.get("tid");
final String name = (String) decodedMap.get("name");
final Boolean isPublic = (Boolean) decodedMap.get("is_public");
final String sid = (String) decodedMap.get("id");
Map options = (Map) decodedMap.get("options");
Boolean isEnabled = (Boolean) options.get("is_enabled");
Boolean autoDownload = (Boolean) options.get("auto_dl");
Map<String, Object> result = new HashMap<>();
if (tid != null)
result.put("tid", tid);
try {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
result.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "updateSubscriptionFailed", result);
} else {
JSONObject payload = new JSONObject();
// change this you need to change create too above
payload.put("engine_id", decodedMap.get("engine_id"));
payload.put("search_term", decodedMap.get("search_term"));
payload.put("filters", decodedMap.get("filters"));
payload.put("schedule", decodedMap.get("schedule"));
payload.put("options", decodedMap.get("options"));
boolean changed = subs.setDetails(name, isPublic.booleanValue(), payload.toString());
subs.getHistory().setDetails(isEnabled == null ? true : isEnabled.booleanValue(), autoDownload == null ? false : autoDownload.booleanValue());
if (changed) {
subs.reset();
try {
subs.getManager().getScheduler().downloadAsync(subs, true);
} catch (Throwable e) {
Debug.out(e);
}
}
result.put("id", subs.getID());
sendBrowserMessage("metasearch", "updateSubscriptionCompleted", result);
}
} catch (Throwable e) {
Debug.out(e);
result.put("error", "update failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "updateSubscriptionFailed", result);
}
} else if (OP_SUBSCRIPTION_SET_AUTODL.equals(opid)) {
Map decodedMap = message.getDecodedMap();
String sid = (String) decodedMap.get("id");
Long tid = (Long) decodedMap.get("tid");
Boolean autoDownload = (Boolean) decodedMap.get("auto_dl");
Map<String, Object> result = new HashMap<>();
if (tid != null)
result.put("tid", tid);
try {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
result.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "setSubscriptionAutoDownloadFailed", result);
} else {
subs.getHistory().setAutoDownload(autoDownload.booleanValue());
sendBrowserMessage("metasearch", "setSubscriptionAutoDownloadCompleted", result);
}
} catch (Throwable e) {
Debug.out(e);
result.put("error", "update failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "setSubscriptionAutoDownloadFailed", result);
}
} else if (OP_READ_SUBSCRIPTION_RESULTS.equals(opid)) {
Map decodedMap = message.getDecodedMap();
final Long tid = (Long) decodedMap.get("tid");
final String sid = (String) decodedMap.get("id");
final Map<String, Object> result = new HashMap<>();
if (tid != null)
result.put("tid", tid);
try {
final Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
result.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "readSubscriptionResultsFailed", result);
} else {
result.put("id", subs.getID());
if (!handleSubscriptionAuth(subs, result)) {
if (subs.getHistory().getLastScanTime() == 0) {
subs.getManager().getScheduler().download(subs, false, new SubscriptionDownloadListener() {
@Override
public void complete(Subscription subs) {
if (!handleSubscriptionAuth(subs, result)) {
encodeResults(subs, result);
sendBrowserMessage("metasearch", "readSubscriptionResultsCompleted", result);
openCloseSearchDetailsListener.resizeMainBrowser();
}
}
@Override
public void failed(Subscription subs, SubscriptionException error) {
Debug.out(error);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(error));
sendBrowserMessage("metasearch", "readSubscriptionResultsFailed", result);
}
});
} else {
encodeResults(subs, result);
sendBrowserMessage("metasearch", "readSubscriptionResultsCompleted", result);
openCloseSearchDetailsListener.resizeMainBrowser();
}
}
}
} catch (Throwable e) {
Debug.out(e);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "readSubscriptionFailed", result);
}
} else if (OP_DELETE_SUBSCRIPTION_RESULTS.equals(opid)) {
Map decodedMap = message.getDecodedMap();
String sid = (String) decodedMap.get("id");
List rids = (List) decodedMap.get("rids");
try {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
Map<String, Object> params = new HashMap<>();
params.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "deleteSubscriptionResultsFailed", params);
} else {
String[] rids_a = (String[]) rids.toArray(new String[rids.size()]);
subs.getHistory().deleteResults(rids_a);
Map<String, Object> result = new HashMap<>();
result.put("rids", rids);
sendBrowserMessage("metasearch", "deleteSubscriptionResultsCompleted", result);
}
} catch (Throwable e) {
Debug.out(e);
Map<String, Object> params = new HashMap<>();
params.put("error", "delete failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "deleteSubscriptionResultsFailed", params);
}
} else if (OP_MARK_SUBSCRIPTION_RESULTS.equals(opid)) {
Map decodedMap = message.getDecodedMap();
String sid = (String) decodedMap.get("id");
List rids = (List) decodedMap.get("rids");
List reads = (List) decodedMap.get("reads");
Map<String, Object> result = new HashMap<>();
try {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
result.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "markSubscriptionResultsFailed", result);
} else {
String[] rids_a = (String[]) rids.toArray(new String[rids.size()]);
boolean[] reads_a = new boolean[reads.size()];
for (int i = 0; i < reads.size(); i++) {
reads_a[i] = ((Boolean) reads.get(i)).booleanValue();
}
subs.getHistory().markResults(rids_a, reads_a);
result.put("rids", rids);
result.put("reads", reads);
sendBrowserMessage("metasearch", "markSubscriptionResultsCompleted", result);
}
} catch (Throwable e) {
Debug.out(e);
result.put("error", "mark failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "markSubscriptionResultsFailed", result);
}
} else if (OP_DOWNLOAD_SUBSCRIPTION.equals(opid)) {
Map decodedMap = message.getDecodedMap();
final Long tid = (Long) decodedMap.get("tid");
final String sid = (String) decodedMap.get("id");
final Map<String, Object> result = new HashMap<>();
if (tid != null)
result.put("tid", tid);
try {
Subscription subs = SubscriptionManagerFactory.getSingleton().getSubscriptionByID(sid);
if (subs == null) {
result.put("error", "Subscription not found");
sendBrowserMessage("metasearch", "downloadSubscriptionFailed", result);
} else {
result.put("id", subs.getID());
if (!handleSubscriptionAuth(subs, result)) {
subs.getManager().getScheduler().download(subs, false, new SubscriptionDownloadListener() {
@Override
public void complete(Subscription subs) {
if (!handleSubscriptionAuth(subs, result)) {
encodeResults(subs, result);
sendBrowserMessage("metasearch", "downloadSubscriptionCompleted", result);
}
}
@Override
public void failed(Subscription subs, SubscriptionException error) {
Debug.out(error);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(error));
sendBrowserMessage("metasearch", "downloadSubscriptionFailed", result);
}
});
}
}
} catch (Throwable e) {
Debug.out(e);
result.put("error", "read failed: " + Debug.getNestedExceptionMessage(e));
sendBrowserMessage("metasearch", "downloadSubscriptionFailed", result);
}
} else if (OP_IS_CUSTOMISED.equals(opid)) {
boolean custom = CustomizationManagerFactory.getSingleton().getActiveCustomization() != null;
Map<String, Object> params = new HashMap<>();
params.put("is_custom", Boolean.valueOf(custom));
sendBrowserMessage("metasearch", "isCustomizedResult", params);
}
}
use of com.biglybt.core.subs.SubscriptionException in project BiglyBT by BiglySoftware.
the class SearchUtils method showCreateSubscriptionDialog.
public static void showCreateSubscriptionDialog(final long engineID, final String searchTerm, final Map optionalFilters) {
final SkinnedDialog dialog = new SkinnedDialog("skin3_dlg_create_search_subscription", "shell", SWT.DIALOG_TRIM);
SWTSkin skin = dialog.getSkin();
final SWTSkinObjectTextbox tb = (SWTSkinObjectTextbox) skin.getSkinObject("sub-name");
final SWTSkinObjectCheckbox cbShare = (SWTSkinObjectCheckbox) skin.getSkinObject("sub-share");
final SWTSkinObjectCheckbox cbAutoDL = (SWTSkinObjectCheckbox) skin.getSkinObject("sub-autodl");
SWTSkinObject soEngineArea = skin.getSkinObject("sub-engine-area");
final SWTSkinObjectCombo soEngines = (SWTSkinObjectCombo) skin.getSkinObject("sub-engine");
if (tb == null || cbShare == null || cbAutoDL == null) {
return;
}
boolean hasEngineID = engineID >= 0;
soEngineArea.setVisible(!hasEngineID);
final Map<Integer, Engine> mapEngines = new HashMap<>();
if (!hasEngineID) {
Engine[] engines = MetaSearchManagerFactory.getSingleton().getMetaSearch().getEngines(true, false);
List<String> list = new ArrayList<>();
int pos = 0;
for (Engine engine : engines) {
mapEngines.put(pos++, engine);
list.add(engine.getName());
}
soEngines.setList(list.toArray(new String[list.size()]));
}
cbShare.setChecked(COConfigurationManager.getBooleanParameter("sub.sharing.default.checked"));
cbAutoDL.setChecked(COConfigurationManager.getBooleanParameter("sub.autodl.default.checked"));
SWTSkinObject soButtonArea = skin.getSkinObject("bottom-area");
if (soButtonArea instanceof SWTSkinObjectContainer) {
StandardButtonsArea buttonsArea = new StandardButtonsArea() {
// @see StandardButtonsArea#clicked(int)
@Override
protected void clicked(int buttonValue) {
if (buttonValue == SWT.OK) {
String name = tb.getText().trim();
boolean isShared = cbShare.isChecked();
boolean autoDL = cbAutoDL.isChecked();
long realEngineID = engineID;
if (engineID <= 0) {
int engineIndex = soEngines.getComboControl().getSelectionIndex();
if (engineIndex < 0) {
// TODO: Flicker combobox
return;
}
realEngineID = mapEngines.get(engineIndex).getId();
}
Map<String, Object> payload = new HashMap<>();
payload.put("engine_id", realEngineID);
payload.put("search_term", searchTerm);
Map<String, Object> mapSchedule = new HashMap<>();
mapSchedule.put("days", Collections.EMPTY_LIST);
// minutes
mapSchedule.put("interval", 120);
payload.put("schedule", mapSchedule);
Map<String, Object> mapOptions = new HashMap<>();
mapOptions.put("auto_dl", autoDL);
payload.put("options", mapOptions);
Map<String, Object> mapFilters = new HashMap<>();
if (optionalFilters != null) {
mapFilters.putAll(optionalFilters);
}
payload.put("filters", mapFilters);
try {
Subscription subs;
subs = SubscriptionManagerFactory.getSingleton().create(name, isShared, JSONUtils.encodeToJSON(payload));
subs.getHistory().setDetails(true, autoDL);
subs.requestAttention();
} catch (SubscriptionException e) {
}
}
dialog.close();
}
};
buttonsArea.setButtonIDs(new String[] { MessageText.getString("Button.add"), MessageText.getString("Button.cancel") });
buttonsArea.setButtonVals(new Integer[] { SWT.OK, SWT.CANCEL });
buttonsArea.swt_createButtons(((SWTSkinObjectContainer) soButtonArea).getComposite());
}
dialog.open();
}
use of com.biglybt.core.subs.SubscriptionException in project BiglyBT by BiglySoftware.
the class SubscriptionBodyImpl method writeVuzeFile.
protected void writeVuzeFile(SubscriptionImpl subs) throws SubscriptionException {
try {
File file = manager.getVuzeFile(subs);
Map details = (Map) map.get("details");
updateDetails(subs, details);
byte[] contents = BEncoder.encode(details);
byte[] new_hash = new SHA1Simple().calculateHash(contents);
byte[] old_hash = (byte[]) map.get("hash");
if (old_hash != null && !Arrays.equals(old_hash, new_hash)) {
Map details_copy = new HashMap(details);
details_copy.remove("az_version");
contents = BEncoder.encode(details_copy);
new_hash = new SHA1Simple().calculateHash(contents);
}
if (old_hash == null || !Arrays.equals(old_hash, new_hash)) {
byte[] private_key = subs.getPrivateKey();
if (private_key == null) {
throw (new SubscriptionException("Only the originator of a subscription can modify it"));
}
map.put("size", new Long(contents.length));
try {
map.put("hash", new_hash);
map.put("sig", sign(private_key, new_hash, version, contents.length));
} catch (Throwable e) {
throw (new SubscriptionException("Crypto failed: " + Debug.getNestedExceptionMessage(e)));
}
}
File backup_file = null;
if (file.exists()) {
backup_file = new File(file.getParent(), file.getName() + ".bak");
backup_file.delete();
if (!file.renameTo(backup_file)) {
throw (new SubscriptionException("Backup failed"));
}
}
try {
VuzeFile vf = VuzeFileHandler.getSingleton().create();
vf.addComponent(VuzeFileComponent.COMP_TYPE_SUBSCRIPTION, map);
vf.write(file);
hash = new_hash;
sig = (byte[]) map.get("sig");
sig_data_size = contents.length;
} catch (Throwable e) {
if (backup_file != null) {
backup_file.renameTo(file);
}
throw (new SubscriptionException("File write failed: " + Debug.getNestedExceptionMessage(e)));
}
} catch (Throwable e) {
rethrow(e);
}
}
Aggregations