###############################################################################
# #
# #  Copyright (C) 2011-2014 Tavendo GmbH
# #
# #  Licensed under the Apache License, Version 2.0 (the "License");
# #  you may not use this file except in compliance with the License.
# #  You may obtain a copy of the License at
# #
# #      http://www.apache.org/licenses/LICENSE-2.0
# #
# #  Unless required by applicable law or agreed to in writing, software
# #  distributed under the License is distributed on an "AS IS" BASIS,
# #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# #  See the License for the specific language governing permissions and
# #  limitations under the License.
# #
###############################################################################

import sys
import csv
import time
    
from twisted.python import log
from twisted.internet import reactor
import argparse


from autobahn.twisted.websocket import WebSocketClientProtocol, WebSocketClientFactory

class MyClientProtocol(WebSocketClientProtocol):

    #def __init__(self, rows):

    def onConnect(self, response):
        print("Server connected: {0}".format(response.peer))
        reactor.callLater(0, self.send_messages)

    def onOpen(self):
        print("WebSocket connection open.")

    def send_messages(self):
        with open("annotsroll_sample_formated.txt", 'rU') as datafile:
            for row in list(datafile):
                self.sendMessage(row, isBinary = False)
                time.sleep(0.1)
            
        #time.sleep(0.1)
        #self.sendMessage('{"user":"Julien"}'.encode('utf8'))
        print("Goodbye")
        self.factory.reactor.callLater(0.1, reactor.stop)

    def onClose(self, wasClean, code, reason):
        print("WebSocket connection closed: {0}".format(reason))



if __name__ == '__main__':
    
    log.startLogging(sys.stdout)
    
    parser = argparse.ArgumentParser(description='Simulate an (osc) pianoroll client.')
    #parser.add_argument('datafile', metavar='DATAFILE', help='The file containing the pianoroll data (CSV).')
    parser.add_argument('-e', '--event', dest='event', metavar='EVENT', required=True, help='the event code.')
    
    args = parser.parse_args()
    
    #with open(args.datafile, 'rU') as datafile:
    #    reader = csv.reader(datafile, delimiter=' ')
    
    factory = WebSocketClientFactory("ws://127.0.0.1:8090/annot?event=%s" % args.event, debug=True)
    factory.protocol = MyClientProtocol
    
    reactor.connectTCP("127.0.0.1", 8090, factory)
    reactor.run()