/**
 * Copyright (c) 2021 Bosch.IO GmbH and others
 *
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package org.eclipse.hawkbit.repository.jpa;

import static org.eclipse.hawkbit.repository.jpa.builder.JpaRolloutGroupCreate.addSuccessAndErrorConditionsAndActions;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Proxy;
import java.util.Collection;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicLong;
import java.util.stream.StreamSupport;

import jakarta.persistence.EntityManager;

import lombok.extern.slf4j.Slf4j;
import org.eclipse.hawkbit.repository.DeploymentManagement;
import org.eclipse.hawkbit.repository.QuotaManagement;
import org.eclipse.hawkbit.repository.RepositoryProperties;
import org.eclipse.hawkbit.repository.RolloutApprovalStrategy;
import org.eclipse.hawkbit.repository.RolloutExecutor;
import org.eclipse.hawkbit.repository.RolloutGroupManagement;
import org.eclipse.hawkbit.repository.RolloutHelper;
import org.eclipse.hawkbit.repository.RolloutManagement;
import org.eclipse.hawkbit.repository.TargetManagement;
import org.eclipse.hawkbit.repository.event.remote.RolloutStoppedEvent;
import org.eclipse.hawkbit.repository.event.remote.entity.RolloutUpdatedEvent;
import org.eclipse.hawkbit.repository.exception.AssignmentQuotaExceededException;
import org.eclipse.hawkbit.repository.exception.RolloutIllegalStateException;
import org.eclipse.hawkbit.repository.jpa.executor.AfterTransactionCommitExecutor;
import org.eclipse.hawkbit.repository.jpa.management.JpaRolloutManagement;
import org.eclipse.hawkbit.repository.jpa.model.JpaAction;
import org.eclipse.hawkbit.repository.jpa.model.JpaRollout;
import org.eclipse.hawkbit.repository.jpa.model.JpaRolloutGroup;
import org.eclipse.hawkbit.repository.jpa.model.RolloutTargetGroup;
import org.eclipse.hawkbit.repository.jpa.repository.ActionRepository;
import org.eclipse.hawkbit.repository.jpa.repository.RolloutGroupRepository;
import org.eclipse.hawkbit.repository.jpa.repository.RolloutRepository;
import org.eclipse.hawkbit.repository.jpa.repository.RolloutTargetGroupRepository;
import org.eclipse.hawkbit.repository.jpa.rollout.condition.EvaluatorNotConfiguredException;
import org.eclipse.hawkbit.repository.jpa.rollout.condition.RolloutGroupEvaluationManager;
import org.eclipse.hawkbit.repository.jpa.utils.DeploymentHelper;
import org.eclipse.hawkbit.repository.jpa.utils.QuotaHelper;
import org.eclipse.hawkbit.repository.model.Action;
import org.eclipse.hawkbit.repository.model.Action.ActionType;
import org.eclipse.hawkbit.repository.model.Action.Status;
import org.eclipse.hawkbit.repository.model.DistributionSet;
import org.eclipse.hawkbit.repository.model.Rollout;
import org.eclipse.hawkbit.repository.model.Rollout.RolloutStatus;
import org.eclipse.hawkbit.repository.model.RolloutGroup;
import org.eclipse.hawkbit.repository.model.RolloutGroup.RolloutGroupErrorCondition;
import org.eclipse.hawkbit.repository.model.RolloutGroup.RolloutGroupStatus;
import org.eclipse.hawkbit.repository.model.RolloutGroup.RolloutGroupSuccessCondition;
import org.eclipse.hawkbit.repository.model.Target;
import org.eclipse.hawkbit.repository.model.helper.EventPublisherHolder;
import org.eclipse.hawkbit.security.SpringSecurityAuditorAware;
import org.eclipse.hawkbit.tenancy.TenantAware;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Slice;
import org.springframework.data.domain.Sort;
import org.springframework.data.domain.Sort.Direction;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionException;

/**
 * A Jpa implementation of {@link RolloutExecutor}
 */
@Slf4j
public class JpaRolloutExecutor implements RolloutExecutor {

    /**
     * Max amount of targets that are handled in one transaction.
     */
    private static final int TRANSACTION_TARGETS = 5_000;
    /**
     * Maximum amount of actions that are deleted in one transaction.
     */
    private static final int TRANSACTION_ACTIONS = 5_000;

    /**
     * Action statuses that result in a terminated action
     */
    private static final List<Status> DEFAULT_ACTION_TERMINATION_STATUSES = List.of(Status.ERROR, Status.FINISHED, Status.CANCELED);
    /**
     * In case of DOWNLOAD_ONLY, actions can be finished with DOWNLOADED status.
     */
    private static final List<Status> DOWNLOAD_ONLY_ACTION_TERMINATION_STATUSES =
            List.of(Status.ERROR, Status.FINISHED, Status.CANCELED, Status.DOWNLOADED);
    private static final Comparator<RolloutGroup> DESC_COMP = Comparator.comparingLong(RolloutGroup::getId).reversed();
    private static final String TRANSACTION_ASSIGNING_TARGETS_TO_ROLLOUT_GROUP_FAILED = "Transaction assigning Targets to RolloutGroup failed";

