import {
  term,
  hint,
  guardrail,
  example,
  explain,
  clarification,
  workflow,
  quirk,
  styleGuide,
  analogy
} from '@deepagents/context';

function term(name: string, definition: string): ContextFragment

term('MRR', 'monthly recurring revenue')
term('ARR', 'annual recurring revenue')
term('CAC', 'customer acquisition cost')

function hint(text: string): ContextFragment

hint('Always exclude test accounts with email ending in @test.com')
hint('Use created_at for order dates, not updated_at')
hint('The users table contains both customers and employees')

function guardrail(config: GuardrailConfig): ContextFragment

interface GuardrailConfig {
  rule: string;
  reason?: string;
  action?: string;
}

guardrail({
  rule: 'Never expose individual salaries',
  reason: 'Confidential HR data',
  action: 'Aggregate by department instead'
})

guardrail({
  rule: 'Never return more than 10,000 rows without aggregation',
  reason: 'Performance protection'
})

function example(config: ExampleConfig): ContextFragment

interface ExampleConfig {
  question: string;
  answer: string;
  note?: string;
}

example({
  question: 'show me churned customers',
  answer: `SELECT * FROM customers WHERE status = 'churned' ORDER BY churned_at DESC`,
  note: 'Status values are lowercase'
})

example({
  question: 'monthly revenue trend',
  answer: `
    SELECT 
      DATE_TRUNC('month', created_at) AS month,
      SUM(amount) AS revenue
    FROM orders
    WHERE status = 'completed'
    GROUP BY month
    ORDER BY month DESC
  `
})

function explain(config: ExplainConfig): ContextFragment

interface ExplainConfig {
  concept: string;
  explanation: string;
  therefore?: string;
}

explain({
  concept: 'soft delete pattern',
  explanation: 'Records are not physically deleted. Instead, deleted_at timestamp is set.',
  therefore: 'Always add WHERE deleted_at IS NULL to exclude soft-deleted records'
})

explain({
  concept: 'status lifecycle',
  explanation: 'Orders move: pending → processing → completed OR cancelled',
  therefore: 'Use status IN ("completed", "cancelled") for finalized orders'
})

function clarification(config: ClarificationConfig): ContextFragment

interface ClarificationConfig {
  when: string;
  ask: string;
  reason: string;
}

clarification({
  when: 'User asks for "revenue" without specifying gross vs net',
  ask: 'Do you mean gross revenue or net revenue (after refunds)?',
  reason: 'Business has separate metrics for gross and net revenue'
})

clarification({
  when: 'User asks for "active users" without timeframe',
  ask: 'Active in the last 30 days, 90 days, or year?',
  reason: 'Multiple definitions of "active" exist'
})

function workflow(config: WorkflowConfig): ContextFragment

interface WorkflowConfig {
  task: string;
  steps: string[];
  triggers?: string[];
  notes?: string;
}

workflow({
  task: 'Cohort retention analysis',
  steps: [
    'Identify cohort by signup month',
    'For each month after signup, count active users',
    'Calculate retention rate: active / cohort_size',
    'Group by cohort and months_since_signup'
  ],
  triggers: ['retention', 'cohort'],
  notes: 'Active = logged in within that month'
})

workflow({
  task: 'Error recovery',
  triggers: ['SQL error', 'query failed'],
  steps: [
    'Classify the error type',
    'For syntax errors: check SQL keywords and quotes',
    'For missing table: re-check schema',
    'Apply targeted fix',
    'Retry with corrected query'
  ],
  notes: 'Maximum 3 retry attempts'
})

function quirk(config: QuirkConfig): ContextFragment

interface QuirkConfig {
  issue: string;
  workaround: string;
}

quirk({
  issue: 'Legacy orders have NULL in created_at column',
  workaround: 'Use COALESCE(created_at, migrated_at) for date filtering'
})

quirk({
  issue: 'Email addresses before 2020 were not normalized to lowercase',
  workaround: 'Use LOWER(email) for case-insensitive matching'
})

function styleGuide(config: StyleGuideConfig): ContextFragment

interface StyleGuideConfig {
  prefer: string;
  never?: string;
  always?: string;
}

styleGuide({
  prefer: 'Full table names as aliases (users AS users)',
  never: 'Abbreviated aliases (u, oi, t1, t2)'
})

styleGuide({
  prefer: 'Explicit column names in SELECT',
  never: 'SELECT * in production queries',
  always: 'Include ORDER BY for deterministic results'
})

function analogy(config: AnalogyConfig): ContextFragment

interface AnalogyConfig {
  concepts: string[];        // At least 2
  relationship: string;
  insight?: string;
  therefore?: string;
  pitfall?: string;
}

analogy({
  concepts: ['customers', 'accounts'],
  relationship: 'One customer can have multiple accounts (personal, business)',
  insight: 'Use customers for billing, accounts for usage tracking',
  therefore: 'Join through customer_id when aggregating usage'
})

analogy({
  concepts: ['orders', 'line_items'],
  relationship: 'Orders are shopping carts, line_items are individual products',
  pitfall: 'Counting orders counts carts, not products sold'
})

function identity(config: IdentityConfig): ContextFragment

interface IdentityConfig {
  name?: string;
  role?: string;
  department?: string;
  [key: string]: any;
}

identity({
  name: 'John Doe',
  role: 'Sales Manager',
  department: 'Sales',
  teamId: 5
})

function preference(text: string): ContextFragment

preference('Always show revenue in USD, rounded to 2 decimals')
preference('Include column headers in all results')

function alias(name: string, expansion: string): ContextFragment

alias('my team', 'sales_team_id = 5')
alias('last quarter', 'created_at >= "2024-10-01" AND created_at < "2025-01-01"')

const saasTeachables = [
  // Define terms
  term('MRR', 'monthly recurring revenue'),
  term('churn', 'customer cancellation'),
  
  // Provide examples
  example({
    question: 'Show MRR trend',
    answer: 'SELECT DATE_TRUNC(...) ...'
  }),
  
  // Add guardrails
  guardrail({
    rule: 'Never expose individual MRR',
    action: 'Aggregate by cohort or plan'
  })
];

// Terms for abbreviations
term('LTV', 'lifetime value'),

// Hints for conventions
hint('Use subscription_start_at for cohort analysis'),

// Quirks for data issues
quirk({
  issue: 'Trial MRR included before March 2024',
  workaround: 'Exclude trials: WHERE plan_name != "trial"'
})

// teachables-v1.ts
export const teachablesV1 = [...];

// teachables-v2.ts (schema changed)
export const teachablesV2 = [
  ...teachablesV1,
  term('NRR', 'net revenue retention'),  // New
  quirk({ issue: 'Legacy MRR changed 2024', ... })  // Updated
];