return undefined if the get_line is called with undefined atom

bragful · Jun 18, 2015 · 05ab5de · 05ab5de
1 parent ebe8c6f
commit 05ab5de
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/ephp_util.erl b/src/ephp_util.erl
@@ -117,7 +117,10 @@ pad_to_bin(Num, Pad) when not is_binary(Num) ->
 pad_to_bin(Num, Pad) ->
     pad_to_bin(<<"0",Num/binary>>, Pad-1).
 
--spec get_line(line()) -> non_neg_integer().
+-spec get_line(line() | undefined) -> non_neg_integer() | undefined.
+
+get_line(undefined) ->
+    undefined;
 
 get_line({{line, Line}, {column, _Column}}) ->
     Line.