More improved typespecs

src/enoise_connection.erl

@@ -57,7 +57,7 @@ start_link(TcpSock, Rx, Tx, Owner, {Active0, Buf}) ->
 send(Noise, Data) ->
     gen_server:call(Noise, {send, Data}).
 
--spec set_active(Noise :: pid(), Active :: boolean()) -> ok | {error, term()}.
+-spec set_active(Noise :: pid(), Active :: true | once) -> ok | {error, term()}.
 set_active(Noise, Active) ->
     gen_server:call(Noise, {active, self(), Active}).
 

src/enoise_hs_state.erl

@@ -90,6 +90,7 @@ read_message(HS = #noise_hs{ msgs = [{in, Msg} | Msgs] }, Message) ->
         Err = {error, _} -> Err
     end.
 
+-spec remote_keys(HS :: state()) -> keypair().
 remote_keys(#noise_hs{ rs = RS }) ->
     RS.