#
# dblatex - Hyphenation classes to provide smart hyphenation of path like
# strings
#
from __future__ import print_function

import re

class Hyphenator:
    def __init__(self, codec=None):
        pass
    
    def hyphenate(self, text):
        return text


class BasicHyphenator(Hyphenator):
    """
    Hyphenates basically by putting an hyphenation point between each character.
    """
    def __init__(self, codec=None):
        self.codec = codec
        self.hyphenchar = b"\-"

    def hyphenate(self, text):
        if self.codec: text = self.codec.decode(text)
        ntext = "\1".join(list(text))
        if self.codec: ntext = self.codec.encode(ntext)
        ntext = re.sub(b"\1? \1?", b" ", ntext)
        ntext = ntext.replace(b"\1", self.hyphenchar)
        return ntext


class UrlHyphenator(Hyphenator):
    """
    Hyphenates <text> so that cutting is easier on URL separators.
    The hyphen chars are expected to be void to prevent from spurious
    characters in displayed filenames or URLs.

    The pathname words can be cut only after the <h_start> first characters
    and before the <h_stop> characters to avoid a cut just after one or two
    chars.

    Tip: the inter-chars can be defined with macros \HO and \HL, to be shorter
    like:

    \def\HL{\penalty9999} (h_char="\HL")
    \def\HO{\penalty5000} (h_sep="\HO")

    By default these shortcuts are not used to avoid some macro declaration in
    existing latex styles.
    """
    def __init__(self, codec=None,
                 h_sep=b"\penalty0{}", h_char=b"\penalty5000{}",
                 h_start=3, h_stop=3):
        self.codec = codec
        self.seps = r":/\@=?#;-."
        self.h_sep = h_sep
        self.h_char = h_char
        self.h_start = (h_start-1)
        self.h_stop = (h_stop-1)

    def _translate(self, text):
        if self.codec:
            return self.codec.encode(text)
        else:
            return text

    def hyphenate(self, text):
        if self.codec: text = self.codec.decode(text)

        vtext = []
        p = "([%s])" % re.escape(self.seps)
        words = re.split(p, text)
        for w in words:
            if not(w):
                continue
            if w in self.seps:
                vtext.append(self._translate(w) + self.h_sep)
            else:
                hword = w[self.h_start:-self.h_stop]
                if len(hword) < 2:
                    vtext.append(self._translate(w))
                else:
                    nw = w[:self.h_start]
                    nw += "\1".join(list(hword))
                    nw += w[-self.h_stop:]
                    nw = self._translate(nw)
                    nw = re.sub(b"\1? \1?", b" ", nw)
                    nw = nw.replace(b"\1", self.h_char)
                    vtext.append(nw)

        ntext = b"".join(vtext)
        return ntext


if __name__ == "__main__":
    url = b"http://www.fg/foobar fun#fght/fkkkf.tz?id=123"
    h1 = BasicHyphenator()
    h2 = UrlHyphenator()
    print(h1.hyphenate(url))
    print(h2.hyphenate(url))
