# Install method for disk image installs (CD & NFS)

from comps import ComponentSet, HeaderList
import os
import rpm
import time
import urllib
import string
import struct
import socket

# we import these explicitly because urllib loads them dynamically, which stinks
import ftplib
import httplib
import StringIO

import todo

FILENAME = 1000000

class InstallMethod:

   def readComps(self, hdlist):
       return ComponentSet(self.baseUrl + '/RedHat/base/comps',
                           hdlist)

   def getFilename(self, h):
       root = "/mnt/sysimage"
       pathlist = [ "/var/tmp", "/tmp",
                    "/." ]
       for p in pathlist:
           if (os.access(root + p, os.X_OK)):
               tmppath = root + p
               break

       file = tmppath + h[FILENAME]

       connected = 0
       while not connected:
           try:
               urllib.urlretrieve(self.baseUrl + "/RedHat/RPMS/" + h[FILENAME],
                                  file)
           except IOError, (errnum, msg):
#                print "IOError occurred, trying again"
               time.sleep(5)
           else:
               connected = 1

       return file

   def unlinkFilename(self, fullName):
       os.remove(fullName)

   def readHeaders(self):

       connected = 0
       while not connected:
           try:
               url = urllib.urlopen(self.baseUrl + "/RedHat/base/hdlist")
           except IOError, (errnum, msg):
#                print "IOError occurred, trying again"
               time.sleep(5)
           else:
               connected = 1

       raw = url.read(16)
       hl = []
       while (raw):
           info = struct.unpack("iiii", raw)
           magic1 = socket.ntohl(info[0]) & 0xffffffff
           if (magic1 != 0x8eade801 or info[1]):
               raise TypeError, "bad magic in header"

           il = socket.ntohl(info[2])
           dl = socket.ntohl(info[3])
           totalSize = il * 16 + dl;
           hdrString = raw[8:] + url.read(totalSize)
           hdr = rpm.headerLoad(hdrString)
           hl.append(hdr)

           raw = url.read(16)

       return HeaderList(hl)

   def targetFstab(self, fstab):
       pass

   def filesDone(self):
       pass

   def __init__(self, url):
       i = string.index(url, '://') + 2
       self.baseUrl = url[0:i]
       rem = url[i:]
       new = string.replace(rem, "//", "/")
       while (new != rem):
           rem = new
           new = string.replace(rem, "//", "/")
       rem = new
       self.baseUrl = self.baseUrl + rem