Fix types (#30)

dsienkiewicz · Feb 15, 2024 · a6c6b9c · a6c6b9c
1 parent 5162832
commit a6c6b9c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/lib/xtb_client/messages/chart_range.ex b/lib/xtb_client/messages/chart_range.ex
@@ -20,10 +20,12 @@ defmodule XtbClient.Messages.ChartRange do
 
     alias XtbClient.Messages.{DateRange, Period}
 
+    import Period
+
     @type t :: %__MODULE__{
             start: integer(),
             end: integer(),
-            period: Period.t(),
+            period: Period.minute_period(),
             symbol: String.t(),
             ticks: integer()
           }
@@ -48,7 +50,7 @@ defmodule XtbClient.Messages.ChartRange do
           period: period,
           symbol: symbol
         })
-        when is_atom(period) and is_binary(symbol) do
+        when is_period(period) and is_binary(symbol) do
       %__MODULE__{
         start: start,
         end: end_value,

diff --git a/lib/xtb_client/messages/period.ex b/lib/xtb_client/messages/period.ex
@@ -18,6 +18,9 @@ defmodule XtbClient.Messages.Period do
     mn1: 43_200
   ]
 
+  defguard is_period(atom)
+           when is_atom(atom) and atom in [:m1, :m5, :m15, :m30, :h1, :h4, :d1, :w1, :mn1]
+
   @doc """
   Parses value given as number of minutes to `Period` atom type.
   """