/*
 * SPDX-License-Identifier: Apache-2.0
 *
 * The OpenSearch Contributors require contributions made to
 * this file be licensed under the Apache-2.0 license or a
 * compatible open source license.
 *
 * Modifications Copyright OpenSearch Contributors. See
 * GitHub history for details.
 */

package org.opensearch.ad;

import static org.opensearch.ad.constant.ADCommonMessages.FAIL_TO_PARSE_DETECTOR_MSG;
import static org.opensearch.core.rest.RestStatus.BAD_REQUEST;
import static org.opensearch.core.rest.RestStatus.INTERNAL_SERVER_ERROR;
import static org.opensearch.core.xcontent.XContentParserUtils.ensureExpectedToken;
import static org.opensearch.timeseries.constant.CommonMessages.FAIL_TO_FIND_CONFIG_MSG;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.core.util.Throwables;
import org.apache.logging.log4j.message.ParameterizedMessage;
import org.opensearch.OpenSearchStatusException;
import org.opensearch.action.ActionListener;
import org.opensearch.action.get.GetRequest;
import org.opensearch.action.search.SearchRequest;
import org.opensearch.action.search.SearchResponse;
import org.opensearch.ad.constant.ADCommonMessages;
import org.opensearch.ad.constant.ADCommonName;
import org.opensearch.ad.model.ADTaskType;
import org.opensearch.ad.model.AnomalyDetector;
import org.opensearch.ad.model.DetectorProfile;
import org.opensearch.ad.model.DetectorProfileName;
import org.opensearch.ad.model.DetectorState;
import org.opensearch.ad.model.InitProgressProfile;
import org.opensearch.ad.settings.ADNumericSetting;
import org.opensearch.ad.settings.AnomalyDetectorSettings;
import org.opensearch.ad.task.ADTaskManager;
import org.opensearch.ad.transport.ProfileAction;
import org.opensearch.ad.transport.ProfileRequest;
import org.opensearch.ad.transport.ProfileResponse;
import org.opensearch.ad.transport.RCFPollingAction;
import org.opensearch.ad.transport.RCFPollingRequest;
import org.opensearch.ad.transport.RCFPollingResponse;
import org.opensearch.client.Client;
import org.opensearch.cluster.node.DiscoveryNode;
import org.opensearch.common.xcontent.LoggingDeprecationHandler;
import org.opensearch.common.xcontent.XContentType;
import org.opensearch.core.xcontent.NamedXContentRegistry;
import org.opensearch.core.xcontent.XContentParser;
import org.opensearch.search.SearchHits;
import org.opensearch.search.aggregations.Aggregation;
import org.opensearch.search.aggregations.AggregationBuilder;
import org.opensearch.search.aggregations.AggregationBuilders;
import org.opensearch.search.aggregations.Aggregations;
import org.opensearch.search.aggregations.bucket.composite.CompositeAggregation;
import org.opensearch.search.aggregations.bucket.composite.TermsValuesSourceBuilder;
import org.opensearch.search.aggregations.metrics.CardinalityAggregationBuilder;
import org.opensearch.search.aggregations.metrics.InternalCardinality;
import org.opensearch.search.builder.SearchSourceBuilder;
import org.opensearch.timeseries.AnalysisType;
import org.opensearch.timeseries.common.exception.NotSerializedExceptionName;
import org.opensearch.timeseries.common.exception.ResourceNotFoundException;
import org.opensearch.timeseries.constant.CommonName;
import org.opensearch.timeseries.model.IntervalTimeConfiguration;
import org.opensearch.timeseries.model.Job;
import org.opensearch.timeseries.util.DiscoveryNodeFilterer;
import org.opensearch.timeseries.util.ExceptionUtil;
import org.opensearch.timeseries.util.MultiResponsesDelegateActionListener;
import org.opensearch.timeseries.util.SecurityClientUtil;
import org.opensearch.transport.TransportService;

