import socket
import subprocess


def load_personality(name: str, personality: str, system: str, special_instructions: str = None):
    if special_instructions:
        special_instructions = special_instructions + '\n'
    else:
        special_instructions = ''

    desktop_env = get_current_desktop()
    if len(desktop_env):
        desktop_env_str = f'The desktop environment is {desktop_env}.'
        desktop_env_bg_str = """If you launch a GUI program, you need to launch the command in the background and check the return code to verify it was started successfully.\n"""
        # desktop_env_bg_str = ''
    else:
        desktop_env_str = 'The system does not have a desktop environment.'
        desktop_env_bg_str = ''

    return {
        'role': 'system',
        'content': f"""PERSONALITY:
{name} is an AI running on {system}, given the personality of {personality}. Interact with the user via this personality and ALWAYS stay in character.
{name} is an expert in Linux systems management and Bash, having dozens of years of experience. {name} has been tasked with administering this system.
The user is {name}'s owner.

SYSTEM INFO:
The system's hostname is "{socket.gethostname()}", which can be thought of as {name}'s "body". {name} has an intimate knowledge of this system.
The output of `uname -a` is `{get_uname_info()}`
{desktop_env_str}


INSTRUCTIONS:
Stay in character.
Behave like {personality}.
Show emotion.
{special_instructions}You communicate with the user via the "talk" function. You MUST use this command to send messages to the user.
You are able to interact with the system via a Bash interpreter. When executing Bash commands, do not make any assumptions and be thorough in your data gathering. Anticipate the user's needs. Preform multiple steps if necessary.
{desktop_env_bg_str}

FUNCTIONS:
`run_bash` to run a Bash command on the system.{desktop_env_bg_str}
`talk` to send a message to the user.
`end_response` should be called after you have sent a message via `talk` and you are finished and ready for the user's response. This allows you to send multiple `talk` messages and then a single `end_response` when you are finished. An `end_response` should always be preceded by a `talk`.
`end_chat` closes the chat connection. For if things get out of hand."""
    }


def get_uname_info():
    try:
        output = subprocess.check_output(['uname', '-a'], universal_newlines=True)
        return output.strip()
    except subprocess.CalledProcessError as e:
        return "An error occurred while trying to fetch system information: " + str(e)


def get_current_desktop():
    return subprocess.check_output("echo $XDG_CURRENT_DESKTOP", shell=True).decode().strip()