import demistomock as demisto  # noqa: F401
import requests
from CommonServerPython import *  # noqa: F401

""" MAIN FUNCTION """


def main() -> None:
    api_token = demisto.params().get("apitoken")
    try:
        if demisto.command() == "line-send-message":
            # This is for sending LINE notification to specific group
            headers = {"Authorization": "Bearer " + api_token, "Content-Type": "application/x-www-form-urlencoded"}
            linemsg = demisto.args().get("message")
            payload = {"message": linemsg}
            r = requests.post("https://notify-api.line.me/api/notify", headers=headers, params=payload)
            return_results(r)

        elif demisto.command() == "test-module":
            headers = {"Authorization": "Bearer " + api_token, "Content-Type": "application/x-www-form-urlencoded"}

            result = requests.get("https://notify-api.line.me/api/status", headers=headers)
            if "200" not in str(result):
                return_results(result)
            else:
                return_results("ok")

    # Log exceptions and return errors
    except Exception as e:
        demisto.error(traceback.format_exc())  # print the traceback
        return_error(f"Failed to execute {demisto.command()} command.\nError:\n{e!s}")


""" ENTRY POINT """

if __name__ in ("__main__", "__builtin__", "builtins"):
    main()