From 8919fe3f14cf2ef20885c35cf04a241d2f842503 Mon Sep 17 00:00:00 2001
From: Pim Kunis <pim@kunis.nl>
Date: Sat, 17 Jul 2021 18:54:49 +0200
Subject: [PATCH] Add more events during room creation

---
 lib/matrix_server/event.ex       | 134 +++++++++++++++++++++++--------
 lib/matrix_server/room_server.ex |  12 +--
 2 files changed, 107 insertions(+), 39 deletions(-)

diff --git a/lib/matrix_server/event.ex b/lib/matrix_server/event.ex
index e4a787c..78244dd 100644
--- a/lib/matrix_server/event.ex
+++ b/lib/matrix_server/event.ex
@@ -42,8 +42,8 @@ defmodule MatrixServer.Event do
       | type: "m.room.create",
         state_key: "",
         content: %{
-          creator: creator,
-          room_version: room_version || MatrixServer.default_room_version()
+          "creator" => creator,
+          "room_version" => room_version || MatrixServer.default_room_version()
         }
     }
   end
@@ -54,51 +54,117 @@ defmodule MatrixServer.Event do
       | type: "m.room.member",
         state_key: sender,
         content: %{
-          membership: "invite"
+          "membership" => "invite"
         }
     }
   end
 
-  def room_creation_create_room(%CreateRoom{room_version: room_version}, %Account{
-        localpart: localpart
+  def power_levels(room_id, sender) do
+    %Event{
+      new(room_id, sender)
+      | type: "m.room.power_levels",
+        state_key: "",
+        content: %{
+          "ban" => 50,
+          "events" => %{},
+          "events_default" => 0,
+          "invite" => 50,
+          "kick" => 50,
+          "redact" => 50,
+          "state_default" => 50,
+          "users" => %{
+            sender => 50
+          },
+          "users_default" => 0,
+          "notifications" => %{
+            "room" => 50
+          }
+        }
+    }
+  end
+
+  def room_name(room_id, sender, name) do
+    %Event{
+      new(room_id, sender)
+      | type: "m.room.name",
+        state_key: "",
+        content: %{
+          "name" => name
+        }
+    }
+  end
+
+  def room_topic(room_id, sender, topic) do
+    %Event{
+      new(room_id, sender)
+      | type: "m.room.topic",
+        state_key: "",
+        content: %{
+          "topic" => topic
+        }
+    }
+  end
+
+  def room_creation_create_room(repo, %{
+        input: %CreateRoom{room_version: room_version},
+        account: %Account{localpart: localpart},
+        room: %Room{id: room_id}
       }) do
-    fn repo, %{room: %Room{id: room_id}} ->
-      # TODO: state resolution
-      create_room(room_id, MatrixServer.get_mxid(localpart), room_version)
-      |> repo.insert()
-    end
+    # TODO: state resolution
+    create_room(room_id, MatrixServer.get_mxid(localpart), room_version)
+    |> repo.insert()
   end
 
-  def room_creation_join_creator do
-    fn repo,
-       %{
-         create_room_event: %Event{sender: creator, event_id: create_room_event_id},
-         room: %Room{id: room_id}
-       } ->
-      # TODO: state resolution
-      join(room_id, creator)
-      |> Map.put(:prev_events, [create_room_event_id])
-      |> Map.put(:auth_events, [create_room_event_id])
-      |> repo.insert()
-    end
+  def room_creation_join_creator(repo, %{
+        room: %Room{id: room_id},
+        create_room_event: %Event{sender: creator}
+      }) do
+    # TODO: state resolution
+    join(room_id, creator)
+    |> repo.insert()
   end
 
-  def room_creation_power_levels(_input) do
-    fn _repo, %{} ->
-      {:ok, :ok}
-    end
+  def room_creation_power_levels(
+        repo,
+        %{
+          room: %Room{id: room_id},
+          create_room_event: %Event{sender: creator}
+        }
+      ) do
+    # TODO: state resolution
+    power_levels(room_id, creator)
+    |> repo.insert()
   end
 
-  def room_creation_name(_input) do
-    fn _repo, %{} ->
-      {:ok, :ok}
-    end
+  def room_creation_name(_repo, %{input: %CreateRoom{name: nil}}), do: {:ok, :noop}
+
+  def room_creation_name(
+        repo,
+        %{
+          input: %CreateRoom{name: name},
+          room: %Room{id: room_id},
+          create_room_event: %Event{sender: creator}
+        }
+      ) do
+    # TODO: state resolution
+    # TODO: check name length
+    room_name(room_id, creator, name)
+    |> repo.insert()
   end
 
-  def room_creation_topic(_input) do
-    fn _repo, %{} ->
-      {:ok, :ok}
-    end
+  def room_creation_topic(_repo, %{input: %CreateRoom{topic: nil}}), do: {:ok, :noop}
+
+  def room_creation_topic(
+        repo,
+        %{
+          input: %CreateRoom{topic: topic},
+          room: %Room{id: room_id},
+          create_room_event: %Event{sender: creator}
+        }
+      ) do
+    # TODO: state resolution
+    room_topic(room_id, creator, topic)
+    |> repo.insert()
   end
 
   def generate_event_id do
diff --git a/lib/matrix_server/room_server.ex b/lib/matrix_server/room_server.ex
index 95fa68a..4ce9715 100644
--- a/lib/matrix_server/room_server.ex
+++ b/lib/matrix_server/room_server.ex
@@ -23,12 +23,14 @@ defmodule MatrixServer.RoomServer do
     # TODO: preset events, initial_state events, invite, invite_3pid
     result =
       Multi.new()
+      |> Multi.put(:input, input)
+      |> Multi.put(:account, account)
       |> Multi.insert(:room, Room.create_changeset(input))
-      |> Multi.run(:create_room_event, Event.room_creation_create_room(input, account))
-      |> Multi.run(:join_creator_event, Event.room_creation_join_creator())
-      |> Multi.run(:power_levels_event, Event.room_creation_power_levels(input))
-      |> Multi.run(:name_event, Event.room_creation_name(input))
-      |> Multi.run(:topic_event, Event.room_creation_topic(input))
+      |> Multi.run(:create_room_event, &Event.room_creation_create_room/2)
+      |> Multi.run(:join_creator_event, &Event.room_creation_join_creator/2)
+      |> Multi.run(:power_levels_event, &Event.room_creation_power_levels/2)
+      |> Multi.run(:name_event, &Event.room_creation_name/2)
+      |> Multi.run(:topic_event, &Event.room_creation_topic/2)
       |> Repo.transaction()
 
     {:reply, result, state}