Contact name setting and property lookup

2 files changed, 64 insertions, 3 deletions

diff --git a/src/contact.cpp b/src/contact.cpp
index 796e896..787831c 100644
--- a/src/contact.cpp
+++ b/src/contact.cpp
@@ -2,8 +2,11 @@
 
 #include "identity.h"
 
+#include <array>
+
 using namespace erebos;
 
+using std::array;
 using std::move;
 
 DEFINE_SHARED_TYPE(Set<Contact>,
@@ -34,6 +37,19 @@ optional<string> Contact::customName() const
 	return p->name;
 }
 
+Contact Contact::customName(const Storage & st, const string & name) const
+{
+	auto cdata = st.store(ContactData {
+		.prev = p->data,
+		.identity = {},
+		.name = name,
+	});
+
+	return Contact(shared_ptr<Contact::Priv>(new Contact::Priv {
+		.data = { cdata },
+	}));
+}
+
 string Contact::name() const
 {
 	if (auto cust = customName())
@@ -59,13 +75,33 @@ vector<Stored<ContactData>> Contact::data() const
 	return p->data;
 }
 
+Digest Contact::leastRoot() const
+{
+	if (p->data.empty())
+		return Digest(array<uint8_t, Digest::size> {});
+
+	vector<Digest> roots;
+	for (const auto & d : p->data)
+		for (const auto & r : d.ref().roots())
+			roots.push_back(r);
+	roots.erase(std::unique(roots.begin(), roots.end()), roots.end());
+	return roots[0];
+}
+
 void Contact::Priv::init()
 {
 	std::call_once(initFlag, [this]() {
-		// TODO: property lookup
-		identity = Identity::load(data[0]->identity);
+		vector<Stored<Signed<IdentityData>>> idata;
+		for (const auto & c : findPropertyComponents<Contact>(data, "identity"))
+			for (const auto & i : c->identity)
+				idata.push_back(i);
+
+		identity = Identity::load(idata);
 		if (identity)
 			name = identity->name();
+
+		if (auto opt = findPropertyComponent<Contact>(data, "name"))
+			name = (*opt)->name;
 	});
 }
 
diff --git a/src/main.cpp b/src/main.cpp
index 12f46ac..edf68d7 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -465,7 +465,7 @@ void contactList(const vector<string> &)
 	};
 	for (const auto & c : h->behavior().lens<SharedState>().lens<Set<Contact>>().get().view(cmp)) {
 		ostringstream ss;
-		ss << "contact-list-item " << c.name();
+		ss << "contact-list-item " << string(c.leastRoot()) << " " << c.name();
 		if (auto id = c.identity())
 			if (auto iname = id->name())
 				ss << " " << *iname;
@@ -474,6 +474,30 @@ void contactList(const vector<string> &)
 	printLine("contact-list-done");
 }
 
+void contactSetName(const vector<string> & args)
+{
+	auto id = args.at(0);
+	auto name = args.at(1);
+
+	auto cmp = [](const Contact & x, const Contact & y) {
+		return x.data() < y.data();
+	};
+	for (const auto & c : h->behavior().lens<SharedState>().lens<Set<Contact>>().get().view(cmp)) {
+		if (string(c.leastRoot()) == id) {
+			auto nh = h->update([&] (const Stored<LocalState> & loc) {
+				auto st = loc.ref().storage();
+				auto cc = c.customName(st, name);
+				auto contacts = loc->shared<Set<Contact>>();
+				return st.store(loc->shared<Set<Contact>>(contacts.add(st, cc)));
+			});
+			if (nh)
+				*h = *nh;
+			break;
+		}
+	}
+	printLine("contact-set-name-done");
+}
+
 vector<Command> commands = {
 	{ "store", store },
 	{ "stored-generation", storedGeneration },
@@ -496,6 +520,7 @@ vector<Command> commands = {
 	{ "contact-accept", contactAccept },
 	{ "contact-reject", contactReject },
 	{ "contact-list", contactList },
+	{ "contact-set-name", contactSetName },
 };
 
 }