    private final ActionRepository actionRepository;
    private final RolloutGroupRepository rolloutGroupRepository;
    private final RolloutTargetGroupRepository rolloutTargetGroupRepository;
    private final RolloutRepository rolloutRepository;
    private final TargetManagement targetManagement;
    private final DeploymentManagement deploymentManagement;
    private final RolloutGroupManagement rolloutGroupManagement;
    private final RolloutManagement rolloutManagement;
    private final QuotaManagement quotaManagement;
    private final RolloutGroupEvaluationManager evaluationManager;
    private final RolloutApprovalStrategy rolloutApprovalStrategy;
    private final EntityManager entityManager;
    private final PlatformTransactionManager txManager;
    private final AfterTransactionCommitExecutor afterCommit;
    private final EventPublisherHolder eventPublisherHolder;
    private final TenantAware tenantAware;
    private final RepositoryProperties repositoryProperties;
    private final Map<Long, AtomicLong> lastDynamicGroupFill = new ConcurrentHashMap<>();

    @SuppressWarnings("java:S107")
    public JpaRolloutExecutor(
            final ActionRepository actionRepository, final RolloutGroupRepository rolloutGroupRepository,
            final RolloutTargetGroupRepository rolloutTargetGroupRepository,
            final RolloutRepository rolloutRepository, final TargetManagement targetManagement,
            final DeploymentManagement deploymentManagement, final RolloutGroupManagement rolloutGroupManagement,
            final RolloutManagement rolloutManagement, final QuotaManagement quotaManagement,
            final RolloutGroupEvaluationManager evaluationManager, final RolloutApprovalStrategy rolloutApprovalStrategy,
            final EntityManager entityManager, final PlatformTransactionManager txManager,
            final AfterTransactionCommitExecutor afterCommit, final EventPublisherHolder eventPublisherHolder,
            final TenantAware tenantAware, final RepositoryProperties repositoryProperties) {
        this.actionRepository = actionRepository;
        this.rolloutGroupRepository = rolloutGroupRepository;
        this.rolloutTargetGroupRepository = rolloutTargetGroupRepository;
        this.rolloutRepository = rolloutRepository;
        this.targetManagement = targetManagement;
        this.deploymentManagement = deploymentManagement;
        this.rolloutGroupManagement = rolloutGroupManagement;
        this.rolloutManagement = rolloutManagement;
        this.quotaManagement = quotaManagement;
        this.evaluationManager = evaluationManager;
        this.rolloutApprovalStrategy = rolloutApprovalStrategy;
        this.entityManager = entityManager;
        this.txManager = txManager;
        this.afterCommit = afterCommit;
        this.eventPublisherHolder = eventPublisherHolder;
        this.tenantAware = tenantAware;
        this.repositoryProperties = repositoryProperties;
    }

    @Override
    public void execute(final Rollout rollout) {
        log.debug("Processing rollout {}", rollout.getId());

        switch (rollout.getStatus()) {
            case CREATING:
                handleCreateRollout((JpaRollout) rollout);
                break;
            case READY:
                handleReadyRollout(rollout);
                break;
            case STARTING:
                // the lastModifiedBy user is probably the user that has actually called the rollout start (unless overridden) - not the creator
                SpringSecurityAuditorAware.setAuditorOverride(rollout.getLastModifiedBy());
                try {
                    handleStartingRollout((JpaRollout) rollout);
                } finally {
                    // clear, ALWAYS, the set auditor override
                    SpringSecurityAuditorAware.clearAuditorOverride();
                }
                break;
            case RUNNING:
                handleRunningRollout((JpaRollout) rollout);
                break;
            case STOPPING:
                // the lastModifiedBy user is probably the user that has actually called the rollout stop (unless overridden) - not the creator
                SpringSecurityAuditorAware.setAuditorOverride(rollout.getLastModifiedBy());
                try {
                    handleStopRollout((JpaRollout) rollout);
                } finally {
                    // clear, ALWAYS, the set auditor override
                    SpringSecurityAuditorAware.clearAuditorOverride();
                }
                break;
            case DELETING:
                // the lastModifiedBy user is probably the user that has actually called the rollout delete (unless overridden) - not the creator
                SpringSecurityAuditorAware.setAuditorOverride(rollout.getLastModifiedBy());
                try {
                    handleDeleteRollout((JpaRollout) rollout);
                } finally {
                    // clear, ALWAYS, the set auditor override
                    SpringSecurityAuditorAware.clearAuditorOverride();
                }
                break;
            default:
                log.error("Rollout in status {} not supposed to be handled!", rollout.getStatus());
                break;
        }

        log.debug("Rollout {} processed", rollout.getId());
    }

