From 8085fb7725f19b18b71ecfdd220cb83a3de15666 Mon Sep 17 00:00:00 2001
From: hyung-hwan <hyunghwan.chung@gmail.com>
Date: Tue, 14 Nov 2023 17:57:20 +0900
Subject: [PATCH] changed class handling to check for the block expression if
 enabled

---
 lib/comp.c    | 15 ++++++++++++---
 lib/hcl-prv.h |  1 +
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/lib/comp.c b/lib/comp.c
index 2017a7e..46a1241 100644
--- a/lib/comp.c
+++ b/lib/comp.c
@@ -2350,12 +2350,14 @@ static int compile_class (hcl_t* hcl, hcl_cnode_t* src)
 		cf = GET_SUBCFRAME(hcl);
 		cf->u._class.nsuperclasses = 0; /* unsed for CLASS_P2 */
 		cf->u._class.start_loc = *HCL_CNODE_GET_LOC(src); /* TODO: use *HCL_CNODE_GET_LOC(cmd) instead? */
+		cf->u._class.cmd_cnode = cmd;
 
 		obj = HCL_CNODE_CONS_CDR(obj);
 		PUSH_SUBCFRAME (hcl, COP_COMPILE_CLASS_P1, obj); /* 2 - variables declaraions and actual body */
 		cf = GET_SUBCFRAME(hcl);
 		cf->u._class.nsuperclasses = 1; /* this one needs to change if we support multiple superclasses... */
 		cf->u._class.start_loc = *HCL_CNODE_GET_LOC(src); /* TODO: use *HCL_CNODE_GET_LOC(cmd) instead? */
+		cf->u._class.cmd_cnode = cmd;
 	}
 	else
 	{
@@ -2364,11 +2366,13 @@ static int compile_class (hcl_t* hcl, hcl_cnode_t* src)
 		cf = GET_TOP_CFRAME(hcl);
 		cf->u._class.nsuperclasses = 0; /* this one needs to change if we support multiple superclasses... */
 		cf->u._class.start_loc = *HCL_CNODE_GET_LOC(src); /* TODO: use *HCL_CNODE_GET_LOC(cmd) instead? */
+		cf->u._class.cmd_cnode = cmd;
 
 		PUSH_SUBCFRAME (hcl, COP_COMPILE_CLASS_P2, class_name); /* 2 */
 		cf = GET_SUBCFRAME(hcl);
 		cf->u._class.nsuperclasses = 0; /* unsed for CLASS_P2 */
 		cf->u._class.start_loc = *HCL_CNODE_GET_LOC(src); /* TODO: use *HCL_CNODE_GET_LOC(cmd) instead? */
+		cf->u._class.cmd_cnode = cmd;
 	}
 
 	return 0;
@@ -2399,13 +2403,13 @@ static HCL_INLINE int compile_class_p1 (hcl_t* hcl)
 		hcl_oow_t dclcount;
 /*
  (defclass X ::: T
-    ::: | a b c | ; class variables
+    ::: | a b c | ## class variables
  )
  (defclass X
-    ::: T | a b c | ; instance varaiables.
+    ::: T | a b c | ## instance varaiables.
  )
  (defclass X
-    ::: | a b c | ; class variables
+    ::: | a b c | ## class variables
  )
 */
 		tmp = HCL_CNODE_CONS_CAR(obj);
@@ -2496,6 +2500,11 @@ static HCL_INLINE int compile_class_p1 (hcl_t* hcl)
 		if (emit_push_literal(hcl, tmp, &cf->u._class.start_loc) <= -1) goto oops;
 	}
 
+	if (hcl->c->flags & HCL_COMPILE_ENABLE_BLOCK)
+	{
+		if (check_block_expression_as_body(hcl, obj, cf->u._class.cmd_cnode, 0) <= -1) return -1;
+	}
+
 	if (push_clsblk(hcl, &cf->u._class.start_loc, nivars, ncvars, &hcl->c->tv.s.ptr[ivar_start], ivar_len, &hcl->c->tv.s.ptr[cvar_start], cvar_len) <= -1) goto oops;
 	if (push_cblk(hcl, &cf->u._class.start_loc, HCL_CBLK_TYPE_CLASS) <= -1) goto oops; /* the class block shall be treated as a control block, too */
 
diff --git a/lib/hcl-prv.h b/lib/hcl-prv.h
index 5321d88..175e5cb 100644
--- a/lib/hcl-prv.h
+++ b/lib/hcl-prv.h
@@ -454,6 +454,7 @@ struct hcl_cframe_t
 		{
 			hcl_ooi_t nsuperclasses;
 			hcl_loc_t start_loc;
+			hcl_cnode_t* cmd_cnode;
 		} _class;
 	} u;
 };