From 1bfcb74a7623f10e6c63bf1ac029912d9302ed58 Mon Sep 17 00:00:00 2001
From: Jason Shin <visualbbasic@gmail.com>
Date: Sun, 4 Aug 2024 12:24:32 +1000
Subject: [PATCH] fix translate column name assignment

---
 .../sql_parser/expressions/translate_expr.rs        |  4 ++--
 src/ts_generator/sql_parser/quoted_strings.rs       |  2 +-
 tests/demo/update/update.ts                         | 13 +++++++++++++
 3 files changed, 16 insertions(+), 3 deletions(-)
 create mode 100644 tests/demo/update/update.ts

diff --git a/src/ts_generator/sql_parser/expressions/translate_expr.rs b/src/ts_generator/sql_parser/expressions/translate_expr.rs
index 68681f21..cf80ea7a 100644
--- a/src/ts_generator/sql_parser/expressions/translate_expr.rs
+++ b/src/ts_generator/sql_parser/expressions/translate_expr.rs
@@ -79,9 +79,9 @@ pub fn translate_column_name_assignment(assignment: &Assignment) -> Option<Strin
   let right = assignment.id.get(1);
 
   if left.is_some() && right.is_some() {
-    return right.map(|x| x.to_string());
+    return right.map(|ident| DisplayIndent(ident).to_string());
   } else if left.is_some() && right.is_none() {
-    return left.map(|x| x.to_string());
+    return left.map(|ident| DisplayIndent(ident).to_string());
   }
   None
 }
diff --git a/src/ts_generator/sql_parser/quoted_strings.rs b/src/ts_generator/sql_parser/quoted_strings.rs
index 470337b7..e1b5f6c3 100644
--- a/src/ts_generator/sql_parser/quoted_strings.rs
+++ b/src/ts_generator/sql_parser/quoted_strings.rs
@@ -1,4 +1,4 @@
-use sqlparser::ast::{Ident, ObjectName, TableAlias};
+use sqlparser::ast::{Assignment, Expr, Ident, ObjectName, TableAlias};
 use std::fmt;
 use std::fmt::{write, Formatter};
 
diff --git a/tests/demo/update/update.ts b/tests/demo/update/update.ts
new file mode 100644
index 00000000..78cb1bb6
--- /dev/null
+++ b/tests/demo/update/update.ts
@@ -0,0 +1,13 @@
+import { sql } from 'sqlx-ts'
+
+const updateQuery = sql`
+UPDATE items
+SET food_type = $1, time_takes_to_cook = $2
+WHERE id = $3
+`
+
+const quotedUpdateQuery = sql`
+UPDATE "items"
+SET "food_type" = "$1", "time_takes_to_cook" = "$2"
+WHERE "id" = $3
+`