import { expect, describe, test } from 'bun:test'

describe('string operations', () => {
  test('to-upper converts to uppercase', async () => {
    await expect(`str.to-upper 'hello'`).toEvaluateTo('HELLO')
    await expect(`str.to-upper 'Hello World!'`).toEvaluateTo('HELLO WORLD!')
  })

  test('to-lower converts to lowercase', async () => {
    await expect(`str.to-lower 'HELLO'`).toEvaluateTo('hello')
    await expect(`str.to-lower 'Hello World!'`).toEvaluateTo('hello world!')
  })

  test('trim removes whitespace', async () => {
    await expect(`str.trim '  hello  '`).toEvaluateTo('hello')
    await expect(`str.trim '\\n\\thello\\t\\n'`).toEvaluateTo('hello')
  })

  test('capitalize makes first char uppercase', async () => {
    await expect(`str.capitalize 'hello'`).toEvaluateTo('Hello')
    await expect(`str.capitalize 'HELLO'`).toEvaluateTo('Hello')
    await expect(`str.capitalize 'hello world'`).toEvaluateTo('Hello world')
  })

  test('titlecase capitalizes each word', async () => {
    await expect(`str.titlecase 'hello world'`).toEvaluateTo('Hello World')
    await expect(`str.titlecase 'HELLO WORLD'`).toEvaluateTo('Hello World')
    await expect(`str.titlecase 'the quick brown fox'`).toEvaluateTo('The Quick Brown Fox')
  })

  test('split divides string by separator', async () => {
    await expect(`str.split 'a,b,c' ','`).toEvaluateTo(['a', 'b', 'c'])
    await expect(`str.split 'hello' ''`).toEvaluateTo(['h', 'e', 'l', 'l', 'o'])
  })

  test('split with comma separator', async () => {
    await expect(`str.split 'a,b,c' ','`).toEvaluateTo(['a', 'b', 'c'])
  })

  test('join combines array elements', async () => {
    await expect(`str.join ['a' 'b' 'c'] '-'`).toEvaluateTo('a-b-c')
    await expect(`str.join ['hello' 'world'] ' '`).toEvaluateTo('hello world')
  })

  test('join with comma separator', async () => {
    await expect(`str.join ['a' 'b' 'c'] ','`).toEvaluateTo('a,b,c')
  })

  test('starts-with? checks string prefix', async () => {
    await expect(`str.starts-with? 'hello' 'hel'`).toEvaluateTo(true)
    await expect(`str.starts-with? 'hello' 'bye'`).toEvaluateTo(false)
  })

  test('ends-with? checks string suffix', async () => {
    await expect(`str.ends-with? 'hello' 'lo'`).toEvaluateTo(true)
    await expect(`str.ends-with? 'hello' 'he'`).toEvaluateTo(false)
  })

  test('contains? checks for substring', async () => {
    await expect(`str.contains? 'hello world' 'o w'`).toEvaluateTo(true)
    await expect(`str.contains? 'hello' 'bye'`).toEvaluateTo(false)
  })

  test('empty? checks if string is empty', async () => {
    await expect(`str.empty? ''`).toEvaluateTo(true)
    await expect(`str.empty? 'hello'`).toEvaluateTo(false)
  })

  test('replace replaces first occurrence', async () => {
    await expect(`str.replace 'hello hello' 'hello' 'hi'`).toEvaluateTo('hi hello')
  })

  test('replace-all replaces all occurrences', async () => {
    await expect(`str.replace-all 'hello hello' 'hello' 'hi'`).toEvaluateTo('hi hi')
  })

  test('slice extracts substring', async () => {
    await expect(`str.slice 'hello' 1 3`).toEvaluateTo('el')
    await expect(`str.slice 'hello' 2 null`).toEvaluateTo('llo')
    await expect(`str.slice 'hello' 2`).toEvaluateTo('llo')
  })

  test('repeat repeats string', async () => {
    await expect(`str.repeat 'ha' 3`).toEvaluateTo('hahaha')
  })

  test('pad-start pads beginning', async () => {
    await expect(`str.pad-start '5' 3 '0'`).toEvaluateTo('005')
  })

  test('pad-end pads end', async () => {
    await expect(`str.pad-end '5' 3 '0'`).toEvaluateTo('500')
  })

  test('lines splits by newlines', async () => {
    await expect(`str.lines 'a\\nb\\nc'`).toEvaluateTo(['a', 'b', 'c'])
  })

  test('chars splits into characters', async () => {
    await expect(`str.chars 'abc'`).toEvaluateTo(['a', 'b', 'c'])
  })

  test('index-of finds substring position', async () => {
    await expect(`str.index-of 'hello world' 'world'`).toEvaluateTo(6)
    await expect(`str.index-of 'hello' 'bye'`).toEvaluateTo(-1)
  })

  test('last-index-of finds last occurrence', async () => {
    await expect(`str.last-index-of 'hello hello' 'hello'`).toEvaluateTo(6)
  })
})

