From 14dcb63c3348e152595a64896b682f9f88ec97f9 Mon Sep 17 00:00:00 2001
From: Yuichi Nishiwaki <yuichi.nishiwaki@gmail.com>
Date: Sun, 27 Oct 2013 02:51:06 +0900
Subject: [PATCH] support `(define (foo . args) . body)` style definition

---
 src/codegen.c | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/codegen.c b/src/codegen.c
index aad6e6ba..eebce9d4 100644
--- a/src/codegen.c
+++ b/src/codegen.c
@@ -200,20 +200,32 @@ codegen(codegen_state *state, pic_value obj)
     proc = pic_car(pic, obj);
     if (pic_eq_p(pic, proc, pic->sDEFINE)) {
       int idx;
-      pic_value var;
+      pic_value var, val;
 
-      if (pic_length(pic, obj) != 3) {
+      if (pic_length(pic, obj) < 3) {
 	pic_error(pic, "syntax error");
       }
 
       var = pic_car(pic, pic_cdr(pic, obj));
+      if (pic_pair_p(var)) {
+	val = pic_cons(pic, pic->sLAMBDA,
+		       pic_cons(pic, pic_cdr(pic, var),
+				pic_cdr(pic, pic_cdr(pic, obj))));
+	var = pic_car(pic, var);
+      }
+      else {
+	if (pic_length(pic, obj) != 3) {
+	  pic_error(pic, "syntax error");
+	}
+	val = pic_car(pic, pic_cdr(pic, pic_cdr(pic, obj)));
+      }
       if (! pic_symbol_p(var)) {
 	pic_error(pic, "syntax error");
       }
 
       idx = scope_global_define(pic, pic_symbol_ptr(var)->name);
 
-      codegen(state, pic_car(pic, pic_cdr(pic, pic_cdr(pic, obj))));
+      codegen(state, val);
       irep->code[irep->clen].insn = OP_GSET;
       irep->code[irep->clen].u.i = idx;
       irep->clen++;