#!/usr/bin/env python2.7
from twisted.python import log

from sqlalchemy import create_engine, pool
from sqlalchemy.exc import ArgumentError
import sys,os

import logging

from iro.controller.pool import dbPool
from iro import config, __version__, install

observer = log.PythonLoggingObserver()
observer.start()

logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(name)s(%(processName)s)-%(levelname)s: %(message)s')

import argparse

parser = argparse.ArgumentParser(description='Iro installer.')
parser.add_argument('-v', '--version', action='version', version="%s %s"%(os.path.basename(sys.argv[0]),__version__))
parser.add_argument('--install', action='store_true',
                           help='will create the right database layout.')
parser.add_argument('--update', action='store_true',
                           help='will install all routes and providers.')
args = parser.parse_args()

if not install.checkConfig():
    install.createSampleConfig()
    logging.info("Please edit iro.conf")
    sys.exit(1)

config.init()

try:
    engine = create_engine(config.main.dburl,
           poolclass = pool.SingletonThreadPool,  pool_size=dbPool.maxthreads, )
except ArgumentError:
    logging.error("Can't connect to database")
    logging.info("Please edit iro.conf")
    sys.exit(1)


if not install.checkDatabaseConnection():
    logging.error("Can't connect to database")
    logging.info("Please edit iro.conf")
    sys.exit(1)
    
if not install.checkDatabase():
    logging.error("Database not in right format.")
    if args.install:
        install.createDatabase()
        logging.info("Database layout created.")
    else:
        logging.info("Please edit iro.conf and run %s --install"%os.path.basename(sys.argv[0]))
        sys.exit(1)

routes = [ s for s in config.configParser.sections() if not s in ["main",]]
ao =  install.getAllRoutes(routes, False)
for o in  ao["orphand"]:
    logging.info("Offer(%s) is orphand (no route using this offer)."%o)
if ao["added"]:
    if args.install or args.update:
        ao = install.getAllRoutes(routes,True)
        for a in  ao["added"]:
            logging.info("Added Offer(%s)"%a)
        logging.info('Updated offerlist.')
    else:
        logging.warning('offerlist is not up-to-date.')
        logging.info("Please run %s --update"%os.path.basename(sys.argv[0]))
        sys.exit(1)

logging.info("You can just start your iro server.")