fix(engine/sqlite): fixed to be able to find relation from WITH clause

fix #2136

Author: orisano

internal/engine/sqlite/convert.go

@@ -331,6 +331,25 @@ func (c *cc) convertMultiSelect_stmtContext(n *parser.Select_stmtContext) ast.No
 	var where ast.Node
 	var groups = []ast.Node{}
 	var having ast.Node
+	var ctes []ast.Node
+
+	if ct := n.Common_table_stmt(); ct != nil {
+		recursive := ct.RECURSIVE_() != nil
+		for _, cte := range ct.AllCommon_table_expression() {
+			tableName := identifier(cte.Table_name().GetText())
+			var cteCols ast.List
+			for _, col := range cte.AllColumn_name() {
+				cteCols.Items = append(cteCols.Items, NewIdentifer(col.GetText()))
+			}
+			ctes = append(ctes, &ast.CommonTableExpr{
+				Ctename:      &tableName,
+				Ctequery:     c.convert(cte.Select_stmt()),
+				Location:     cte.GetStart().GetStart(),
+				Cterecursive: recursive,
+				Ctecolnames:  &cteCols,
+			})
+		}
+	}
 
 	for _, icore := range n.AllSelect_core() {
 		core, ok := icore.(*parser.Select_coreContext)
@@ -377,6 +396,9 @@ func (c *cc) convertMultiSelect_stmtContext(n *parser.Select_stmtContext) ast.No
 		LimitCount:   limitCount,
 		LimitOffset:  limitOffset,
 		ValuesLists:  &ast.List{},
+		WithClause: &ast.WithClause{
+			Ctes: &ast.List{Items: ctes},
+		},
 	}
 }