import os
import requests

from command import CommandRegistry
from commands import core
from little_shit import get_source

__registry__ = cr = CommandRegistry()


@cr.register('tuling123', 'chat', '聊天')
def tuling123(args_text, ctx_msg, internal=False):
    url = 'http://www.tuling123.com/openapi/api'
    data = {
        'key': os.environ.get('TURING123_API_KEY'),
        'info': args_text,
        'userid': get_source(ctx_msg)
    }
    resp = requests.post(url, data=data)
    if resp.status_code == 200:
        json = resp.json()
        if internal:
            return json
        if int(json.get('code', 0)) == 100000:
            reply = json.get('text', '')
        else:
            # Is not text type
            reply = '腊鸡图灵机器人返回了一堆奇怪的东西，就不发出来了'
    else:
        if internal:
            return None
        reply = '腊鸡图灵机器人出问题了，先不管他，过会儿再玩他'
    core.echo(reply, ctx_msg)

# TODO: 加入微信消息源之后修改
# @cr.register('xiaoice', '小冰')
# def xiaoice(args_text, ctx_msg, internal=False):
#     resp = api.wx_consult(account='xiaoice-ms', content=args_text)
#     if resp:
#         json = resp.json()
#         if json and json.get('reply'):
#             reply = json['reply']
#             core.echo(reply, ctx_msg, internal)
#             return reply
#     core.echo('小冰没有回复，请稍后再试', ctx_msg, internal)
#     return None