fix(sqlite): introduce "update from..." to the parser

Author: IdoKendo

internal/engine/sqlite/convert.go

@@ -1021,7 +1021,7 @@ func (c *cc) convertTablesOrSubquery(n []parser.ITable_or_subqueryContext) []ast
 }
 
 type Update_stmt interface {
-	Qualified_table_name() parser.IQualified_table_nameContext
+	Table_name() parser.ITable_nameContext
 	GetStart() antlr.Token
 	AllColumn_name() []parser.IColumn_nameContext
 	WHERE_() antlr.TerminalNode
@@ -1035,7 +1035,7 @@ func (c *cc) convertUpdate_stmtContext(n Update_stmt) ast.Node {
 	}
 
 	relations := &ast.List{}
-	tableName := identifier(n.Qualified_table_name().GetText())
+	tableName := n.Table_name().GetText()
 	rel := ast.RangeVar{
 		Relname:  &tableName,
 		Location: n.GetStart().GetStart(),

internal/engine/sqlite/parser/SQLiteParser.g4

@@ -468,9 +468,9 @@ compound_operator:
 update_stmt:
     with_clause? UPDATE_ (
         OR_ (ROLLBACK_ | ABORT_ | REPLACE_ | FAIL_ | IGNORE_)
-    )? qualified_table_name SET_ (column_name | column_name_list) ASSIGN expr (
+    )? table_name (AS_ table_alias)? SET_ (column_name | column_name_list) ASSIGN expr (
         COMMA (column_name | column_name_list) ASSIGN expr
-    )* (WHERE_ expr)? returning_clause?
+    )* (FROM_ table_or_subquery)? (WHERE_ expr)? returning_clause?
 ;
 
 column_name_list:
@@ -480,7 +480,7 @@ column_name_list:
 update_stmt_limited:
     with_clause? UPDATE_ (
         OR_ (ROLLBACK_ | ABORT_ | REPLACE_ | FAIL_ | IGNORE_)
-    )? qualified_table_name SET_ (column_name | column_name_list) ASSIGN expr (
+    )? table_name (AS_ table_alias)? SET_ (column_name | column_name_list) ASSIGN expr (
         COMMA (column_name | column_name_list) ASSIGN expr
     )* (WHERE_ expr)? (order_by_stmt? limit_stmt)?
 ;