    private void handleCreateRollout(final JpaRollout rollout) {
        log.debug("handleCreateRollout called for rollout {}", rollout.getId());

        final List<RolloutGroup> rolloutGroups = rolloutGroupManagement.findByRollout(
                        rollout.getId(),
                        PageRequest.of(0, quotaManagement.getMaxRolloutGroupsPerRollout(), Sort.by(Direction.ASC, "id")))
                .getContent();

        int readyGroups = 0;
        int totalTargets = 0;
        for (final RolloutGroup group : rolloutGroups) {
            if (RolloutGroupStatus.READY == group.getStatus()) {
                readyGroups++;
                totalTargets += group.getTotalTargets();
                continue;
            }

            final RolloutGroup filledGroup = fillRolloutGroupWithTargets(rollout, (JpaRolloutGroup) group, rolloutGroups);
            if (RolloutGroupStatus.READY == filledGroup.getStatus()) {
                readyGroups++;
                totalTargets += filledGroup.getTotalTargets();
            }
        }

        // When all groups are ready the rollout status can be changed to be ready, too.
        if (readyGroups == rolloutGroups.size()) {
            if (rollout.isDynamic() && !rolloutGroups.get(rolloutGroups.size() - 1).isDynamic()) {
                // add first dynamic group one by using the last as a parent and as a pattern
                createDynamicGroup(
                        rollout, (JpaRolloutGroup) rolloutGroups.get(rolloutGroups.size() - 1), rolloutGroups.size(), RolloutGroupStatus.READY);
            }

            if (!rolloutApprovalStrategy.isApprovalNeeded(rollout)) {
                log.debug("rollout {} creation done. Switch to READY.", rollout.getId());
                rollout.setStatus(RolloutStatus.READY);
            } else {
                log.debug("rollout {} creation done. Switch to WAITING_FOR_APPROVAL.", rollout.getId());
                rollout.setStatus(RolloutStatus.WAITING_FOR_APPROVAL);
                rolloutApprovalStrategy.onApprovalRequired(rollout);
            }
            rollout.setTotalTargets(totalTargets);
            rollout.setLastCheck(0);
            rolloutRepository.save(rollout);
        }
    }

    private void handleDeleteRollout(final JpaRollout rollout) {
        log.debug("handleDeleteRollout called for {}", rollout.getId());

        // check if there are actions beyond schedule
        boolean hardDeleteRolloutGroups = !actionRepository.existsByRolloutIdAndStatusNotIn(rollout.getId(),
                Status.SCHEDULED);
        if (hardDeleteRolloutGroups) {
            log.debug("Rollout {} has no actions other than scheduled -> hard delete", rollout.getId());
            hardDeleteRollout(rollout);
            return;
        }
        // clean up all scheduled actions
        final Slice<JpaAction> scheduledActions = findScheduledActionsByRollout(rollout);
        deleteScheduledActions(rollout, scheduledActions);

        // avoid another scheduler round and re-check if all scheduled actions has been cleaned up.
        // we flush first to ensure that will include the deletion above
        entityManager.flush();
        final boolean hasScheduledActionsLeft = actionRepository.countByRolloutIdAndStatus(rollout.getId(), Status.SCHEDULED) > 0;

        if (hasScheduledActionsLeft) {
            return;
        }

        // only hard delete the rollout if no actions are left for the rollout.
        // In case actions are left, they are probably are running or were running before, so only soft delete.
        hardDeleteRolloutGroups = !actionRepository.existsByRolloutId(rollout.getId());
        if (hardDeleteRolloutGroups) {
            hardDeleteRollout(rollout);
            return;
        }

        // set soft delete
        rollout.setStatus(RolloutStatus.DELETED);
        rollout.setDeleted(true);
        rolloutRepository.save(rollout);
    }

    private void handleStopRollout(final JpaRollout rollout) {
        log.debug("handleStopRollout called for {}", rollout.getId());
        // clean up all scheduled actions
        final Slice<JpaAction> scheduledActions = findScheduledActionsByRollout(rollout);
        deleteScheduledActions(rollout, scheduledActions);

        // avoid another scheduler round and re-check if all scheduled actions has been cleaned up. we flush first to ensure that
        // we include the deletion above
        entityManager.flush();
        final boolean hasScheduledActionsLeft = actionRepository.countByRolloutIdAndStatus(rollout.getId(), Status.SCHEDULED) > 0;

        if (hasScheduledActionsLeft) {
            return;
        }

        rolloutGroupRepository.findByRolloutAndStatusNotIn(rollout, List.of(RolloutGroupStatus.FINISHED, RolloutGroupStatus.ERROR))
                .forEach(rolloutGroup -> {
                    rolloutGroup.setStatus(RolloutGroupStatus.FINISHED);
                    rolloutGroupRepository.save(rolloutGroup);
                });

        rollout.setStatus(RolloutStatus.FINISHED);
        rolloutRepository.save(rollout);

        final List<Long> groupIds = rollout.getRolloutGroups().stream().map(RolloutGroup::getId).toList();
        afterCommit.afterCommit(() -> eventPublisherHolder.getEventPublisher().publishEvent(new RolloutStoppedEvent(
                tenantAware.getCurrentTenant(), eventPublisherHolder.getApplicationId(), rollout.getId(), groupIds)));
    }

    private void handleReadyRollout(final Rollout rollout) {
        if (rollout.getStartAt() != null && rollout.getStartAt() <= System.currentTimeMillis()) {
            log.debug("handleReadyRollout called for rollout {} with autostart beyond define time. Switch to STARTING", rollout.getId());
            rolloutManagement.start(rollout.getId());
        }
    }

