from waveapi import events

from waveapi import model

from waveapi import robot

from waveapi import document

import logging

import re

logger = logging.getLogger('AltLawBot')

logger.setLevel(logging.DEBUG)

cases_re = re.compile('\[\[(.+?)\]\]')

def OnRobotAdded(properties, context):

  """Invoked when the robot has been added."""

  root_wavelet = context.GetRootWavelet()

  root_wavelet.CreateBlip().GetDocument().SetText("Hello I am the AltLaw Bot!")

def OnBlipSubmitted(properties, context):

  blip = context.GetBlipById(properties['blipId'])

  doc = blip.GetDocument()

  text = doc.GetText()

  left = 0

  try:

      logger.debug('creator: %s' % blip.GetCreator())

      logger.debug('text: %s' % text)

  except:

      pass

  cases = cases_re.findall(text)

  for case in cases:

      left = text.find(case, left)

      replaceRange = document.Range(left-2, left+len(case)+2)

      linkText = "Replaced"

      linkLocation = "http://www.google.com"

      doc.SetTextInRange(replaceRange, linkText)

      doc.SetAnnotation(document.Range(left-2, left+len(linkText)-2), "link/manual", linkLocation)

      newBlip = doc.AppendInlineBlip()

      newBlip.GetDocument().SetText(linkLocation)

      #text = text.replace(text[replaceRange.start:replaceRange.end], linkText, 1)

      #left += len(linkTxt)

if __name__ == '__main__':

  myRobot = robot.Robot('AltLawBot', 

      image_url='http://altlawbot.appspot.com/assets/icon.png',

      version='2.06',

      profile_url='http://altlawbot.appspot.com/')

  myRobot.RegisterHandler(events.WAVELET_SELF_ADDED, OnRobotAdded)

  myRobot.RegisterHandler(events.BLIP_SUBMITTED, OnBlipSubmitted)

  myRobot.Run(debug=True)

application: altlawbot 

version: 1

runtime: python

api_version: 1

handlers:

- url: /_wave/.*

  script: altlawbot.py

- url: /assets

  static_dir: assets