reset PREFIX after switch stream
[rainbowstream.git] / rainbowstream / draw.py
index 9f8a0eb..a947506 100644 (file)
@@ -15,6 +15,7 @@ from .c_image import *
 from .colors import *
 from .config import *
 from .py3patch import *
+from .emoji import *
 
 # Draw global variables
 dg = {}
@@ -37,6 +38,7 @@ def start_cycle():
     """
     dg['cyc'] = init_cycle()
     dg['cache'] = {}
+    dg['humanize_unsupported'] = False
 
 
 def order_rainbow(s):
@@ -145,7 +147,34 @@ def color_func(func_name):
     return globals()[func_name]
 
 
-def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
+def fallback_humanize(date, fallback_format=None, use_fallback=False):
+    """
+    Format date with arrow and a fallback format
+    """
+    # Convert to local timezone
+    date = arrow.get(date).to('local')
+    # Set default fallback format
+    if not fallback_format:
+        fallback_format = '%Y/%m/%d %H:%M:%S'
+    # Determine using fallback format or not by a variable
+    if use_fallback:
+        return date.datetime.strftime(fallback_format)
+    try:
+        # Use Arrow's humanize function
+        lang, encode = locale.getdefaultlocale()
+        clock = date.humanize(locale=lang)
+    except:
+        # Notice at the 1st time only
+        if not dg['humanize_unsupported']:
+            dg['humanize_unsupported'] = True
+            printNicely(
+                light_magenta('Humanized date display method does not support your $LC_ALL.'))
+        # Fallback when LC_ALL is not supported
+        clock = date.datetime.strftime(fallback_format)
+    return clock
+
+
+def draw(t, keyword=None, humanize=True, noti=False, fil=[], ig=[]):
     """
     Draw the rainbow
     """
@@ -161,28 +190,38 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
     favorited = t['favorited']
     retweet_count = t['retweet_count']
     favorite_count = t['favorite_count']
+    client = t['source']
     date = parser.parse(created_at)
-    date = arrow.get(date).to('local')
-    if humanize:
-        lang, encode = locale.getdefaultlocale()
-        clock = arrow.get(date).to('local').humanize(locale=lang)
-    else:
-        try:
-            clock_format = c['FORMAT']['TWEET']['CLOCK_FORMAT']
-        except:
-            clock_format = '%Y/%m/%d %H:%M:%S'
-        clock = date.datetime.strftime(clock_format)
+    try:
+        clock_format = c['FORMAT']['TWEET']['CLOCK_FORMAT']
+    except:
+        clock_format = '%Y/%m/%d %H:%M:%S'
+    clock = fallback_humanize(date, clock_format, not humanize)
 
     # Pull extended retweet text
     try:
         text = 'RT @' + t['retweeted_status']['user']['screen_name'] + ': ' +\
             t['retweeted_status']['text']
+        # Display as a notification
+        target = t['retweeted_status']['user']['screen_name']
+        if all([target == c['original_name'], not noti]):
+            # Add to evens for 'notification' command
+            t['event'] = 'retweet'
+            c['events'].append(t)
+            notify_retweet(t)
+            return
     except:
         pass
 
     # Unescape HTML character
     text = unescape(text)
 
+    # Get client name
+    try:
+        client = client.split('>')[-2].split('<')[0]
+    except:
+        client = None
+
     # Get expanded url
     try:
         expanded_url = []
@@ -205,6 +244,7 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
         media_url = None
 
     # Filter and ignore
+    mytweet = screen_name == c['original_name']
     screen_name = '@' + screen_name
     fil = list(set((fil or []) + c['ONLY_LIST']))
     ig = list(set((ig or []) + c['IGNORE_LIST']))
@@ -222,7 +262,10 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
 
     # Format info
     name = cycle_color(name)
-    nick = color_func(c['TWEET']['nick'])(screen_name)
+    if mytweet:
+        nick = color_func(c['TWEET']['mynick'])(screen_name)
+    else:
+        nick = color_func(c['TWEET']['nick'])(screen_name)
     clock = clock
     id = str(rid)
     fav = ''
@@ -249,7 +292,7 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
     # Highlight link
     tweet = lmap(
         lambda x: color_func(c['TWEET']['link'])(x)
-        if x[0:4] == 'http'
+        if x.startswith('http')
         else x,
         tweet)
     # Highlight hashtag
@@ -258,6 +301,15 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
         if x.startswith('#')
         else x,
         tweet)
+    # Highlight my tweet
+    if mytweet:
+        tweet = [color_func(c['TWEET']['mytweet'])(x)
+                 for x in tweet
+                 if not any([
+                     x == 'RT',
+                     x.startswith('http'),
+                     x.startswith('#')])
+                 ]
     # Highlight keyword
     tweet = ' '.join(tweet)
     if keyword:
@@ -276,6 +328,7 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
         formater = nick.join(formater.split('#nick'))
         formater = fav.join(formater.split('#fav'))
         formater = tweet.join(formater.split('#tweet'))
+        formater = emojize(formater)
         # Change clock word
         word = [wo for wo in formater.split() if '#clock' in wo][0]
         delimiter = color_func(c['TWEET']['clock'])(
@@ -295,9 +348,21 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
         delimiter = color_func(c['TWEET']['favorite_count'])(
             str(favorite_count).join(word.split('#fa_count')))
         formater = delimiter.join(formater.split(word))
+        # Change client word
+        word = [wo for wo in formater.split() if '#client' in wo][0]
+        delimiter = color_func(c['TWEET']['client'])(
+            client.join(word.split('#client')))
+        formater = delimiter.join(formater.split(word))
     except:
         pass
 
+    # Add spaces in begining of line if this is inside a notification
+    if noti:
+        formater = '\n  '.join(formater.split('\n'))
+        # Reformat
+        if formater.startswith('\n'):
+            formater = formater[1:]
+
     # Draw
     printNicely(formater)
 
@@ -383,6 +448,7 @@ def print_right_message(m):
     h, w = os.popen('stty size', 'r').read().split()
     w = int(w)
     frame_width = w // 3 - dg['frame_margin']
+    frame_width = max(c['THREAD_MIN_WIDTH'], frame_width)
     step = frame_width - 2 * dg['frame_margin']
     slicing = textwrap.wrap(m['text'], step)
     spaces = w - frame_width - dg['frame_margin']
@@ -431,6 +497,7 @@ def print_right_message(m):
         word = [wo for wo in formater.split() if '#id' in wo][0]
         delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
         formater = delimiter.join(formater.split(word))
+        formater = emojize(formater)
     except Exception:
         printNicely(red('Wrong format in config.'))
         return
@@ -446,6 +513,7 @@ def print_left_message(m):
     h, w = os.popen('stty size', 'r').read().split()
     w = int(w)
     frame_width = w // 3 - dg['frame_margin']
+    frame_width = max(c['THREAD_MIN_WIDTH'], frame_width)
     step = frame_width - 2 * dg['frame_margin']
     slicing = textwrap.wrap(m['text'], step)
     spaces = dg['frame_margin']
@@ -494,6 +562,7 @@ def print_left_message(m):
         word = [wo for wo in formater.split() if '#id' in wo][0]
         delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
         formater = delimiter.join(formater.split(word))
+        formater = emojize(formater)
     except Exception:
         printNicely(red('Wrong format in config.'))
         return
@@ -559,6 +628,7 @@ def print_message(m):
         word = [wo for wo in formater.split() if '#id' in wo][0]
         delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
         formater = delimiter.join(formater.split(word))
+        formater = emojize(formater)
     except:
         printNicely(red('Wrong format in config.'))
         return
@@ -567,6 +637,264 @@ def print_message(m):
     printNicely(formater)
 
 
+def notify_retweet(t):
+    """
+    Notify a retweet
+    """
+    source = t['user']
+    created_at = t['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'retweeted your tweet')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    draw(t=t['retweeted_status'], noti=True)
+
+
+def notify_favorite(e):
+    """
+    Notify a favorite event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = e['target_object']
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'favorited your tweet')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    draw(t=target_object, noti=True)
+
+
+def notify_unfavorite(e):
+    """
+    Notify a unfavorite event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = e['target_object']
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'unfavorited your tweet')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    draw(t=target_object, noti=True)
+
+
+def notify_follow(e):
+    """
+    Notify a follow event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'followed you')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+
+
+def notify_list_member_added(e):
+    """
+    Notify a list_member_added event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'added you to a list')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def notify_list_member_removed(e):
+    """
+    Notify a list_member_removed event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'removed you from a list')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def notify_list_user_subscribed(e):
+    """
+    Notify a list_user_subscribed event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'subscribed to your list')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def notify_list_user_unsubscribed(e):
+    """
+    Notify a list_user_unsubscribed event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'unsubscribed from your list')
+    date = parser.parse(created_at)
+    clock = fallback_humanize(date)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    meta = emojize(meta)
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def print_event(e):
+    """
+    Notify an event
+    """
+    event_dict = {
+        'retweet': notify_retweet,
+        'favorite': notify_favorite,
+        'unfavorite': notify_unfavorite,
+        'follow': notify_follow,
+        'list_member_added': notify_list_member_added,
+        'list_member_removed': notify_list_member_removed,
+        'list_user_subscribed': notify_list_user_subscribed,
+        'list_user_unsubscribed': notify_list_user_unsubscribed,
+    }
+    event_dict.get(e['event'], lambda *args: None)(e)
+
+
 def show_profile(u):
     """
     Show a profile
@@ -607,8 +935,7 @@ def show_profile(u):
     location = 'Location : ' + color_func(c['PROFILE']['location'])(location)
     url = 'URL : ' + (color_func(c['PROFILE']['url'])(url) if url else '')
     date = parser.parse(created_at)
-    lang, encode = locale.getdefaultlocale()
-    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = fallback_humanize(date)
     clock = 'Join at ' + color_func(c['PROFILE']['clock'])(clock)
 
     # Format
@@ -665,7 +992,7 @@ def print_trends(trends):
     printNicely('')
 
 
-def print_list(group):
+def print_list(group, noti=False):
     """
     Display a list
     """
@@ -686,15 +1013,18 @@ def print_list(group):
         mode = color_func(c['GROUP']['mode'])('Type: ' + mode)
         created_at = grp['created_at']
         date = parser.parse(created_at)
-        lang, encode = locale.getdefaultlocale()
-        clock = arrow.get(date).to('local').humanize(locale=lang)
+        clock = fallback_humanize(date)
         clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
 
+        prefix = ' ' * 2
+        # Add spaces in begining of line if this is inside a notification
+        if noti:
+            prefix = ' ' * 2 + prefix
         # Create lines
-        line1 = ' ' * 2 + name + member + '  ' + subscriber
-        line2 = ' ' * 4 + description
-        line3 = ' ' * 4 + mode
-        line4 = ' ' * 4 + clock
+        line1 = prefix + name + member + '  ' + subscriber
+        line2 = prefix + ' ' * 2 + description
+        line3 = prefix + ' ' * 2 + mode
+        line4 = prefix + ' ' * 2 + clock
 
         # Display
         printNicely('')
@@ -703,7 +1033,8 @@ def print_list(group):
         printNicely(line3)
         printNicely(line4)
 
-    printNicely('')
+    if not noti:
+        printNicely('')
 
 
 def show_calendar(month, date, rel):
@@ -747,6 +1078,7 @@ def format_quote(tweet):
         formater = screen_name.join(formater.split('#owner'))
         formater = text.join(formater.split('#tweet'))
         formater = u2str(formater)
+        formater = emojize(formater)
     except:
         pass
     # Highlight like a tweet