#!/usr/bin/env python
# -*- coding: utf-8 -*-
######################################################
##
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published
#  by the Free Software Foundation; version 2 only.
#
#  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 General Public License for more details.
##
######################################################
## 
#  Project: AptOnCd
#  File: RepDownload.py
#  Author: Laudeci Oliveira <laudeci@gmail.com>
#  Creation: 07/12/2006
#  Changed: 
#  Purpose: Class that will control repositoy downloads
##
######################################################
import CreateAptOncd
import utils
import os
import xmlfile
import parsegz
import config
import ProgressWindow
import config
from messageBox import MessageBox
import time
import gtk
import gtk.glade
import pygtk
import DownloadGUI
import download
import msg
import gzip
import MediaSplitter

class RepDownload:
    def __init__(self,gladeFileName):
        self.gladeFile = gladeFileName
        self.totaltime = 0
        self.totalsize = 0
        self.speed = 0
        self.totalread = 0
        self.filecount = 0
        self.downloadedFiles = 0
        self.treeline = -1
        self.DownloadResult = gtk.RESPONSE_NONE
    
    def load_conf(self, file):
        XMLFile = xmlfile.XMLFile()
        self.node_text = XMLFile.parse(file)
        return XMLFile.load_conf(file)
     
    def CalcSpeed(self, bytesRead):
        if time.time() != self.initialtime:
            TimeFraction = ((self.totalsize - self.totalread )* (time.time() - self.initialtime)) / self.totalread
            speed = self.totalread / (time.time() - self.initialtime)
            
            self.speed = speed
            self.totaltime = TimeFraction 
            self.labelestimatedtime.set_text(str(utils.TimeFormat(self.totaltime)) )
            self.labeldownloadrate.set_text(utils.fileSizeFormat(speed))
            
            fraction = (float(self.totalread) / self.totalsize)
            self.progressbar.set_fraction(fraction)  
            
            self.progressbar.set_text("%d%%" %(100 * float(self.totalread) / self.totalsize) )
            
            
    def scrollToPath(self, index):
        try :
            self.treeprogress.scroll_to_cell(index)
        except:
            pass

    def pCancelEvent(self, widgets = None):
        self.DownloadResult = gtk.RESPONSE_CANCEL
        self.Packages.cancelDownload()
    
    def show_progress(self):
        self.progressgui = gtk.glade.XML(self.gladeFile, "frmFetchRepo")
        
        self.frmFetchRepo = self.progressgui.get_widget("frmFetchRepo")
        self.frmFetchRepo.set_position(gtk.WIN_POS_CENTER)
        self.frmFetchRepo.set_modal(True)
        
        self.frmFetchRepo.connect("delete-event", self.close_fetchwindow)
        
        self.labeltotalsize = self.progressgui.get_widget("lblTotalSize")
        self.labeltotalsize.set_text(utils.fileSizeFormat(self.totalsize))
            
        self.labelnumfiles = self.progressgui.get_widget("lblNumberFiles")
        self.labelnumfiles.set_text( str(self.downloadedFiles) + '/' + str(self.filecount) )
        
        self.labelestimatedtime = self.progressgui.get_widget("lblEstimatedTime")
        self.labelestimatedtime.set_text(str("%0.2f" % (self.totaltime))) 
        
        self.labeldownloadrate = self.progressgui.get_widget("lblDownloadRate")
        self.labeldownloadrate.set_text(utils.fileSizeFormat(self.speed))
        
        self.labeldist = self.progressgui.get_widget("lblDistribuition")
        self.labeldist.set_text( self.conf[3])
        
        self.labelversion = self.progressgui.get_widget("lblVersion")
        self.labelversion.set_text(self.conf[4])

        self.labelsection = self.progressgui.get_widget("lblSection")
        self.labelsection.set_text(self.conf[5])

        self.labelarch = self.progressgui.get_widget("lblArchitecture")
        self.labelarch.set_text(self.conf[6])

        self.labelhost = self.progressgui.get_widget("lblMirror")
        self.labelhost.set_text(self.conf[2])
        
        self.button_cancel = self.progressgui.get_widget("btnCancel")
        self.button_cancel.connect("clicked", self.close_fetchwindow)
        
        self.progressbar = self.progressgui.get_widget("pbDownloadTotal")
        
        self.treeprogress = self.progressgui.get_widget("tvDownload")
        bar = gtk.CellRendererProgress()
        
        self.mgrade = gtk.ListStore(str, str, int)
        
        self.treeprogress.set_model(self.mgrade)
        column1 = gtk.TreeViewColumn(msg.MESSAGE_0060, gtk.CellRendererText(), text = 0)
        column2 = gtk.TreeViewColumn(msg.MESSAGE_0061, gtk.CellRendererText(), text = 1)
        column3 = gtk.TreeViewColumn(msg.MESSAGE_0062, bar, value = 2)
        column1.set_resizable(True)
        column2.set_resizable(True)
        column3.set_resizable(True)
        column1.set_sort_column_id(0)
        column2.set_sort_column_id(1)
        column3.set_sort_column_id(2)
        self.treeprogress.append_column(column1)
        self.treeprogress.append_column(column2)
        self.treeprogress.append_column(column3)
        
        self.frmFetchRepo.show()

    def close_fetchwindow(self, widget, data = None):
        self.DownloadResult = gtk.RESPONSE_CANCEL
        self.frmFetchRepo.destroy()       
         
    def SetOptions(self):
        self.DownloadGUI = DownloadGUI.DownloadGUI(self.gladeFile)
        result = self.DownloadGUI.run()
        column3.set_resizable(True)
        column1.set_sort_column_id(0)
        column2.set_sort_column_id(1)
        column3.set_sort_column_id(2)
        self.treeprogress.append_column(column1)
        self.treeprogress.append_column(column2)
        self.treeprogress.append_column(column3)

    def close_fetchwindow(self, widget, data = None):
        self.DownloadResult = gtk.RESPONSE_CANCEL
        self.frmFetchRepo.destroy()       
    
    def on_downloadStatus(self,widget, filedown, size, bytesRead, percent):
        if "Packages.gz" not in filedown:
            if self.DownloadResult == gtk.RESPONSE_CANCEL:
                widget.cancelDownload()
            self.totalread+= bytesRead
            self.mgrade[self.treeline][2] = percent
            self.CalcSpeed(bytesRead)
        else:
            self.totalread+= bytesRead
            fraction = (float(self.totalread) / self.totalsize)
            #fraction =  (float(percent)/100)
            self.wProgress.updateFraction(fraction)
            time.sleep(0.2)
        utils.updateUI()
        return
    
    def Download_TarGZ(self, PackFiles):
        self.initialtime = time.time()
        self.totalread = 0

        self.wProgress.connectEvent(self.pCancelEvent)
        
        downloaded = []
        downresult = False
        for files in PackFiles:
            try:
                self.Packages = download.Download(files[2], files[1])
                self.wProgress.set_task(msg.MESSAGE_0063 % self.conf[5])
                #self.wProgress.TaskLenght(Packages.getDownloadSize())
                self.Packages.connect('download_status',self.on_downloadStatus)
                downresult = self.Packages.download(True)
                if self.DownloadResult == gtk.RESPONSE_CANCEL:
                   self. Packages.cancelDownload()
                   downloaded = []