describe('boolean logic', () => {
  test('not negates value', async () => {
    await expect(`not true`).toEvaluateTo(false)
    await expect(`not false`).toEvaluateTo(true)
    await expect(`not 42`).toEvaluateTo(false)
    await expect(`not null`).toEvaluateTo(true)
  })

  test('not works with function calls', async () => {
    await expect(`equals = do x y: x == y end; not equals 5 5`).toEvaluateTo(false)
    await expect(`equals = do x y: x == y end; not equals 5 10`).toEvaluateTo(true)
  })

  test('not works with binary operations and comparisons', async () => {
    await expect(`not 5 > 10`).toEvaluateTo(true)
    await expect(`not 10 > 5`).toEvaluateTo(false)
    await expect(`not true and false`).toEvaluateTo(true)
  })
})

describe('utilities', () => {
  test('inc increments by 1', async () => {
    await expect(`inc 5`).toEvaluateTo(6)
    await expect(`inc -1`).toEvaluateTo(0)
  })

  test('dec decrements by 1', async () => {
    await expect(`dec 5`).toEvaluateTo(4)
    await expect(`dec 0`).toEvaluateTo(-1)
  })

  test('identity returns value as-is', async () => {
    await expect(`identity 42`).toEvaluateTo(42)
    await expect(`identity 'hello'`).toEvaluateTo('hello')
  })
})

