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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
defmodule SApp.Identity do
use GenServer
require Logger
defmodule Manifest do
defstruct [:pk]
end
defmodule State do
defstruct [:info, :rev, :signed]
end
defimpl Shard.Manifest, for: Manifest do
def start(m) do
DynamicSupervisor.start_child(Shard.DynamicSupervisor, {SApp.Identity, m.pk})
end
end
def start_link(pk) do
GenServer.start_link(__MODULE__, pk)
end
def init(pk) do
manifest = %Manifest{pk: pk}
id = SData.term_hash manifest
case Shard.Manager.register(id, manifest, self()) do
:ok ->
Shard.Manager.dispatch_to(id, nil, self())
state = case Shard.Manager.load_state(id) do
nil ->
info = %{nick: default_nick(pk)}
SData.SignRev.new info
st ->
st
end
GenServer.cast(self(), :init_pull)
{:ok, %{pk: pk, id: id, state: state}}
:redundant ->
exit(:redundant)
end
end
def default_nick(pk) do
nick_suffix = Shard.Keys.pk_display pk
"Anon" <> nick_suffix
end
def find_proc(pk) do
manifest = %Manifest{pk: pk}
id = SData.term_hash manifest
case Shard.Manager.find_proc id do
nil ->
Shard.Manifest.start manifest
pid -> pid
end
end
def get_nick(pk) do
pid = find_proc pk
info = GenServer.call(pid, :get_info)
info.nick
end
def handle_call(:manifest, _from, state) do
{:replyl, state.manifest, state}
end
def handle_call(:get_info, _from, state) do
{:reply, SData.SignRev.get(state.state), state}
end
def handle_call({:set_info, new_info}, _from, state) do
case SData.SignRev.set(state.state, new_info, state.pk) do
{:ok, st2} ->
Shard.Manager.save_state(state.id, st2)
state = put_in(state.state, st2)
bcast_state(state)
{:reply, :ok, state}
err ->
{:reply, err, state}
end
end
def handle_cast(:init_pull, state) do
for {_, pid, _} <- SNet.Manager.list_connections do
GenServer.cast(pid, {:send_msg, {:interested, [state.id]}})
end
{:noreply, state}
end
def handle_cast({:interested, peer_pid, _auth}, state) do
SNet.Manager.send_pid(peer_pid, {state.id, nil, {:update, SData.SignRev.signed(state.state)}})
{:noreply, state}
end
def handle_cast({:msg, conn_pid, _auth, _shard_id, nil, msg}, state) do
state = case msg do
{:update, signed} when signed != nil ->
case SData.SignRev.merge(state.state, signed, state.pk) do
{true, st2} ->
Shard.Manager.save_state(state.id, st2)
state = put_in(state.state, st2)
bcast_state(state, [GenServer.call(conn_pid, :get_peer_info)])
state
{false, _} ->
state
end
_ -> state
end
{:noreply, state}
end
def bcast_state(state, exclude \\ []) do
for peer_id <- Shard.Manager.get_shard_peers(state.id) do
if not Enum.member? exclude, peer_id do
SNet.Manager.send(peer_id, {state.id, nil, {:update, SData.SignRev.signed(state.state)}})
end
end
end
end
|
