# -*- coding: utf-8 -*-
"""
    sphinx.builders.webapp
    ~~~~~~~~~~~~~~~~~~~~~~

    A web application builder.

    :copyright: Copyright 2007-2009 by the Sphinx team, see AUTHORS.
    :license: BSD, see LICENSE for details.
"""

import os
import sys
import codecs
import shutil
import cPickle
from os import path
from hashlib import md5
from types import ListType, TupleType
import re


import jinja2 as j2

#from mercurial import commands, ui
#from mercurial.hg import repository

from sphinx.builders.html import StandaloneHTMLBuilder
from sphinx.errors import SphinxError
from sphinx.web.dbutils import PidDb
from sphinx.web.webconfig import WebConfig
from sphinx.writers.html import HTMLTranslator






class SimpleCommentHTMLBuilder(StandaloneHTMLBuilder):
    """ Static html pages with paragraph ids. 
    """
    def init(self):
        self.app.add_javascript('simplecomment.js')
        self.id_file_loc=os.path.join(self.outdir,'paragraph_id.py')
        self.id_file=open(self.id_file_loc,'w')
        self.beginning="p_list= "
        self.id_dictionary={}

        StandaloneHTMLBuilder.init(self)

    def get_target_uri(self, docname, typ=None):
        return docname + self.link_suffix

    def write_doc(self, docname, doctree):
        StandaloneHTMLBuilder.write_doc(self, docname, doctree)

    def prepare_writing(self, docnames):
        StandaloneHTMLBuilder.prepare_writing(self, docnames)

    def finish(self):
        StandaloneHTMLBuilder.finish(self)
        self.add_pids_to_paragraphs()
        self.id_file.write(self.beginning+str(self.id_dictionary))
       

    def add_pids_to_paragraphs(self):
        all_files=[]
        for path,dir,filenames in os.walk(self.outdir):
            for filename in filenames:
                all_files.append(os.path.join(path, filename))

            for element in all_files :
                if element.split('.')[1]=='html':
                    self.id_list=[]
                    self.taggen(element)
                    self.id_dictionary[self.chapter.split('/')[-1]]=self.id_list
                
                else :
                    pass

    def retag(self,s):
        self.biggest_id += 1
        id_name="%s_%x" % (self.chapter.split('/')[-1],self.biggest_id)   
        self.id_list.append(id_name) 
         
        return '<p id="%s">' %id_name




    def taggen(self,html_file_name):
        tagged = re.compile('<p id="x_([0-9a-f]+)"[^>]*>', re.M)
        self.biggest_id=0 
        self.chapter=html_file_name.split('.')[0]    
       
        try:    
            f = open(html_file_name).read()
            f1 = re.sub('<p>',self.retag, f )
            
           
            if f1 != f:
                tmpname = html_file_name+ '.tmp'
                fp = open(tmpname, 'w')
                fp.write(f1)
                fp.close()
                os.rename(tmpname,html_file_name)
        
        except IOError:
            pass