add var? and var

src/prelude/index.ts

@@ -1,7 +1,7 @@
 // The prelude creates all the builtin Shrimp functions.
 
 import {
-  type Value, toValue,
+  type Value, type VM, toValue,
   extractParamInfo, isWrapped, getOriginalFunction,
 } from 'reefvm'
 
@@ -34,13 +34,11 @@ export const globals = {
     const val = toValue(v)
     return `#<${val.type}: ${formatValue(val)}>`
   },
-  length: (v: any) => {
-    const value = toValue(v)
-    switch (value.type) {
-      case 'string': case 'array': return value.value.length
-      case 'dict': return value.value.size
-      default: throw new Error(`length: expected string, array, or dict, got ${value.type}`)
-    }
+  var: function (this: VM, v: any) {
+    return typeof v === 'string' ? this.scope.get(v) : v
+  },
+  'var?': function (this: VM, v: string) {
+    return typeof v !== 'string' || this.scope.has(v)
   },
 
   // type predicates
@@ -65,6 +63,14 @@ export const globals = {
   identity: (v: any) => v,
 
   // collections
+  length: (v: any) => {
+    const value = toValue(v)
+    switch (value.type) {
+      case 'string': case 'array': return value.value.length
+      case 'dict': return value.value.size
+      default: throw new Error(`length: expected string, array, or dict, got ${value.type}`)
+    }
+  },
   at: (collection: any, index: number | string) => {
     const value = toValue(collection)
     if (value.type === 'string' || value.type === 'array') {

src/prelude/tests/info.test.ts

@@ -0,0 +1,79 @@
+import { expect, describe, test } from 'bun:test'
+import { globals } from '#prelude'
+
+describe('var and var?', () => {
+  test('var? checks if a variable exists', async () => {
+    await expect(`var? 'nada'`).toEvaluateTo(false, globals)
+    await expect(`var? 'info'`).toEvaluateTo(false, globals)
+    await expect(`abc = abc; var? 'abc'`).toEvaluateTo(true, globals)
+    await expect(`var? 'var?'`).toEvaluateTo(true, globals)
+
+    await expect(`var? 'dict'`).toEvaluateTo(true, globals)
+    await expect(`var? dict`).toEvaluateTo(true, globals)
+  })
+
+  test('var returns a value or null', async () => {
+    await expect(`var 'nada'`).toEvaluateTo(null, globals)
+    await expect(`var nada`).toEvaluateTo(null, globals)
+    await expect(`var 'info'`).toEvaluateTo(null, globals)
+    await expect(`abc = my-string; var 'abc'`).toEvaluateTo('my-string', globals)
+    await expect(`abc = my-string; var abc`).toEvaluateTo(null, globals)
+  })
+})
+
+describe('type predicates', () => {
+  test('string? checks for string type', async () => {
+    await expect(`string? 'hello'`).toEvaluateTo(true, globals)
+    await expect(`string? 42`).toEvaluateTo(false, globals)
+  })
+
+  test('number? checks for number type', async () => {
+    await expect(`number? 42`).toEvaluateTo(true, globals)
+    await expect(`number? 'hello'`).toEvaluateTo(false, globals)
+  })
+
+  test('boolean? checks for boolean type', async () => {
+    await expect(`boolean? true`).toEvaluateTo(true, globals)
+    await expect(`boolean? 42`).toEvaluateTo(false, globals)
+  })
+
+  test('array? checks for array type', async () => {
+    await expect(`array? [1 2 3]`).toEvaluateTo(true, globals)
+    await expect(`array? 42`).toEvaluateTo(false, globals)
+  })
+
+  test('dict? checks for dict type', async () => {
+    await expect(`dict? [a=1]`).toEvaluateTo(true, globals)
+    await expect(`dict? []`).toEvaluateTo(false, globals)
+  })
+
+  test('null? checks for null type', async () => {
+    await expect(`null? null`).toEvaluateTo(true, globals)
+    await expect(`null? 42`).toEvaluateTo(false, globals)
+  })
+
+  test('some? checks for non-null', async () => {
+    await expect(`some? 42`).toEvaluateTo(true, globals)
+    await expect(`some? null`).toEvaluateTo(false, globals)
+  })
+})
+
+describe('introspection', () => {
+  test('type returns proper types', async () => {
+    await expect(`type 'hello'`).toEvaluateTo('string', globals)
+    await expect(`type 42`).toEvaluateTo('number', globals)
+    await expect(`type true`).toEvaluateTo('boolean', globals)
+    await expect(`type false`).toEvaluateTo('boolean', globals)
+    await expect(`type null`).toEvaluateTo('null', globals)
+    await expect(`type [1 2 3]`).toEvaluateTo('array', globals)
+    await expect(`type [a=1 b=2]`).toEvaluateTo('dict', globals)
+  })
+
+  test('inspect formats values', async () => {
+    await expect(`inspect 'hello'`).toEvaluateTo("\u001b[32m'hello\u001b[32m'\u001b[0m", globals)
+  })
+
+  test('describe describes values', async () => {
+    await expect(`describe 'hello'`).toEvaluateTo("#<string: \u001b[32m'hello\u001b[32m'\u001b[0m>", globals)
+  })
+})

src/prelude/tests/prelude.test.ts

@@ -98,43 +98,6 @@ describe('string operations', () => {
   })
 })
 
-describe('type predicates', () => {
-  test('string? checks for string type', async () => {
-    await expect(`string? 'hello'`).toEvaluateTo(true, globals)
-    await expect(`string? 42`).toEvaluateTo(false, globals)
-  })
-
-  test('number? checks for number type', async () => {
-    await expect(`number? 42`).toEvaluateTo(true, globals)
-    await expect(`number? 'hello'`).toEvaluateTo(false, globals)
-  })
-
-  test('boolean? checks for boolean type', async () => {
-    await expect(`boolean? true`).toEvaluateTo(true, globals)
-    await expect(`boolean? 42`).toEvaluateTo(false, globals)
-  })
-
-  test('array? checks for array type', async () => {
-    await expect(`array? [1 2 3]`).toEvaluateTo(true, globals)
-    await expect(`array? 42`).toEvaluateTo(false, globals)
-  })
-
-  test('dict? checks for dict type', async () => {
-    await expect(`dict? [a=1]`).toEvaluateTo(true, globals)
-    await expect(`dict? []`).toEvaluateTo(false, globals)
-  })
-
-  test('null? checks for null type', async () => {
-    await expect(`null? null`).toEvaluateTo(true, globals)
-    await expect(`null? 42`).toEvaluateTo(false, globals)
-  })
-
-  test('some? checks for non-null', async () => {
-    await expect(`some? 42`).toEvaluateTo(true, globals)
-    await expect(`some? null`).toEvaluateTo(false, globals)
-  })
-})
-
 describe('boolean logic', () => {
   test('not negates value', async () => {
     await expect(`not true`).toEvaluateTo(false, globals)
@@ -161,17 +124,7 @@ describe('utilities', () => {
   })
 })
 
-describe('introspection', () => {
-  test('type returns proper types', async () => {
-    await expect(`type 'hello'`).toEvaluateTo('string', globals)
-    await expect(`type 42`).toEvaluateTo('number', globals)
-    await expect(`type true`).toEvaluateTo('boolean', globals)
-    await expect(`type false`).toEvaluateTo('boolean', globals)
-    await expect(`type null`).toEvaluateTo('null', globals)
-    await expect(`type [1 2 3]`).toEvaluateTo('array', globals)
-    await expect(`type [a=1 b=2]`).toEvaluateTo('dict', globals)
-  })
-
+describe('collections', () => {
   test('length', async () => {
     await expect(`length 'hello'`).toEvaluateTo(5, globals)
     await expect(`length [1 2 3]`).toEvaluateTo(3, globals)
@@ -184,20 +137,6 @@ describe('introspection', () => {
     await expect(`try: length null catch e: 'error' end`).toEvaluateTo('error', globals)
   })
 
-  test('inspect formats values', async () => {
-    // Just test that inspect returns something for now
-    // (we'd need more complex assertion to check the actual format)
-    await expect(`type (inspect 'hello')`).toEvaluateTo('string', globals)
-  })
-
-  test('describe describes values', async () => {
-    // Just test that inspect returns something for now
-    // (we'd need more complex assertion to check the actual format)
-    await expect(`describe 'hello'`).toEvaluateTo("#<string: \u001b[32m'hello\u001b[32m'\u001b[0m>", globals)
-  })
-})
-
-describe('collections', () => {
   test('literal array creates array from arguments', async () => {
     await expect(`[ 1 2 3 ]`).toEvaluateTo([1, 2, 3], globals)
     await expect(`['a' 'b']`).toEvaluateTo(['a', 'b'], globals)