Move Inspect implementation insdie Matrex module.

versilov · May 9, 2018 · 5b193b9 · 5b193b9
1 parent 946f5f4
commit 5b193b9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 5 deletions.
diff --git a/lib/matrex.ex b/lib/matrex.ex
@@ -109,6 +109,11 @@ defmodule Matrex do
       ),
       do: {:ok, columns}
 
+  defimpl Inspect do
+    def inspect(%Matrex{} = matrex, %{width: screen_width}),
+      do: Matrex.Inspect.do_inspect(matrex, screen_width)
+  end
+
   @doc """
   Adds two matrices
   """

diff --git a/lib/matrex/inspect.ex b/lib/matrex/inspect.ex
@@ -1,13 +1,20 @@
-defimpl Inspect, for: Matrex do
-  def inspect(
+defmodule Matrex.Inspect do
+  def inspect(%Matrex{} = matrex) do
+    IO.puts(do_inspect(matrex))
+    matrex
+  end
+
+  def do_inspect(matrex, screen_width \\ 80)
+
+  def do_inspect(
         %Matrex{
           data: <<
             rows::unsigned-integer-little-32,
             columns::unsigned-integer-little-32,
             _rest::binary
           >>
         } = matrex,
-        %{width: screen_width} = _opts
+        screen_width
       )
       when columns < screen_width / 8 and rows <= 21 do
     rows_as_strings =
@@ -30,15 +37,15 @@ defimpl Inspect, for: Matrex do
   @element_byte_size 4
   @element_chars_size 8
 
-  def inspect(
+  def do_inspect(
         %Matrex{
           data: <<
             rows::unsigned-integer-little-32,
             columns::unsigned-integer-little-32,
             body::binary
           >>
         } = matrex,
-        %{width: screen_width} = _opts
+        screen_width
       )
       when columns >= screen_width / 8 or rows > 21 do
     suffix_size = prefix_size = div(screen_width, 16)