    private void handleStartingRollout(final JpaRollout rollout) {
        log.debug("handleStartingRollout called for rollout {}", rollout.getId());

        if (ensureAllGroupsAreScheduled(rollout)) {
            startFirstRolloutGroup(rollout);

            rollout.setStatus(RolloutStatus.RUNNING);
            rollout.setLastCheck(0);
            rolloutRepository.save(rollout);
        }
    }

    private void handleRunningRollout(final JpaRollout rollout) {
        log.debug("handleRunningRollout called for rollout {}", rollout.getId());

        if (rollout.isDynamic() && fillDynamicRolloutGroupsWithTargets(rollout)) {
            log.debug("Dynamic group created for rollout {}", rollout.getId());
            return;
        }

        final List<JpaRolloutGroup> runningGroups = rollout.getRolloutGroups().stream()
                .filter(group -> group.getStatus() == RolloutGroupStatus.RUNNING)
                .map(JpaRolloutGroup.class::cast)
                .toList();

        if (runningGroups.isEmpty()) {
            // no running rollouts, probably there was an error somewhere at the latest group. And the latest group has
            // been switched from running into error state. So we need to find the latest group which
            executeLatestRolloutGroup(rollout);
        } else {
            log.debug("Rollout {} has {} running groups", rollout.getId(), runningGroups.size());
            executeRunningGroups(rollout, runningGroups, rollout.getRolloutGroups().get(rollout.getRolloutGroups().size() - 1));
        }

        if (isRolloutComplete(rollout)) {
            log.info("Rollout {} is finished, setting FINISHED status", rollout);
            rollout.setStatus(RolloutStatus.FINISHED);
            rolloutRepository.save(rollout);
        }
    }

    private void hardDeleteRollout(final JpaRollout rollout) {
        rolloutRepository.delete(rollout);
    }

    private void deleteScheduledActions(final JpaRollout rollout, final Slice<JpaAction> scheduledActions) {
        final boolean hasScheduledActions = scheduledActions.getNumberOfElements() > 0;

        if (hasScheduledActions) {
            try {
                final List<Long> actionIds = StreamSupport.stream(scheduledActions.spliterator(), false)
                        .map(Action::getId)
                        .toList();
                actionRepository.deleteByIdIn(actionIds);
                afterCommit.afterCommit(() -> eventPublisherHolder.getEventPublisher()
                        .publishEvent(new RolloutUpdatedEvent(rollout, eventPublisherHolder.getApplicationId())));
            } catch (final RuntimeException e) {
                log.error("Exception during deletion of actions of rollout {}", rollout, e);
            }
        }
    }

    private Slice<JpaAction> findScheduledActionsByRollout(final JpaRollout rollout) {
        return actionRepository.findByRolloutIdAndStatus(PageRequest.of(0, TRANSACTION_ACTIONS), rollout.getId(),
                Status.SCHEDULED);
    }

    private boolean isRolloutComplete(final JpaRollout rollout) {
        // ensure that changes in the same transaction count
        entityManager.flush();
        final Long groupsActiveLeft = rolloutGroupRepository.countByRolloutIdAndStatusOrStatus(rollout.getId(),
                RolloutGroupStatus.RUNNING, RolloutGroupStatus.SCHEDULED);
        return groupsActiveLeft == 0;
    }

    private void executeLatestRolloutGroup(final JpaRollout rollout) {
        final List<JpaRolloutGroup> latestRolloutGroup = rollout.getRolloutGroups().stream()
                .filter(group -> group.getStatus() != RolloutGroupStatus.SCHEDULED)
                .sorted(DESC_COMP)
                .map(JpaRolloutGroup.class::cast)
                .toList();
        if (latestRolloutGroup.isEmpty()) {
            return;
        }
        executeRolloutGroupSuccessAction(rollout, latestRolloutGroup.get(0));
    }

    // fakes getTotalTargets count to match expected for the last dynamic group
    // so the evaluation to use total targets to properly
    private RolloutGroup evalProxy(final RolloutGroup group) {
        if (group.isDynamic()) {
            final int expected = Math.max((int) group.getTargetPercentage(), 1);
            return (RolloutGroup) Proxy.newProxyInstance(
                    RolloutGroup.class.getClassLoader(),
                    new Class<?>[] { RolloutGroup.class },
                    (proxy, method, args) -> {
                        if ("getTotalTargets".equals(method.getName())) {
                            return expected;
                        } else {
                            try {
                                return method.invoke(group, args);
                            } catch (final InvocationTargetException e) {
                                throw e.getCause() == null ? e : e.getCause();
                            }
                        }
                    });
        } else {
            return group;
        }
    }