describe('collections', () => {
  test('length', async () => {
    await expect(`length 'hello'`).toEvaluateTo(5)
    await expect(`length [1 2 3]`).toEvaluateTo(3)
    await expect(`length [a=1 b=2]`).toEvaluateTo(2)
  })

  test('length throws on invalid types', async () => {
    await expect(`try: length 42 catch e: 'error' end`).toEvaluateTo('error')
    await expect(`try: length true catch e: 'error' end`).toEvaluateTo('error')
    await expect(`try: length null catch e: 'error' end`).toEvaluateTo('error')
  })

  test('literal array creates array from arguments', async () => {
    await expect(`[ 1 2 3 ]`).toEvaluateTo([1, 2, 3])
    await expect(`['a' 'b']`).toEvaluateTo(['a', 'b'])
    await expect(`[]`).toEvaluateTo([])
  })

  test('literal dict creates object from named arguments', async () => {
    await expect(`[ a=1 b=2 ]`).toEvaluateTo({ a: 1, b: 2 })
    await expect(`[=]`).toEvaluateTo({})
  })

  test('at retrieves element at index', async () => {
    await expect(`at [10 20 30] 0`).toEvaluateTo(10)
    await expect(`at [10 20 30] 2`).toEvaluateTo(30)
  })

  test('at retrieves property from object', async () => {
    await expect(`at [name='test'] 'name'`).toEvaluateTo('test')
  })

  test('slice extracts array subset', async () => {
    await expect(`list.slice [1 2 3 4 5] 1 3`).toEvaluateTo([2, 3])
    await expect(`list.slice [1 2 3 4 5] 2 5`).toEvaluateTo([3, 4, 5])
  })

  test('range creates number sequence', async () => {
    await expect(`range 0 5`).toEvaluateTo([0, 1, 2, 3, 4, 5])
    await expect(`range 3 6`).toEvaluateTo([3, 4, 5, 6])
  })

  test('range with single argument starts from 0', async () => {
    await expect(`range 3 null`).toEvaluateTo([0, 1, 2, 3])
    await expect(`range 0 null`).toEvaluateTo([0])
  })

  test('empty? checks if list, dict, string is empty', async () => {
    await expect(`empty? []`).toEvaluateTo(true)
    await expect(`empty? [1]`).toEvaluateTo(false)

    await expect(`empty? [=]`).toEvaluateTo(true)
    await expect(`empty? [a=true]`).toEvaluateTo(false)

    await expect(`empty? ''`).toEvaluateTo(true)
    await expect(`empty? 'cat'`).toEvaluateTo(false)
    await expect(`empty? meow`).toEvaluateTo(false)
  })

  test('list.filter keeps matching elements', async () => {
    await expect(`
      is-positive = do x:
        x == 3 or x == 4 or x == 5
      end
      list.filter [1 2 3 4 5] is-positive
    `).toEvaluateTo([3, 4, 5])
  })

  test('list.reject doesnt keep matching elements', async () => {
    await expect(`
      is-even = do x:
        (x % 2) == 0
      end
      list.reject [1 2 3 4 5] is-even
    `).toEvaluateTo([1, 3, 5])
  })

  test('list.reduce accumulates values', async () => {
    await expect(`
      add = do acc x:
        acc + x
      end
      list.reduce [1 2 3 4] add 0
    `).toEvaluateTo(10)
  })

  test('list.find returns first match', async () => {
    await expect(`
      is-four = do x:
        x == 4
      end
      list.find [1 2 4 5] is-four
    `).toEvaluateTo(4)
  })

  test('list.find returns null if no match', async () => {
    await expect(`
      is-ten = do x: x == 10 end
      list.find [1 2 3] is-ten
    `).toEvaluateTo(null)
  })

  test('list.empty? checks if list is empty', async () => {
    await expect(`list.empty? []`).toEvaluateTo(true)
    await expect(`list.empty? [1]`).toEvaluateTo(false)
  })

  test('list.contains? checks for element', async () => {
    await expect(`list.contains? [1 2 3] 2`).toEvaluateTo(true)
    await expect(`list.contains? [1 2 3] 5`).toEvaluateTo(false)
  })

  test('list.reverse reverses array', async () => {
    await expect(`list.reverse [1 2 3]`).toEvaluateTo([3, 2, 1])
  })

  test('list.concat combines arrays', async () => {
    await expect(`list.concat [1 2] [3 4]`).toEvaluateTo([1, 2, 3, 4])
  })

  test('list.flatten flattens nested arrays', async () => {
    await expect(`list.flatten [[1 2] [3 4]] 1`).toEvaluateTo([1, 2, 3, 4])
  })

  test('list.unique removes duplicates', async () => {
    await expect(`list.unique [1 2 2 3 1]`).toEvaluateTo([1, 2, 3])
  })

  test('list.zip combines two arrays', async () => {
    await expect(`list.zip [1 2] [3 4]`).toEvaluateTo([[1, 3], [2, 4]])
  })

  test('list.first returns first element', async () => {
    await expect(`list.first [1 2 3]`).toEvaluateTo(1)
    await expect(`list.first []`).toEvaluateTo(null)
  })

  test('list.last returns last element', async () => {
    await expect(`list.last [1 2 3]`).toEvaluateTo(3)
    await expect(`list.last []`).toEvaluateTo(null)
  })

  test('list.rest returns all but first', async () => {
    await expect(`list.rest [1 2 3]`).toEvaluateTo([2, 3])
  })

  test('list.take returns first n elements', async () => {
    await expect(`list.take [1 2 3 4 5] 3`).toEvaluateTo([1, 2, 3])
  })

  test('list.drop skips first n elements', async () => {
    await expect(`list.drop [1 2 3 4 5] 2`).toEvaluateTo([3, 4, 5])
  })

  test('list.append adds to end', async () => {
    await expect(`list.append [1 2] 3`).toEvaluateTo([1, 2, 3])
  })

  test('list.prepend adds to start', async () => {
    await expect(`list.prepend [2 3] 1`).toEvaluateTo([1, 2, 3])
  })

  test('list.index-of finds element index', async () => {
    await expect(`list.index-of [1 2 3] 2`).toEvaluateTo(1)
    await expect(`list.index-of [1 2 3] 5`).toEvaluateTo(-1)
  })

  test('list.push adds to end and mutates array', async () => {
    await expect(`arr = [1 2]; list.push arr 3; arr`).toEvaluateTo([1, 2, 3])
  })

  test('list.push returns the size of the array', async () => {
    await expect(`arr = [1 2]; arr | list.push 3`).toEvaluateTo(3)
  })

  test('list.pop removes from end and mutates array', async () => {
    await expect(`arr = [1 2 3]; list.pop arr; arr`).toEvaluateTo([1, 2])
  })

  test('list.pop returns removed element', async () => {
    await expect(`list.pop [1 2 3]`).toEvaluateTo(3)
  })

  test('list.pop returns null for empty array', async () => {
    await expect(`list.pop []`).toEvaluateTo(null)
  })

  test('list.shift removes from start and mutates array', async () => {
    await expect(`arr = [1 2 3]; list.shift arr; arr`).toEvaluateTo([2, 3])
  })

  test('list.shift returns removed element', async () => {
    await expect(`list.shift [1 2 3]`).toEvaluateTo(1)
  })

  test('list.shift returns null for empty array', async () => {
    await expect(`list.shift []`).toEvaluateTo(null)
  })

  test('list.unshift adds to start and mutates array', async () => {
    await expect(`arr = [2 3]; list.unshift arr 1; arr`).toEvaluateTo([1, 2, 3])
  })

  test('list.unshift returns the length of the array', async () => {
    await expect(`arr = [2 3]; arr | list.unshift 1`).toEvaluateTo(3)
  })

  test('list.splice removes elements and mutates array', async () => {
    await expect(`arr = [1 2 3 4 5]; list.splice arr 1 2; arr`).toEvaluateTo([1, 4, 5])
  })

  test('list.splice returns removed elements', async () => {
    await expect(`list.splice [1 2 3 4 5] 1 2`).toEvaluateTo([2, 3])
  })

  test('list.splice from start', async () => {
    await expect(`list.splice [1 2 3 4 5] 0 2`).toEvaluateTo([1, 2])
  })

  test('list.splice to end', async () => {
    await expect(`arr = [1 2 3 4 5]; list.splice arr 3 2; arr`).toEvaluateTo([1, 2, 3])
  })

  test('list.insert adds element at index and mutates array', async () => {
    await expect(`arr = [1 2 4 5]; list.insert arr 2 3; arr`).toEvaluateTo([1, 2, 3, 4, 5])
  })

  test('list.insert returns array length', async () => {
    await expect(`list.insert [1 2 4] 2 3`).toEvaluateTo(4)
  })

  test('list.insert at start', async () => {
    await expect(`arr = [2 3]; list.insert arr 0 1; arr`).toEvaluateTo([1, 2, 3])
  })

  test('list.insert at end', async () => {
    await expect(`arr = [1 2]; list.insert arr 2 99; arr`).toEvaluateTo([1, 2, 99])
  })

  test('list.sort with no callback sorts ascending', async () => {
    await expect(`list.sort [3 1 4 1 5] null`).toEvaluateTo([1, 1, 3, 4, 5])
  })

  test('list.sort with callback sorts using comparator', async () => {
    await expect(`
      desc = do a b:
        b - a
      end
      list.sort [3 1 4 1 5] desc
    `).toEvaluateTo([5, 4, 3, 1, 1])
  })

  test('list.sort with callback for strings by length', async () => {
    await expect(`
      by-length = do a b:
        (length a) - (length b)
      end
      list.sort ['cat' 'a' 'dog' 'elephant'] by-length
    `).toEvaluateTo(['a', 'cat', 'dog', 'elephant'])
  })

  test('list.any? checks if any element matches', async () => {
    await expect(`
      gt-three = do x: x > 3 end
      list.any? [1 2 4 5] gt-three
    `).toEvaluateTo(true)
    await expect(`
      gt-ten = do x: x > 10 end
      list.any? [1 2 3] gt-ten
    `).toEvaluateTo(false)
  })

  test('list.all? checks if all elements match', async () => {
    await expect(`
      positive = do x: x > 0 end
      list.all? [1 2 3] positive
    `).toEvaluateTo(true)
    await expect(`
      positive = do x: x > 0 end
      list.all? [1 -2 3] positive
    `).toEvaluateTo(false)
  })

  test('list.sum adds all numbers', async () => {
    await expect(`list.sum [1 2 3 4]`).toEvaluateTo(10)
    await expect(`list.sum []`).toEvaluateTo(0)
  })

  test('list.count counts matching elements', async () => {
    await expect(`
      gt-two = do x: x > 2 end
      list.count [1 2 3 4 5] gt-two
    `).toEvaluateTo(3)
  })

  test('list.partition splits array by predicate', async () => {
    await expect(`
      gt-two = do x: x > 2 end
      list.partition [1 2 3 4 5] gt-two
    `).toEvaluateTo([[3, 4, 5], [1, 2]])
  })

  test('list.compact removes null values', async () => {
    await expect(`list.compact [1 null 2 null 3]`).toEvaluateTo([1, 2, 3])
  })

  test('list.group-by groups by key function', async () => {
    await expect(`
      get-type = do x:
        if (string? x):
          'str'
        else:
          'num'
        end
      end
      list.group-by ['a' 1 'b' 2] get-type
    `).toEvaluateTo({ str: ['a', 'b'], num: [1, 2] })
  })
})

