yukigo-haskell-parser 0.1.0

package/src/utils/helpers.ts

@@ -0,0 +1,29 @@
+import { inspect } from "util";
+import { AST } from "yukigo-ast";
+import { Function } from "yukigo-ast";
+
+export function groupFunctionDeclarations(ast: AST): AST {
+  const groups: Record<string, Function[]> = {};
+  const others: AST = [];
+
+  for (const node of ast) {
+    if (node instanceof Function) {
+      const name = node.identifier.value;
+      if (!groups[name]) {
+        groups[name] = [];
+      }
+      groups[name].push(node);
+    } else {
+      others.push(node);
+    }
+  }
+
+  // Merge equations for each function name
+  const functionGroups: Function[] = Object.values(groups).map((functions) => {
+    const identifier = functions[0].identifier;
+    const allEquations = functions.flatMap((f) => f.equations);
+    return new Function(identifier, allEquations, identifier.loc);
+  });
+
+  return [...others, ...functionGroups];
+}

package/src/utils/types.ts

@@ -0,0 +1,56 @@
+import { YukigoPrimitive } from "yukigo-ast";
+
+export const keywords = [
+  "type",
+  "where",
+  "in",
+  "if",
+  "else",
+  "then",
+  "data",
+  "case",
+  "class",
+  "do",
+  "default",
+  "deriving",
+  "import",
+  "infix",
+  "infixl",
+  "infixr",
+  "instance",
+  "let",
+  "module",
+  "newtype",
+  "of",
+  "qualified",
+  "hiding",
+  "foreign",
+];
+
+export const typeMappings: { [key: string]: YukigoPrimitive } = {
+  Float: "YuNumber",
+  Double: "YuNumber",
+  Int: "YuNumber",
+  Integer: "YuNumber",
+  String: "YuString",
+  Char: "YuChar",
+  Boolean: "YuBoolean",
+  Bool: "YuBoolean",
+};
+
+export const typeClasses: Map<string, string[]> = new Map([
+  ["Bounded", ["YuChar", "YuNumber"]],
+  ["Enum", ["YuChar", "YuNumber"]],
+  ["Ord", ["YuNumber", "YuString", "YuChar"]],
+  ["Eq", ["YuNumber", "YuString", "YuChar", "YuBoolean"]],
+  ["Floating", ["YuNumber"]],
+  ["Fractional", ["YuNumber"]],
+  ["Integral", ["YuNumber"]],
+  ["Num", ["YuNumber"]],
+  ["Random", ["YuBoolean", "YuChar", "YuNumber"]],
+  ["Read", ["YuList", "YuChar", "YuNumber"]],
+  ["Real", ["YuNumber"]],
+  ["RealFloat", ["YuNumber"]],
+  ["RealFrac", ["YuNumber"]],
+  ["Show", ["YuNumber", "YuString", "YuChar", "YuList"]],
+]);