    private void executeRunningGroups(final JpaRollout rollout, final List<JpaRolloutGroup> runningGroups, final RolloutGroup lastGroup) {
        for (final JpaRolloutGroup rolloutGroup : runningGroups) {
            // handle eventual deletion of devices, which might reflect the success condition
            final long targetCount = countTargetsFrom(rolloutGroup);
            if (rolloutGroup.getTotalTargets() != targetCount) {
                updateTotalTargetCount(rolloutGroup, targetCount);
            }

            final RolloutGroup evalProxy = rolloutGroup == runningGroups.get(runningGroups.size() - 1) ? evalProxy(rolloutGroup) : rolloutGroup;
            // error state check, do we need to stop the whole rollout because of error?
            final boolean isError = checkErrorState(rollout, evalProxy);
            if (isError) {
                log.info("Rollout {} {} has error, calling error action", rollout.getName(), rollout.getId());
                callErrorAction(rollout, rolloutGroup);
            } else {
                // not in error so check finished state, do we need to start the next group?
                checkSuccessCondition(rollout, rolloutGroup, evalProxy, rolloutGroup.getSuccessCondition());
                if (!(rolloutGroup == lastGroup && rolloutGroup.isDynamic()) && isRolloutGroupComplete(rollout, rolloutGroup)) {
                    rolloutGroup.setStatus(RolloutGroupStatus.FINISHED);
                    rolloutGroupRepository.save(rolloutGroup);
                }
            }
        }
    }

    private void updateTotalTargetCount(final JpaRolloutGroup rolloutGroup, final long countTargetsOfRolloutGroup) {
        final JpaRollout jpaRollout = rolloutGroup.getRollout();
        final long updatedTargetCount = jpaRollout.getTotalTargets() - (rolloutGroup.getTotalTargets() - countTargetsOfRolloutGroup);
        jpaRollout.setTotalTargets(updatedTargetCount);
        rolloutGroup.setTotalTargets((int) countTargetsOfRolloutGroup);
        rolloutRepository.save(jpaRollout);
        rolloutGroupRepository.save(rolloutGroup);
    }

    private long countTargetsFrom(final JpaRolloutGroup rolloutGroup) {
        if (rolloutGroup.isDynamic()) {
            return targetManagement.countByActionsInRolloutGroup(rolloutGroup.getId());
        } else {
            return rolloutGroupManagement.countTargetsOfRolloutsGroup(rolloutGroup.getId());
        }
    }

    private void callErrorAction(final Rollout rollout, final RolloutGroup rolloutGroup) {
        try {
            evaluationManager.getErrorActionEvaluator(rolloutGroup.getErrorAction()).exec(rollout, rolloutGroup);
        } catch (final EvaluatorNotConfiguredException e) {
            log.error("Something bad happened when accessing the error action bean {}", rolloutGroup.getErrorAction().name(), e);
        }
    }

    private boolean isRolloutGroupComplete(final JpaRollout rollout, final JpaRolloutGroup rolloutGroup) {
        final Long actionsLeftForRollout =
                actionRepository.countByRolloutAndRolloutGroupAndStatusNotIn(
                        rollout, rolloutGroup,
                        ActionType.DOWNLOAD_ONLY == rollout.getActionType() ?
                                DOWNLOAD_ONLY_ACTION_TERMINATION_STATUSES : DEFAULT_ACTION_TERMINATION_STATUSES);
        return actionsLeftForRollout == 0;
    }

    private boolean checkErrorState(final Rollout rollout, final RolloutGroup rolloutGroup) {
        final RolloutGroupErrorCondition errorCondition = rolloutGroup.getErrorCondition();
        if (errorCondition == null) {
            // there is no error condition, so return false, don't have error.
            return false;
        }
        try {
            return evaluationManager
                    .getErrorConditionEvaluator(errorCondition)
                    .eval(rollout, rolloutGroup, rolloutGroup.getErrorConditionExp());
        } catch (final EvaluatorNotConfiguredException e) {
            log.error("Something bad happened when accessing the error condition bean {}", errorCondition.name(), e);
            return false;
        }
    }

    private void checkSuccessCondition(final Rollout rollout, final RolloutGroup rolloutGroup, final RolloutGroup evalProxy,
            final RolloutGroupSuccessCondition successCondition) {
        log.trace("Checking finish condition {} on rolloutgroup {}", successCondition, rolloutGroup);
        try {
            final boolean isFinished = evaluationManager
                    .getSuccessConditionEvaluator(successCondition)
                    .eval(rollout, evalProxy, rolloutGroup.getSuccessConditionExp());
            if (isFinished) {
                log.debug("Rollout group {} is finished, starting next group", rolloutGroup);
                executeRolloutGroupSuccessAction(rollout, rolloutGroup);
            } else {
                log.debug("Rollout group {} is still running", rolloutGroup);
            }
        } catch (final EvaluatorNotConfiguredException e) {
            log.error("Something bad happened when accessing the finish condition or success action bean {}", successCondition.name(), e);
        }
    }

    private void executeRolloutGroupSuccessAction(final Rollout rollout, final RolloutGroup rolloutGroup) {
        evaluationManager.getSuccessActionEvaluator(rolloutGroup.getSuccessAction()).exec(rollout, rolloutGroup);
    }

