# -* coding: utf8 *-

import Invoice
import re

# our page size and margins
from metrics import *
# our custom page style
from custom_elements import basicPage, firstPage, address_header

# reportlab imports
from reportlab.lib.units import cm
from reportlab.pdfgen import canvas as Canvas


def _formatPrice(price, symbol='€'):
  '''_formatPrice(price, symbol='€'):
  Gets a floating point value and returns a formatted price, suffixed by 'symbol'. '''
  s = ("%.2f" % price).replace('.', ',')
  pat = re.compile(r'([0-9])([0-9]{3}[.,])')
  while pat.search(s):
    s = pat.sub(r'\1.\2', s)
  return s+' '+symbol

def _niceCount(value):
  '''_niceCount(value):
  Returns a tuple (integer , decimals) where decimals can be None'''
  if type(value) == int:
    return ('%i' % value, None)
  if round(value, 2) == int(value):
    return ('%i' % int(value), None)
  s = '%.2f' % value
  (integer, decimals) = s.split('.', 1)
  if decimals[-1] == '0':
    decimals = decimals[:-1]
  return (integer, decimals)


def _splitToWidth(canvas, text, width, font, size):
  '''_splitToWidth(canvas, text, width, font, size)
  Split a string to several lines of a given width.'''
  lines = []
  paras = text.split('\n')
  for para in paras:
    words = para.split(' ')
    while len(words) > 0:
      mywords = [words[0], ]
      del words[0]
      while len(words) > 0 and canvas.stringWidth(' '.join(mywords) + ' ' + words[0], font, size) <= width:
        mywords.append(words[0])
        del words[0]
      lines.append(' '.join(mywords))
  return lines


def _drawJustifiedString(x, y, text, canvas, width, font, size):
  text = text.strip()
  if canvas.stringWidth(text, font, size) > width:
    canvas.drawString(x, y, text)
    # too long line, I cannot handle this
    return
  if not ' ' in text:
    canvas.drawString(x, y, text)
    # no space in there, nothing to justify
    return
  
  words = [ '%s' % w for w in text.split(' ')]
  words_width = 0.0
  for word in words:
    words_width += canvas.stringWidth(word, font, size)
  
  available_space = width - words_width
  available_each = available_space / float((len(words) - 1))
 
  my_x = x
  for idx in range(len(words)):
    word = words[idx]
    canvas.drawString(my_x, y, word)
    my_x += canvas.stringWidth(word, font, size) + available_each
  return


def _PageWrap(canvas):
  '''Seitenumbruch'''
  canvas.showPage()
  basicPage(canvas)


def address(canvas, lines):
  x = 2.0 * cm
  y = page_height - 5.0*cm
  canvas.setFont(font, 8)
  canvas.drawString(x+0.5*cm, y+0.1*cm, 'schokokeks.org · Köchersberg 25 · 71540 Murrhardt')
  canvas.setLineWidth(1)
  canvas.line(x+0.4*cm, y, x + address_width, y)
  y = y - 0.2*cm

  line_height = 11 + 0.1*cm
  y -= line_height
  canvas.setFont(font, 11)
  for line in lines:
    fontsize = 11
    if canvas.stringWidth(line, font, fontsize) > address_width:
      # Wenn es in zwei Zeilen passt, dann ist alles okay, ansonsten verkleinern
      if canvas.stringWidth(line, font, fontsize) > 2*address_width:
        for fontsize in [10.5, 10, 9.5, 9, 8.5]:
          if canvas.stringWidth(line, font, fontsize) <= 2*address_width:
            break
      mylines = _splitToWidth(canvas, line, address_width, font, fontsize)
      for l in mylines:
        canvas.drawString(x+0.5*cm, y, l)
        y -= line_height
    else:
      canvas.drawString(x+0.5*cm, y, line)
      y -= line_height




