from rich import print
import os

uname: str = os.getlogin()
configPath = f'/home/{uname}/.config/rofi/'

def reconfigureRofi() -> None:
    """
    Create a Rofi theme if one is not present with desired defaults to work with InstantRice.
    Will pull source file from data dir
    """    
    print('[bold red]Initializing Rofi Theme')
    
    # copy the default config from data to .config
    dirExists = os.path.isdir(configPath)
    if dirExists:
        print('path exists')
        if os.path.exists(f'{configPath}config.rasi'):
            print('config present')
            with open(f'{configPath}config.rasi', 'r') as file:
                data = file.readlines()
            themeSet = False
            for i, line in enumerate(data):
                if '@theme' in line:
                    themeSet = True
                    data[i] = '@theme "theme.rasi"'
            if not themeSet:
                data.append('@theme "theme.rasi"')
            with open(f'{configPath}config.rasi', 'w') as file:
                file.writelines(data)
        else:
            # create the rasi config
            file = open(rf'{configPath}config.rasi', 'w')
            file.write('@theme "theme.rasi"')
            file.close()
    else:
        print('path doesnt exist')
    # add line to rofi config 
    # (this means we need might need to make the rofi config file, and the rofi directory)
    # (it also will be necessary to grab the current user's username to access their .config)
    print(uname)
    print(dirExists)

def dropRofiTheme():
    with open('../data/theme.rasi', 'r') as file:
        data = file.readlines()

    with open(f'{configPath}theme.rasi', 'w') as file:
        file.writelines(data)
    

reconfigureRofi()
dropRofiTheme()