import io
import string

import hserver.utils

class MIMEHeader:

	def __init__(self,line):
		self.__line = line
		self.__parameters ={}
		
		self.__parseline()
	
	def name(self):
		return self.__name
		
	def rawvalue(self):
		return self.__rawvalue
		
	def value(self):
		return self.__value
		
	def parameters(self):
		return self.__parameters
	
	
	def __parseparameters(self,parameters):
		p = 0
		while (p < len(parameters)):
			pe = parameters.find("=",p)
			if (pe < 0):
				return
				
			name = parameters[p:pe].strip()
			
			pe = pe + 1
			
			if (parameters[pe] == '"'):
				pe = pe + 1
				pq = parameters.find('"',pe)
				pr = parameters.find(';',pq)
				value = parameters[ pe : pq]
			else:
				pq = parameters.find(";",pe)
				if (pq < 0):
					pq = len(parameters)
				pr = pq
				value = parameters[ pe : pq].strip()
			
			self.__parameters[name] = value
			if (pr < 0):
				return
			p = pr + 1
	
	def __parseline(self):
		i = self.__line.find(":")
		if (i > 0):
			self.__name = self.__line[:i].upper()
			self.__rawvalue = self.__line[ i+1 : ].strip()
			
			i = self.__rawvalue.find(";")
			if (i>0):
				self.__parseparameters(self.__rawvalue[ i+1 : ])
				self.__value = self.__rawvalue[ :i ]
			else:
				self.__value = self.__rawvalue
			
			
#			print("MIMEHeader: %s: %s %s" % (self.__name,self.__value,self.__parameters))
			


class MIMEHeaders:

	def __init__(self, fp=None, src=None):
		self.__items = {}
		self.__parameters = {}

		if not fp is None:
			self.__read(fp)
			
	def get(self,name):
		name = name.upper()
		if not name in self.__items.keys():
			return None
		
		return self.__items[name]
			
	def __getitem__(self,name):
		name = name.upper()
		if name in self.__items.keys():
			return self.__items[name].rawvalue()
		raise KeyError()
			
	def value(self,name,default=None):
		h = self.get(name)
		if h is None:
			return default
		return h.value()

	def rawvalue(self,name,default=None):
		h = self.get(name)
		if h is None:
			return default
		return h.rawvalue()
		
	def parameters(self,name):
		h = self.get(name)
		if h is None:
			return {}
		return h.parameters()
	
			
	def __read(self,fp):
		cont = True
		lines = []
		
		while (cont):
			l = str(fp.readline(),"iso-8859-1").strip()
			if (l.strip() == ""):
				cont = False
			else:
				if (l[0] in string.whitespace ):
					lines[-1] = lines[-1] + " " + l.lstrip()
				else:
					lines.append(l)
					
		for line in lines:
			h = MIMEHeader(line)
			self.__items[ h.name() ] = h

		if self.get("CONTENT-TYPE") is None:
			self.__items[ "CONTENT-TYPE" ] = MIMEHeader("Content-Type: text/plain")