From 9dde77a63bb8edbaed02dfaceac9c245631a99ff Mon Sep 17 00:00:00 2001
From: LiberH <liberh@gmx.com>
Date: Fri, 1 Aug 2014 20:09:16 +0200
Subject: [PATCH] Dirty hack for compatibility with new web profiles.

---
 modules/twitter.py | 34 +++++++++++++++++++++++++---------
 1 file changed, 25 insertions(+), 9 deletions(-)

diff --git a/modules/twitter.py b/modules/twitter.py
index afd1d5c8e..2c2e00ec2 100755
--- a/modules/twitter.py
+++ b/modules/twitter.py
@@ -12,7 +12,8 @@
 
 r_username = re.compile(r'^[a-zA-Z0-9_]{1,15}$')
 r_link = re.compile(r'^https?://twitter.com/\S+$')
-r_p = re.compile(r'(?ims)(<p class="js-tweet-text.*?</p>)')
+r_p1 = re.compile(r'(?ims)(<p class="ProfileTweet-text js-tweet-text.*?</p>)')
+r_p2 = re.compile(r'(?ims)(<p class="js-tweet-text.*?</p>)')
 r_tag = re.compile(r'(?ims)<[^>]+>')
 r_anchor = re.compile(r'(?ims)(<a.*?</a>)')
 r_expanded = re.compile(r'(?ims)data-expanded-url=["\'](.*?)["\']')
@@ -33,13 +34,29 @@ def replacement(match):
       return r_tag.sub('', anchor)
    return r_anchor.sub(replacement, tweet)
 
-def read_tweet(url):
+def read_tweet1(url):
    bytes = web.get(url)
-   shim = '<div class="content clearfix">'
+   shim = '<div class="ProfileTweet-contents">'
    if shim in bytes:
       bytes = bytes.split(shim, 1).pop()
 
-   for text in r_p.findall(bytes):
+   for text in r_p1.findall(bytes):
+      text = expand(text)
+      text = r_tag.sub('', text)
+      text = text.strip()
+      text = r_whiteline.sub(' ', text)
+      text = r_breaks.sub(' ', text)
+      return decode(text)
+   return "Sorry, couldn't get a tweet from %s" % url
+
+def read_tweet2(url):
+   bytes = web.get(url)
+   abc = '/'.join(url.split('/')[3:])
+   shim = '<a href="/%s"' % abc
+   if shim in bytes:
+      bytes = bytes.split(shim, 1).pop()
+
+   for text in r_p2.findall(bytes):
       text = expand(text)
       text = r_tag.sub('', text)
       text = text.strip()
@@ -52,7 +69,7 @@ def format(tweet, username):
    return '%s (@%s)' % (tweet, username)
 
 def user_tweet(username):
-   tweet = read_tweet('https://twitter.com/' + username + "?" + str(time.time()))
+   tweet = read_tweet1('https://twitter.com/' + username + "?" + str(time.time()))
    return format(tweet, username)
 
 def id_tweet(tid):
@@ -63,7 +80,7 @@ def id_tweet(tid):
       url = message.get("Location")
       if not url: return "Sorry, couldn't get a tweet from %s" % link
       username = url.split('/')[3]
-      tweet = read_tweet(url)
+      tweet = read_tweet2(url)
       return format(tweet, username)
    return "Sorry, couldn't get a tweet from %s" % link
 
@@ -81,9 +98,8 @@ def twitter(phenny, input):
    elif r_username.match(arg):
       phenny.say(user_tweet(arg))
    elif r_link.match(arg):
-      username = arg.split('/')[3]
-      tweet = read_tweet(arg)
-      phenny.say(format(tweet, username))
+      tid = arg.split('/')[5]
+      phenny.say(id_tweet(tid))
    else: phenny.reply("Give me a link, a username, or a tweet id")
 
 twitter.commands = ['tw', 'twitter']