#!/usr/bin/env python2.7

# Copyright (c) 2012 netzguerilla.net <iro@netzguerilla.net>

#

# This file is part of Iro.

#

# Permission is hereby granted, free of charge, to any person obtaining a copy of

# this software and associated documentation files (the "Software"), to deal in

# the Software without restriction, including without limitation the rights to use,

# copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the

# #Software, and to permit persons to whom the Software is furnished to do so,

# subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in

# all copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,

# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A

# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION

# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE

# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

from twisted.python import log

from sqlalchemy import create_engine, pool

from sqlalchemy.exc import ArgumentError

import sys,os

import random

import logging

from iro.controller.pool import dbPool

from iro import config, __version__, install

from iro.model.schema import Offer, User, Userright

from iro.model.utils import WithSession

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 adduser.')

parser.add_argument('-v', '--version', action='version', version="%s %s"%(os.path.basename(sys.argv[0]),__version__))

parser.add_argument('name', help="username")

parser.add_argument('offers', nargs='+', help="offers for user")

args = parser.parse_args()

if not install.checkConfig():

    logging.error("iro.conf is not in right format.")

    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.")

    logging.info("Please edit iro.conf and run iro-install --install")

    sys.exit(1)

#test offers

with WithSession(engine) as session:

        available_offers = [i[0] for i in session.query(Offer.name)]

error = False

for o in args.offers:

    if o not in available_offers:

        logging.error("Unknown offer: '%s'"%o)

	error = True

if error:

    sys.exit(1)

#create apikey

apikey = "".join([random.choice("0123456789abcdef") for i in range(15)])

#adduser to database

with WithSession(engine, False) as session:

    u = User(name=args.name, apikey=apikey)

    session.add(u)

    for i,offer in enumerate(args.offers):

        o = session.query(Offer).filter_by(name=offer).first()

        u.rights.append(Userright(o,default=i+1))

    session.commit()

logging.info("created user %s:"% args.name)

logging.info("apikey = %s"% apikey)

logging.info("offers = %s"% ", ".join(args.offers))