feat: improve support for sqlite named parameters

Author: orisano

internal/endtoend/testdata/invalid_group_by_reference/sqlite/stderr.txt

@@ -1,2 +1,2 @@
 # package querytest
-query.sql:9:1: column reference "invalid_reference" not found
+query.sql:11:10: column reference "invalid_reference" not found

internal/endtoend/testdata/invalid_table_alias/sqlite/stderr.txt

@@ -1,2 +1,2 @@
 # package querytest
-query.sql:9:1: table alias "p" does not exist
+query.sql:11:9: table alias "p" does not exist

internal/endtoend/testdata/join_left/sqlite/go/query.sql.go

@@ -223,6 +223,7 @@ func (c *cc) convertFuncContext(n *parser.Expr_functionContext) ast.Node {
 				Args:        args,
 				AggOrder:    &ast.List{},
 				AggDistinct: n.DISTINCT_() != nil,
+				Location:    n.GetStart().GetStart(),
 			}
 		}
 	}
@@ -253,6 +254,7 @@ func (c *cc) convertColumnNameExpr(n *parser.Expr_qualified_column_nameContext)
 		Fields: &ast.List{
 			Items: items,
 		},
+		Location: n.GetStart().GetStart(),
 	}
 }
 
@@ -554,11 +556,38 @@ func (c *cc) convertBinaryNode(n *parser.Expr_binaryContext) ast.Node {
 
 func (c *cc) convertParam(n *parser.Expr_bindContext) ast.Node {
 	if n.BIND_PARAMETER() != nil {
-		// Parameter numbers start at one
-		c.paramCount += 1
-		return &ast.ParamRef{
-			Number:   c.paramCount,
-			Location: n.GetStart().GetStart(),
+		text := n.GetText()
+		if strings.ContainsRune("@$:", rune(text[0])) {
+			loc := n.GetStart().GetStart()
+			return &ast.A_Expr{
+				Kind: ast.A_Expr_Kind(1),
+				Name: &ast.List{
+					Items: []ast.Node{&ast.String{Str: text[:1]}},
+				},
+				Rexpr: &ast.ColumnRef{
+					Name: text[1:],
+					Fields: &ast.List{
+						Items: []ast.Node{&ast.String{Str: text[1:]}},
+					},
+					Location: loc + 1,
+				},
+				Location: loc,
+			}
+		}
+		if text == "?" {
+			// Parameter numbers start at one
+			c.paramCount += 1
+			return &ast.ParamRef{
+				Number:   c.paramCount,
+				Location: n.GetStart().GetStart(),
+			}
+		} else {
+			argn, _ := strconv.Atoi(text[1:])
+			return &ast.ParamRef{
+				Number:   argn,
+				Location: n.GetStart().GetStart(),
+				Dollar:   true,
+			}
 		}
 	}
 	return todo(n)

internal/engine/sqlite/convert.go

@@ -102,6 +102,8 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt, numbs map[int]bool,
 				} else {
 					replace = "?"
 				}
+			} else if engine == config.EngineSQLite {
+				replace = fmt.Sprintf("?%d", argn)
 			} else {
 				replace = fmt.Sprintf("$%d", argn)
 			}
@@ -130,6 +132,8 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt, numbs map[int]bool,
 			var replace string
 			if engine == config.EngineMySQL || !dollar {
 				replace = "?"
+			} else if engine == config.EngineSQLite {
+				replace = fmt.Sprintf("?%d", argn)
 			} else {
 				replace = fmt.Sprintf("$%d", argn)
 			}
@@ -156,6 +160,8 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt, numbs map[int]bool,
 			var replace string
 			if engine == config.EngineMySQL || !dollar {
 				replace = "?"
+			} else if engine == config.EngineSQLite {
+				replace = fmt.Sprintf("?%d", argn)
 			} else {
 				replace = fmt.Sprintf("$%d", argn)
 			}