Fixes

3 files changed, 31 insertions, 2 deletions

diff --git a/shard/lib/app/chat.ex b/shard/lib/app/chat.ex
index c7df62e..52ad920 100644
--- a/shard/lib/app/chat.ex
+++ b/shard/lib/app/chat.ex
@@ -51,9 +51,17 @@ defmodule SApp.Chat do
       :ok ->
         Shard.Manager.dispatch_to(id, nil, self())
         {:ok, page_store} = SApp.PageStore.start_link(id, :page_store)
+        root = Shard.Manager.load_state id
+        root = cond do
+          root == nil -> nil
+          GenServer.call(page_store, {:have_rec, root}) -> root
+          true ->
+            Logger.warn "Not all pages for saved root were saved, restarting from an empty state!"
+            nil
+        end
         mst = %MST{store: %SApp.PageStore{pid: page_store},
                    cmp: &msg_cmp/2,
-                   root: Shard.Manager.load_state(id)}
+                   root: root}
         GenServer.cast(self(), :init_pull)
         {:ok,
           %{channel: channel,
diff --git a/shard/lib/app/pagestore.ex b/shard/lib/app/pagestore.ex
index 7962084..f093ed4 100644
--- a/shard/lib/app/pagestore.ex
+++ b/shard/lib/app/pagestore.ex
@@ -66,6 +66,10 @@ defmodule SApp.PageStore do
     {:reply, hash, state}
   end
 
+  def handle_call({:have_rec, root}, _from, state) do
+    {:reply, have_rec(state, root), state}
+  end
+
   defp add_request(state, key, from) do
     reqs_key = case state.reqs[key] do
       nil ->
@@ -89,6 +93,16 @@ defmodule SApp.PageStore do
     end
   end
 
+  defp have_rec(state, root) do
+    case :dets.lookup state.store, root do
+      [{_, _, bin}] ->
+        pg = SData.term_unbin bin
+        pg |> SData.Page.refs |> Enum.map(&(have_rec(state, &1))) |> Enum.all?
+      _ ->
+        false
+    end
+  end
+
   defp init_rec_pull(state, key, why, prefer_ask) do
     case prefer_ask do
       [_|_] ->
diff --git a/shard/lib/keys.ex b/shard/lib/keys.ex
index de054e1..b6ff461 100644
--- a/shard/lib/keys.ex
+++ b/shard/lib/keys.ex
@@ -72,7 +72,7 @@ defmodule Shard.Keys do
   List the public keys of all identities for which we have a secret key
   """
   def list_identities() do
-    for [{pk, _sk}] <- :dets.match(@key_db, :"$1"), do: pk
+    for [pk, _sk] <- :dets.match(@key_db, {:"$1", :"$2"}), do: pk
   end
 
   @doc"""
@@ -103,6 +103,13 @@ defmodule Shard.Keys do
     end
   end
 
+  def have_sk?(pk) do
+    case :dets.lookup @key_db, pk do
+      [{^pk, _sk}] -> true
+      _ -> false
+    end
+  end
+
   @doc"""
     Lookup the secret key for a pk and generate a detached signature for a message.