public class AnomalyDetectorProfileRunner extends AbstractProfileRunner {
    private final Logger logger = LogManager.getLogger(AnomalyDetectorProfileRunner.class);
    private Client client;
    private SecurityClientUtil clientUtil;
    private NamedXContentRegistry xContentRegistry;
    private DiscoveryNodeFilterer nodeFilter;
    private final TransportService transportService;
    private final ADTaskManager adTaskManager;
    private final int maxTotalEntitiesToTrack;

    public AnomalyDetectorProfileRunner(
        Client client,
        SecurityClientUtil clientUtil,
        NamedXContentRegistry xContentRegistry,
        DiscoveryNodeFilterer nodeFilter,
        long requiredSamples,
        TransportService transportService,
        ADTaskManager adTaskManager
    ) {
        super(requiredSamples);
        this.client = client;
        this.clientUtil = clientUtil;
        this.xContentRegistry = xContentRegistry;
        this.nodeFilter = nodeFilter;
        if (requiredSamples <= 0) {
            throw new IllegalArgumentException("required samples should be a positive number, but was " + requiredSamples);
        }
        this.transportService = transportService;
        this.adTaskManager = adTaskManager;
        this.maxTotalEntitiesToTrack = AnomalyDetectorSettings.MAX_TOTAL_ENTITIES_TO_TRACK;
    }

    public void profile(String detectorId, ActionListener<DetectorProfile> listener, Set<DetectorProfileName> profilesToCollect) {
        if (profilesToCollect.isEmpty()) {
            listener.onFailure(new IllegalArgumentException(ADCommonMessages.EMPTY_PROFILES_COLLECT));
            return;
        }
        calculateTotalResponsesToWait(detectorId, profilesToCollect, listener);
    }

    private void calculateTotalResponsesToWait(
        String detectorId,
        Set<DetectorProfileName> profilesToCollect,
        ActionListener<DetectorProfile> listener
    ) {
        GetRequest getDetectorRequest = new GetRequest(CommonName.CONFIG_INDEX, detectorId);
        client.get(getDetectorRequest, ActionListener.wrap(getDetectorResponse -> {
            if (getDetectorResponse != null && getDetectorResponse.isExists()) {
                try (
                    XContentParser xContentParser = XContentType.JSON
                        .xContent()
                        .createParser(xContentRegistry, LoggingDeprecationHandler.INSTANCE, getDetectorResponse.getSourceAsString())
                ) {
                    ensureExpectedToken(XContentParser.Token.START_OBJECT, xContentParser.nextToken(), xContentParser);
                    AnomalyDetector detector = AnomalyDetector.parse(xContentParser, detectorId);
                    prepareProfile(detector, listener, profilesToCollect);
                } catch (Exception e) {
                    logger.error(FAIL_TO_PARSE_DETECTOR_MSG + detectorId, e);
                    listener.onFailure(new OpenSearchStatusException(FAIL_TO_PARSE_DETECTOR_MSG + detectorId, BAD_REQUEST));
                }
            } else {
                listener.onFailure(new OpenSearchStatusException(FAIL_TO_FIND_CONFIG_MSG + detectorId, BAD_REQUEST));
            }
        }, exception -> {
            logger.error(FAIL_TO_FIND_CONFIG_MSG + detectorId, exception);
            listener.onFailure(new OpenSearchStatusException(FAIL_TO_FIND_CONFIG_MSG + detectorId, INTERNAL_SERVER_ERROR));
        }));
    }

