"""
All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
its licensors.

For complete copyright and license terms please see the LICENSE at the root of this
distribution (the "License"). All use of this software is governed by the License,
or, if provided, by the license below or the license accompanying this file. Do not
remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
"""

# Tests the Python API for Entity CRUD while the Editor is running

import azlmbr.bus as bus
import azlmbr.editor as editor
from azlmbr.entity import EntityId

createdEntityIds = []


def onEditorEntityCreated(parameters):
    global createdEntityIds

    entityId = parameters[0]
    createdEntityIds.append(entityId)


def onEditorEntityDeleted(parameters):
    global createdEntityIds

    deletedEntityId = parameters[0]
    for entityId in createdEntityIds:
        if (entityId.invoke("Equal", deletedEntityId)):
            createdEntityIds.remove(entityId)
            break


# Open a level (any level should work)
editor.EditorToolsApplicationRequestBus(bus.Broadcast, 'OpenLevelNoPrompt', 'WaterSample')

# Listen for notifications when entities are created/deleted
handler = bus.NotificationHandler('EditorEntityContextNotificationBus')
handler.connect()
handler.add_callback('OnEditorEntityCreated', onEditorEntityCreated)
handler.add_callback('OnEditorEntityDeleted', onEditorEntityDeleted)

# Create a new Entity at the root level
rootEntityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', EntityId())
if (rootEntityId):
    print("New entity with no parent created")
if (rootEntityId.invoke("Equal", createdEntityIds[0])):
    print("New root entity matches entity received from notification")

# Create a new Entity parented to the first Entity we created
firstChildEntityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', rootEntityId)
if (firstChildEntityId):
    print("New entity with valid parent created")
if (firstChildEntityId.invoke("Equal", createdEntityIds[1])):
    print("First child entity matches entity received from notification")

# Create another Entity parented to the first Entity we created
secondChildEntityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', rootEntityId)
if (secondChildEntityId):
    print("Another new entity with valid parent created")
if (secondChildEntityId.invoke("Equal", createdEntityIds[2])):
    print("Second child entity matches entity received from notification")

# Create two more entities and then delete them using the API that takes a list instead of a single entity
thirdChildEntityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', secondChildEntityId)
fourthChildEntityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', secondChildEntityId)
editor.ToolsApplicationRequestBus(bus.Broadcast, 'DeleteEntities', [thirdChildEntityId, fourthChildEntityId])

# Delete the second child Entity we created
editor.ToolsApplicationRequestBus(bus.Broadcast, 'DeleteEntityById', secondChildEntityId)

# Delete the root Entity we created and all its children (so this should also delete the firstChildEntityId)
editor.ToolsApplicationRequestBus(bus.Broadcast, 'DeleteEntityAndAllDescendants', rootEntityId)

# There should be no more entities left that we created
if (len(createdEntityIds) == 0):
    print("Deleted all entities we created")

# Stop listening for entity creation/deletion notifications
handler.disconnect()

# Create new Entity
entityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', EntityId())

# Get current name
oldName = editor.EditorEntityInfoRequestBus(bus.Event, 'GetName', entityId);

# Set a new name
editor.EditorEntityAPIBus(bus.Event, 'SetName', entityId, "TestName")

# Get new name
newName = editor.EditorEntityInfoRequestBus(bus.Event, 'GetName', entityId);

if not(oldName == newName):
    print("GetName and SetName work")

# Create new Entity
parentId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', EntityId())
 
# Create new Entity with parentId as parent
childId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', parentId)

getId = editor.EditorEntityInfoRequestBus(bus.Event, 'GetParent', childId);

if(getId.Equal(parentId)):
    print("GetParent works")

# Find the entity in the scene

import azlmbr.bus as bus
import azlmbr.entity

searchFilter = azlmbr.entity.SearchFilter()
searchFilter.names = ['TestName']

# Search by name
searchEntityIdList = azlmbr.entity.SearchBus(bus.Broadcast, 'SearchEntities', searchFilter)

for entityId in searchEntityIdList:
    entityName = editor.EditorEntityInfoRequestBus(bus.Event, 'GetName', entityId)
    if(entityName == 'TestName'):
        print("TestName entity found")

# Search by name path (DAG)
editor.EditorEntityAPIBus(bus.Event, 'SetName', parentId, "TestParent")
editor.EditorEntityAPIBus(bus.Event, 'SetName', childId, "TestChild")

searchFilter = azlmbr.entity.SearchFilter()
searchFilter.names = ['TestParent|TestChild']

searchEntityIdList = azlmbr.entity.SearchBus(bus.Broadcast, 'SearchEntities', searchFilter)

for entityId in searchEntityIdList:
    entityName = editor.EditorEntityInfoRequestBus(bus.Event, 'GetName', entityId)
    if(entityName == 'TestChild'):
        print("TestChild entity found")

# Search using wildcard
searchFilter = azlmbr.entity.SearchFilter()
searchFilter.names = ['Test*']

searchEntityIdList = azlmbr.entity.SearchBus(bus.Broadcast, 'SearchEntities', searchFilter)

if(len(searchEntityIdList) == 3):
    print("Test* 3 entities found")

# Close Editor
editor.EditorToolsApplicationRequestBus(bus.Broadcast, 'ExitNoPrompt')