// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.microsoft.aad.msal4j;

class TelemetryHelper implements AutoCloseable {
    private Event eventToEnd;
    private String requestId;
    private String clientId;
    private ITelemetry telemetry;
    private Boolean shouldFlush;

    TelemetryHelper(ITelemetry telemetry,
                    String requestId,
                    String clientId,
                    Event event,
                    Boolean shouldFlush) {

        this.telemetry = telemetry;
        this.requestId = requestId;
        this.clientId = clientId;
        this.eventToEnd = event;
        this.shouldFlush = shouldFlush;

        if (telemetry != null) {
            telemetry.startEvent(requestId, event);
        }
    }

    public void close() {
        if (telemetry != null) {
            telemetry.stopEvent(requestId, eventToEnd);
            if (shouldFlush) {
                telemetry.flush(requestId, clientId);
            }
        }
    }
}