poe.trade.assist.service.AutoSearchService.java Source code

Introduction

Here is the source code for poe.trade.assist.service.AutoSearchService.java

Source

/*
 * Copyright (C) 2015 thirdy
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
package poe.trade.assist.service;

import static java.lang.String.format;
import static org.apache.commons.lang3.StringUtils.isNotBlank;

import java.io.File;
import java.util.UUID;
import java.util.function.Consumer;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.math.NumberUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javafx.application.Platform;
import javafx.beans.property.ListProperty;
import javafx.beans.property.SimpleListProperty;
import javafx.beans.property.SimpleStringProperty;
import javafx.beans.property.StringProperty;
import javafx.concurrent.Service;
import javafx.concurrent.Task;
import poe.trade.assist.Search;
import poe.trade.assist.scraper.BackendClient;
import poe.trade.assist.util.Dialogs;

public class AutoSearchService extends Service<Void> {

    private final Logger logger = LoggerFactory.getLogger(this.getClass().getName());

    private ListProperty<Search> searches = new SimpleListProperty<>();

    public ListProperty<Search> searchesProperty() {
        return searches;
    }

    //    private StringProperty minsToSleep = new SimpleStringProperty();
    //    public StringProperty minsToSleepProperty() {return minsToSleep;}

    private static final BackendClient backendClient = new BackendClient();

    // Just a way to provide notifications when searches' elements has been updated
    private Consumer<Integer> callback;

    private boolean autoEnabled;
    private float searchMins;

    public AutoSearchService(boolean autoEnabled, float searchMins) {
        this.autoEnabled = autoEnabled;
        this.searchMins = searchMins;
        setOnSucceeded(e -> restart());
        setOnFailed(e -> {
            getException().printStackTrace();
            Dialogs.showError(getException());
            try {
                Thread.sleep(5000);
            } catch (Exception e1) {
                e1.printStackTrace();
                return;
            }
            restart();
        });
    }

    @Override
    public void restart() {
        if (autoEnabled) {
            super.restart();
        }
    }

    @Override
    protected Task<Void> createTask() {
        return new Task<Void>() {

            @Override
            protected Void call() throws Exception {
                int numberOfItemsFound = 0;
                for (Search search : searches) {
                    String url = search.getUrl();
                    if (isNotBlank(url) && search.getAutoSearch()) {
                        update(format("Downloading... %s %s", search.getName(), url));
                        String html = doDownload(url, search.getSort());
                        //                         FileUtils.writeStringToFile(new File(search.getName()), html);
                        update(format("%s for %s %s", html.isEmpty() ? "Failure" : "Success", search.getName(),
                                url));
                        search.setHtml(html);
                        search.parseHtml();
                        if (search.getResultList() != null) { // i'm not sure if list will get null, the bane of java..
                            numberOfItemsFound += search.getResultList().stream().filter(r -> r.getIsNew()).count();
                        }
                        Thread.sleep(250); // a little delay between downloads
                    }
                }
                callback.accept(numberOfItemsFound);

                int mins = Math.round(60 * searchMins);

                for (int i = mins; i >= 0; i--) {
                    update("Sleeping... " + i);
                    Thread.sleep(1000);
                }
                return null;
            }

            private void update(String msg) {
                Platform.runLater(() -> updateMessage(msg));
            }
        };
    }

    public static String doDownload(String url, String sort) {
        int count = 0;
        int maxTries = 3;
        while (true) {
            try {
                return backendClient.postXMLHttpRequest(url, "sort=" + sort + "&bare=true");
            } catch (Exception e) {
                e.printStackTrace();
                if (++count == maxTries)
                    break;
            }
        }

        return "";
    }

    public void setCallback(Consumer<Integer> callback) {
        this.callback = callback;
    }

}