import json
import sys
import os

from devchat.llm import chat_json

sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), ".."))

from common_util import ui_edit, assert_exit  # noqa: E402
from git_api import check_git_installed, create_and_checkout_branch, is_issue_url, read_issue_by_url  # noqa: E402


# Function to generate a random branch name
PROMPT = (
    "Give me 5 different git branch names, "
    "mainly hoping to express: {task}, "
    "Good branch name should looks like: <type>/<main content>-#<issue id>,"
    "<issue id> is optional, add it only when you know the issue id clearly, "
    "don't miss '#' before issue id. "
    "the final result is output in JSON format, "
    'as follows: {{"names":["name1", "name2", .. "name5"]}}\n'
)
@chat_json(prompt=PROMPT, model="gpt-4-1106-preview")
def generate_branch_name(task):
    pass


@ui_edit(ui_type="radio", description="Select a branch name")
def select_branch_name_ui(branch_names):
    pass

def select_branch_name(branch_names):
    [branch_selection] = select_branch_name_ui(branch_names)
    assert_exit(branch_selection is None, "No branch selected.", exit_code=0)
    return branch_names[branch_selection]


def get_issue_or_task(task):
    if is_issue_url(task):
        issue = read_issue_by_url(task.strip())
        assert_exit(not issue, "Failed to read issue.", exit_code=-1)
     
        return json.dumps({"id": issue["number"], "title": issue["title"], "body": issue["body"]})
    else:
        return task


# Main function
def main():
    print("Start create branch ...", end="\n\n", flush=True)
    
    is_git_installed = check_git_installed()
    assert_exit(not is_git_installed, "Git is not installed.", exit_code=-1)
    
    task = sys.argv[1]
    assert_exit(not task, "You need input something about the new branch, or input a issue url.", exit_code=-1)
    
    # read issue by url
    task = get_issue_or_task(task)

    # Generate 5 branch names
    print("Generating branch names ...", end="\n\n", flush=True)
    branch_names = generate_branch_name(task = task)
    assert_exit(not branch_names, "Failed to generate branch names.", exit_code=-1)
    branch_names = branch_names["names"]

    # Select branch name
    selected_branch = select_branch_name(branch_names)

    # create and checkout branch
    print(f"Creating and checking out branch: {selected_branch}")
    create_and_checkout_branch(selected_branch)
    print("Branch has create and checkout")


if __name__ == "__main__":
    main()