fix multiple argument

Author: codehz

src/easy_sqlite3/macros.nim

@@ -26,20 +26,21 @@ proc injectDbFetch(procbody: var NimNode, sql: string, db_ident, st_ident: NimNo
 proc injectDbArguments(procbody: var NimNode, body, st_ident: NimNode): seq[tuple[name: string, idxnode, param: NimNode]] =
   result = newSeq[tuple[name: string, idxnode, param: NimNode]]()
   procbody.addTree(nnkVarSection, varsec):
-    for arg in body[3][1..^1]:
-      let arg_name = $arg[0]
-      let arg_ident = genSym(nskVar, arg_name & "_idx")
-      result.add (name: arg_name, idxnode: arg_ident, param: arg[0])
-      varsec.add nnkIdentDefs.newTree(
-        nnkPragmaExpr.newTree(
-          arg_ident,
-          nnkPragma.newTree(
-            ident "threadvar"
-          )
-        ),
-        ident "int",
-        newEmptyNode()
-      )
+    for args in body[3][1..^1]:
+      for arg in args[0..^3]:
+        let arg_name = $arg
+        let arg_ident = genSym(nskVar, arg_name & "_idx")
+        result.add (name: arg_name, idxnode: arg_ident, param: arg)
+        varsec.add nnkIdentDefs.newTree(
+          nnkPragmaExpr.newTree(
+            arg_ident,
+            nnkPragma.newTree(
+              ident "threadvar"
+            )
+          ),
+          ident "int",
+          newEmptyNode()
+        )
   for it in result:
     procbody.add nnkIfStmt.newTree(
       nnkElifBranch.newTree(

tests/test_basic.nim

@@ -4,6 +4,7 @@ import easy_sqlite3
 import easy_sqlite3/[memfs, logfs]
 
 proc select_1(arg: int): tuple[value: int] {.importdb: "SELECT $arg".}
+proc select_2(arg1, arg2: int): tuple[value: int, value2: int] {.importdb: "SELECT $arg1, $arg2".}
 
 proc insert_data(name: string, value: int) {.importdb: """
   INSERT INTO mydata(name, value) VALUES ($name, $value);
@@ -19,6 +20,10 @@ test "simple":
   var db = initDatabase(":memory:")
   check db.select_1(1) == (value: 1)
 
+test "multiple":
+  var db = initDatabase(":memory:")
+  check db.select_2(1, 2) == (value: 1, value2: 2)
+
 test "full":
   const dataset = {
     "A": 0,