add file manager

Author: DecafMango

lib/backend/application.ex

@@ -17,7 +17,9 @@ defmodule Backend.Application do
       # Start a worker by calling: Backend.Worker.start_link(arg)
       # {Backend.Worker, arg},
       # Start to serve requests, typically the last entry
-      BackendWeb.Endpoint
+      BackendWeb.Endpoint,
+      Backend.Files.FileRegistry,
+      Backend.Files.FileManager
     ]
 
     # See https://hexdocs.pm/elixir/Supervisor.html

lib/backend/files/file.ex

@@ -12,9 +12,8 @@ defmodule Backend.Files.File do
     field :content, :string
   end
 
-  def changeset(file, attrs) do
-    file
-    |> cast(attrs, [:name, :content], empty_values: [:content])
-    |> validate_required([:name])
+  def changeset(file, attr) do
+    cast(file, attr, [:name, :content])
   end
+
 end

lib/backend/files/file_manager.ex

@@ -0,0 +1,53 @@
+defmodule Backend.Files.FileManager do
+  alias Backend.Files.FileProcess
+
+  def start_link() do
+    IO.puts("Starting file manager")
+
+    DynamicSupervisor.start_link(
+      name: __MODULE__,
+      strategy: :one_for_one
+    )
+  end
+
+  defp start_child(file_id) do
+    DynamicSupervisor.start_child(
+      __MODULE__,
+      {Backend.Files.FileProcess, file_id}
+    )
+  end
+
+  @spec child_spec(any()) :: %{
+          id: Backend.Files.FileManager,
+          start: {Backend.Files.FileManager, :start_link, []},
+          type: :supervisor
+        }
+  def child_spec(_arg) do
+    %{
+      id: __MODULE__,
+      start: {__MODULE__, :start_link, []},
+      type: :supervisor
+    }
+  end
+
+  def get_or_start_process(file_id) do
+    case Registry.lookup(Backend.Files.FileRegistry, file_id) do
+      [] ->
+        # Процесс не существует — запускаем новый
+        start_child(file_id)
+        :ok
+
+      [{_pid, _}] ->
+        # Процесс уже существует
+        :ok
+    end
+  end
+
+  def update_file_content(file_id, content) do
+    # Убедимся, что процесс существует
+    get_or_start_process(file_id)
+
+    # Обновляем содержимое файла в процессе
+    FileProcess.update_content(file_id, content)
+  end
+end

lib/backend/files/file_process.ex

@@ -0,0 +1,43 @@
+defmodule Backend.Files.FileProcess do
+  use GenServer, restart: :temporary
+  alias Backend.Files
+
+  @timeout 5_000 # Время бездействия (например, 5 секунд)
+
+  # API
+  def start_link(file_id) do
+    GenServer.start_link(__MODULE__, file_id, name: via_tuple(file_id))
+  end
+
+  def update_content(file_id, new_content) do
+    GenServer.call(via_tuple(file_id), {:update_content, new_content})
+  end
+
+  defp via_tuple(file_id) do
+    {:via, Registry, {Backend.Files.FileRegistry, file_id}}
+  end
+
+  # Callbacks
+  def init(file_id) do
+    file = Files.get_file(file_id) # Загружаем файл из БД при запуске
+    {:ok, %{file: file, timer: reset_timer(nil)}}
+  end
+
+  def handle_call({:update_content, new_content}, _from, %{file: file, timer: timer} = state) do
+    new_file = %{file | content: new_content}
+    {:reply, :ok, %{state | file: new_file, timer: reset_timer(timer)}}
+  end
+
+  def handle_info(:timeout, %{file: file} = state) do
+    IO.inspect file
+    Files.save_file(file) # Сохраняем в БД перед завершением
+    {:stop, :normal, state}
+  end
+
+  defp reset_timer(timer) do
+    # Отменяем старый таймер, если он существует
+    if timer, do: Process.cancel_timer(timer)
+    # Устанавливаем новый таймер
+    Process.send_after(self(), :timeout, @timeout)
+  end
+end

lib/backend/files/file_registry.ex

@@ -0,0 +1,21 @@
+defmodule Backend.Files.FileRegistry do
+
+  def start_link do
+    IO.puts("Starting file registry")
+
+    Registry.start_link(name: __MODULE__,keys: :unique)
+  end
+
+  def via_tuple(key) do
+    {:via, Registry, {__MODULE__, key}}
+  end
+
+  def child_spec(_) do
+    Supervisor.child_spec(
+      Registry,
+      id: __MODULE__,
+      start: {__MODULE__, :start_link, []}
+    )
+  end
+
+end

lib/backend/files/files.ex

@@ -16,9 +16,13 @@ defmodule Backend.Files do
   end
 
   def save_file(file) do
+    # Это сделано специально, так как без обнуления контента cast думает, что ничего не изменилось
+    content = file.content
+    file = %{file | content: ""}
+
     file
-    |> File.changeset(%{})
-    |> Repo.update()
+    |> File.changeset(%{content: content})
+    |> Repo.update!()
   end
 
   def delete_file(file) do

lib/backend_web/channels/room_channel.ex

@@ -2,14 +2,25 @@ defmodule BackendWeb.FileChannel do
   @moduledoc """
   Websocket channel for each .mds file
   """
+  alias Backend.Files
+  alias Backend.Files.FileManager
   use Phoenix.Channel
 
-  def join("file:" <> _room_id, _params, socket) do
-    {:ok, socket}
+  def join("file:" <> file_id, _params, socket) do
+    case Files.get_file(file_id) do
+      nil -> {:error, %{reason: "File with id " <> file_id <> " does not exist"}}
+      _ -> {:ok, assign(socket, :file_id, file_id)}
+    end
   end
 
   def handle_in("edit", %{"content" => content}, socket) do
-    # Рассылка обновлений всем клиентам
+    # Получаем file_id из socket.assigns
+    file_id = socket.assigns.file_id
+
+    # Вызываем FileManager для обновления содержимого файла
+    FileManager.update_file_content(file_id, content)
+
+    # Рассылаем обновления всем клиентам
     broadcast!(socket, "update", %{content: content})
     {:noreply, socket}
   end

priv/repo/migrations/20241129212656_add_file.exs

@@ -4,7 +4,7 @@ defmodule Backend.Repo.Migrations.AddFile do
   def change do
     create table(:files) do
       add :name, :string, null: false
-      add :content, :binary
+      add :content, :string
     end
   end
 end