from pytrace.client import TraceClient

def main():
    """
    An example script demonstrating how to use the TraceClient to register
    a custom node with the TraceStudio Server.
    """
    # Point the client to the server
    client = TraceClient(base_url="http://127.0.0.1:8000")
    
    # 1. Register our custom node with the server
    print("Registering custom 'StringConcatNode'...")
    try:
        registration_result = client.register_node_from_file(
            "example_nodes/string_nodes.py"
        )
        print("Registration successful:", registration_result)
    except Exception as e:
        print(f"Failed to register node: {e}")
        return

    # 2. Get the list of all available plugins from the server to verify
    #    that our new node is now listed.
    print("\nFetching all available plugins...")
    try:
        plugins = client.get_plugins()
        print(f"Successfully fetched {plugins['total']} plugins.")
        
        custom_node_type = registration_result.get("node_type")
        if custom_node_type in plugins.get("plugins", {}):
            print(f"SUCCESS: Found registered node '{custom_node_type}' in server's plugin list.")
        else:
            print(f"WARNING: Could not find '{custom_node_type}' in server's plugin list.")

    except Exception as e:
        print(f"Failed to get plugins: {e}")
        return

if __name__ == "__main__":
    main()