# -*- coding: utf-8 -*-

"""

	fstab.py - Abstraction class for fstab file.

	© Jaime Soriano Pastor <kronoss@kronoss.org>
	
"""

import re
import os

def uuidToSpec(uuid):
	ls = os.popen("readlink -f /dev/disk/by-uuid/" + uuid)
	return ls.readline().rsplit()[0]

class Filesystem:
	spec = ""
	file = ""
	vfstype = ""
	mntops = []
	freq = ""
	passno = ""
	uuid = None

	def __init__(self, spec, file, vfstype, mntops, freq, passno):
		self.file = file
		self.vfstype = vfstype
		self.mntops = mntops
		self.freq = freq
		self.passno = passno

		exp = "^UUID=(\S+)$"
		fields = re.compile(exp).match(spec)
		if fields == None:
			self.uuid = None
			self.spec = spec
		else:
			self.uuid = fields.groups()[0]
			self.spec = uuidToSpec(self.uuid)

	def toTab(self):
		if len(self.mntops) > 0:
			ops = self.mntops[0]

			for op in self.mntops[1:]:
				ops = ops + "," + op
		else:
			ops = "defaults"
			self.mntops = "defaults"

		ret = ""
		if self.uuid == None:
			ret = self.spec
		else:
			ret = "UUID=" + self.uuid
		ret += "\t" + self.file + "\t" + self.vfstype + "\t" + ops + "\t" + self.freq + "\t" + self.passno
		return ret

	def mount(self):
		ops = ""
		if len(self.mntops) > 0:
			ops = self.mntops[0]

			for op in self.mntops[1:]:
				ops = ops + "," + op
		return os.system("mount " + self.spec + " " + self.file + " -t " + self.vfstype + " -o " + ops)

	def umount(self):
		return os.system("umount " + self.spec)

	def is_mounted(self):
		mtab = file("/etc/mtab", "r")
		lines = mtab.readlines()
		for line in lines:
			if line.find(self.spec) >= 0:
				return True
		return False

	def __repr__(self):
		return self.toTab()



class Fstab:
	"Abstraction class for fstab file"

	def __init__(self, fstab):
		self.filesystems = []
		self.fromFile(fstab)


	def addFS(self, fs):
		self.filesystems.append(fs)


	def getFS(self, spec, file):
		for fs in self.filesystems:
			if type(fs) == str: continue
			if (fs.spec==spec and (fs.file==file or file==None)) or (fs.file==file and spec==None):
				return fs
		return None

	def delFS(self, spec, file):
		i=0
		for fs in self.filesystems:
			if type(fs) != str:
				if fs.spec==spec and fs.file==file:
					deleted = self.filesystems[i]
					del self.filesystems[i]
					return deleted
			i = i + 1
		return None

	def parseFS(self, fs):
		exp = "^(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\d)\s+(\d)\s+$"
		fields = re.compile(exp).match(fs)
		if fields==None:
			return None
		else:
			fields = fields.groups()
			return Filesystem(fields[0], fields[1], fields[2], re.split(",",fields[3]), fields[4], fields[5]);


	def fromFile(self, fstab):
		self.lines = []
		fstabfile = file(fstab, "r")
		n = 0

		for line in fstabfile.readlines():
			parsed = self.parseFS(line)
			if parsed != None:
				self.addFS(parsed)
			else:
				self.addFS(line)

		fstabfile.close()


	def toTab(self):
		ret = ""
		col_l = []
		for fs in self.filesystems:
			if type(fs) == str: continue
			cols = re.compile("\t").split(fs.toTab())
			if len(col_l)==0: col_l = [0]*len(cols)
			for i in range(len(cols)):
				if col_l[i] < len(cols[i]): col_l[i] = len(cols[i])
		for fs in self.filesystems:
			if type(fs) == str:
				ret = ret + fs
			else:
				cols = re.compile("\t").split(fs.toTab())
				for i in range(len(cols)):
					ret = ret + cols[i].ljust(col_l[i] + 2)
				ret = ret + "\n"
		return ret

	def	toFile(self, fstab):
		fstabfile = file(fstab, "w")
		fstabfile.write(self.toTab())
		fstabfile.close()


	def __repr__(self):
		return self.toTab()