import { IObsidianMetricsAPI, MetricInstance, ObsidianMetricsPlugin } from './obsidian-metrics';

class MyPlugin extends Plugin {
  private metricsApi: IObsidianMetricsAPI | undefined;
  private myGauge: MetricInstance | undefined;

  async onload() {
    // Listen for metrics API becoming available (handles load order and reloads)
    this.registerEvent(
      this.app.workspace.on('obsidian-metrics:ready', (api: IObsidianMetricsAPI) => {
        this.initializeMetrics(api);
      })
    );

    // Also try to get it immediately in case metrics plugin loaded first
    const metricsPlugin = this.app.plugins.plugins['obsidian-metrics'] as ObsidianMetricsPlugin | undefined;
    if (metricsPlugin?.api) {
      this.initializeMetrics(metricsPlugin.api);
    }
  }

  private initializeMetrics(api: IObsidianMetricsAPI) {
    this.metricsApi = api;

    // Metric creation is idempotent - safe to call multiple times
    this.myGauge = api.createGauge({
      name: 'my_document_size_bytes',
      help: 'Size of documents in bytes',
      labelNames: ['document']
    });
  }

  updateDocumentSize(doc: string, bytes: number) {
    this.myGauge?.labels({ document: doc }).set(bytes);
  }
}

const pageViewCounter = api.createCounter({
  name: 'page_views_total',
  help: 'Total number of page views',
  labelNames: ['page_type', 'source']
});

// Increment
pageViewCounter.inc();
pageViewCounter.inc(5);
pageViewCounter.inc(1, { page_type: 'note', source: 'search' });

// Or use fluent labels() API
pageViewCounter.labels({ page_type: 'note', source: 'search' }).inc();

const activeNotesGauge = api.createGauge({
  name: 'active_notes_count',
  help: 'Number of currently active notes'
});

// Set value
activeNotesGauge.set(42);
activeNotesGauge.inc();
activeNotesGauge.dec(5);

// With labels
activeNotesGauge.labels({ workspace: 'main' }).set(10);

const loadTimeHistogram = api.createHistogram({
  name: 'page_load_duration_seconds',
  help: 'Page load duration in seconds',
  buckets: [0.1, 0.5, 1, 2, 5]
});

// Observe values
loadTimeHistogram.observe(1.2);
loadTimeHistogram.observe(0.8, { page_type: 'canvas' });

// Time operations
const timer = loadTimeHistogram.startTimer();
// ... do work ...
timer(); // Automatically observes the duration

const responseSummary = api.createSummary({
  name: 'api_response_duration_seconds',
  help: 'API response duration in seconds',
  percentiles: [0.5, 0.9, 0.95, 0.99]
});

responseSummary.observe(0.234);

// Quick counter creation
const counter = api.counter('button_clicks', 'Button click count', 1);

// Quick gauge creation
const gauge = api.gauge('memory_usage', 'Memory usage in bytes', 1024);

// Quick histogram creation
const hist = api.histogram('request_duration', 'Request duration');

// Measure async functions
const result = await api.measureAsync('async_operation_duration', async () => {
  return await someAsyncOperation();
});

// Measure sync functions
const result = api.measureSync('sync_operation_duration', () => {
  return someCalculation();
});

// Manual timing
const timer = api.createTimer('custom_operation_duration');
// ... do work ...
const durationMs = timer(); // Returns duration in milliseconds