"""
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 Entity Selection Python API while the Editor is running

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

def createTestEntities(count):
    testEntityIds = []
    
    for i in range(count):
        entityId = editor.ToolsApplicationRequestBus(bus.Broadcast, 'CreateNewEntity', EntityId())
        testEntityIds.append(entityId)
        print("Entity " + str(i)  + " created.")
        editor.EditorEntityAPIBus(bus.Event, 'SetName', entityId, "TestEntity")

    return testEntityIds

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


# Create new test entities at the root level
numTestEntities = 10
testEntityIds = createTestEntities(numTestEntities)

# Select all test entities
editor.ToolsApplicationRequestBus(bus.Broadcast, 'SetSelectedEntities', testEntityIds)

# Get all test entities selected and check if any entity selected
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
areAnyEntitiesSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'AreAnyEntitiesSelected')

if(len(testEntityIds) == len(selectedTestEntityIds) and areAnyEntitiesSelected): 
    print("SUCCESS: All Test Entities Selected")

# Mark first test entity deselected
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntityDeselected', testEntityIds[0])
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
isSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'IsSelected', testEntityIds[0])

if(len(testEntityIds) - 1 == len(selectedTestEntityIds) and not isSelected): 
    print("SUCCESS: One Test Entity Marked as DeSelected")

# Mark first test entity selected
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntitySelected', testEntityIds[0])
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
isSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'IsSelected', testEntityIds[0])

if(len(testEntityIds) == len(selectedTestEntityIds) and isSelected): 
    print("SUCCESS: One New Test Entity Marked as Selected")

# Mark first half of test entities as deselected
halfNumTestEntities = len(testEntityIds)//2

editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntitiesDeselected', testEntityIds[:halfNumTestEntities])
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')

if(len(testEntityIds) - halfNumTestEntities == len(selectedTestEntityIds)): 
    print("SUCCESS: Half of Test Entity Marked as DeSelected")

# Mark first half test entity as selected
editor.ToolsApplicationRequestBus(bus.Broadcast, 'MarkEntitiesSelected', testEntityIds[:halfNumTestEntities])
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')

if(len(testEntityIds) == len(selectedTestEntityIds)): 
    print("SUCCESS: All Test Entities Marked as Selected")

# Clear all test entities selected
editor.ToolsApplicationRequestBus(bus.Broadcast, 'SetSelectedEntities', [])
selectedTestEntityIds = editor.ToolsApplicationRequestBus(bus.Broadcast, 'GetSelectedEntities')
areAnyEntitiesSelected = editor.ToolsApplicationRequestBus(bus.Broadcast, 'AreAnyEntitiesSelected')

if(len(selectedTestEntityIds) == 0 and not areAnyEntitiesSelected): 
    print("SUCCESS: Clear All Test Entities Selected")


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