import json
import requests
import os
import sys
import time
from os.path import expanduser
from AuthToken import AuthToken
import click
from rich import print
from rich.panel import Panel
from rich.text import Text
import pyperclip
try:
    import gnureadline as readline
except ImportError:
    import readline

# get a suggestion using github copilot
def get_suggestion(prompt, token):
    headers = {
        'Content-Type': 'application/json',
        'Openai-Organization': 'copilot-ghost',
        'OpenAI-Intent': 'github-copilot',
        'Authorization': f'Bearer {token}',
    }

    data = {
        'prompt': prompt,
        'stream': True,
        'temperature': 0.3,
        'max_tokens': 50,
        'top_p': 0.5,
        'n': 1,
        'logprobs': 2,
    }

    response = requests.post('https://copilot-proxy.githubusercontent.com/v1/engines/copilot-codex/completions', headers=headers, data=json.dumps(data))
    # get the first completion from the response
    data = response.text

    # split by \n
    data = data.split("\n")

    # filter where start with data: and remove it
    data = [x.replace("data: ", "") for x in data if x.startswith("data: ")]
    
    # filter [Done]
    data = [x for x in data if x != "[DONE]"]

    choices: dict(int, str) = {}
    for x in data:
        # x to json
        x = json.loads(x)
        c = x["choices"][0]
        
        if c["index"] not in choices:
            choices[c["index"]] = c["text"]
        else:
            choices[c["index"]] += c["text"]
    
    # get the first choice else return "Command not found"
    if 0 in choices:
        choice = choices[0]
        
        # remove unwanted characters from front (#, $, \n)
        while choice.startswith("#") or choice.startswith("$") or choice.startswith("\n") or choice.startswith(" "):
            choice = choice[1:]

        # strip \n
        choice = choice.strip()

        return choice.split("\n")[0]

    return "Command not found"

@click.group()
def main():
    pass

@main.command()
@click.argument('prompt', nargs=-1)
def generate_suggestion(prompt):
    prompt = " ".join(prompt)
    prompt = "!/bin/bash\n\n" + prompt + ":\n"

    authToken = AuthToken()
    token = authToken.get_github_api_token()

    suggestion = get_suggestion(prompt, token)
    panel = Panel(Text(suggestion, justify="center"), title_align="center", title="Execute? \[y/n/e/c]", expand=False)
    print(panel)
    # read a character from stdin without displaying it
    c = click.getchar()
    if c == "y":
        # execute the suggestion
        os.system(suggestion)
    elif c == "e":
        # prompt the suggestion by using redisplay
        def pre_input_hook():
            readline.insert_text(suggestion)
            readline.redisplay()


        readline.set_pre_input_hook(pre_input_hook)
        line = input()
        
        # execute line
        os.system(line)
        # remove pre input hook
        readline.set_pre_input_hook()

        # set history
        readline.clear_history()
        readline.add_history(line)
    elif c == "c":
        # copy the suggestion to clipboard
        pyperclip.copy(suggestion)
        print("Copied to clipboard")