from __future__ import unicode_literals
import logging
import selenium.webdriver

# set WebElement hash to default to use set(x)
selenium.webdriver.remote.webelement.WebElement.__hash__ = lambda self: hash(self.id)

def diff(a, b):
  b = set(b)
  return [aa for aa in a if aa not in b]

logger = logging.getLogger('rschat.' + __name__)

class ChatInterface():
  def __init__(self, chat_type, username, password):
    self.chat_type = chat_type
    self.username = username
    self.password = password
    self.browser = None

  def __enter__(self):
    logger.info('Starting webdriver')
    options = selenium.webdriver.ChromeOptions()
    options.add_argument('--disable-file-system')

    self.browser = selenium.webdriver.Chrome('/usr/lib64/chromium-browser/chromedriver', chrome_options=options)

    self.browser.implicitly_wait(30)

    logger.debug('Navigating to companion app')
    self.browser.get('http://www.runescape.com/companion/comapp.ws')

    logger.debug('Switching to iframe')
    self.browser.switch_to_frame(self.browser.find_element_by_tag_name('iframe'))

# login
    logger.debug('Logging in as %s' % self.username)
    self.browser.find_element_by_id('username').send_keys(self.username)
    self.browser.find_element_by_id('password').send_keys(self.password)
    self.browser.find_element_by_tag_name('form').submit()

# decline remember login
    logger.debug('Declining remember login')
    self.browser.find_element_by_partial_link_text('No').click()

    logger.debug('Switching to %s chat' % self.chat_type)
    self.browser.find_element_by_css_selector('.chat-nav .icon-%schat' % self.chat_type).click()

    self.browser.implicitly_wait(0)

    return self

  lastmsgs = []

  def send(self, line):
    self.browser.find_element_by_id('message').send_keys(line)
    self.browser.find_element_by_tag_name('form').submit()

  def poll(self):
    try:
      allmsgs = self.browser.find_elements_by_css_selector('#content-%s-chat .message:not(.my-message) .message-content' % self.chat_type)
      newmsgs = diff(allmsgs, self.lastmsgs)

      self.lastmsgs = allmsgs

      return [(e.find_element_by_class_name('author').text.replace(' -', ''), e.find_element_by_tag_name('p').text) for e in newmsgs]
    except selenium.webdriver.common.exceptions.NoSuchElementException:
      pass

  def __exit__(self, *_):
    self.browser.quit()