Search in sources :

Example 1 with SubscriptionDownloadListener

use of com.biglybt.core.subs.SubscriptionDownloadListener 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);
    }
}
Also used : SubscriptionException(com.biglybt.core.subs.SubscriptionException) WebEngine(com.biglybt.core.metasearch.impl.web.WebEngine) SubscriptionDownloadListener(com.biglybt.core.subs.SubscriptionDownloadListener) Subscription(com.biglybt.core.subs.Subscription) WebEngine(com.biglybt.core.metasearch.impl.web.WebEngine)

Example 2 with SubscriptionDownloadListener

use of com.biglybt.core.subs.SubscriptionDownloadListener 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);
    }
}
Also used : SubscriptionResult(com.biglybt.core.subs.SubscriptionResult) SubscriptionDownloadListener(com.biglybt.core.subs.SubscriptionDownloadListener) Subscription(com.biglybt.core.subs.Subscription) WebEngine(com.biglybt.core.metasearch.impl.web.WebEngine) SubscriptionException(com.biglybt.core.subs.SubscriptionException) SubscriptionHistory(com.biglybt.core.subs.SubscriptionHistory) WebEngine(com.biglybt.core.metasearch.impl.web.WebEngine) JSONObject(org.json.simple.JSONObject) JSONObject(org.json.simple.JSONObject) VuzeFile(com.biglybt.core.vuzefile.VuzeFile) File(java.io.File) VuzeFileComponent(com.biglybt.core.vuzefile.VuzeFileComponent) URL(java.net.URL) Shell(org.eclipse.swt.widgets.Shell) VuzeFile(com.biglybt.core.vuzefile.VuzeFile) JSONArray(org.json.simple.JSONArray) SubscriptionException(com.biglybt.core.subs.SubscriptionException) FileDialog(org.eclipse.swt.widgets.FileDialog) VuzeFileHandler(com.biglybt.core.vuzefile.VuzeFileHandler)

Aggregations

WebEngine (com.biglybt.core.metasearch.impl.web.WebEngine)2 Subscription (com.biglybt.core.subs.Subscription)2 SubscriptionDownloadListener (com.biglybt.core.subs.SubscriptionDownloadListener)2 SubscriptionException (com.biglybt.core.subs.SubscriptionException)2 SubscriptionHistory (com.biglybt.core.subs.SubscriptionHistory)1 SubscriptionResult (com.biglybt.core.subs.SubscriptionResult)1 VuzeFile (com.biglybt.core.vuzefile.VuzeFile)1 VuzeFileComponent (com.biglybt.core.vuzefile.VuzeFileComponent)1 VuzeFileHandler (com.biglybt.core.vuzefile.VuzeFileHandler)1 File (java.io.File)1 URL (java.net.URL)1 FileDialog (org.eclipse.swt.widgets.FileDialog)1 Shell (org.eclipse.swt.widgets.Shell)1 JSONArray (org.json.simple.JSONArray)1 JSONObject (org.json.simple.JSONObject)1