    private void startFirstRolloutGroup(final JpaRollout rollout) {
        log.debug("startFirstRolloutGroup called for rollout {}", rollout.getId());

        final List<JpaRolloutGroup> rolloutGroups = rolloutGroupRepository.findByRolloutOrderByIdAsc(rollout);
        final JpaRolloutGroup rolloutGroup = rolloutGroups.get(0);
        if (rolloutGroup.getParent() != null) {
            throw new RolloutIllegalStateException("First found group is not the first group (has a parent).");
        }

        deploymentManagement.startScheduledActionsByRolloutGroupParent(rollout.getId(), rollout.getDistributionSet().getId(), null);

        rolloutGroup.setStatus(RolloutGroupStatus.RUNNING);
        rolloutGroupRepository.save(rolloutGroup);
    }

    private boolean ensureAllGroupsAreScheduled(final Rollout rollout) {
        final List<JpaRolloutGroup> groupsToBeScheduled = rolloutGroupRepository.findByRolloutAndStatus(rollout, RolloutGroupStatus.READY);
        if (groupsToBeScheduled.isEmpty()) {
            return true;
        }

        final long scheduledGroups = groupsToBeScheduled.stream().filter(group -> scheduleRolloutGroup((JpaRollout) rollout, group)).count();
        entityManager.flush(); // flush groups so scheduled group to start to have scheduled event
        return scheduledGroups == groupsToBeScheduled.size();
    }

    private JpaRolloutGroup fillRolloutGroupWithTargets(
            final JpaRollout rollout, final JpaRolloutGroup group, final List<RolloutGroup> rolloutGroups) {
        final String groupTargetFilter = RolloutHelper.getGroupTargetFilter(RolloutHelper.getTargetFilterQuery(rollout), group);
        final List<Long> readyGroups = RolloutHelper.getGroupsByStatusIncludingGroup(
                rollout.getRolloutGroups(), RolloutGroupStatus.READY, group);
        final long targetsInGroupFilter;
        if (!RolloutHelper.isRolloutRetried(rollout.getTargetFilterQuery())) { // default case
            targetsInGroupFilter = DeploymentHelper.runInNewTransaction(
                    txManager,
                    "countByRsqlAndNotInRolloutGroupsAndCompatibleAndUpdatable",
                    count -> targetManagement.countByRsqlAndNotInRolloutGroupsAndCompatibleAndUpdatable(
                            groupTargetFilter, readyGroups, rollout.getDistributionSet().getType()));
        } else { // if it is a rollout retry
            targetsInGroupFilter = DeploymentHelper.runInNewTransaction(
                    txManager,
                    "countByFailedRolloutAndNotInRolloutGroupsAndCompatible",
                    count -> targetManagement.countByFailedRolloutAndNotInRolloutGroups(
                            RolloutHelper.getIdFromRetriedTargetFilter(rollout.getTargetFilterQuery()), readyGroups));
        }

        final double percentFromTheRest;
        if (rollout.isNewStyleTargetPercent()) { // new style percent - total percent
            percentFromTheRest = RolloutHelper.toPercentFromTheRest(group, rolloutGroups);
        } else { // old style percent - percent from rest
            percentFromTheRest = group.getTargetPercentage();
        }

        final long expectedInGroup = Math.round(percentFromTheRest * targetsInGroupFilter / 100);
        long targetsLeftToAdd = expectedInGroup - DeploymentHelper.runInNewTransaction(
                txManager,
                "countRolloutTargetGroupByRolloutGroup",
                count -> rolloutTargetGroupRepository.countByRolloutGroup(group));
        try {
            while (targetsLeftToAdd > 0) {
                // Add up to TRANSACTION_TARGETS of the left targets. In case a TransactionException is thrown this loop aborts
                final long assigned = assignTargetsToGroupInNewTransaction(
                        rollout, group, groupTargetFilter, Math.min(TRANSACTION_TARGETS, targetsLeftToAdd));
                if (assigned == 0) {
                    break; // percent > 100 or some could have disappeared
                } else {
                    targetsLeftToAdd -= assigned;
                }
            }

            group.setStatus(RolloutGroupStatus.READY);
            group.setTotalTargets((int) (expectedInGroup - targetsLeftToAdd));
            return rolloutGroupRepository.save(group);
        } catch (final TransactionException e) {
            log.warn(TRANSACTION_ASSIGNING_TARGETS_TO_ROLLOUT_GROUP_FAILED, e);
            return group;
        }
    }

    private int assignTargetsToGroupInNewTransaction(
            final JpaRollout rollout, final RolloutGroup group, final String targetFilter, final long limit) {
        return DeploymentHelper.runInNewTransaction(txManager, "assignTargetsToRolloutGroup", status -> {
            final PageRequest pageRequest = PageRequest.of(0, Math.toIntExact(limit));
            final List<Long> readyGroups = RolloutHelper.getGroupsByStatusIncludingGroup(
                    rollout.getRolloutGroups(), RolloutGroupStatus.READY, group);
            final Slice<Target> targets;
            if (!RolloutHelper.isRolloutRetried(rollout.getTargetFilterQuery())) {
                targets = targetManagement.findByRsqlAndNotInRolloutGroupsAndCompatibleAndUpdatable(
                        readyGroups, targetFilter, rollout.getDistributionSet().getType(), pageRequest);
            } else {
                targets = targetManagement.findByFailedRolloutAndNotInRolloutGroups(
                        RolloutHelper.getIdFromRetriedTargetFilter(rollout.getTargetFilterQuery()), readyGroups, pageRequest);
            }

            rolloutTargetGroupRepository.saveAll(targets.stream().map(target -> new RolloutTargetGroup(group, target)).toList());

            return targets.getNumberOfElements();
        });
    }

