Handle links inside link texts

1 files changed, 20 insertions, 7 deletions

diff --git a/wikimarkup.py b/wikimarkup.py
index 46aa839..429a02a 100644
--- a/wikimarkup.py
+++ b/wikimarkup.py
@@ -26,8 +26,8 @@ eltre = re.compile("(\\[\\[)|(\\{\\{)|\\[|(\\'\\'\\'?)")
 delims = { "[[" : re.compile("\\||(\\]\\])"),
            "{{" : re.compile("\\||(\\}\\})") }
 term   = { "[[" : "]]" , "{{" : "}}" }                             
-ends   = { "[[" : re.compile("\\]\\]"),
-           "{{" : re.compile("\\}\\}") }                  
+ends   = { "[[" : re.compile("(\\[\\[)|(\\]\\])"),
+           "{{" : re.compile("(\\{\\{)|(\\}\\})") }                  
 itend  = re.compile("\\'\\'($|[^\\'])")
 boend  = re.compile("\\'\\'\\'($|[^\\'])")
 
@@ -115,6 +115,21 @@ It handles:
             else:
                 pos = d.start(0) + 1
 
+    def linkend(self, paren, line, pos):
+        r = ends[paren]
+        count = 1
+        while count > 0:
+            m = r.search(line, pos);
+            if not m:
+                return len(line), len(line)
+            else:
+                pos = m.end(0)
+                if m.group(0) == paren:
+                    count += 1
+                else:
+                    count -= 1
+        return m.start(0), m.end(0)
+                                            
     la = None
     def putback(self, line):
         self.la = line
@@ -172,8 +187,7 @@ It handles:
                     self.putback(line)
                     break
                 yield(self.getkn(line))
-                
-                                            
+
     def getkn(self, line):
         pos = 0
         while 1:
@@ -189,10 +203,9 @@ It handles:
                 if m.group(0) == "[[" or m.group(0) == "{{":
                     d = delims[m.group(0)].search(line, pos)
                     if d.group(0) == "|":
-                        e = ends[m.group(0)].search(line, d.end(0))
                         target = (self.TEXT, line[pos:d.start(0)])
-                        text = (self.SEQ, self.getkn(line[d.end(0):e.start(0)]))
-                        pos = e.end(0)
+                        (start,pos) = self.linkend(m.group(0), line, m.end(0))
+                        text = (self.SEQ, self.getkn(line[d.end(0):start]))
                     elif d.group(0) == term[m.group(0)]:
                         target = (self.TEXT, line[pos:d.start(0)])
                         text = (self.NIL,)