import getpass
from mastodon import Mastodon
from mastodon.Mastodon import MastodonMalformedEventError, MastodonNetworkError, MastodonReadTimeout, MastodonAPIError, MastodonIllegalArgumentError
import fileinput,re
import os
import sys

def create_dir():
    if not os.path.exists('secrets'):
        os.makedirs('secrets')

def create_file():
    if not os.path.exists('secrets/secrets.txt'):
        with open('secrets/secrets.txt', 'w'): pass
        print(secrets_filepath + " created!")

def create_config():
    if not os.path.exists('config'):
        os.makedirs('config')
    if not os.path.exists(config_filepath):
        print(config_filepath + " created!")
        with open('config/config.txt', 'w'): pass

def create_lang_config():
    if not os.path.exists(lang_config_filepath):
        print(lang_config_filepath + " created!")
        with open('config/lang_config.txt', 'w'): pass

def write_params():
    with open(secrets_filepath, 'a') as the_file:
        print("Writing secrets parameter names to " + secrets_filepath)
        the_file.write('uc_client_id: \n'+'uc_client_secret: \n'+'uc_access_token: \n')

def write_config():
    with open(config_filepath, 'a') as the_file:
        print("Writing parameters names 'mastodon_hostname', 'bot_username', 'gitea_hostname' and 'gitea_access_token' to " + config_filepath)
        the_file.write('mastodon_hostname: \n' + 'bot_username: \n' + 'gitea_hostname: \n' + 'gitea_access_token: \n')

def read_client_lines(self):
    client_path = 'app_clientcred.txt'
    with open(client_path) as fp:
        line = fp.readline()
        cnt = 1
        while line:
            if cnt == 1:
                print("Writing client id to " + secrets_filepath)
                modify_file(secrets_filepath, "uc_client_id: ", value=line.rstrip())
            elif cnt == 2:
                print("Writing client secret to " + secrets_filepath)
                modify_file(secrets_filepath, "uc_client_secret: ", value=line.rstrip())
            line = fp.readline()
            cnt += 1

def read_token_line(self):
    token_path = 'app_usercred.txt'
    with open(token_path) as fp:
        line = fp.readline()
        print("Writing access token to " + secrets_filepath)
        modify_file(secrets_filepath, "uc_access_token: ", value=line.rstrip())

def read_config_line():
    with open(config_filepath) as fp:
        line = fp.readline()
        modify_file(config_filepath, "mastodon_hostname: ", value=hostname)
        modify_file(config_filepath, "bot_username: ", value=bot_username)
        modify_file(config_filepath, "gitea_hostname: ", value=gitea_hostname)
        modify_file(config_filepath, "gitea_access_token: ", value=gitea_access_token)

def log_in():
    error = 0
    try:
        global hostname, bot_username, gitea_hostname, gitea_access_token
        hostname = input("Enter Mastodon hostname: ")
        user = input("User name, ex. user@" + hostname +"? ")
        user_password = getpass.getpass("User password? ")
        bot_username = input("Bot's username, ex. welcome: ")
        app_name = input("This app name? ")
        gitea_hostname = input("Enter Gitea hostname: ")
        gitea_access_token = input(f"{gitea_hostname} access token: ")

        Mastodon.create_app(app_name, scopes=["read","write"],
            to_file="app_clientcred.txt", api_base_url=hostname)
        mastodon = Mastodon(client_id = "app_clientcred.txt", api_base_url = hostname)
        mastodon.log_in(
            user,
            user_password,
            scopes = ["read", "write"],
            to_file = "app_usercred.txt"
        )
    except MastodonIllegalArgumentError as i_error:
        error = 1
        if os.path.exists("secrets/secrets.txt"):
            print("Removing secrets/secrets.txt file..")
            os.remove("secrets/secrets.txt")
        if os.path.exists("app_clientcred.txt"):
            print("Removing app_clientcred.txt file..")
            os.remove("app_clientcred.txt")
        sys.exit(i_error)
    except MastodonNetworkError as n_error:
        error = 1
        if os.path.exists("secrets/secrets.txt"):
            print("Removing secrets/secrets.txt file..")
            os.remove("secrets/secrets.txt")
        if os.path.exists("app_clientcred.txt"):
            print("Removing app_clientcred.txt file..")
            os.remove("app_clientcred.txt")
        sys.exit(n_error)
    except MastodonReadTimeout as r_error:
        error = 1
        if os.path.exists("secrets/secrets.txt"):
            print("Removing secrets/secrets.txt file..")
            os.remove("secrets/secrets.txt")
        if os.path.exists("app_clientcred.txt"):
            print("Removing app_clientcred.txt file..")
            os.remove("app_clientcred.txt")
        sys.exit(r_error)
    except MastodonAPIError as a_error:
        error = 1
        if os.path.exists("secrets/secrets.txt"):
            print("Removing secrets/secrets.txt file..")
            os.remove("secrets/secrets.txt")
        if os.path.exists("app_clientcred.txt"):
            print("Removing app_clientcred.txt file..")
            os.remove("app_clientcred.txt")
        sys.exit(a_error)
    finally:
        if error == 0:

            create_dir()
            create_file()
            write_params()
            client_path = 'app_clientcred.txt'
            read_client_lines(client_path)
            token_path = 'app_usercred.txt'
            read_token_line(token_path)
            if os.path.exists("app_clientcred.txt"):
                print("Removing app_clientcred.txt temp file..")
                os.remove("app_clientcred.txt")
            if os.path.exists("app_usercred.txt"):
                print("Removing app_usercred.txt temp file..")
                os.remove("app_usercred.txt")
            print("Secrets setup done!\n")

def modify_file(file_name,pattern,value=""):
    fh=fileinput.input(file_name,inplace=True)
    for line in fh:
        replacement=pattern + value
        line=re.sub(pattern,replacement,line)
        sys.stdout.write(line)
    fh.close()

# Returns the parameter from the specified file
def get_parameter( parameter, file_path ):
    # Check if secrets file exists
    if not os.path.isfile(file_path):
        print("File %s not found, creating it."%file_path)
        log_in()

    # Find parameter in file
    with open( file_path ) as f:
        for line in f:
            if line.startswith( parameter ):
                return line.replace(parameter + ":", "").strip()

    # Cannot find parameter, exit
    print(file_path + " Missing parameter %s "%parameter)
    sys.exit(0)

# Returns the parameter from the specified file
def get_hostname( parameter, config_filepath ):
    # Check if secrets file exists
    if not os.path.isfile(config_filepath):
        print("File %s not found, creating it."%config_filepath)
        create_config()

    # Find parameter in file
    with open( config_filepath ) as f:
        for line in f:
            if line.startswith( parameter ):
                return line.replace(parameter + ":", "").strip()

    # Cannot find parameter, exit
    print(config_filepath + "  Missing parameter %s "%parameter)
    write_config()
    read_config_line()
    print("setup done!")
    sys.exit(0)

###############################################################################
# main

if __name__ == '__main__':

    # Load secrets from secrets file
    secrets_filepath = "secrets/secrets.txt"
    uc_client_id     = get_parameter("uc_client_id",     secrets_filepath)
    uc_client_secret = get_parameter("uc_client_secret", secrets_filepath)
    uc_access_token  = get_parameter("uc_access_token",  secrets_filepath)

    # Load configuration from config file
    config_filepath = "config/config.txt"
    mastodon_hostname = get_hostname("mastodon_hostname", config_filepath)
    bot_username = get_parameter("bot_username", config_filepath)