#!/usr/bin/env python
#
# Copyright (c) 2013 Martin Abente Lahaye. - tch@sugarlabs.org

#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.

import pycurl
import urllib

from gi.repository import GObject

from twr_account import TwrAccount


class TwrObject(GObject.GObject):

    __gsignals__ = {
        'transfer-completed': (GObject.SignalFlags.RUN_FIRST, None, ([str])),
        'transfer-progress': (GObject.SignalFlags.RUN_FIRST, None, \
                             ([float, float, str])),
        'transfer-failed': (GObject.SignalFlags.RUN_FIRST, None, ([str])),
        'transfer-started': (GObject.SignalFlags.RUN_FIRST, None, ([]))}

    def _gen_header(self, method, url, params=[]):
        authorization = TwrAccount.authorization_header(method, url, params)
        headers = ['Host: api.twitter.com',
                   'Authorization: %s' % authorization]

        return headers

    def _update_cb(self, down_total, down_done, up_total, up_done, states):

        if 2 in states:
            return

        total = up_total
        done = up_done
        mode = 'upload'

        if 1 in states:
            total = down_total
            done = down_done
            mode = 'download'

        if total == 0:
            return

        if 0 not in states:
            self.emit('transfer-started')
            states.append(0)

        self.emit('transfer-progress', total, done, mode)

        state = states[-1]
        if total == done and state in states and len(states) == state + 1:
            states.append(state + 1)

    def _write_cb(self, buffer):
        self.emit('transfer-completed', buffer)

    def request(self, method, url, params, filepath=None):
        c = pycurl.Curl()

        if method == 'POST':
            c.setopt(c.POST, 1)
            c.setopt(c.HTTPHEADER, self._gen_header(method, url))

            if filepath is not None:
                params += [("media", (c.FORM_FILE, filepath))]

            if params is not None:
                c.setopt(c.HTTPPOST, params)
            else:
                c.setopt(c.POSTFIELDS, '')
        else:
            c.setopt(c.HTTPGET, 1)
            c.setopt(c.HTTPHEADER, self._gen_header(method, url, params))
            url += '?%s' % urllib.urlencode(params)

        # XXX hack to trace transfer states
        states = []

        def pre_update_cb(*args):
            args = list(args) + [states]
            self._update_cb(*args)

        c.setopt(c.URL, url)
        c.setopt(c.NOPROGRESS, 0)
        c.setopt(c.PROGRESSFUNCTION, pre_update_cb)
        c.setopt(c.WRITEFUNCTION, self._write_cb)
        #c.setopt(c.VERBOSE, True)

        try:
            c.perform()
        except pycurl.error, e:
            self.emit('transfer-failed', str(e))
        else:
            code = c.getinfo(c.HTTP_CODE)
            if code != 200:
                self.emit('transfer-failed', 'HTTP code %s' % code)
        finally:
            c.close()

@@ -2534,90 +2534,8 @@
 1)%0A%0A
-    def _write_cb(self, buffer):%0A        self.emit('transfer-completed', buffer)%0A%0A
     
@@ -3334,24 +3334,154 @@
 _cb(*args)%0A%0A
+        #XXX hack to write multiple responses%0A        buffer = %5B%5D%0A%0A        def __write_cb(data):%0A            buffer.append(data)%0A%0A
         c.se
@@ -3617,21 +3617,17 @@
 NCTION, 
-self.
+_
 _write_c
@@ -3952,16 +3952,77 @@
 inally:%0A
+            self.emit('transfer-completed', ''.join(buffer))%0A
         

Fix buffer write