import { InstrumentationConfigMap } from '@opentelemetry/auto-instrumentations-node';
import { Instrumentation } from '@opentelemetry/instrumentation';
import { Sampler } from '@opentelemetry/sdk-trace-base';
import { JWTInput } from 'google-auth-library';

/**
 * Copyright 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/** Configuration options for the Google Cloud plugin. */
interface GcpTelemetryConfigOptions {
    /**
     * Google Cloud Project ID. If provided, will take precedence over the
     * projectId inferred from the application credential and/or environment.
     * Required when providing an external credential (e.g. Workload Identity
     * Federation.)
     */
    projectId?: string;
    /**
     * Credentials for authenticating with Google Cloud. Primarily intended for
     * use in environments outside of GCP. On GCP credentials will typically be
     * inferred from the environment via Application Default Credentials (ADC).
     */
    credentials?: JWTInput;
    /**
     * OpenTelemetry sampler; controls the number of traces collected and exported
     * to Google Cloud. Defaults to AlwaysOnSampler, which will collect and export
     * all traces.
     *
     * There are four built-in samplers to choose from:
     *
     * - {@link https://github.com/open-telemetry/opentelemetry-js/blob/main/packages/opentelemetry-sdk-trace-base/src/sampler/AlwaysOnSampler.ts | AlwaysOnSampler} - samples all traces
     * - {@link https://github.com/open-telemetry/opentelemetry-js/blob/main/packages/opentelemetry-sdk-trace-base/src/sampler/AlwaysOffSampler.ts | AlwaysOffSampler} - samples no traces
     * - {@link https://github.com/open-telemetry/opentelemetry-js/blob/main/packages/opentelemetry-sdk-trace-base/src/sampler/ParentBasedSampler.ts | ParentBasedSampler} - samples based on parent span
     * - {@link https://github.com/open-telemetry/opentelemetry-js/blob/main/packages/opentelemetry-sdk-trace-base/src/sampler/TraceIdRatioBasedSampler.ts | TraceIdRatioBasedSampler} - samples a configurable percentage of traces
     */
    sampler?: Sampler;
    /**
     * Enabled by default, OpenTelemetry will automatically collect telemetry for
     * popular libraries via auto instrumentations without any additional code
     * or configuration. All available instrumentations will be collected, unless
     * otherwise specified via {@link autoInstrumentationConfig}.
     *
     * @see https://opentelemetry.io/docs/zero-code/js/
     */
    autoInstrumentation?: boolean;
    /**
     * Map of auto instrumentations and their configuration options. Available
     * options will vary by instrumentation.
     *
     * @see https://opentelemetry.io/docs/zero-code/js/
     */
    autoInstrumentationConfig?: InstrumentationConfigMap;
    /**
     * Additional OpenTelemetry instrumentations to include, beyond those
     * provided by auto instrumentations.
     */
    instrumentations?: Instrumentation[];
    /**
     * Metrics export interval in milliseconds; Google Cloud requires a minimum
     * value of 5000ms.
     */
    metricExportIntervalMillis?: number;
    /**
     * Timeout for the metrics export in milliseconds.
     */
    metricExportTimeoutMillis?: number;
    /**
     * If set to true, metrics will not be exported to Google Cloud. Traces and
     * logs may still be exported.
     */
    disableMetrics?: boolean;
    /**
     * If set to true, traces will not be exported to Google Cloud. Metrics and
     * logs may still be exported.
     */
    disableTraces?: boolean;
    /**
     * If set to true, input and output logs will not be collected.
     */
    disableLoggingInputAndOutput?: boolean;
    /**
     * If set to true, telemetry data will be exported in the Genkit `dev`
     * environment. Useful for local testing and troubleshooting; default is
     * false.
     */
    forceDevExport?: boolean;
}
/** Internal telemetry configuration. */
interface GcpTelemetryConfig {
    projectId?: string;
    credentials?: JWTInput;
    sampler: Sampler;
    autoInstrumentation: boolean;
    autoInstrumentationConfig: InstrumentationConfigMap;
    metricExportIntervalMillis: number;
    metricExportTimeoutMillis: number;
    instrumentations: Instrumentation[];
    disableMetrics: boolean;
    disableTraces: boolean;
    exportInputAndOutput: boolean;
    export: boolean;
}
interface GcpPrincipal {
    projectId?: string;
    serviceAccountEmail?: string;
}

export type { GcpPrincipal, GcpTelemetryConfig, GcpTelemetryConfigOptions };