conditional in formatter

1 files changed, 37 insertions, 1 deletions

diff --git a/misc.py b/misc.py
index 6ead246..4195851 100644
--- a/misc.py
+++ b/misc.py
@@ -1,4 +1,5 @@
 from PyQt4 import QtCore, QtGui
+import re
 
 appIcon=QtGui.QIcon('gfx/icon.png')
 
@@ -23,7 +24,6 @@ def format(string, song=None, xtra_tags={}):
 	"""Replace all tags in $str with their respective value."""
 	# what tags are available?
 	if song:
-		print song
 		tags=song._data
 	else:
 		tags={}
@@ -31,6 +31,42 @@ def format(string, song=None, xtra_tags={}):
 		tags[tag]=xtra_tags[tag]
 	
 	ret=string
+	# first perform some functions: $name(value)
+	func=re.compile('\$[a-z]{2}\(', re.MULTILINE|re.IGNORECASE)
+	start=0
+	loops=20	# my stupidity is endless, so we better make sure to never always loop!
+	while True and loops>0:
+		loops-=1
+		match=func.search(ret[start:])
+		if not match:
+			break
+		# we've found a match!
+		# look for matching parenthesis!
+		start+=match.start()	# we have to add, because we start search from $start!
+		end=ret.find('(', start)+1
+		balance=1
+		while balance>0 and end<len(ret):
+			if ret[end]==')':balance-=1
+			if ret[end]=='(':balance+=1
+			end+=1	# can do better
+		# whole function is in ret[start:end]
+		# find function-name
+		name=ret[start+1:ret.find('(', start)]
+		result=None	# result of this function
+		if name=='if':
+			# $if(param,if-true)
+			comma=ret.find(',',start)
+			param1=ret[start+len('$if('):comma]
+			param2=ret[comma+1:end-1]
+			result=''
+			if param1[1:] in tags:
+				result=param2
+		else:
+			start+=1
+		if result!=None:
+			ret=("%s%s%s")%(ret[0:start], result, ret[end:])
+	
+	# perform all replacements!
 	for tag in tags:
 		ret=ret.replace('$%s'%(tag), str(tags[tag]))
 	return ret