#!/usr/bin/env python
# Copyright(C) 2012 thomasv@gitorious

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public
# License along with this program.  If not, see
# <http://www.gnu.org/licenses/agpl.html>.

import time, json, socket, operator, thread, ast, sys, re, traceback
import ConfigParser
from json import dumps, loads
import urllib
import threading

config = ConfigParser.ConfigParser()
# set some defaults, which will be overwritten by the config file
config.add_section('server')
config.set('server','banner', 'Welcome to Electrum!')
config.set('server', 'host', 'localhost')
config.set('server', 'port', '50000')
config.set('server', 'password', '')
config.set('server', 'irc', 'yes')
config.set('server', 'ircname', 'Electrum server')
config.add_section('database')
config.set('database', 'type', 'psycopg2')
config.set('database', 'database', 'abe')

try:
    f = open('/etc/electrum.conf','r')
    config.readfp(f)
    f.close()
except:
    print "Could not read electrum.conf. I will use the default values."

try:
    f = open('/etc/electrum.banner','r')
    config.set('server','banner', f.read())
    f.close()
except:
    pass


password = config.get('server','password')


from processor import Shared, Processor, Dispatcher
from stratum_http import HttpServer
from stratum import TcpServer
from native import NativeServer
from irc import Irc
from abe_backend import AbeStore

class AbeProcessor(Processor):
    def process(self,request):
        message_id = request['id']
        method = request['method']
        params = request.get('params',[])
        #print request

        result = ''
        if method == 'numblocks.subscribe':
            result = store.block_number
        elif method == 'address.subscribe':
            address = params[0]
            store.watch_address(address)
            status = store.get_status(address)
            result = status
        elif method == 'client.version':
            #session.version = params[0]
            pass
        elif method == 'server.banner':
            result = config.get('server','banner').replace('\\n','\n')
        elif method == 'server.peers':
            result = irc.get_peers()
        elif method == 'address.get_history':
            address = params[0]
            result = store.get_history( address ) 
        elif method == 'transaction.broadcast':
            txo = store.send_tx(params[0])
            print "sent tx:", txo
            result = txo 
        else:
            print "unknown method", request

        if result!='':
            response = { 'id':message_id, 'method':method, 'params':params, 'result':result }
            self.push_response(response)

    def get_status(self,addr):
        return store.get_status(addr)






if __name__ == '__main__':

    if len(sys.argv)>1:
        import jsonrpclib
        server = jsonrpclib.Server('http://%s:8081'%config.get('server','host'))
        cmd = sys.argv[1]
        if cmd == 'stop':
            out = server.stop(password)
        else:
            out = "Unknown command: '%s'" % cmd
        print out
        sys.exit(0)

    processor = AbeProcessor()
    shared = Shared()
    # Bind shared to processor since constructor is user defined
    processor.shared = shared
    processor.start()

    irc = Irc(processor, config.get('server','host'), config.get('server','ircname'))
    if (config.get('server','irc') == 'yes' ): irc.start()

    # backend
    store = AbeStore(config)

    # dispatcher
    dispatcher = Dispatcher(shared, processor)
    dispatcher.start()

    host = config.get('server','host')
    # Create various transports we need
    transports = [ NativeServer(shared, store, irc, config.get('server','banner'), host, 50000),
                   TcpServer(shared, processor, host, 50001),
                   HttpServer(shared, processor, host, 8081),
                   ]
    for server in transports:
        server.start()

    print "starting Electrum server on", host
    store.run(processor)
    print "server stopped"