1 files changed, 149 insertions, 0 deletions
diff --git a/src/honeysuckle/hs_parse_args.c b/src/honeysuckle/hs_parse_args.c
new file mode 100644
index 0000000..d6d3b3c
--- /dev/null
+++ b/src/honeysuckle/hs_parse_args.c
@@ -0,0 +1,149 @@
+#include <stdio.h>
+
+#include "honeysuckle.h"
+
+static bool check_parse(lua_State *L, int index, struct hs_arg *expected)
+{
+   switch(expected->type) {
+   case HS_BOOL:
+      if (!lua_isboolean(L, index))
+	 return false;
+      *(expected->ptr.boolean) = lua_toboolean(L, index);
+      return true;
+
+   case HS_INT:
+      if (!lua_isnumber(L, index))
+	 return false;
+      *(expected->ptr.integer) = lua_tointeger(L, index);
+      return true;
+
+   case HS_NUM:
+      if (!lua_isnumber(L, index))
+	 return false;
+      *(expected->ptr.number) = lua_tonumber(L, index);
+      return true;
+
+   case HS_STR:
+      if (!lua_isstring(L, index) || lua_isnumber(L, index))
+	 return false;
+      *(expected->ptr.string) = (char *) lua_tostring(L, index);
+      return true;
+
+   case HS_TBL:
+      if (!lua_istable(L, index))
+	 return false;
+      *(expected->ptr.stack_index) = index;
+      return true;
+
+   case HS_FUNC:
+      if (!lua_isfunction(L, index))
+	 return false;
+      *(expected->ptr.stack_index) = index;
+      return true;
+
+   case HS_CFUNC:
+      if (!lua_iscfunction(L, index))
+	 return false;
+      *(expected->ptr.function) = lua_tocfunction(L, index);
+      return true;
+
+   case HS_USER:
+      if (!lua_isuserdata(L, index))
+	 return false;
+      *(expected->ptr.userdata) = lua_touserdata(L, index);
+      return true;
+
+   case HS_LIGHT:
+      if (!lua_islightuserdata(L, index))
+	 return false;
+      *(expected->ptr.userdata) = lua_touserdata(L, index);
+      return true;
+
+   case HS_NIL:
+      if (!lua_isnil(L, index))
+	 return false;
+      *(expected->ptr.stack_index) = index;
+
+   case HS_ANY:
+      *(expected->ptr.stack_index) = index;
+      return true;
+
+   default:
+      return false;
+   }
+}
+
+
+static bool try_parse_args(lua_State *L, int n_args, struct hs_arg *arguments)
+{
+   int args_provided = lua_gettop(L);
+   if (args_provided != n_args)
+      return false;
+
+   for (int i=0; i<n_args; i++) {
+      bool success = check_parse(L, i+1, arguments + i);
+      if (!success)
+	 return false;
+   }
+   return true;
+}
+
+
+void hs_parse_args_(lua_State *L, int n_args, struct hs_arg *arguments)
+{
+   bool success = try_parse_args(L, n_args, arguments);
+   if (!success) {
+      lua_pushstring(L, "expected arguments of the form (");
+      for (int i=0; i<n_args; i++) {
+	 lua_pushstring(L, hs_type_to_string(arguments[i].type));
+	 lua_pushstring(L, ", ");
+      }
+      lua_pop(L, 1);
+      lua_pushstring(L, "); received (");
+      const int n_provided = lua_gettop(L);
+      for (int i=0; i<n_provided; i++) {
+	 lua_pushstring(L, lua_typename(L, lua_type(L, i+1)));
+	 lua_pushstring(L, ", ");
+      }
+      lua_pop(L, 1);
+      lua_pushstring(L, ") instead");
+      lua_concat(L, 1 + (2*n_args) + (2*n_provided));
+      lua_error(L);
+   }
+}
+
+
+// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+static void hs_overloaded_error(lua_State *L, va_list args)
+{
+   lua_pushstring(L, "hs_overloaded failed");
+   lua_error(L);
+}
+
+
+int hs_parse_overloaded_(lua_State *L, ...)
+{
+   va_list args, args_error;
+   va_start(args, L);
+   va_copy(args_error, args);
+
+   int choice = 0;
+
+   while(true) {
+      int n_args = va_arg(args, int);
+      if (n_args == -1)
+	 break;
+      else {
+	 struct hs_arg *arguments = va_arg(args, struct hs_arg *);
+	 if (try_parse_args(L, n_args, arguments))
+	    return choice;
+      }
+      choice++;
+   }
+
+   hs_overloaded_error(L, args_error);
+   
+   va_end(args);
+   va_end(args_error);
+}