import {
  requireNativeModule,
  requireNativeViewManager,
  type NativeModule,
} from 'expo-modules-core';
import type { ViewProps } from 'react-native';

/** Number of spectrum bins returned by getSpectrumFrame (fftSize/2, fftSize=2048). */
export const SPECTRUM_BINS = 1024;

/** Spectrum values are dB magnitudes in this range (silence ~ -100). */
export const SPECTRUM_DB_MIN = -100;
export const SPECTRUM_DB_MAX = 0;

/** Render-ready oscilloscope point count requested by the mobile UI. */
export const OSCILLOSCOPE_POINTS = 256;

declare class AstraScopeModuleType extends NativeModule {
  /** Gate the audio-thread PCM tap. Off when backgrounded/paused/reduced-motion. */
  setActive(active: boolean): void;
  /**
   * Fill `out` (length should be {@link SPECTRUM_BINS}) with the latest dB
   * spectrum magnitudes in place; returns the number of bins written. `smoothing`
   * is the previous-frame retention in [0, 0.99]. Call once per render frame
   * from the JS thread.
   */
  getSpectrumFrame(out: Float32Array, smoothing: number): number;
  /**
   * Fill `out` with render-ready, evenly spaced points from the latest
   * pitch-locked oscilloscope window. Values are interpolated visual samples in
   * ~[-1, 1]. Returns the number of points written (0 before warmup).
   */
  getOscilloscopeFrame(out: Float32Array): number;
  /** Gate the post-EQ tap (true only while the EQ screen is visible). */
  setActivePostEq(active: boolean): void;
  /**
   * Like {@link getSpectrumFrame} but for the POST-EQ tap (ring #2) — feeds the
   * EQ screen's spectrum behind the response curve. Returns bins written.
   */
  getSpectrumFramePostEq(out: Float32Array, smoothing: number): number;

  // --- M4 EQ + per-track gain (params pushed from JS; biquad coeffs computed
  // natively at the real stream sample rate) ---

  /** Master EQ bypass. When false the EqAudioProcessor is passthrough. */
  setEqEnabled(enabled: boolean): void;
  /** EQ preamp as a linear amplitude (1 = unity). */
  setEqPreamp(linear: number): void;
  /**
   * Flat band params: 5 values per band — [typeOrdinal, frequency, gain, Q,
   * enabled?1:0]. Native recomputes biquad coefficients at the stream rate.
   */
  setEqBands(params: number[]): void;
  /** Glide the active normalization gain to an explicit linear value (1 = unity). */
  setNormalizationGain(linear: number): void;
  /**
   * Register a queued track's gain by URL. The player switches the active gain to the
   * matching entry natively at the media-item transition (no JS round-trip).
   */
  setTrackGain(url: string, linear: number): void;
  /**
   * Bulk-register queued tracks' gains (url -> linear) in one bridge call. With
   * `clearExisting` the native map is cleared first (bounds it to the live queue).
   */
  setTrackGains(entries: Record<string, number>, clearExisting: boolean): void;
  /** Glide to the registered gain for this URL now (mount/settings/late measurement). */
  activateTrackGain(url: string): void;
  /** Apply the registered gain immediately while playback is still paused. */
  primeTrackGain(url: string): void;
  /**
   * Conservative temp gain (linear) applied when a media-item transition hits a URL
   * with no registered gain (unanalyzed track). Keep at 1 while normalization is off.
   */
  setFallbackGain(linear: number): void;
}

export const AstraScope = requireNativeModule<AstraScopeModuleType>('AstraScope');

/** Internal prop contract for the allocation-free Android scope surface. */
export interface AstraScopeViewProps extends ViewProps {
  mode: 'spectrum' | 'oscilloscope';
  source?: 'pre' | 'post';
  active: boolean;
  reducedMotion?: boolean;
  /** Positive milliseconds for a fixed cadence; 0 enables native display sync. */
  frameMs: number;
  analysisFrameMs?: number;
  smoothing?: number;
  pointCount?: number;
  dbMin?: number;
  dbMax?: number;
  tiltDbPerOctave?: number;
  /** Android ARGB color produced by React Native's processColor(). */
  color: number;
  lineWidth?: number;
  lineOpacity?: number;
  fillOpacity?: number;
  glow?: boolean;
  glowOpacity?: number;
  edgeFade?: boolean;
  edgeFadeWidth?: number;
  gain?: number;
  values?: number[];
}

export const AstraScopeView =
  requireNativeViewManager<AstraScopeViewProps>('AstraScope');