import time
import os
from api_hook_client import ApiHookClient
from simulation.user_agent import UserSimAgent

class WorkflowSimulator:
    def __init__(self, hook_client: ApiHookClient):
        self.client = hook_client
        self.user_agent = UserSimAgent(hook_client)

    def setup_new_project(self, name, git_dir):
        print(f"Setting up new project: {name}")
        self.client.click("btn_project_new")
        time.sleep(1)
        self.client.set_value("project_git_dir", git_dir)
        self.client.click("btn_project_save")
        time.sleep(1)

    def run_discussion_turn(self, user_message=None):
        if user_message is None:
            # Generate from AI history
            session = self.client.get_session()
            entries = session.get('session', {}).get('entries', [])
            user_message = self.user_agent.generate_response(entries)

        print(f"\n[USER]: {user_message}")
        self.client.set_value("ai_input", user_message)
        self.client.click("btn_gen_send")
        
        # Wait for AI
        return self.wait_for_ai_response()

    def wait_for_ai_response(self, timeout=60):
        print("Waiting for AI response...", end="", flush=True)
        start_time = time.time()
        last_count = len(self.client.get_session().get('session', {}).get('entries', []))
        
        while time.time() - start_time < timeout:
            time.sleep(1)
            print(".", end="", flush=True)
            entries = self.client.get_session().get('session', {}).get('entries', [])
            if len(entries) > last_count:
                last_entry = entries[-1]
                if last_entry.get('role') == 'AI' and last_entry.get('content'):
                    print(f"\n[AI]: {last_entry.get('content')[:100]}...")
                    return last_entry
            
        print("\nTimeout waiting for AI")
        return None