1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
defmodule Shard.Identity do
use Agent
require Salty.Sign.Ed25519, as: Sign
require Logger
@identity_db [Application.get_env(:shard, :data_path), "identity_db"] |> Path.join |> String.to_atom
def start_link(_) do
Agent.start_link(__MODULE__, :init, [], name: __MODULE__)
end
def init() do
:dets.start
{:ok, @identity_db} = :dets.open_file @identity_db, type: :set
case :dets.match @identity_db, :"$1" do
[] ->
Logger.info "Generating keypair..."
{pk, sk} = gen_keypair(Application.get_env(:shard, :peer_id_suffix))
nick_suffix = pk
|> binary_part(0, 3)
|> Base.encode16
|> String.downcase
nick = "Anon" <> nick_suffix
:dets.insert @identity_db, {pk, sk, nick}
%{
keypair: {pk, sk},
nickname: nick
}
[[{pk, sk, nick}] | _] ->
%{
keypair: {pk, sk},
nickname: nick
}
end
end
defp gen_keypair(suffix, n \\ 0) do
{:ok, pk, sk} = Sign.keypair
if rem(n, 10000) == 0 do
Logger.info "#{n}... expected #{:math.pow(256, byte_size(suffix))}"
end
if :binary.longest_common_suffix([pk, suffix]) == byte_size(suffix) do
{pk, sk}
else
gen_keypair(suffix, n+1)
end
end
def get_keypair() do
Agent.get(__MODULE__, &(&1.keypair))
end
def get_nickname() do
Agent.get(__MODULE__, &(&1.nickname))
end
def set_nickname(newnick) do
Agent.update(__MODULE__, fn state ->
{pk, sk} = state.keypair
:dets.insert @identity_db, {pk, sk, newnick}
%{state | nickname: newnick}
end)
end
end
|
