# This program is free software. It comes without any warranty, to
# the extent permitted by applicable law. You can redistribute it
# and/or modify it under the terms of the Do What The Fuck You Want
# To Public License, Version 2, as published by Sam Hocevar. See
# http://sam.zoy.org/wtfpl/COPYING for more details.

try:
    
    import locale
    
    try: locale.setlocale( locale.LC_ALL, '' )
    except: pass
    
    from include import HydrusConstants as HC
    
    import os
    import sys
    import time
    
    from include import ServerController
    import threading
    from twisted.internet import reactor
    from include import HydrusExceptions
    from include import HydrusGlobals
    import traceback
    
    HydrusGlobals.instance = HC.HYDRUS_SERVER
    
    action = ServerController.GetStartingAction()
    
    if action is None or action == 'help':
    
        print( 'This is the hydrus server. It accepts these commands:' )
        print( '' )
        print( 'server start - runs the server' )
        print( 'server stop - stops an already running server' )
        print( 'server restart - stops an already running server, then runs itself' )
        print( '' )
        print( 'You can also run \'server\' without arguments. Depending on what is going on, it will try to start or it will ask you if you want to stop or restart.' )
        print( 'You can also stop the server just by hitting your keyboard interrupt (usually Ctrl+C).')
        print( '' )
        print( 'PROTIP: stop and restart don\'t work yet lol')
        
    else:
        
        if action in ( 'start', 'restart' ):
            
            print( 'Running...' )
            
        
        error_occured = False
        
        initial_sys_stdout = sys.stdout
        initial_sys_stderr = sys.stderr
        
        with open( HC.LOGS_DIR + os.path.sep + 'server.log', 'a' ) as f:
            
            sys.stdout = f
            sys.stderr = f
            
            try:
                
                print( 'hydrus server started at ' + time.ctime() )
                
                threading.Thread( target = reactor.run, kwargs = { 'installSignalHandlers' : 0 } ).start()
                
                controller = ServerController.Controller()
                
                controller.Run( action )
                
            except HydrusExceptions.PermissionException as e:
                
                error_occured = True
                error = str( e )
                
                print( error )
                
            except:
                
                error_occured = True
                error = traceback.format_exc()
                
                print( 'hydrus server failed at ' + time.ctime() )
                
                print( traceback.format_exc() )
                
            finally:
                
                HydrusGlobals.view_shutdown = True
                HydrusGlobals.model_shutdown = True
                
                try: controller.pubimmediate( 'wake_daemons' )
                except: pass
                
                reactor.callFromThread( reactor.stop )
                
                print( 'hydrus server shut down at ' + time.ctime() )
                
            
        
        sys.stdout = initial_sys_stdout
        sys.stderr = initial_sys_stderr
        
        if error_occured:
            
            print( error )
            
        
        if action in ( 'start', 'restart' ):
            
            print( 'Finished.' )
            
        
    
except HydrusExceptions.PermissionException as e:
    
    print( e )
    
except:
    
    import traceback
    
    print( 'Critical error occured! Details written to crash.log!' )
    
    with open( 'crash.log', 'wb' ) as f: f.write( traceback.format_exc() )