"""cubicweb-clt handlers for twisted
"""

import sys

from cubicweb.toolsutils import CommandHandler
from cubicweb.web.webctl import WebCreateHandler

# trigger configuration registration
import cubicweb.etwist.twconfig # pylint: disable-msg=W0611


class TWCreateHandler(WebCreateHandler):
    cfgname = 'twisted'

    def bootstrap(self, cubes, inputlevel=0):
        """bootstrap this configuration"""
        print '** twisted configuration'
        mainpyfile = self.config.server_file()
        mainpy = open(mainpyfile, 'w')
        mainpy.write('''
from cubicweb.etwist import server
application = server.main(%r, %r)
''' % (self.config.appid, self.config.name))
        mainpy.close()
        print 'application\'s twisted file %s generated' % mainpyfile
        super(TWCreateHandler, self).bootstrap(cubes, inputlevel)


class TWStartHandler(CommandHandler):
    cmdname = 'start'
    cfgname = 'twisted'

    def start_command(self, config, debug):
        command = ['%s `which twistd`' % sys.executable]
        for ctl_opt, server_opt in (('pid-file', 'pidfile'),
                                    ('uid', 'uid'),
                                    ('log-file', 'logfile',)):
            value = config[ctl_opt]
            if not value or (debug and ctl_opt == 'log-file'):
                continue
            command.append('--%s %s' % (server_opt, value))
        if debug:
            command.append('-n')
        if config['profile']:
            command.append('-p %s --savestats' % config['profile'])
        command.append('-oy')
        command.append(self.config.server_file())
        return ' '.join(command)


class TWStopHandler(CommandHandler):
    cmdname = 'stop'
    cfgname = 'twisted'
    
    
try:
    from cubicweb.server import serverctl

    class AllInOneCreateHandler(serverctl.RepositoryCreateHandler, TWCreateHandler):
        """configuration to get a web application running in a twisted web
        server integrating a repository server in the same process
        """
        cfgname = 'all-in-one'

        def bootstrap(self, cubes, inputlevel=0):
            """bootstrap this configuration"""
            serverctl.RepositoryCreateHandler.bootstrap(self, cubes, inputlevel)
            TWCreateHandler.bootstrap(self, cubes, inputlevel)
            
    class AllInOneStartHandler(TWStartHandler):
        cmdname = 'start'
        cfgname = 'all-in-one'
        subcommand = 'cubicweb-twisted'

    class AllInOneStopHandler(serverctl.RepositoryStopHandler):
        cmdname = 'stop'
        cfgname = 'all-in-one'
        subcommand = 'cubicweb-twisted'

except ImportError:
    pass