# -*- coding: utf-8 -*-

# あらかじめ役に立つモジュールを追加してますが好きなようにカスタマイスしてください。
# from mastodon import * 及び import threading は必須となります。
from mastodon import *
from time import sleep
from time import time
import re, sys, os, csv, json, codecs, io, gc
import threading, random, traceback

# プロンプトで起動したい人のための装置
sys.stdout = io.TextIOWrapper(sys.stdout.buffer,
                              encoding=sys.stdout.encoding,
                              errors='backslashreplace',
                              line_buffering=sys.stdout.line_buffering)

url_ins = 'https://knzk.me'

mastodon = Mastodon(
    client_id='client.secret',
    access_token='user.secret',
    api_base_url=url_ins)

class User(StreamListener):  # ホームでフォローした人と通知を監視します
    def on_notification(self, notification):  # 通知を監視します
        """ここに機能を追加します"""
        print(str(notification["type"]))
        pass
class Local(StreamListener):  # ここではLTLを監視する継承クラスになります。
    def on_update(self, status):  # StreamingAPIがリアルタイムにトゥート情報を吐き出してくれます。
        """ここに機能を追加します"""
        print("update "+str(status["id"]))
        pass

    def on_delete(self, status_id):  # トゥー消し警察の監視場になります。
        print("del "+str(status_id))
        pass

class Loading():
    def go_local():  # listenerオブジェクトには監視させるものを（続く）
        try:
            listener = Local()
            mastodon.stream_local(listener)
        except:
            print("例外情報\n" + traceback.format_exc())
            pass

    def go_user():  # （続き）継承で組み込んだものを追加するようにします。
        try:
            listener = User()
            mastodon.stream_user(listener)
        except:
            print("例外情報\n" + traceback.format_exc())
            sleep(180)
            pass

if __name__ == '__main__':  # ファイルから直接開いたら動くよ！
    try:
        user = threading.Thread(target=Loading.go_user)
        user.start()
        local = threading.Thread(target=Loading.go_local)
        local.start()
    except:
        sleep(3)
        bot.toot("すみません、ログアウトするかもしれません。")

listener = Local() #　ローカルタイムライン
mastodon.stream_local(listener)

listener = User() #　ホーム及び通知
mastodon.stream_user(listener)

class bot():
    def func01(status):
        content = status['content']
        if re.compile("コーヒー(ひとつ|くださ[あ～ー]*い|お(願|ねが)い"
                      "|ちょ[うぉお～ー]*だい)").search(content):  #前回より正規表現を駆使して強化です
            post = 'どうぞ☕'
            time.sleep(3)
            return mastodon.status_post(post,visibility='public')
        pass

    def on_update(self, status):  # StreamingAPIがリアルタイムにトゥート情報を吐き出してくれます。
        if if account["acct"] != 'ここにアカウントのID名':  # 自分のtootに反応しないようにするための処理です
            bot.func01(status)
        pass

class User(StreamListener):  # ホームでフォローした人と通知を監視するStreamingAPIの継承クラスです。
    def on_notification(self, notification):  # 通知を監視します。
        account = notification["account"]
        if notification["type"] == "mention":  # 通知がリプだった場合です。
            status = notification["status"]
            if account["acct"] != "ここにBotのアカウントID名":
                post = bot.func(status)
                bot.men(post, status)
        pass

class bot():
    def func01(status):
        content = status['content']
        if re.compile("コーヒー(ひとつ|くださ[あ～ー]*い|お(願|ねが)い"
                      "|ちょ[うぉお～ー]*だい)").search(content):
            return 'どうぞ☕'
        else:
            return None

    def men(post, status):
        account = status["account"]
        toot = '@' + account["acct"] + ' ' + post
        time.sleep(3)
        return mastodon.status_post(toot,
                                    visibility=status["visibility"],
                                    in_reply_to_id=status["id"])

class User(StreamListener):
    def on_notification(self, notification):
        account = notification["account"]
        if notification["type"] == "mention":
            status = notification["status"]
            time.sleep(2)
            mastodon.status_favourite(status["id"])  # ←これを追加します
            if account["acct"] != "ここにBotのアカウントID名":
                post = bot.func(status)
                bot.men(post, status)
        pass

class User(StreamListener):
    def on_notification(self, notification):
        account = notification["account"]
            if notification["type"] == "follow":  # 通知がフォローだった場合はフォロバします。
                sleep(2)
                mastodon.account_follow(account["id"])