describe('enumerables', () => {
  test('map transforms array elements', async () => {
    await expect(`
      double = do x: x * 2 end
      list.map [1 2 3] double
    `).toEvaluateTo([2, 4, 6])
  })

  test('map handles empty array', async () => {
    await expect(`
      double = do x: x * 2 end
      list.map [] double
    `).toEvaluateTo([])
  })

  test('each iterates over array', async () => {
    // Note: each doesn't return the results, it returns null
    // We can test it runs by checking the return value
    await expect(`
      double = do x: x * 2 end
      each [1 2 3] double
    `).toEvaluateTo([1, 2, 3])
  })

  test('each handles empty array', async () => {
    await expect(`
      fn = do x: x end
      each [] fn
    `).toEvaluateTo([])
  })
})

describe('dict operations', () => {
  test('dict.keys returns all keys', async () => {
    await expect(`dict.keys [a=1 b=2 c=3] | list.sort`).toEvaluateTo(['a', 'b', 'c'].sort())
  })

  test('dict.values returns all values', async () => {
    await expect('dict.values [a=1 b=2] | list.sort').toEvaluateTo([1, 2].sort())
  })

  test('dict.has? checks for key', async () => {
    await expect(`dict.has? [a=1 b=2] 'a'`).toEvaluateTo(true)
    await expect(`dict.has? [a=1 b=2] 'c'`).toEvaluateTo(false)
  })

  test('dict.get retrieves value with default', async () => {
    await expect(`dict.get [a=1] 'a' 0`).toEvaluateTo(1)
    await expect(`dict.get [a=1] 'b' 99`).toEvaluateTo(99)
    await expect(`dict.get [a=1] 'b'`).toEvaluateTo(null)
  })

  test('dict.set sets value', async () => {
    await expect(`map = [a=1]; dict.set map 'b' 99; map.b`).toEvaluateTo(99)
    await expect(`map = [a=1]; dict.set map 'a' 100; map.a`).toEvaluateTo(100)
  })

  test('dict.empty? checks if dict is empty', async () => {
    await expect(`dict.empty? [=]`).toEvaluateTo(true)
    await expect(`dict.empty? [a=1]`).toEvaluateTo(false)
  })

  test('dict.merge combines dicts', async () => {
    await expect(`dict.merge [a=1] [b=2]`).toEvaluateTo({ a: 1, b: 2 })
  })

  test('dict.map transforms values', async () => {
    await expect(`
      double = do v k: v * 2 end
      dict.map [a=1 b=2] double
    `).toEvaluateTo({ a: 2, b: 4 })
  })

  test('dict.filter keeps matching entries', async () => {
    await expect(`
      gt-one = do v k: v > 1 end
      dict.filter [a=1 b=2 c=3] gt-one
    `).toEvaluateTo({ b: 2, c: 3 })
  })

  test('dict.from-entries creates dict from array', async () => {
    await expect(`dict.from-entries [['a' 1] ['b' 2]]`).toEvaluateTo({ a: 1, b: 2 })
  })
})

