# Written by Arno Bakker 
# see LICENSE.txt for license information

from Tribler.Core.Session import Session
from Tribler.Core.SessionConfig import *
from Tribler.Core.TorrentDef import TorrentDef
from Tribler.Core.DownloadConfig import *
from Tribler.Policies.RateManager import UserDefinedMaxAlwaysOtherwiseEquallyDividedRateManager
from Tribler.Policies.UploadLimitation import TestUploadLimitation
import time
from Tribler.Core.APIImplementation.UserCallbackHandler import Notifier
from Tribler.Core.Utilities.utilities import show_permid_short
from threading import currentThread

sscfg = SessionStartupConfig()
if sys.platform == 'win32':
    s = Session()
else:
    sscfg.set_state_dir('/tmp/statedir')
    sscfg.set_install_dir('.')
    sscfg.set_overlay(0)
    sscfg.set_listen_port(7001)
    s = Session(sscfg)
    


r = UserDefinedMaxAlwaysOtherwiseEquallyDividedRateManager()
uploadLimitation = TestUploadLimitation(s,r)
t = 0
count = 0

def testfunc(subj, change, obj_id, *rest):
            #if subj == NTFY_PEERS:
            obj_id = show_permid_short(obj_id) # also infohash :)
            print 'Observer: %s %s %s: %s' % (subj, change, obj_id, rest)
            
def states_callback(dslist):
    global s
    global r
    global t
    global count
    
    adjustspeeds = False
#    r.set_global_max_speed(DOWNLOAD,10000)
#    if count > 10:
#        r.set_global_max_speed(DOWNLOAD,15)
#    if count > 20:
#        count = 0
#    count += 1
#    
    for ds in dslist:
        d = ds.get_download()
        print >>sys.stderr,"main: Stats",`d.get_def().get_name()`,dlstatus_strings[ds.get_status()],ds.get_progress(),"%",ds.get_error(),"up",ds.get_current_speed(UPLOAD),"down",ds.get_current_speed(DOWNLOAD),ds.get_num_seeds_peers(), currentThread().getName()
        
        #complete = ds.get_pieces_complete()
        #print >>sys.stderr,"main: Pieces completed",`d.get_def().get_name()`,"len",len(complete)
        #print >>sys.stderr,"main: Pieces completed",`d.get_def().get_name()`,complete[:60]
        
        """
        if ds.get_status() == DLSTATUS_SEEDING:
            print >>sys.stderr,"main: Syncing download because complete"
            d.checkpoint()
        """
        
        
        
    if adjustspeeds:
        r.add_downloadstatelist(dslist)
        r.adjust_speeds()
        
    #time.sleep(10)
    return (20.0,True)



def state_callback(ds):
    d = ds.get_download()
    print >>sys.stderr,"main: SingleStats",`d.get_def().get_name()`,dlstatus_strings[ds.get_status()],ds.get_progress(),"%",ds.get_error(),"up",ds.get_current_speed(UPLOAD),"down",ds.get_current_speed(DOWNLOAD),currentThread().getName()
    return (1.0,False)


def vod_ready_callback(mimetype,stream):
    print >>sys.stderr,"main: VOD ready callback called",currentThread().getName(),"###########################################################",mimetype


if __name__ == "__main__":
    
    s.set_download_states_callback(states_callback,getpeerlist=True)
    
    # For testing only! 

    #s.add_observer(testfunc, NTFY_PEERS)
    #s.add_observer(testfunc, NTFY_TORRENTS)

    #s.remove_observer(testfunc)
     
    # Torrent 1
    if sys.platform == 'win32':
        tdef = TorrentDef.load('bla.torrent')
    else:
        #tdef = TorrentDef.load('/tmp/bla3multi.torrent')
        tdef = TorrentDef.load('/tmp/bla.torrent')
        
    dcfg = DownloadStartupConfig()
    dcfg.set_max_rate_period(4.0)
    #dcfg.set_dest_dir('/arno/tmp/scandir')
    """
    dcfg.set_video_start_callback(vod_ready_callback)
    #dcfg.set_selected_files('star-wreck-in-the-pirkinning.txt') # play this video
    dcfg.set_selected_files('star_wreck_in_the_pirkinning_subtitled_xvid.avi') # play this video
    """
    d = s.start_download(tdef,dcfg)
    
    # Torrent 2
    """
    if sys.platform == 'win32':
        tdef = TorrentDef.load('bla2.torrent')
    else:
        tdef = TorrentDef.load('/tmp/bla2.torrent')
    d2 = s.start_download(tdef)
    d2.set_state_callback(state_callback)
    """

    
    #s.shutdown()
    #s.remove_download(d,removecontent=True)

    time.sleep(2500) # TODO: make sure we don't quit before shutdown checkpoint complete
    print 'end'