From 6e0c555588f6142c4bf10b2f11bd8d79530e2aaa Mon Sep 17 00:00:00 2001
From: Bunny Lushington <bunny@bapi.us>
Date: Mon, 13 May 2024 14:59:17 -0500
Subject: [PATCH 1/2] fix(bug): allow "complex" filter with update statement

---
 lib/moebius/query.ex         | 32 +++++++++-----------------------
 test/moebius/update_test.exs | 34 +++++++++++++++++++++++++++++++---
 2 files changed, 40 insertions(+), 26 deletions(-)

diff --git a/lib/moebius/query.ex b/lib/moebius/query.ex
index 8cc8bbc..74d72de 100644
--- a/lib/moebius/query.ex
+++ b/lib/moebius/query.ex
@@ -432,31 +432,17 @@ defmodule Moebius.Query do
     cols = Keyword.keys(criteria)
     vals = Keyword.values(criteria)
 
-    {cols, col_count} =
-      Enum.map_reduce(cols, 1, fn col, acc ->
-        {"#{col} = $#{acc}", acc + 1}
-      end)
-
-    # here's something for John to clean up :):)
-    where =
-      cond do
-        length(cmd.where_columns) > 0 ->
-          {filters, _count} =
-            Enum.map_reduce(cmd.where_columns, col_count, fn col, acc ->
-              {"#{col} = $#{acc}", acc + 1}
-            end)
-
-          " where " <> Enum.join(filters, " and ")
-
-        cmd.where ->
-          cmd.where
-      end
-
-    # add the filter criteria to the update list
-    params = vals ++ cmd.params
+    first_available_param = length(cmd.params) + 1
+    {cols, _col_count} =
+      Enum.map_reduce(cols, first_available_param,
+                      fn col, acc ->
+                        {"#{col} = $#{acc}", acc + 1}
+                      end)
+
+    params = cmd.params ++ vals
     columns = Enum.join(cols, ", ")
 
-    sql = "update #{cmd.table_name} set #{columns}#{where} returning *;"
+    sql = "update #{cmd.table_name} set #{columns}#{cmd.where} returning *;"
     %{cmd | sql: sql, type: :update, params: params}
   end
 
diff --git a/test/moebius/update_test.exs b/test/moebius/update_test.exs
index 42c2939..3646cdc 100644
--- a/test/moebius/update_test.exs
+++ b/test/moebius/update_test.exs
@@ -13,8 +13,9 @@ defmodule Moebius.UpdateTest do
   end
 
   test "a basic user update", %{cmd: cmd} do
-    assert cmd.sql == "update users set email = $1 where id = $2 returning *;"
+    assert cmd.sql == "update users set email = $2 where id = $1 returning *;"
     assert length(cmd.params) == 2
+    assert cmd.params == [1, "maggot@test.com"]
   end
 
   test "a basic user insert has params set", %{cmd: cmd} do
@@ -34,13 +35,40 @@ defmodule Moebius.UpdateTest do
   test "a bulk update with a string filter and params" do
     cmd =
       db(:users)
-      |> filter("email LIKE %$2", "test")
+      |> filter("email LIKE %$1", "test")
       |> update(email: "ox@test.com")
 
-    assert cmd.sql == "update users set email = $1 where email LIKE %$2 returning *;"
+    assert cmd.sql == "update users set email = $2 where email LIKE %$1 returning *;"
     assert length(cmd.params) == 2
+    assert cmd.params == ["test", "ox@test.com"]
   end
 
+
+  test "basic update with 'in' filter" do
+    names = ["Super", "Mike"]
+    cmd =
+      db(:users)
+      |> filter(:first, in: names)
+      |> update(roles: ["newrole"])
+
+    assert cmd.sql == "update users set roles = $3 where first IN($1, $2) returning *;"
+    assert length(cmd.params) == 3
+    assert cmd.params == names ++ [["newrole"]]
+  end
+
+
+  test "basic update with '>' filter" do
+    cmd =
+      db(:users)
+      |> filter(:order_count, gt: 5)
+      |> update(roles: ["newrole"])
+    assert cmd.sql == "update users set roles = $2 where order_count > $1 returning *;"
+    assert length(cmd.params) == 2
+    assert cmd.params == [5, ["newrole"]]
+  end
+
+
+
   # TODO: Move this to date tests
 
   # test "it actually works" do

From 972b8606d7075d5a450fd4730196d35b018f0411 Mon Sep 17 00:00:00 2001
From: Bunny Lushington <bunny@bapi.us>
Date: Mon, 13 May 2024 18:48:44 -0500
Subject: [PATCH 2/2] fix(format): adjust formatting

---
 lib/moebius/query.ex         | 8 ++++----
 test/moebius/update_test.exs | 6 ++----
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/lib/moebius/query.ex b/lib/moebius/query.ex
index 74d72de..a404c06 100644
--- a/lib/moebius/query.ex
+++ b/lib/moebius/query.ex
@@ -433,11 +433,11 @@ defmodule Moebius.Query do
     vals = Keyword.values(criteria)
 
     first_available_param = length(cmd.params) + 1
+
     {cols, _col_count} =
-      Enum.map_reduce(cols, first_available_param,
-                      fn col, acc ->
-                        {"#{col} = $#{acc}", acc + 1}
-                      end)
+      Enum.map_reduce(cols, first_available_param, fn col, acc ->
+        {"#{col} = $#{acc}", acc + 1}
+      end)
 
     params = cmd.params ++ vals
     columns = Enum.join(cols, ", ")
diff --git a/test/moebius/update_test.exs b/test/moebius/update_test.exs
index 3646cdc..79d6c1a 100644
--- a/test/moebius/update_test.exs
+++ b/test/moebius/update_test.exs
@@ -43,9 +43,9 @@ defmodule Moebius.UpdateTest do
     assert cmd.params == ["test", "ox@test.com"]
   end
 
-
   test "basic update with 'in' filter" do
     names = ["Super", "Mike"]
+
     cmd =
       db(:users)
       |> filter(:first, in: names)
@@ -56,19 +56,17 @@ defmodule Moebius.UpdateTest do
     assert cmd.params == names ++ [["newrole"]]
   end
 
-
   test "basic update with '>' filter" do
     cmd =
       db(:users)
       |> filter(:order_count, gt: 5)
       |> update(roles: ["newrole"])
+
     assert cmd.sql == "update users set roles = $2 where order_count > $1 returning *;"
     assert length(cmd.params) == 2
     assert cmd.params == [5, ["newrole"]]
   end
 
-
-
   # TODO: Move this to date tests
 
   # test "it actually works" do