From 10f1f06e556231b5878faee2fb9eda27764aa5ce Mon Sep 17 00:00:00 2001
From: Steven Smith <stevensmith.ome@gmail.com>
Date: Thu, 27 Jun 2013 18:34:06 +0800
Subject: [PATCH] Update imdb.py

Add autoresponse
---
 plugins/imdb.py | 41 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

diff --git a/plugins/imdb.py b/plugins/imdb.py
index 5d015fa..65e73e2 100755
--- a/plugins/imdb.py
+++ b/plugins/imdb.py
@@ -1,10 +1,26 @@
-# IMDb lookup plugin by Ghetto Wizard (2011).
+# IMDb lookup plugin by Ghetto Wizard (2011) and blha303 (2013)
 
 from util import hook, http
 import re
+import json
 
 id_re = re.compile("tt\d+")
 
+def truncate(msg):
+    nmsg = msg.split(" ")
+    out = None
+    x = 0
+    for i in nmsg:
+      if x <= 7:
+        if out:
+          out = out + " " + nmsg[x]
+        else:
+          out = nmsg[x]
+      x = x + 1
+    if x <= 7:
+      return out
+    else:
+      return out + "..."
 
 @hook.command
 def imdb(inp):
@@ -32,3 +48,26 @@ def imdb(inp):
         return out % content
     else:
         return 'Unknown error.'
+
+imdb_re = (r'(.*:)//(imdb.com|www.imdb.com)(:[0-9]+)?(.*)', re.I)
+
+@hook.regex(*imdb_re)
+def imdb_url(match):
+    id = match.group(4).split('/')[-1]
+    if id == "":
+      id = match.group(4).split('/')[-2]
+    content = http.get_json("http://www.omdbapi.com/", i=id)
+    if content.get('Error', None) == 'Movie not found!':
+        return 'Movie not found!'
+    elif content['Response'] == 'True':
+        content['URL'] = 'http://www.imdb.com/title/%(imdbID)s' % content
+        content['Plot'] = truncate(content['Plot'])
+        out = '\x02%(Title)s\x02 (%(Year)s) (%(Genre)s): %(Plot)s'
+        if content['Runtime'] != 'N/A':
+            out += ' \x02%(Runtime)s\x02.'
+        if content['imdbRating'] != 'N/A' and content['imdbVotes'] != 'N/A':
+            out += ' \x02%(imdbRating)s/10\x02 with \x02%(imdbVotes)s\x02' \
+                   ' votes.'
+        return out % content
+    else:
+        return 'Unknown error.'