describe('math operations', () => {
  test('math.abs returns absolute value', async () => {
    await expect(`math.abs -5`).toEvaluateTo(5)
    await expect(`math.abs 5`).toEvaluateTo(5)
  })

  test('math.floor rounds down', async () => {
    await expect(`math.floor 3.7`).toEvaluateTo(3)
  })

  test('math.ceil rounds up', async () => {
    await expect(`math.ceil 3.2`).toEvaluateTo(4)
  })

  test('math.round rounds to nearest', async () => {
    await expect(`math.round 3.4`).toEvaluateTo(3)
    await expect(`math.round 3.6`).toEvaluateTo(4)
  })

  test('math.min returns minimum', async () => {
    await expect(`math.min 5 2 8 1`).toEvaluateTo(1)
  })

  test('math.max returns maximum', async () => {
    await expect(`math.max 5 2 8 1`).toEvaluateTo(8)
  })

  test('math.pow computes power', async () => {
    await expect(`math.pow 2 3`).toEvaluateTo(8)
  })

  test('math.sqrt computes square root', async () => {
    await expect(`math.sqrt 16`).toEvaluateTo(4)
  })

  test('math.even? checks if even', async () => {
    await expect(`math.even? 4`).toEvaluateTo(true)
    await expect(`math.even? 5`).toEvaluateTo(false)
  })

  test('math.odd? checks if odd', async () => {
    await expect(`math.odd? 5`).toEvaluateTo(true)
    await expect(`math.odd? 4`).toEvaluateTo(false)
  })

  test('math.positive? checks if positive', async () => {
    await expect(`math.positive? 5`).toEvaluateTo(true)
    await expect(`math.positive? -5`).toEvaluateTo(false)
    await expect(`math.positive? 0`).toEvaluateTo(false)
  })

  test('math.negative? checks if negative', async () => {
    await expect(`math.negative? -5`).toEvaluateTo(true)
    await expect(`math.negative? 5`).toEvaluateTo(false)
  })

  test('math.zero? checks if zero', async () => {
    await expect(`math.zero? 0`).toEvaluateTo(true)
    await expect(`math.zero? 5`).toEvaluateTo(false)
  })

  test('math.clamp restricts value to range', async () => {
    await expect(`math.clamp 5 0 10`).toEvaluateTo(5)
    await expect(`math.clamp -5 0 10`).toEvaluateTo(0)
    await expect(`math.clamp 15 0 10`).toEvaluateTo(10)
  })

  test('math.sign returns sign of number', async () => {
    await expect(`math.sign 5`).toEvaluateTo(1)
    await expect(`math.sign -5`).toEvaluateTo(-1)
    await expect(`math.sign 0`).toEvaluateTo(0)
  })

  test('math.trunc truncates decimal', async () => {
    await expect(`math.trunc 3.7`).toEvaluateTo(3)
    await expect(`math.trunc -3.7`).toEvaluateTo(-3)
  })
})

// describe('echo', () => {
//   test('echo returns null value', async () => {
//     await expect(`echo 'hello' 'world'`).toEvaluateTo(null, globalFunctions)
//   })

//   test('echo with array', async () => {
//     await expect(`echo [1 2 3]`).toEvaluateTo(null, globalFunctions)
//   })

//   test('echo with multiple arguments', async () => {
//     await expect(`echo 'test' 42 true`).toEvaluateTo(null, globalFunctions)
//   })
// })