stone-lang 0.1.0

package/frontend/types/index.js

@@ -0,0 +1,5 @@
+/**
+ * Frontend Types - Barrel exports
+ */
+export * from './types.js';
+export * from './operatorMap.js';

package/frontend/types/operatorMap.js

@@ -0,0 +1,134 @@
+/**
+ * Stone Operator to Function Mapping
+ *
+ * Maps operators to function names for desugaring.
+ * Operators are transformed into function calls which are then
+ * resolved via the overload system.
+ *
+ * Example: a + b  →  add(a, b)
+ */
+
+/**
+ * Binary operators mapped to function names
+ *
+ * Scalar operators (+, -, *, /) only work on scalars.
+ * Elementwise operators (.+, .-, .*, ./) work on arrays.
+ * Using scalar operators on arrays is a type error.
+ */
+export const BINARY_OP_NAMES = {
+  // Scalar arithmetic (num, complex, string only)
+  "+": "add",
+  "-": "sub",
+  "*": "mul",
+  "/": "div",
+  "%": "mod",
+  "^": "pow",
+
+  // Elementwise arithmetic (works on arrays)
+  ".+": "elem_add",
+  ".-": "elem_sub",
+  ".*": "elem_mul",
+  "./": "elem_div",
+  ".^": "elem_pow",
+
+  // Matrix operations
+  "@": "matmul",
+
+  // Comparison
+  "<": "lt",
+  ">": "gt",
+  "<=": "lte",
+  ">=": "gte",
+  "==": "eq",
+  "!=": "neq",
+};
+
+/**
+ * Unary operators mapped to function names
+ */
+export const UNARY_OP_NAMES = {
+  "-": "neg",
+  "!": "not",
+};
+
+/**
+ * Postfix operators mapped to function names
+ */
+export const POSTFIX_OP_NAMES = {};
+
+/**
+ * Index access operator
+ * arr[i]  →  index_get(arr, i)
+ */
+export const INDEX_OP_NAME = "index_get";
+
+/**
+ * Operators that do NOT desugar to functions
+ * These have special semantics that require direct handling.
+ */
+export const SPECIAL_CASES = {
+  // Short-circuit evaluation - cannot be desugared
+  "&&": { reason: "short-circuit evaluation" },
+  "||": { reason: "short-circuit evaluation" },
+
+  // Static member access - resolved at compile time
+  ".": { reason: "static member access" },
+
+  // Assignment - modifies environment
+  "=": { reason: "assignment" },
+
+  // Destructuring assignment
+  ":=": { reason: "destructuring assignment" },
+};
+
+/**
+ * Check if an operator should be desugared to a function call
+ *
+ * @param {string} op - Operator symbol
+ * @returns {boolean}
+ */
+export function shouldDesugar(op) {
+  return !SPECIAL_CASES.hasOwnProperty(op);
+}
+
+/**
+ * Get the function name for a binary operator
+ *
+ * @param {string} op - Operator symbol
+ * @returns {string|null} Function name or null if not found
+ */
+export function getBinaryOpName(op) {
+  return BINARY_OP_NAMES[op] || null;
+}
+
+/**
+ * Get the function name for a unary operator
+ *
+ * @param {string} op - Operator symbol
+ * @returns {string|null} Function name or null if not found
+ */
+export function getUnaryOpName(op) {
+  return UNARY_OP_NAMES[op] || null;
+}
+
+/**
+ * Get the function name for a postfix operator
+ *
+ * @param {string} op - Operator symbol
+ * @returns {string|null} Function name or null if not found
+ */
+export function getPostfixOpName(op) {
+  return POSTFIX_OP_NAMES[op] || null;
+}
+
+export default {
+  BINARY_OP_NAMES,
+  UNARY_OP_NAMES,
+  POSTFIX_OP_NAMES,
+  INDEX_OP_NAME,
+  SPECIAL_CASES,
+  shouldDesugar,
+  getBinaryOpName,
+  getUnaryOpName,
+  getPostfixOpName,
+};