import { trace } from "@opentelemetry/api";
import getObservabilityDbPath from "../core/db-path.js";
import { AIGNEObserverOptionsSchema } from "../core/type.js";
import { isBlocklet } from "../core/util.js";
import { initOpenTelemetry } from "../opentelemetry/instrument/init.js";
export class AIGNEObserver {
    storage;
    tracer = trace.getTracer("aigne-tracer");
    processor;
    exporter;
    sdk;
    sdkServerStarted;
    static exportFn;
    static setExportFn(exportFn) {
        AIGNEObserver.exportFn = exportFn;
    }
    static updateFn;
    static setUpdateFn(updateFn) {
        AIGNEObserver.updateFn = updateFn;
    }
    constructor(options) {
        const parsed = AIGNEObserverOptionsSchema.parse(options);
        this.storage = parsed?.storage ?? (!isBlocklet ? getObservabilityDbPath() : undefined);
    }
    async serve() {
        this.sdkServerStarted ??= this._serve();
        return this.sdkServerStarted;
    }
    async _serve() {
        if (!this.storage && !isBlocklet) {
            throw new Error("Server storage is not configured");
        }
        const { sdk, spanProcessor, traceExporter } = await initOpenTelemetry({
            dbPath: this.storage,
            exportFn: AIGNEObserver.exportFn,
            updateFn: AIGNEObserver.updateFn,
        });
        this.sdk = sdk;
        this.processor = spanProcessor;
        this.exporter = traceExporter;
    }
    async flush(span) {
        await this.processor?.onEnd(span);
        await this.processor?.forceFlush();
    }
    async update(id, data) {
        await this.exporter?.update(id, data);
    }
    async close() {
        try {
            // Shutdown processor
            await this.processor?.shutdown();
            // Shutdown exporter
            await this.exporter?.shutdown();
            // Shutdown SDK
            await this.sdk?.shutdown();
        }
        catch (error) {
            console.error("[Observability] Error during shutdown:", error);
        }
    }
}
