From e89a8ea5dbf26d6c3488f43c304fedbf8db93684 Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Fri, 4 Aug 2023 21:10:36 +0300
Subject: [PATCH] Update Links.create_one to use a function.

---
 CHANGELOG.org                     | 1 +
 lib/link_shortener/links/links.ex | 7 +++----
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG.org b/CHANGELOG.org
index 3bfc343..41fbf7c 100644
--- a/CHANGELOG.org
+++ b/CHANGELOG.org
@@ -46,3 +46,4 @@
 
 ** 0.5.1 <2023-08-04 Fri>
     Add a generator for Link with random shorten.
+    Update Links.create_one to use a function.
diff --git a/lib/link_shortener/links/links.ex b/lib/link_shortener/links/links.ex
index 1092aac..e5b301a 100644
--- a/lib/link_shortener/links/links.ex
+++ b/lib/link_shortener/links/links.ex
@@ -4,13 +4,12 @@ defmodule LinkShortener.Links do
   import Ecto.Query
   alias LinkShortener.Repo
   alias LinkShortener.Links.Link
-  alias LinkShortener.Generators.SafeString
+  alias LinkShortener.Generators.LinkWithRandomShorten, as: LinkGenerator
 
   def new_one(), do: Link.changeset(%Link{}, %{})
 
-  def create_one(attrs, length \\ 10, generator \\ SafeString) do
-    shorten = generator.generate(length)
-    Map.put(attrs, :shorten, shorten)
+  def create_one(attrs, length \\ 10, generator \\ &LinkGenerator.generate_one/2) do
+    generator.(attrs, length)
     |> insert_one()
   end