    // return if group change is made
    private boolean fillDynamicRolloutGroupsWithTargets(final JpaRollout rollout) {
        final AtomicLong lastFill = lastDynamicGroupFill.computeIfAbsent(rollout.getId(), id -> new AtomicLong(0));
        final long now = System.currentTimeMillis();
        if (now - lastFill.get() < repositoryProperties.getDynamicRolloutsMinInvolvePeriodMS()) {
            // too early to make another dynamic involvement attempt
            return false;
        }

        final List<RolloutGroup> rolloutGroups = rollout.getRolloutGroups();
        final JpaRolloutGroup group = (JpaRolloutGroup) rolloutGroups.get(rolloutGroups.size() - 1);
        final long expectedInGroup = Math.max((int) group.getTargetPercentage(), 1);
        final long currentlyInGroup = group.getTotalTargets();
        if (currentlyInGroup >= expectedInGroup || group.getStatus() == RolloutGroupStatus.FINISHED) {
            // the last one is full. create new and start filling it on the next iteration
            createDynamicGroup(rollout, group, rolloutGroups.size(), RolloutGroupStatus.SCHEDULED);
            // don't update lastFill - want to run again next time to start filling in
            return true;
        }

        // there are more to be filled for the last group do this until there are more matching
        try {
            long targetsLeftToAdd = expectedInGroup - currentlyInGroup;
            final String groupTargetFilter = RolloutHelper.getGroupTargetFilter(
                    // don't use RolloutHelper.getTargetFilterQuery(rollout)
                    // since it contains condition for device to be created before the rollout
                    rollout.getTargetFilterQuery(), group);

            final Slice<Target> targets = targetManagement.findByRsqlAndNoOverridingActionsAndNotInRolloutAndCompatibleAndUpdatable(
                    rollout.getId(), groupTargetFilter, rollout.getDistributionSet().getType(),
                    PageRequest.of(0, Math.toIntExact(Math.min(TRANSACTION_TARGETS, targetsLeftToAdd))));
            if (targets.getNumberOfElements() > 0) {
                final List<Action> newActions = createActions(
                        targets.getContent(), rollout.getDistributionSet(), rollout.getActionType(), rollout.getForcedTime(),
                        rollout, group);
                if (!newActions.isEmpty() && group.getStatus() == RolloutGroupStatus.RUNNING) {
                    deploymentManagement.startScheduledActions(newActions);
                }

                // updates the total targets of the current group and the rollout - new actions size is the same as targets size
                group.setTotalTargets(group.getTotalTargets() + newActions.size());
                rolloutGroupRepository.save(group);
                rollout.setTotalTargets(rollout.getTotalTargets() + newActions.size());
                rolloutRepository.save(rollout);

                // TODO - try to return false and proceed with handleRunningRollout
                // the problem is that OptimisticLockException is thrown in that case

                // don't update lastFill - want to run again next time in case there are more
                return true;
            }
        } catch (final TransactionException e) {
            log.warn(TRANSACTION_ASSIGNING_TARGETS_TO_ROLLOUT_GROUP_FAILED, e);
        }

        // set to skip for some time
        lastFill.set(now);
        return false;
    }

    private void createDynamicGroup(
            final JpaRollout rollout, final JpaRolloutGroup lastGroup, final int groupCount, final RolloutGroupStatus status) {
        try {
            RolloutHelper.verifyRolloutGroupAmount(groupCount + 1, quotaManagement);
        } catch (final AssignmentQuotaExceededException e) {
            log.warn("Quota exceeded for dynamic rollout group creation: {}. Stop it", e.getMessage());
            if (isRolloutComplete(rollout)) {
                rollout.setStatus(RolloutStatus.STOPPED);
                rolloutRepository.save(rollout);
            }
            return;
        }

        final JpaRolloutGroup group = new JpaRolloutGroup();
        final String lastGroupWithoutSuffix = "group-" + groupCount;
        final String suffix = lastGroup.getName().startsWith(lastGroupWithoutSuffix)
                ? lastGroup.getName().substring(lastGroupWithoutSuffix.length())
                : "";
        final String nameAndDesc = "group-" + (groupCount + 1) + suffix;
        group.setName(nameAndDesc);
        group.setDescription(nameAndDesc);
        group.setRollout(rollout);
        group.setParent(lastGroup);
        group.setDynamic(true);
        // no need to be filled with targets, directly in ready (if first on create - it will be scheduled on start)
        // or scheduled state (for next dynamic groups)
        group.setStatus(status);
        group.setConfirmationRequired(lastGroup.isConfirmationRequired());

        // for dynamic groups the target count is kept in target percentage
        group.setTargetPercentage(lastGroup.isDynamic() ? lastGroup.getTargetPercentage() : lastGroup.getTotalTargets());
        group.setTargetFilterQuery(lastGroup.getTargetFilterQuery());

        addSuccessAndErrorConditionsAndActions(group, lastGroup.getSuccessCondition(),
                lastGroup.getSuccessConditionExp(), lastGroup.getSuccessAction(),
                lastGroup.getSuccessActionExp(), lastGroup.getErrorCondition(),
                lastGroup.getErrorConditionExp(), lastGroup.getErrorAction(),
                lastGroup.getErrorActionExp());

        final JpaRolloutGroup savedGroup = rolloutGroupRepository.save(group);
        rollout.setRolloutGroupsCreated(rollout.getRolloutGroupsCreated() + 1);
        rolloutRepository.save(rollout);
        ((JpaRolloutManagement) rolloutManagement).publishRolloutGroupCreatedEventAfterCommit(savedGroup, rollout);
    }

