#!/usr/bin/python3
# coding: utf-8
# -*- coding: utf-8 -*-


from mastodon import Mastodon
import os
import random
import sys
import argparse
from configparser import ConfigParser

__prog_name__ = 'img2toot'
__version__ = '0.3'
__description__ = 'Toot bot for random local NSFW image for Mastodon'


if __name__ == "__main__":
    # Read arguments
    parser = argparse.ArgumentParser(prog=__prog_name__,
                                        description=__description__)
    parser.add_argument('--version', action='version', version=__version__)
    parser.add_argument('-d', '--dir', required=True, help="Image\'s directory")
    parser.add_argument('-c', '--config', default='config.ini',
                            help="Config file (default: %(default)s)")
    args = parser.parse_args()
    config_filepath = args.config

    # Check if config file exists
    if not os.path.isfile(config_filepath):
        print("Config file %s not found, exiting." % config_filepath)
        sys.exit(0)
    else:
        # Read config file
        conf = ConfigParser()
        conf.read(config_filepath)

    # Need a directory
    if not args.dir:
        print("Image\'s directory argument missing")
        sys.exit(0)
    image_dir = args.dir + "/"

    # Overwrite default settings
    if conf.has_option('toot', 'visibility'):
        toot_visibility = conf['toot']['visibility']
    else:
        toot_visibility = ''

    if conf.has_option('toot', 'sensitive'):
        toot_sensitive = conf['toot']['sensitive']
    else:
        toot_sensitive = True

    if conf.has_option('toot', 'spoiler'):
        toot_spoiler = conf['toot']['spoiler']
    else:
        toot_spoiler = None

    # Log into Mastodon if enabled in settings
    mastodon_hostname = conf['mastodon']['mastodon_hostname']
    try:
        mastodonAPI = Mastodon(
            client_id=conf['mastodon']['client_id'],
            client_secret=conf['mastodon']['client_secret'],
            access_token=conf['mastodon']['access_token'],
            api_base_url='https://' + mastodon_hostname)
        masto_username = mastodonAPI.account_verify_credentials()['username']
        print ('[ OK ] Sucessfully authenticated on ' + mastodon_hostname
                + ' as @' + masto_username)
    except BaseException as e:
        print ('[ERROR] Error while logging into Mastodon:', str(e))
        sys.exit(0)

    #Prepare media IDs
    try:
        mfile = random.choice(os.listdir(image_dir))
        image_byte = open(image_dir + mfile, "rb").read()

        if mfile[-3:] == "jpe":
            mime = "image/jpeg"
        elif mfile[-3:] == "jpg":
            mime = "image/jpeg"
        elif mfile[-3:] == "png":
            mime = "image/png"
        elif mfile[-3:] == "gif":
            mime = "image/gif"
        elif mfile[-3:] == "gifv":
            mime = "video/mp4"
        elif mfile[-3:] == "mp4":
            mime = "video/mp4"
        else:
            print("Incorrect media file format")

        media_dict = mastodonAPI.media_post(image_byte, mime)
    except BaseException as e:
        print ('[ERROR] Error while reading media file : ' + str(e))
        sys.exit(0)

    # Post the toot
    toot_status = conf['toot']['status']
    toot_status += '\n'
    toot_status += conf['toot']['hashtag']
    try:
        status = mastodonAPI.status_post(
                            status=toot_status,
                            in_reply_to_id=None,
                            media_ids=[media_dict],
                            sensitive=toot_sensitive,
                            visibility=toot_visibility,
                            spoiler_text=toot_spoiler
                            )
        print (
            '[ OK ] Posting this on Mastodon account with media attachment : '
            + status['url'])
    except BaseException as e:
        print ('[ERROR] Error while posting toot:' + str(e))