"""
DM All
"""

# from masto_tools.script import Script

# class MassDM(Script):
#     """
#     Send a message to everyone on the instance :)
#
#     fuck it getting tired come back to making formal script mode later
#     just do functional mode for now
#     """
from typing import List, Optional
from argparse import ArgumentParser, Namespace
from masto_tools.init import log_in
from warnings import warn

from mastodon import Mastodon

def get_accounts(client) -> List[dict]:
    # get first page of accounts
    first_accounts = client.admin_accounts()
    # then paginate
    accounts = client.fetch_remaining(first_accounts)
    return accounts

def filter_accounts(accounts:List[dict]) -> List[dict]:
    # filter by approval and suspensions status
    accounts = [a for a in accounts if \
                a['approved'] is True and \
                a['disabled'] is False and \
                a['suspended'] is False and \
                a['silenced'] is False]

    return accounts

def send_dm(client:Mastodon, message:str, account:dict) -> dict:
    # prepend @ to message
    message = f"@{account['username']}\n" + message
    response = client.status_post(message, visibility='direct')
    return response


def send_to_all(client:Mastodon, message:str, accounts:List[dict], message_id:str, message_tag:str="DMAnnouncement") -> List[dict]:
    """
    Send message to all, returning dictionary of account IDs and

    :param message: The message to send
    :param accounts: List of accounts to send to
    :param message_id: Unique ID to label messages with (to be able to find this specific message later
    :param message_tag: General tag to label all mass DMs with
    :return:
    """
    # Postpend the tags to the message
    message += "\n\n" + f"#{message_tag.replace(' ', '_')} " + f"#{message_id.replace(' ', '_')}"

    responses = []
    for account in accounts:
        try:
            response = send_dm(client, message, account)
            responses.append(response)
        except Exception as e:
            # TODO: Logging and error handling lmao
            warn(f"Could not send dm to username: {account['username']}: {str(e)}")

    return responses

def argparser() -> ArgumentParser:
    parser = ArgumentParser()
    parser.add_argument('-m', '--message', help="The Message to send", type=str)
    parser.add_argument('-i', '--id', help="Unique message ID")
    parser.add_argument('-t', '--tag', help="Tag for all mass DMs", type=str, default="DMAnnouncement")

    return parser


def main() -> List[dict]:
    args = argparser().parse_args()

    client = log_in()
    accounts = get_accounts(client)
    accounts = filter_accounts(accounts)
    responses = send_to_all(client, args.message, accounts, message_id=args.id, message_tag=args.tag)
    return responses


if __name__ == "__main__":
    responses = main()