use curl::easy::{Easy, WriteError};

use curl::Error;

use dbus::{BusType, Connection, Message};

use log::{debug, info, warn};

use scraper::{Html, Selector};

use std::fs;

use std::str;

use std::thread;

use std::time;

fn inform(url: &str, title: &str, author: &str) {

    let c = Connection::get_private(BusType::Session).unwrap();

    let mut m = Message::new_method_call(

        "org.kde.konversation",

        "/irc",

        "org.kde.konversation",

        "say",

    )

    .unwrap();

    let msg_str = &format!("[{}] {}: {}", url, author, title)[..];

    info!("Notification: {}", msg_str);

    m.append_items(&["freenode".into(), "#hedgewars".into(), msg_str.into()]);

    let r = c.send(m).unwrap();

}

fn handle_page(url: &str, data: &[u8]) -> Result<usize, WriteError> {

    let page = str::from_utf8(data).unwrap();

    let document = Html::parse_document(page);

    let title_selector = Selector::parse("title").unwrap();

    let mut titles = document.select(&title_selector);

    if let Some(title_str) = titles.next() {

        let t = &title_str.inner_html();

        let title = &t[..t.len() - 12];

        //info!("Topic is '{}'", title);

        let author_selector = Selector::parse("div.forum-post-wrapper div.forum-post-panel-sub div.author-pane div.author-pane-inner div.author-pane-name-status.author-pane-section div.author-pane-line.author-name").unwrap();

        let mut authors = document.select(&author_selector);

        if let Some(author_name) = authors.next() {

            let author = &author_name.inner_html();

            //info!("Author is '{}'", author);

            inform(url, title, &author[1..author.len() - 1]);

        } else {

            warn!("Author not found");

        }

    } else {

        warn!("No title found");

    }

    Ok(data.len())

}

fn query(url: &str) -> Result<(), Error> {

    let mut easy = Easy::new();

    easy.url(url)?;

    easy.fail_on_error(true);

    let mut transfer = easy.transfer();

    transfer.write_function(|data| handle_page(url, data))?;

    transfer.perform()

}

fn main() {

    stderrlog::new()

        .verbosity(3)

        .timestamp(stderrlog::Timestamp::Second)

        .module(module_path!())

        .init()

        .unwrap();

    let short_delay = time::Duration::from_secs(3);

    let long_delay = time::Duration::from_secs(7 * 60);

    let file_name = "/usr/home/unC0Rr/.local/unC0Rr/forumchecker/node.txt";

    let mut node_id: u32 = fs::read_to_string(file_name).unwrap().parse().unwrap();

    info!("Starting from node id {}", node_id);

    loop {

        debug!("Quering node id {}", node_id);

        let res = query(&format!("https://www.hedgewars.org/node/{}", node_id));

        if let Ok(_) = res {

            node_id += 1;

            fs::write(file_name, node_id.to_string()).unwrap();

            thread::sleep(short_delay);

        } else {

            thread::sleep(long_delay);

        }

    }

}