from mastodon import Mastodon
import random
import os
from urllib import parse
from worker import conn
import psycopg2
from utils import first_time_setup, update_toots, markov_toot
from rq import Queue

# setting up the redis, mastodon and postgres connections
q = Queue(connection=conn)
mastodon = Mastodon(
    client_id=os.environ['client_id'],
    client_secret=os.environ['client_secret'],
    access_token=os.environ['access_token'],
    api_base_url=os.environ['instance']
)
parse.uses_netloc.append("postgres")
url = parse.urlparse(os.environ["DATABASE_URL"])
dbconn = psycopg2.connect(
    database=url.path[1:],
    user=url.username,
    password=url.password,
    host=url.hostname,
    port=url.port
)
cur = dbconn.cursor()


# this is a check to see if it's the first time a user's run the bot, so the script knows to do db setup
try:
    cur.execute("select value from config_data where id=%s", ("setup",))
except psycopg2.ProgrammingError:
    first_time = True
else:
    first_time = False


if first_time:
    q.enqueue(first_time_setup, timeout=1800)
else:
    cur.execute("select value from config_data where id=%s", ("userid",))
    userid = cur.fetchone()
    userid = userid[0]
    q.enqueue(update_toots, userid, timeout=900)
    if random.randint(1, 4) == 1:
        q.enqueue(markov_toot, timeout=900)
        print("generating toot")
    else:
        print("Not tooting this hour.")

dbconn.close()