    private void prepareProfile(
        AnomalyDetector detector,
        ActionListener<DetectorProfile> listener,
        Set<DetectorProfileName> profilesToCollect
    ) {
        String detectorId = detector.getId();
        GetRequest getRequest = new GetRequest(CommonName.JOB_INDEX, detectorId);
        client.get(getRequest, ActionListener.wrap(getResponse -> {
            if (getResponse != null && getResponse.isExists()) {
                try (
                    XContentParser parser = XContentType.JSON
                        .xContent()
                        .createParser(xContentRegistry, LoggingDeprecationHandler.INSTANCE, getResponse.getSourceAsString())
                ) {
                    ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.nextToken(), parser);
                    Job job = Job.parse(parser);
                    long enabledTimeMs = job.getEnabledTime().toEpochMilli();

                    boolean isMultiEntityDetector = detector.isHighCardinality();

                    int totalResponsesToWait = 0;
                    if (profilesToCollect.contains(DetectorProfileName.ERROR)) {
                        totalResponsesToWait++;
                    }

                    // total number of listeners we need to define. Needed by MultiResponsesDelegateActionListener to decide
                    // when to consolidate results and return to users
                    if (isMultiEntityDetector) {
                        if (profilesToCollect.contains(DetectorProfileName.TOTAL_ENTITIES)) {
                            totalResponsesToWait++;
                        }
                        if (profilesToCollect.contains(DetectorProfileName.COORDINATING_NODE)
                            || profilesToCollect.contains(DetectorProfileName.SHINGLE_SIZE)
                            || profilesToCollect.contains(DetectorProfileName.TOTAL_SIZE_IN_BYTES)
                            || profilesToCollect.contains(DetectorProfileName.MODELS)
                            || profilesToCollect.contains(DetectorProfileName.ACTIVE_ENTITIES)
                            || profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)
                            || profilesToCollect.contains(DetectorProfileName.STATE)) {
                            totalResponsesToWait++;
                        }
                        if (profilesToCollect.contains(DetectorProfileName.AD_TASK)) {
                            totalResponsesToWait++;
                        }
                    } else {
                        if (profilesToCollect.contains(DetectorProfileName.STATE)
                            || profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)) {
                            totalResponsesToWait++;
                        }
                        if (profilesToCollect.contains(DetectorProfileName.COORDINATING_NODE)
                            || profilesToCollect.contains(DetectorProfileName.SHINGLE_SIZE)
                            || profilesToCollect.contains(DetectorProfileName.TOTAL_SIZE_IN_BYTES)
                            || profilesToCollect.contains(DetectorProfileName.MODELS)) {
                            totalResponsesToWait++;
                        }
                        if (profilesToCollect.contains(DetectorProfileName.AD_TASK)) {
                            totalResponsesToWait++;
                        }
                    }

                    MultiResponsesDelegateActionListener<DetectorProfile> delegateListener =
                        new MultiResponsesDelegateActionListener<DetectorProfile>(
                            listener,
                            totalResponsesToWait,
                            ADCommonMessages.FAIL_FETCH_ERR_MSG + detectorId,
                            false
                        );
                    if (profilesToCollect.contains(DetectorProfileName.ERROR)) {
                        adTaskManager.getAndExecuteOnLatestDetectorLevelTask(detectorId, ADTaskType.REALTIME_TASK_TYPES, adTask -> {
                            DetectorProfile.Builder profileBuilder = new DetectorProfile.Builder();
                            if (adTask.isPresent()) {
                                long lastUpdateTimeMs = adTask.get().getLastUpdateTime().toEpochMilli();

                                // if state index hasn't been updated, we should not use the error field
                                // For example, before a detector is enabled, if the error message contains
                                // the phrase "stopped due to blah", we should not show this when the detector
                                // is enabled.
                                if (lastUpdateTimeMs > enabledTimeMs && adTask.get().getError() != null) {
                                    profileBuilder.error(adTask.get().getError());
                                }
                                delegateListener.onResponse(profileBuilder.build());
                            } else {
                                // detector state for this detector does not exist
                                delegateListener.onResponse(profileBuilder.build());
                            }
                        }, transportService, false, delegateListener);
                    }

                    // total number of listeners we need to define. Needed by MultiResponsesDelegateActionListener to decide
                    // when to consolidate results and return to users
                    if (isMultiEntityDetector) {
                        if (profilesToCollect.contains(DetectorProfileName.TOTAL_ENTITIES)) {
                            profileEntityStats(delegateListener, detector);
                        }
                        if (profilesToCollect.contains(DetectorProfileName.COORDINATING_NODE)
                            || profilesToCollect.contains(DetectorProfileName.SHINGLE_SIZE)
                            || profilesToCollect.contains(DetectorProfileName.TOTAL_SIZE_IN_BYTES)
                            || profilesToCollect.contains(DetectorProfileName.MODELS)
                            || profilesToCollect.contains(DetectorProfileName.ACTIVE_ENTITIES)
                            || profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)
                            || profilesToCollect.contains(DetectorProfileName.STATE)) {
                            profileModels(detector, profilesToCollect, job, true, delegateListener);
                        }
                        if (profilesToCollect.contains(DetectorProfileName.AD_TASK)) {
                            adTaskManager.getLatestHistoricalTaskProfile(detectorId, transportService, null, delegateListener);
                        }
                    } else {
                        if (profilesToCollect.contains(DetectorProfileName.STATE)
                            || profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)) {
                            profileStateRelated(detector, delegateListener, job.isEnabled(), profilesToCollect);
                        }
                        if (profilesToCollect.contains(DetectorProfileName.COORDINATING_NODE)
                            || profilesToCollect.contains(DetectorProfileName.SHINGLE_SIZE)
                            || profilesToCollect.contains(DetectorProfileName.TOTAL_SIZE_IN_BYTES)
                            || profilesToCollect.contains(DetectorProfileName.MODELS)) {
                            profileModels(detector, profilesToCollect, job, false, delegateListener);
                        }
                        if (profilesToCollect.contains(DetectorProfileName.AD_TASK)) {
                            adTaskManager.getLatestHistoricalTaskProfile(detectorId, transportService, null, delegateListener);
                        }
                    }

                } catch (Exception e) {
                    logger.error(ADCommonMessages.FAIL_TO_GET_PROFILE_MSG, e);
                    listener.onFailure(e);
                }
            } else {
                onGetDetectorForPrepare(detectorId, listener, profilesToCollect);
            }
        }, exception -> {
            if (ExceptionUtil.isIndexNotAvailable(exception)) {
                logger.info(exception.getMessage());
                onGetDetectorForPrepare(detectorId, listener, profilesToCollect);
            } else {
                logger.error(ADCommonMessages.FAIL_TO_GET_PROFILE_MSG + detectorId);
                listener.onFailure(exception);
            }
        }));
    }

    private void profileEntityStats(MultiResponsesDelegateActionListener<DetectorProfile> listener, AnomalyDetector detector) {
        List<String> categoryField = detector.getCategoryFields();
        if (!detector.isHighCardinality() || categoryField.size() > ADNumericSetting.maxCategoricalFields()) {
            listener.onResponse(new DetectorProfile.Builder().build());
        } else {
            if (categoryField.size() == 1) {
                // Run a cardinality aggregation to count the cardinality of single category fields
                SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
                CardinalityAggregationBuilder aggBuilder = new CardinalityAggregationBuilder(ADCommonName.TOTAL_ENTITIES);
                aggBuilder.field(categoryField.get(0));
                searchSourceBuilder.aggregation(aggBuilder);

                SearchRequest request = new SearchRequest(detector.getIndices().toArray(new String[0]), searchSourceBuilder);
                final ActionListener<SearchResponse> searchResponseListener = ActionListener.wrap(searchResponse -> {
                    Map<String, Aggregation> aggMap = searchResponse.getAggregations().asMap();
                    InternalCardinality totalEntities = (InternalCardinality) aggMap.get(ADCommonName.TOTAL_ENTITIES);
                    long value = totalEntities.getValue();
                    DetectorProfile.Builder profileBuilder = new DetectorProfile.Builder();
                    DetectorProfile profile = profileBuilder.totalEntities(value).build();
                    listener.onResponse(profile);
                }, searchException -> {
                    logger.warn(ADCommonMessages.FAIL_TO_GET_TOTAL_ENTITIES + detector.getId());
                    listener.onFailure(searchException);
                });
                // using the original context in listener as user roles have no permissions for internal operations like fetching a
                // checkpoint
                clientUtil
                    .<SearchRequest, SearchResponse>asyncRequestWithInjectedSecurity(
                        request,
                        client::search,
                        detector.getId(),
                        client,
                        AnalysisType.AD,
                        searchResponseListener
                    );
            } else {
                // Run a composite query and count the number of buckets to decide cardinality of multiple category fields
                AggregationBuilder bucketAggs = AggregationBuilders
                    .composite(
                        ADCommonName.TOTAL_ENTITIES,
                        detector
                            .getCategoryFields()
                            .stream()
                            .map(f -> new TermsValuesSourceBuilder(f).field(f))
                            .collect(Collectors.toList())
                    )
                    .size(maxTotalEntitiesToTrack);
                SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder().aggregation(bucketAggs).trackTotalHits(false).size(0);
                SearchRequest searchRequest = new SearchRequest()
                    .indices(detector.getIndices().toArray(new String[0]))
                    .source(searchSourceBuilder);
                final ActionListener<SearchResponse> searchResponseListener = ActionListener.wrap(searchResponse -> {
                    DetectorProfile.Builder profileBuilder = new DetectorProfile.Builder();
                    Aggregations aggs = searchResponse.getAggregations();
                    if (aggs == null) {
                        // This would indicate some bug or some opensearch core changes that we are not aware of (we don't keep up-to-date
                        // with
                        // the large amounts of changes there). For example, they may change to if there are results return it; otherwise
                        // return
                        // null instead of an empty Aggregations as they currently do.
                        logger.warn("Unexpected null aggregation.");
                        listener.onResponse(profileBuilder.totalEntities(0L).build());
                        return;
                    }

                    Aggregation aggrResult = aggs.get(ADCommonName.TOTAL_ENTITIES);
                    if (aggrResult == null) {
                        listener.onFailure(new IllegalArgumentException("Fail to find valid aggregation result"));
                        return;
                    }

                    CompositeAggregation compositeAgg = (CompositeAggregation) aggrResult;
                    DetectorProfile profile = profileBuilder.totalEntities(Long.valueOf(compositeAgg.getBuckets().size())).build();
                    listener.onResponse(profile);
                }, searchException -> {
                    logger.warn(ADCommonMessages.FAIL_TO_GET_TOTAL_ENTITIES + detector.getId());
                    listener.onFailure(searchException);
                });
                // using the original context in listener as user roles have no permissions for internal operations like fetching a
                // checkpoint
                clientUtil
                    .<SearchRequest, SearchResponse>asyncRequestWithInjectedSecurity(
                        searchRequest,
                        client::search,
                        detector.getId(),
                        client,
                        AnalysisType.AD,
                        searchResponseListener
                    );
            }

        }
    }

    private void onGetDetectorForPrepare(String detectorId, ActionListener<DetectorProfile> listener, Set<DetectorProfileName> profiles) {
        DetectorProfile.Builder profileBuilder = new DetectorProfile.Builder();
        if (profiles.contains(DetectorProfileName.STATE)) {
            profileBuilder.state(DetectorState.DISABLED);
        }
        if (profiles.contains(DetectorProfileName.AD_TASK)) {
            adTaskManager.getLatestHistoricalTaskProfile(detectorId, transportService, profileBuilder.build(), listener);
        } else {
            listener.onResponse(profileBuilder.build());
        }
    }

    /**
     * We expect three kinds of states:
     *  -Disabled: if get ad job api says the job is disabled;
     *  -Init: if rcf model's total updates is less than required
     *  -Running: if neither of the above applies and no exceptions.
     * @param detector anomaly detector
     * @param listener listener to process the returned state or exception
     * @param enabled whether the detector job is enabled or not
     * @param profilesToCollect target profiles to fetch
     */
    private void profileStateRelated(
        AnomalyDetector detector,
        MultiResponsesDelegateActionListener<DetectorProfile> listener,
        boolean enabled,
        Set<DetectorProfileName> profilesToCollect
    ) {
        if (enabled) {
            RCFPollingRequest request = new RCFPollingRequest(detector.getId());
            client.execute(RCFPollingAction.INSTANCE, request, onPollRCFUpdates(detector, profilesToCollect, listener));
        } else {
            DetectorProfile.Builder builder = new DetectorProfile.Builder();
            if (profilesToCollect.contains(DetectorProfileName.STATE)) {
                builder.state(DetectorState.DISABLED);
            }
            listener.onResponse(builder.build());
        }
    }

    private void profileModels(
        AnomalyDetector detector,
        Set<DetectorProfileName> profiles,
        Job job,
        boolean forMultiEntityDetector,
        MultiResponsesDelegateActionListener<DetectorProfile> listener
    ) {
        DiscoveryNode[] dataNodes = nodeFilter.getEligibleDataNodes();
        ProfileRequest profileRequest = new ProfileRequest(detector.getId(), profiles, forMultiEntityDetector, dataNodes);
        client.execute(ProfileAction.INSTANCE, profileRequest, onModelResponse(detector, profiles, job, listener));// get init progress
    }

    private ActionListener<ProfileResponse> onModelResponse(
        AnomalyDetector detector,
        Set<DetectorProfileName> profilesToCollect,
        Job job,
        MultiResponsesDelegateActionListener<DetectorProfile> listener
    ) {
        boolean isMultientityDetector = detector.isHighCardinality();
        return ActionListener.wrap(profileResponse -> {
            DetectorProfile.Builder profile = new DetectorProfile.Builder();
            if (profilesToCollect.contains(DetectorProfileName.COORDINATING_NODE)) {
                profile.coordinatingNode(profileResponse.getCoordinatingNode());
            }
            if (profilesToCollect.contains(DetectorProfileName.SHINGLE_SIZE)) {
                profile.shingleSize(profileResponse.getShingleSize());
            }
            if (profilesToCollect.contains(DetectorProfileName.TOTAL_SIZE_IN_BYTES)) {
                profile.totalSizeInBytes(profileResponse.getTotalSizeInBytes());
            }
            if (profilesToCollect.contains(DetectorProfileName.MODELS)) {
                profile.modelProfile(profileResponse.getModelProfile());
                profile.modelCount(profileResponse.getModelCount());
            }
            if (isMultientityDetector && profilesToCollect.contains(DetectorProfileName.ACTIVE_ENTITIES)) {
                profile.activeEntities(profileResponse.getActiveEntities());
            }

            if (isMultientityDetector
                && (profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)
                    || profilesToCollect.contains(DetectorProfileName.STATE))) {
                profileMultiEntityDetectorStateRelated(job, profilesToCollect, profileResponse, profile, detector, listener);
            } else {
                listener.onResponse(profile.build());
            }
        }, listener::onFailure);
    }

    private void profileMultiEntityDetectorStateRelated(
        Job job,
        Set<DetectorProfileName> profilesToCollect,
        ProfileResponse profileResponse,
        DetectorProfile.Builder profileBuilder,
        AnomalyDetector detector,
        MultiResponsesDelegateActionListener<DetectorProfile> listener
    ) {
        if (job.isEnabled()) {
            if (profileResponse.getTotalUpdates() < requiredSamples) {
                // need to double check since what ProfileResponse returns is the highest priority entity currently in memory, but
                // another entity might have already been initialized and sit somewhere else (in memory or on disk).
                long enabledTime = job.getEnabledTime().toEpochMilli();
                long totalUpdates = profileResponse.getTotalUpdates();
                ProfileUtil
                    .confirmDetectorRealtimeInitStatus(
                        detector,
                        enabledTime,
                        client,
                        onInittedEver(enabledTime, profileBuilder, profilesToCollect, detector, totalUpdates, listener)
                    );
            } else {
                createRunningStateAndInitProgress(profilesToCollect, profileBuilder);
                listener.onResponse(profileBuilder.build());
            }
        } else {
            if (profilesToCollect.contains(DetectorProfileName.STATE)) {
                profileBuilder.state(DetectorState.DISABLED);
            }
            listener.onResponse(profileBuilder.build());
        }
    }

    private ActionListener<SearchResponse> onInittedEver(
        long lastUpdateTimeMs,
        DetectorProfile.Builder profileBuilder,
        Set<DetectorProfileName> profilesToCollect,
        AnomalyDetector detector,
        long totalUpdates,
        MultiResponsesDelegateActionListener<DetectorProfile> listener
    ) {
        return ActionListener.wrap(searchResponse -> {
            SearchHits hits = searchResponse.getHits();
            if (hits.getTotalHits().value == 0L) {
                processInitResponse(detector, profilesToCollect, totalUpdates, false, profileBuilder, listener);
            } else {
                createRunningStateAndInitProgress(profilesToCollect, profileBuilder);
                listener.onResponse(profileBuilder.build());
            }
        }, exception -> {
            if (ExceptionUtil.isIndexNotAvailable(exception)) {
                // anomaly result index is not created yet
                processInitResponse(detector, profilesToCollect, totalUpdates, false, profileBuilder, listener);
            } else {
                logger
                    .error(
                        "Fail to find any anomaly result with anomaly score larger than 0 after AD job enabled time for detector {}",
                        detector.getId()
                    );
                listener.onFailure(exception);
            }
        });
    }

    /**
     * Listener for polling rcf updates through transport messaging
     * @param detector anomaly detector
     * @param profilesToCollect profiles to collect like state
     * @param listener delegate listener
     * @return Listener for polling rcf updates through transport messaging
     */
    private ActionListener<RCFPollingResponse> onPollRCFUpdates(
        AnomalyDetector detector,
        Set<DetectorProfileName> profilesToCollect,
        MultiResponsesDelegateActionListener<DetectorProfile> listener
    ) {
        return ActionListener.wrap(rcfPollResponse -> {
            long totalUpdates = rcfPollResponse.getTotalUpdates();
            if (totalUpdates < requiredSamples) {
                processInitResponse(detector, profilesToCollect, totalUpdates, false, new DetectorProfile.Builder(), listener);
            } else {
                DetectorProfile.Builder builder = new DetectorProfile.Builder();
                createRunningStateAndInitProgress(profilesToCollect, builder);
                listener.onResponse(builder.build());
            }
        }, exception -> {
            // we will get an AnomalyDetectionException wrapping the real exception inside
            Throwable cause = Throwables.getRootCause(exception);

            // exception can be a RemoteTransportException
            Exception causeException = (Exception) cause;
            if (ExceptionUtil
                .isException(
                    causeException,
                    ResourceNotFoundException.class,
                    NotSerializedExceptionName.RESOURCE_NOT_FOUND_EXCEPTION_NAME_UNDERSCORE.getName()
                )
                || (ExceptionUtil.isIndexNotAvailable(causeException)
                    && causeException.getMessage().contains(ADCommonName.CHECKPOINT_INDEX_NAME))) {
                // cannot find checkpoint
                // We don't want to show the estimated time remaining to initialize
                // a detector before cold start finishes, where the actual
                // initialization time may be much shorter if sufficient historical
                // data exists.
                processInitResponse(detector, profilesToCollect, 0L, true, new DetectorProfile.Builder(), listener);
            } else {
                logger.error(new ParameterizedMessage("Fail to get init progress through messaging for {}", detector.getId()), exception);
                listener.onFailure(exception);
            }
        });
    }

    private void createRunningStateAndInitProgress(Set<DetectorProfileName> profilesToCollect, DetectorProfile.Builder builder) {
        if (profilesToCollect.contains(DetectorProfileName.STATE)) {
            builder.state(DetectorState.RUNNING).build();
        }

        if (profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)) {
            InitProgressProfile initProgress = new InitProgressProfile("100%", 0, 0);
            builder.initProgress(initProgress);
        }
    }

    private void processInitResponse(
        AnomalyDetector detector,
        Set<DetectorProfileName> profilesToCollect,
        long totalUpdates,
        boolean hideMinutesLeft,
        DetectorProfile.Builder builder,
        MultiResponsesDelegateActionListener<DetectorProfile> listener
    ) {
        if (profilesToCollect.contains(DetectorProfileName.STATE)) {
            builder.state(DetectorState.INIT);
        }

        if (profilesToCollect.contains(DetectorProfileName.INIT_PROGRESS)) {
            if (hideMinutesLeft) {
                InitProgressProfile initProgress = computeInitProgressProfile(totalUpdates, 0);
                builder.initProgress(initProgress);
            } else {
                long intervalMins = ((IntervalTimeConfiguration) detector.getInterval()).toDuration().toMinutes();
                InitProgressProfile initProgress = computeInitProgressProfile(totalUpdates, intervalMins);
                builder.initProgress(initProgress);
            }
        }

        listener.onResponse(builder.build());
    }
}