import i3
import time

"""
Py3status plugin - shows current window title.

DEPENDENCIES:
    Depends on i3-py!
    pip install i3-py

CONFIG:
    MAX_WIDTH - max title width
    CACHED_TIME - in seconds

If payload from server contains wierd utf-8 (for example one window have something bad in title) - plugin will give empty output UNTIL this window will be closed. I can't fix or workaround that in PLUGIN, problem is in i3-py library. 
"""

POSITION = 0
MAX_WIDTH = 120
CACHED_TIME = 0.5

def find_focused(tree):
    if type(tree) == list:
        for el in tree:
            res = find_focused(el)
            if res:
                return res

    elif type(tree) == dict:
        if tree["focused"]:
            return tree
        else:
            return find_focused(tree["nodes"] + tree["floating_nodes"])

class Py3status:
    def __init__(self, *args, **kwargs):
        self.text = ""
        super(Py3status).__init__(*args, **kwargs)

    def currentTitle(self, i3_status_output_json, i3status_config):
        window = find_focused(i3.get_tree())

        transformed = False
        if window and "name" in window and window["name"] != self.text: 
            self.text = len(window["name"]) > MAX_WIDTH and "..." + window["name"][-(MAX_WIDTH-3):] or window["name"]
            transformed = True

        return (POSITION, {
            'full_text': self.text,
            'transformed': transformed,
            'name': 'current-title',
            'cached_until': time.time() + CACHED_TIME,
        })