import type {
  VAD,
  AppraisalVector,
  AgentResponse,
  State,
  SoulStage,
  NeedState,
  CatastropheState,
  Episode,
} from '@molroo-ai/sdk';

interface VAD {
  /** Valence: negative (-1) to positive (+1) */
  V: number;
  /** Arousal: calm (0) to activated (1) */
  A: number;
  /** Dominance: submissive (-1) to dominant (+1) */
  D: number;
}

interface AppraisalVector {
  /** How relevant is this to the persona's goals? [-1, 1] */
  goal_relevance: number;
  /** Does this help (+1) or hinder (-1) goals? [-1, 1] */
  goal_congruence: number;
  /** How expected was this? [0, 1] */
  expectedness: number;
  /** How much control does the persona have? [0, 1] */
  controllability: number;
  /** Who caused this — self (+1) or other (-1)? [-1, 1] */
  agency: number;
  /** Compatible with the persona's norms? [-1, 1] */
  norm_compatibility: number;
}

interface AgentResponse {
  emotion: {
    vad: VAD;
    discrete: {
      primary: string;
      secondary?: string;
      intensity: number;
    };
  };
  mood?: {
    vad: VAD;
    valence: 'negative' | 'neutral' | 'positive';
    arousal: 'calm' | 'moderate' | 'activated';
    dominance: 'submissive' | 'balanced' | 'dominant';
  };
  somatic?: string[];
  narrative?: {
    tone: number;      // -1 (decline) to +1 (recovery)
    agency: number;    // -1 (reactive) to +1 (agentic)
    coherence: number; // 0 to 1
  };
  text?: string;
  action?: string;
  memoryEpisode?: Episode;
  socialUpdates?: SocialUpdate[];
  reflectionPrompt?: ReflectionPrompt;
  stageTransition?: boolean;
  maskExposure?: { integrity: number; state: string };
  goalChanges?: { achieved: string[]; blocked: string[] };
}

interface State {
  emotion: VAD;
  velocity: Velocity;
  emotion_history: Array<VAD & { timestamp: number }>;
  body_budget: number;
  soul_stage: SoulStage;
  prediction_model: AppraisalVector;
  prev_emotion_delta?: VAD;
  need_state?: NeedState;
  lastUpdated?: number;
  catastrophe: CatastropheState;
  metacog: MetacogState;
  affect_dynamics: AffectDynamicsState;
  interpersonal: InterpersonalState;
  regulation: RegulationState;
  prevMaskIntegrity?: number;
}

interface SoulStage {
  id: number;                // [1, 15]
  name: string;
  tolerance: number;
  stress_modifier: number;
  appraisal_bias: Partial<AppraisalVector>;
  avoidance_mod: number;
  blend_table: { rational: number; irrational: number };
  cumulative_load: number;
  turns_in_stage: number;
}

interface NeedState {
  autonomy: number;     // [0, 1]
  competence: number;   // [0, 1]
  relatedness: number;  // [0, 1]
}

interface CatastropheState {
  alpha: number;           // Asymmetry factor
  beta: number;            // Bifurcation factor
  z: number;               // State variable
  hysteresis_bound: number;
}

interface Episode {
  id: string;
  timestamp: number;
  sourceEntity?: string;
  context?: string;
  appraisal: AppraisalVector;
  emotionSnapshot: VAD;
  intensity: number;
  importance: number;
}

interface SocialUpdate {
  entityId: string;
  field: string;
  oldValue: number;
  newValue: number;
}

interface PersonaSnapshot {
  state: State;
  relationships: Record<string, Relationship>;
  goals: Goal[];
  selfConcept?: string;
  mood?: VAD;
  moodHistory?: Array<{ emotion: VAD; intensity: number }>;
  somatic?: {
    regions: {
      head: number; neck: number; chest: number;
      stomach: number; upperBack: number; hands: number; legs: number;
    };
    interoception: number;
  };
  narrative?: {
    tone: number;
    coherence: { temporal: number; causal: number; thematic: number };
    agency: number;
    communion: number;
    arcs: Array<{
      id: string;
      type: 'decline' | 'recovery' | 'stability' | 'growth';
      strength: number;
      episodeIds: string[];
      theme: string;
    }>;
  };
  tmt?: {
    mortalitySalience: number;
    deathAnxiety: number;
    proximity: 'immediate' | 'near' | 'distant' | 'abstract';
    proximalDefense: boolean;
    distalDefense: boolean;
    defenseTarget: 'in_group' | 'out_group' | 'values' | 'self_esteem' | null;
    defenseStrength: number;
    lastReminderTime: number;
  };
  selfEsteem?: { global: number; contingencies: Record<string, number> };
  motivationContexts?: Record<string, MotivationContext>;
}