# coding=utf-8

# WolframAlpha module for drastikbot
#
# It makes use of the Short Answers API to get answers for user
# queries.
#
# You need to use you own AppID to use this module. Find out more:
# http://products.wolframalpha.com/short-answers-api/documentation/

'''
Copyright (C) 2019 drastik.org

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, version 3 only.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
'''

import urllib.parse
import requests


class Module:
    def __init__(self):
        self.commands = ['wa', 'wolfram', 'wolframalpha']
        self.manual = {
            "desc": "Get results from the Wolfram|Alpha short answers API.",
            "bot_commands": {
                "wa": {"usage": lambda x: f"{x}wa <query>",
                       "alias": ["wolfram", "wolframalpha"]},
                "wolfram": {"usage": lambda x: f"{x}wolfram <query>",
                       "alias": ["w", "wolframalpha"]},
                "wolframalpha": {"usage": lambda x: f"{x}wolframalpha <query>",
                       "alias": ["wolfram", "w"]}
            }
        }


AppID = "Enter your AppID here"


def short_answers(query):
    url = f"http://api.wolframalpha.com/v1/result?appid={AppID}&i={query}"
    try:
        r = requests.get(url, timeout=10)
    except Exception:
        return False
    return r.text


def main(i, irc):
    query = urllib.parse.quote_plus(i.msg_nocmd)
    r = short_answers(query)
    r = f"Wolfram|Alpha: {r}"
    irc.privmsg(i.channel, r)