import json
import sys
import re

from urllib.request import urlopen

Usage = """
Usage: python3 dict.sh [word]
"""

HAS_RICH = 0
try:
    from rich.console import Console
    from rich.panel import Panel
    from rich.text import Text
    from rich import print
    HAS_RICH = 1
except:
    pass

def basic_paring(data):
    """ Display data in the most simple way
    """
    word = data["word"]
    defs = data["def"]
    samples = data["sample_sentence"]
    ponetic = data["phonetic"]
    
    print(word)

    if len(ponetic.items()) > 0:
        for _, v in ponetic.items():
            print(v, end="  ")
        print()
    print()

    if len(defs) > 0:
        for k in defs:
            print(k, defs[k])
        print()

    for s in samples[:2]:
        print("-", end=" ")
        print(s['en'])
        print()

def rich_paring(data):
    console = Console()
    word = data["word"]
    defs = data["def"]
    ponetic = data["phonetic"]
    samples = data["sample_sentence"]

    console.print(word, style="bold magenta")
    print()

    if len(ponetic.items()) > 0:
        for _, v in ponetic.items():
            #v = v.replace('[', '\[').replace(']', ']')
            console.print(Text(v, style="yellow"))
        print()
    
    if len(defs) > 0:
        for k in defs:
            console.print(Text(k, style="green"))
            console.print(Text(defs[k], style="cyan"))
        print()

    for s in samples[:2]:
        ens = s['en'].replace('"', '\"')
        ens = re.sub(r'\*\*([^ ]*?)\*\*', r'[/][red]\1[/][yellow4]', ens)

        text = Text.from_markup("[yellow4]" + ens + "[/]", justify="left")
        panel = Panel(text)

        console.print(panel)


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print(Usage)
        exit(1)

    word = sys.argv[1]
    url = "http://do.useless.pw/{}".format(word)
    r = urlopen(url)
    content = r.read()
    json_content = json.loads(content)
    if not HAS_RICH:
        basic_paring(json_content)
    else:
        rich_paring(json_content)