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


from mastodon import Mastodon
import os
import time
import sys
import argparse
from configparser import ConfigParser
import ytscraper

__prog_name__ = 'tootytvideo'
__version__ = '0.1'
__description__ = 'Toot a youtube video on 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('-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)

    # Overwrite default settings
    if conf.has_option('toot', 'status'):
        toot_status = conf['toot']['status'] + '\n'
    else:
        toot_status = ''

    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 = False

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

    # Log into Mastodon if enabled in settings
    print('[ OK ] {} Login'.format(time.strftime('%H:%M')))
    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 {} as @{}'.format(
            time.strftime('%H:%M'), mastodon_hostname, masto_username))
    except BaseException as e:
        print('[ERROR] Error while logging into Mastodon:', str(e))
        sys.exit(0)
    # Run
    resp = True

    while resp:
        print('+- New toot -+')
        yturl = input('Youtube URL : ')
        comment = input('Add a comment : ')
        hashtag = input('Add hastag : ')

        # Prepare the toot
        if conf.has_option('toot', 'status'):
            toot_status = conf['toot']['status'] + '\n'
        else:
            toot_status = ''
        # video commentary
        toot_status += comment
        toot_status += '\n'
        # video title
        video = ytscraper.scrape_url(yturl)
        toot_status += video.title
        toot_status += '\n'
        toot_status += video.yturl
        # add hashtag
        toot_status += '\n'
        toot_status += conf['toot']['hashtag']
        toot_status += ' '
        toot_status += hashtag

        # Prepare media IDs
        try:
            mfile = ytscraper.download_image(
                    video.thumbnail,
                    video.videoid,
                    '.'
                    )
            image_byte = open(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
        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 : {}'.format(
                time.strftime('%H:%M'), status['url']))
        except BaseException as e:
            print('[ERROR] Error while posting toot:' + str(e))
            sys.exit(0)

        # make a pause
        r = input('Continue (y/n)? ')
        if r.lower() == 'y':
            resp = True
        else:
            resp = False

    # The End
    print('[ OK ] {} The show is over !'.format(time.strftime('%H:%M')))