from flask import Flask, abort
import os
import psycopg2
import pymysql
import pymongo
import redis
import pyes

if os.getenv("DOCKERBUILD") == "True":
    import newsblur_web.docker_local_settings as settings
else:
    import flask_monitor.flask_settings as settings

app = Flask(__name__)

@app.route("/db_check/postgres")
def db_check_postgres():
    connect_params = "dbname='%s' user='%s' password='%s' host='%s' port='%s'" % (
        settings.DATABASES['default']['NAME'],
        settings.DATABASES['default']['USER'],
        settings.DATABASES['default']['PASSWORD'],
        settings.DATABASES['default']['HOST'],
        settings.DATABASES['default']['PORT'],
    )
    try:
        conn = psycopg2.connect(connect_params)
    except:
        print(" ---> Postgres can't connect to the database: %s" % connect_params)
        abort(502)

    cur = conn.cursor()
    cur.execute("""SELECT id FROM feeds ORDER BY feeds.id DESC LIMIT 1""")
    rows = cur.fetchall()
    for row in rows:
        return str(row[0])
    
    abort(404)

@app.route("/db_check/mysql")
def db_check_mysql():
    connect_params = "dbname='%s' user='%s' password='%s' host='%s' port='%s'" % (
        settings.DATABASES['default']['NAME'],
        settings.DATABASES['default']['USER'],
        settings.DATABASES['default']['PASSWORD'],
        settings.DATABASES['default']['HOST'],
        settings.DATABASES['default']['PORT'],
    )
    try:

        conn = pymysql.connect(host=settings.DATABASES['default']['HOST'],
                               port=settings.DATABASES['default']['PORT'],
                               user=settings.DATABASES['default']['USER'],
                               passwd=settings.DATABASES['default']['PASSWORD'],
                               db=settings.DATABASES['default']['NAME'])
    except:
        print(" ---> Mysql can't connect to the database: %s" % connect_params)
        abort(502)

    cur = conn.cursor()
    cur.execute("""SELECT id FROM feeds ORDER BY feeds.id DESC LIMIT 1""")
    rows = cur.fetchall()
    for row in rows:
        return str(row[0])
    
    abort(404)

@app.route("/db_check/mongo")
def db_check_mongo():
    try:
        client = pymongo.MongoClient('mongodb://%s' % settings.MONGO_DB['host'])
        db = client.newsblur
    except:
        abort(502)
    
    return str(db.stories.count())

@app.route("/db_check/redis")
def db_check_redis():
    redis_host = getattr(settings, 'REDIS', {'host': 'db_redis'})
    try:
        r = redis.Redis(redis_host['host'], db=0)
    except:
        abort(502)
    
    randkey = r.randomkey()
    if randkey:
        return str(randkey)
    else:
        abort(404)

@app.route("/db_check/redis_story")
def db_check_redis_story():
    redis_host = getattr(settings, 'REDIS', {'host': 'db_redis_story'})
    try:
        r = redis.Redis(redis_host['host'], db=1)
    except:
        abort(502)
    
    randkey = r.randomkey()
    if randkey:
        return str(randkey)
    else:
        abort(404)

@app.route("/db_check/redis_pubsub")
def db_check_redis_pubsub():
    redis_host = getattr(settings, 'REDIS', {'host': 'db_redis_pubsub'})
    try:
        r = redis.Redis(redis_host['host'], db=1)
    except:
        abort(502)
    
    pubsub_numpat = r.pubsub_numpat()
    if pubsub_numpat:
        return str(pubsub_numpat)
    else:
        abort(404)

    

@app.route("/db_check/redis_sessions")
def db_check_redis_sessions():
    redis_host = getattr(settings, 'REDIS', {'host': 'db_redis_sessions'})
    try:
        r = redis.Redis(redis_host['host'], db=5)
    except:
        abort(502)
    
    randkey = r.randomkey()
    if randkey:
        return str(randkey)
    else:
        abort(404)

@app.route("/db_check/elasticsearch")
def db_check_elasticsearch():
    es_host = getattr(settings, 'ELASTICSEARCH_FEED_HOSTS', ['db_search_feed:9200'])
    try:
        conn = pyes.ES(es_host)
    except:
        abort(502)
    
    if conn.indices.exists_index('feeds-index'):
        return str("Index exists, but didn't try search")
        # query = pyes.query.TermQuery("title", "daring fireball")
        # results = conn.search(query=query, size=1, doc_types=['feeds-type'], sort="num_subscribers:desc")
        # for result in results:
        #     return unicode(result)
        # else:
        #     abort(404)
    else:
        abort(404)    

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5579)