import requests 
import sys

import os.path as op
from mastodon import Mastodon

# --------------------------------------------------

def main():

    mastodon = Mastodon(
        access_token = 'token.dat',
        api_base_url = 'https://udongein.xyz/'
    )

    with open('tags.dat', 'r') as dat:
        tags = dat.readlines()

    URL = "https://danbooru.donmai.us/posts.json"
    LIMIT = 10
    MIN_SCORE = 25
    SAFETY = 's'
    TAGS_POST = tags[0].strip()
    TAGS_FORBID = tags[1].strip().split()
    TAGS_SENSITIVE = tags[2].strip().split()

    PARAMS = { 'tags': TAGS_POST,
               'limit': LIMIT,
               'random': True } 

    print('[start] Settings:')
    print('LIMIT = ' + str(LIMIT) + ' | MIN_SCORE = ' + str(MIN_SCORE) + ' | SAFETY = ' + SAFETY)
    print('TAGS_POST=' + str(TAGS_POST))
    print('TAGS_FORBID=' + str(TAGS_FORBID))
    print('TAGS_SENSITIVE=' + str(TAGS_SENSITIVE) + '\n')

# --------------------------------------------------

    counter = 1
    b_search = True
    while b_search:
        r = requests.get(url = URL, params = PARAMS)
        print('[get] Attempt N' + str(counter) + '.')
        data = r.json()
        for i in range(0, LIMIT):
            fileurl = data[i]['file_url']
            print('url ', fileurl)
            fileid = data[i]['id']
            print('id ', fileid)
            filescore = data[i]['fav_count']
            print('score ', filescore)
            filesafe = data[i]['rating']
            print('rating ', filesafe)
            filetagstring = data[i]['tag_string']
            print('tags ', filetagstring)
            pulledtags = filetagstring.split()

            if (filesafe == SAFETY and filescore >= MIN_SCORE
                and not set(pulledtags).intersection(TAGS_FORBID)):
                print('[success] Found!')
                b_search = False
                break

# --------------------------------------------------

    fformat = op.splitext(fileurl)[1][1:]
    if (fformat == 'jpg'):
        fformat = 'jpeg'

    media = mastodon.media_post(requests.get(fileurl).content, f'image/{fformat}')
    toot  = f'This is just a test post nevermind thisjksajskkj.... https://danbooru.donmai.us/posts/{fileid}'

    b_sensetive = bool(set(pulledtags).intersection(TAGS_SENSITIVE))

    if (b_sensetive):
        print('[success] Marked as sensitive.')

    mastodon.status_post(toot, media_ids=[media], visibility='unlisted', sensitive=b_sensetive)
    print('[success] Posted!\n----------------------------------\n')

if __name__ == '__main__':
    sys.exit(main())