def InvoiceToPDF(iv):
  from StringIO import StringIO
  fd = StringIO()
  canvas = Canvas.Canvas(fd, pagesize=A4)
  
  if iv.tender:
    canvas.setTitle("Angebot von schokokeks.org")
  else:
    canvas.setTitle("Rechnung von schokokeks.org")
  
  canvas.setFont(font, 12)

  num_pages = 1

  # Waehrungssysmbol
  symbol = '€'
  y = topcontent
  font_size = default_font_size
  font_height = 0.35*cm
  line_padding = 0.1*cm
  line_height = font_height+0.1*cm

  def _partHeight(part):
    height = 0
    if type(part) == Invoice.Text:
      left, right = leftcontent, rightcontent
      if part.urgent:
        left += 1.5*cm
        right -= 1.5*cm
        height += len(part.paragraphs) * 3 * line_padding
        # Rechne eine Zeile mehr für den Rahmen
        height += line_height
      if part.headline:
        height += (len(_splitToWidth(canvas, part.headline, right-left, font+'-Bold', default_font_size+1)) * line_height) + line_padding
      for para in part.paragraphs:  
        height += (len(_splitToWidth(canvas, para, right-left, font, default_font_size)) * line_height) + line_padding
    elif type(part) == Invoice.Table:
      # Eine Zeile plus 2 mal line_padding für Tabellenkopf
      height = line_height + 2 * line_padding
      for el in part.entries:
        # Die Abstände oben und unten
        height += 2 * line_padding
        # Die Breite ist konservativ
        height += line_height*len(_splitToWidth(canvas, el['subject'], 9.3*cm, font, font_size))
        if 'desc' in el and el['desc'] != '':
          height += line_height * len(_splitToWidth(canvas, el['desc'], 11*cm, font, font_size))
      if part.vatType == 'net':
        # Eine Zeile mehr
        height += line_height + line_padding
      # Für die MwSt-Summen
      height += (line_height + line_padding) * len(part.vat)
      # Für den Rechnungsbetrag
      height += line_height + line_padding
    return height


   
  address(canvas, iv.addresslines)

  font_size = default_font_size
  y = firstPage(canvas)

  canvas.setFont(font+'-Bold', font_size+3)
  min_y = y
  if iv.caption:
    canvas.drawString(leftcontent, y, iv.caption)
    min_y -= (font_size + 3) + 0.5*cm

  if type(iv) == Invoice.Tender:
    canvas.setFont(font, font_size)
    canvas.drawString(rightcolumn, y, "Erstellungsdatum:")
    canvas.drawRightString(rightcontent, y, "%s" % iv.date.strftime('%d. %m. %Y'))
    y -= (font_size + 0.1*cm)
  elif type(iv) == Invoice.Generic:
    canvas.setFont(font, font_size)
    canvas.drawString(rightcolumn, y, "Datum:")
    canvas.drawRightString(rightcontent, y, "%s" % iv.date.strftime('%d. %m. %Y'))
    y -= (font_size + 0.1*cm)
  elif type(iv) == Invoice.Invoice:
    canvas.setFont(font+'-Bold', font_size)
    canvas.drawString(rightcolumn, y, "Bei Fragen bitte immer angeben:")
    y -= (font_size + 0.2*cm)
    canvas.setFont(font, font_size)
    canvas.drawString(rightcolumn, y, "Rechnungsdatum:")
    canvas.drawRightString(rightcontent, y, "%s" % iv.date.strftime('%d. %m. %Y'))
    y -= (font_size + 0.1*cm)
    canvas.drawString(rightcolumn, y, "Rechnungsnummer:")
    canvas.drawRightString(rightcontent, y, "%i" % iv.id)
    y -= (font_size + 0.1*cm)
  if iv.customerno:
    canvas.drawString(rightcolumn, y, "Kundennummer:")
    canvas.drawRightString(rightcontent, y, "%s" % iv.customerno)
    y -= (font_size + 0.5*cm)
  canvas.setFont(font, font_size)
  y = min(min_y, y)

  if iv.salutation:
    canvas.drawString(leftcontent, y, iv.salutation)
    y -= font_size + 0.2*cm
    if type(iv) in [Invoice.Tender, Invoice.Invoice]:
      introText = 'hiermit stellen wir Ihnen die nachfolgend genannten Leistungen in Rechnung.'
      if type(iv) == Invoice.Tender:
        introText = 'hiermit unterbreiten wir Ihnen folgendes Angebot.'
      intro = _splitToWidth(canvas, introText, rightcontent - leftcontent, font, font_size)
      for line in intro:
        canvas.drawString(leftcontent, y, line)
        y -= font_size + 0.1*cm
      y -= font_size + 0.1*cm

  
  font_size = default_font_size
  for part in iv.parts:
    if y - _partHeight(part) < (bottomcontent + (0.5*cm)):
      num_pages += 1
      y = bottomcontent + (0.5*cm)
      canvas.setFont(font, default_font_size-2)
      canvas.drawRightString(rightcontent, bottomcontent + line_padding, 'Fortsetzung auf Seite %i' % num_pages)
      _PageWrap(canvas)
      y = topcontent - font_size
      canvas.setFillColor((0,0,0))
      canvas.setFont(font, font_size-2)
      canvas.drawCentredString(leftcontent + (rightcontent - leftcontent) / 2, y, '- Seite %i -' % num_pages)
      y -= line_padding*3
    # Debug: Was hat die Höhenbestimmung für diesen Teil als Höhe herausgefunden?
    #canvas.line(leftcontent, y-_partHeight(part), rightcontent, y-_partHeight(part))

    if type(part) == Invoice.Table:
        
      left = leftcontent
      right = rightcontent
      top = topcontent
      bottom = bottomcontent
      canvas.setFont(font, font_size)
      canvas.drawString(left+(0.1*cm), y-line_height+line_padding, 'Anz.')
      canvas.drawString(left+(1.6*cm), y-line_height+line_padding, 'Beschreibung')
      if len(part.vat) == 1:
        canvas.drawRightString(left+(14.3*cm), y-line_height+line_padding, 'Einzelpreis')
      else:
        canvas.drawRightString(left+(13.7*cm), y-line_height+line_padding, 'Einzelpreis')
      canvas.drawRightString(left+(16.8*cm), y-line_height+line_padding, 'Gesamtpreis')
      canvas.setLineWidth(0.01*cm)
      canvas.line(left, y - line_height, right, y - line_height)
      y -= line_height + 0.02*cm
      odd=True
      for el in part.entries:
        subject = []
        if len(part.vat) == 1:
          subject = _splitToWidth(canvas, el['subject'], 10.3*cm, font, font_size)
        else:
          subject = _splitToWidth(canvas, el['subject'], 9.3*cm, font, font_size)
        desc = []
        if 'desc' in el and el['desc'] != '':
          desc = _splitToWidth(canvas, el['desc'], 11*cm, font, font_size)

        # draw the background
        if not odd:
          canvas.setFillColorRGB(0.9, 0.9, 0.9)
        else:
          canvas.setFillColorRGB(1, 1, 1)
        need_lines = len(subject) + len(desc)
        canvas.rect(left, y - (need_lines*line_height)-(2*line_padding), height = (need_lines*line_height)+(2*line_padding), width = right-left, fill=1, stroke=0)
        canvas.setFillColorRGB(0, 0, 0)
        y -= line_padding
        (integer, decimals) = _niceCount(el['count'])
        canvas.drawRightString(left+0.8*cm, y-font_height, integer)
        if decimals:
          canvas.drawString(left+0.8*cm, y-font_height, ',%s' % decimals)
        if len(part.vat) < 2:
          canvas.drawString(left+1.7*cm, y-font_height, subject[0])
          canvas.drawRightString(left+14.3*cm, y-font_height, _formatPrice(el['price']))
          if el['tender']:  
            canvas.drawRightString(left+16.8*cm, y-font_height, 'eventual')
          else:
            canvas.drawRightString(left+16.8*cm, y-font_height, _formatPrice(el['total']))
          subject = subject[1:]
          x = 1
          for line in subject:
            canvas.drawString(left+1.7*cm, y-(x * line_height)-font_height, line)
            x += 1
          for line in desc[:-1]:
            _drawJustifiedString(left+1.7*cm, y-(x * line_height)-font_height, line, canvas, 11*cm, font, font_size)
            x += 1
          canvas.drawString(left+1.7*cm, y-(x * line_height)-font_height, desc[-1])
          x += 1
        else:
          canvas.drawString(left+1.7*cm, y-font_height, subject[0])
          canvas.drawRightString(left+13.3*cm, y-font_height, _formatPrice(el['price']))
          canvas.drawString(left+13.7*cm, y-font_height, str(part.vat[el['vat']][1]))
          if el['tender']:  
            canvas.drawRightString(left+16.8*cm, y-font_height, 'eventual')
          else:
            canvas.drawRightString(left+16.8*cm, y-font_height, _formatPrice(el['total']))
          subject = subject[1:]
          x = 1
          for line in subject:
            canvas.drawString(left+1.7*cm, y-(x * line_height)-font_height, line)
            x += 1
          for line in desc:
            canvas.drawString(left+1.7*cm, y-(x * line_height)-font_height, line)
            x += 1
        odd = not odd
        y -= (need_lines * line_height) + line_padding
      if part.summary:
        y -= (0.3*cm)
        if part.vatType == 'gross':
          canvas.setFont(font+'-Bold', font_size)
          if iv.tender:
            canvas.drawRightString(left + 14.5*cm, y-font_height, 'Gesamtbetrag:')
          else:
            canvas.drawRightString(left + 14.5*cm, y-font_height, 'Rechnungsbetrag:')
          canvas.drawRightString(left + 16.8*cm, y-font_height, _formatPrice(part.sum))
          canvas.setFont(font, font_size)
          y -= line_height + line_padding
          summaries = []
          if len(part.vat) == 1:
            vat = part.vat.keys()[0]
            (integer, decimals) = _niceCount( (vat * 100) )
            vatstr = '%s' % integer
            if decimals:
              vatstr += ',%s' % decimals
            if iv.tender:
              summaries.append(('Im Gesamtbetrag sind %s%% MwSt enthalten:' % vatstr, _formatPrice((part.sum/(vat+1))*vat)))
            else:
              summaries.append(('Im Rechnungsbetrag sind %s%% MwSt enthalten:' % vatstr, _formatPrice((part.sum/(vat+1))*vat)))
          else:
            for vat, vatdata in part.vat.iteritems():
              (integer, decimals) = _niceCount( (vat * 100) )
              vatstr = '%s' % integer
              if decimals:
                vatstr += ',%s' % decimals
              summaries.append(('%s: Im Teilbetrag von %s sind %s%% MwSt enthalten:' % (vatdata[1], _formatPrice(vatdata[0]), vatstr), _formatPrice((vatdata[0]/(vat+1))*vat)))
          summaries.sort()
          for line in summaries:
            canvas.drawRightString(left + 14.5*cm, y-font_height, line[0])
            canvas.drawRightString(left + 16.8*cm, y-font_height, line[1])
            y -= line_height
        else:
          canvas.drawRightString(left + 14.5*cm, y-font_height, 'Nettobetrag:')
          canvas.drawRightString(left + 16.8*cm, y-font_height, _formatPrice(part.sum))
          y -= line_height
          summaries = []
          if len(part.vat) == 1:
            vat = part.vat.keys()[0]
            (integer, decimals) = _niceCount( (vat * 100) )
            vatstr = '%s' % integer
            if decimals:
              vatstr += ',%s' % decimals
            summaries.append(('zzgl. %s%% MwSt:' % vatstr, _formatPrice(vat*part.sum)))
          elif len(part.vat) > 1:
            for vat, vatdata in part.vat.iteritems():
              (integer, decimals) = _niceCount( (vat * 100) )
              vatstr = '%s' % integer
              if decimals:
                vatstr += ',%s' % decimals
              summaries.append(('zzgl. %s%% MwSt (%s):' % (vatstr, vatdata[1]), _formatPrice(vat*vatdata[0])))
          summaries.sort()
          for line in summaries:
            canvas.drawRightString(left + 14.5*cm, y-font_height, line[0])
            canvas.drawRightString(left + 16.8*cm, y-font_height, line[1])
            y -= line_height
          sum = part.sum
          for vat, vatdata in part.vat.iteritems():
            sum += vat*vatdata[0]
          canvas.setFont(font+'-Bold', font_size)
          if iv.tender:
            canvas.drawRightString(left + 14.5*cm, y-font_height, 'Gesamtbetrag:')
          else:
            canvas.drawRightString(left + 14.5*cm, y-font_height, 'Rechnungsbetrag:')
          canvas.drawRightString(left + 16.8*cm, y-font_height, _formatPrice(sum))
          canvas.setFont(font, font_size)
          y -= line_height + line_padding
    elif type(part) == Invoice.Text:
      my_font_size = font_size
      canvas.setFont(font, my_font_size)
      left, right = leftcontent, rightcontent
      firsttime = True
      headlines = []
      if part.urgent:
        left += 1.5*cm
        right -= 1.5*cm
      if part.headline:
        headlines = _splitToWidth(canvas, part.headline, right-left, font, my_font_size)
      for para in part.paragraphs:
        lines = _splitToWidth(canvas, para, right-left, font, my_font_size)
        if part.urgent:
          need_height = len(lines) * line_height
          if len(headlines) > 0:
            need_height += len(headlines) * (line_height + 1) + line_padding
          canvas.setFillColorRGB(0.95, 0.95, 0.95)
          canvas.rect(left-0.5*cm, y - (need_height+(6*line_padding)), height = need_height+(6*line_padding), width = right-left+1*cm, fill=1, stroke=1)
          canvas.setFillColorRGB(0, 0, 0)
          y -= line_padding*3
        if part.headline and firsttime:
          firsttime = False
          canvas.setFont(font+'-Bold', my_font_size+1)
          for line in headlines:
            canvas.drawString(left, y-(font_height+1), line)
            y -= line_height + 1
          y -= line_padding
          canvas.setFont(font, my_font_size)
        for line in lines[:-1]:
          _drawJustifiedString(left, y-font_height, line, canvas, right-left, font, my_font_size)
          y -= line_height
        canvas.drawString(left, y-font_height, lines[-1])
        y -= line_height
        
        y -= line_padding*3
      left, right = leftcontent, rightcontent
    else:
      raise NotImplementedError("Cannot handle part of type %s" % type(part))
    y -= (0.5*cm)
    
  
  canvas.showPage()
  canvas.save()
  pdfdata = fd.getvalue()
  return pdfdata