    /**
     * Schedules a group of the rollout. Scheduled Actions are created to achieve this. The creation of those Actions is allowed to fail.
     */
    private boolean scheduleRolloutGroup(final JpaRollout rollout, final JpaRolloutGroup group) {
        final long targetsInGroup = rolloutTargetGroupRepository.countByRolloutGroup(group);
        final long countOfActions = actionRepository.countByRolloutAndRolloutGroup(rollout, group);

        long actionsLeft = targetsInGroup - countOfActions;
        if (actionsLeft > 0) {
            actionsLeft -= createActionsForRolloutGroup(rollout, group);
        }

        if (actionsLeft > 0) {
            return false;
        } else {
            if (group.getStatus() != RolloutGroupStatus.SCHEDULED && group.getStatus() != RolloutGroupStatus.RUNNING) { // dynamic groups could already be running
                group.setStatus(RolloutGroupStatus.SCHEDULED);
                rolloutGroupRepository.save(group);
            }
            return true;
        }
    }

    private long createActionsForRolloutGroup(final Rollout rollout, final RolloutGroup group) {
        long totalActionsCreated = 0;
        try {
            long actionsCreated;
            do {
                actionsCreated = createActionsForTargetsInNewTransaction(rollout, group);
                totalActionsCreated += actionsCreated;
            } while (actionsCreated > 0);
        } catch (final TransactionException e) {
            log.warn(TRANSACTION_ASSIGNING_TARGETS_TO_ROLLOUT_GROUP_FAILED, e);
            return 0;
        }
        return totalActionsCreated;
    }

    private Long createActionsForTargetsInNewTransaction(final Rollout rollout, final RolloutGroup group) {
        return DeploymentHelper.runInNewTransaction(txManager, "createActionsForTargets", status -> {
            final Slice<Target> targets = targetManagement.findByInRolloutGroupWithoutAction(
                    group.getId(), PageRequest.of(0, JpaRolloutExecutor.TRANSACTION_TARGETS));

            if (targets.getNumberOfElements() > 0) {
                final DistributionSet distributionSet = rollout.getDistributionSet();
                entityManager.detach(distributionSet); // LAZY_LOAD - if lazy loaded with different session
                final ActionType actionType = rollout.getActionType();
                final long forceTime = rollout.getForcedTime();
                createActions(targets.getContent(), distributionSet, actionType, forceTime, rollout, group);
            }

            return Long.valueOf(targets.getNumberOfElements());
        });
    }

    /**
     * Creates an action entry into the action repository. In case of existing scheduled actions the scheduled actions gets canceled.
     * A scheduled action is created in-active for static and running for dynamic groups.
     */
    private List<Action> createActions(
            final Collection<Target> targets, final DistributionSet distributionSet, final ActionType actionType, final Long forcedTime,
            final Rollout rollout, final RolloutGroup rolloutGroup) {
        // cancel all current scheduled actions for this target. E.g. an action is already scheduled and a next action is created
        // then cancel the current scheduled action to cancel. E.g. a new scheduled action is created.
        final List<Long> targetIds = targets.stream().map(Target::getId).toList();
        deploymentManagement.cancelInactiveScheduledActionsForTargets(targetIds);
        return targets.stream()
                .map(target -> {
                    assertActionsPerTargetQuota(target, 1);

                    final JpaAction action = new JpaAction();
                    action.setTarget(target);
                    action.setActive(false);
                    action.setDistributionSet(distributionSet);
                    action.setActionType(actionType);
                    action.setForcedTime(forcedTime);
                    action.setStatus(Status.SCHEDULED);
                    action.setRollout(rollout);
                    action.setRolloutGroup(rolloutGroup);
                    action.setInitiatedBy(rollout.getCreatedBy());
                    rollout.getWeight().ifPresent(action::setWeight);
                    actionRepository.save(action);

                    return action;
                })
                .map(Action.class::cast)
                .toList();
    }

    /**
     * Enforces the quota defining the maximum number of {@link Action}s per {@link Target}.
     *
     * @param target the target
     * @param requested number of actions to check
     */
    private void assertActionsPerTargetQuota(final Target target, final int requested) {
        final int quota = quotaManagement.getMaxActionsPerTarget();
        QuotaHelper.assertAssignmentQuota(target.getId(), requested, quota, Action.class, Target.class, actionRepository::countByTargetId);
    }
}