#! /usr/bin/env python2
import logging
from time import sleep


logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# FILTERS = ["tangodown", "malware", "spam", "hacker", "anonymous", "cyberwar"]
FILTERS = ["malware"]
DATABASE = {
    'name': 'blackforest',
    # 'engine': 'peewee.PostgresqlDatabase',
    'user': 'postgres',
    'password': 'testing',
    'host': 'localhost'
}

# TWITTER =====================================================================
from collectors.twitter import Twitter
from config import (KESTREL_SERVER_IP, KESTREL_SERVER_PORT, TWITTER_ACCESS_KEY,
                    TWITTER_ACCESS_SECRET, TWITTER_CONSUMER_KEY,
                    TWITTER_CONSUMER_SECRET)

config = {
    "consumer_key": TWITTER_CONSUMER_KEY,
    "consumer_secret": TWITTER_CONSUMER_SECRET,
    "access_key": TWITTER_ACCESS_KEY,
    "access_secret": TWITTER_ACCESS_SECRET,
    "kestrel_conn_str": ':'.join([KESTREL_SERVER_IP,
                                  KESTREL_SERVER_PORT]),
    "db_conn_dict": DATABASE,
    "filters": FILTERS
}

logger.info("Running Twitter collector...")
Twitter.run(config)

while(True):
    try:
        sleep(10)
    except KeyboardInterrupt:
        logger.info("Shutting down")
        break

# twitter.disconnect()


# PASTEBIN=====================================================================
# logger.info("Running pastebin collector...")
# from collectors.pastebin import Pastebin
#  config = {
#     "filters":FILTERS,
#     "db_conn_dict":DATABASE
# }
# Pastebin.run(config)
# pastebin.run()

# IRC =========================================================================
# from collectors.irc import Irc
# logger.info("Running IRC collector...")
# #irc = Irc(kestrel_ip=KESTREL_SERVER_IP, kestrel_port=KESTREL_SERVER_PORT)
#
# irc.add_connection("chat.freenode.net", "6667",
# "#archlinux", "jinglejangle")
# irc.add_connection("chat.freenode.net", "6667",
# "#bot_testing", "jinglejangle2")
# config = {"kestrel_conn_str": ':'.join([KESTREL_SERVER_IP,
#                                        KESTREL_SERVER_PORT])}
# config["connections"] = [{
#    "channel_list": ["#bot_testing", "#bot_testing2"],
#    "server": "irc.freenode.net",
#    "port": "6667",
#    "nickname": 'GravCat'
# }]
##
# Irc.run(config)

# GOOGLE ======================================================================
# from collectors.google import Google
# config = {'filters': FILTERS,
#           'db_conn_dict': DATABASE,
#           'kestrel_conn_str': "127.0.0.1:22133"}
# Google.run(config)

# TwitterFollow ==============================================================
from collectors.twitter_follower import TwitterFollower
logger.info("Running Twitter Follower collector...")
config = {
    "kestrel_conn_str": ':'.join([KESTREL_SERVER_IP,
                                  KESTREL_SERVER_PORT]),
    "consumer_key": TWITTER_CONSUMER_KEY,
    "consumer_secret": TWITTER_CONSUMER_SECRET,
    "access_key": TWITTER_ACCESS_KEY,
    "access_secret": TWITTER_ACCESS_SECRET,
}
TwitterFollower.run(config)