from HTMLComponent import *
from GUIComponent import *
import re

from MenuList import MenuList

from Tools.Directories import *

from enigma import *

RT_HALIGN_LEFT = 0
RT_HALIGN_RIGHT = 1
RT_HALIGN_CENTER = 2
RT_HALIGN_BLOCK = 4

RT_VALIGN_TOP = 0
RT_VALIGN_CENTER = 8
RT_VALIGN_BOTTOM = 16

EXTENSIONS = {
		"mp3": "music",
		"wav": "music",
		"jpg": "picture",
		"jpeg": "picture",
		"png": "picture",
		"ts": "movie",
		"avi": "movie",
		"mpg": "movie",
		"mpeg": "movie",
	}

def FileEntryComponent(name, absolute, isDir = False):
	res = [ (absolute, isDir) ]
	res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 200, 20, 0, RT_HALIGN_LEFT ,name))
	if isDir:
		png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/extensions/directory.png"))
	else:
		extension = name.split('.')
		extension = extension[-1]
		if EXTENSIONS.has_key(extension):
			png = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "/extensions/" + EXTENSIONS[extension] + ".png"))
	if png is not None:
		res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
	
	return res

class FileList(HTMLComponent, GUIComponent, MenuList):
	def __init__(self, directory, showDirectories = True, showFiles = True, matchingPattern = None):
		GUIComponent.__init__(self)
		self.l = eListboxPythonMultiContent()

		self.showDirectories = showDirectories
		self.showFiles = showFiles
		# example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
		self.matchingPattern = matchingPattern
		self.changeDir(directory)

		self.l.setFont(0, gFont("Regular", 18))
		
	def getSelection(self):
		return self.l.getCurrentSelection()[0]
	
	def changeDir(self, directory):
		self.list = []
		
		files = os.listdir(directory)
		files.sort()
		
		if directory != "/" and self.showDirectories:
			self.list.append(FileEntryComponent(name = "..", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
			
		directories = []
		for x in files:
			if os.path.isdir(directory + x):
				directories.append(x)
				files.remove(x)

		if self.showDirectories:
			for x in directories:
				self.list.append(FileEntryComponent(name = x, absolute = directory + x + "/" , isDir = True))

					
		if self.showFiles:
			for x in files:
				if self.matchingPattern is not None:
					if re.compile(self.matchingPattern).search(x):
						self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False))
				else:
					self.list.append(FileEntryComponent(name = x, absolute = directory + x , isDir = False))
				
		self.l.setList(self.list)
				
	def GUIcreate(self, parent):
		self.instance = eListbox(parent)
		self.instance.setContent(self.l)
		self.instance.setItemHeight(23)