From ae5d96a645bd6e158d2e52dfdfaf18b65ce2ef82 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Mon, 11 Nov 2013 19:27:06 +0100
Subject: [PATCH] Allow method chaining when changing identities.

---
 .../sone/freenet/wot/DefaultIdentity.java            | 18 ++++++++++++------
 .../sone/freenet/wot/DefaultOwnIdentity.java         | 20 ++++++++++++++++++++
 .../net/pterodactylus/sone/freenet/wot/Identity.java | 12 ++++++------
 .../pterodactylus/sone/freenet/wot/OwnIdentity.java  |  5 +++++
 4 files changed, 43 insertions(+), 12 deletions(-)

diff --git a/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultIdentity.java b/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultIdentity.java
index 3e03f79..dc44aab 100644
--- a/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultIdentity.java
+++ b/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultIdentity.java
@@ -101,13 +101,15 @@ public class DefaultIdentity implements Identity {
 	}
 
 	@Override
-	public void addContext(String context) {
+	public Identity addContext(String context) {
 		contexts.add(context);
+		return this;
 	}
 
 	@Override
-	public void removeContext(String context) {
+	public Identity removeContext(String context) {
 		contexts.remove(context);
+		return this;
 	}
 
 	@Override
@@ -127,13 +129,15 @@ public class DefaultIdentity implements Identity {
 	}
 
 	@Override
-	public void setProperty(String name, String value) {
+	public Identity setProperty(String name, String value) {
 		properties.put(name, value);
+		return this;
 	}
 
 	@Override
-	public void removeProperty(String name) {
+	public Identity removeProperty(String name) {
 		properties.remove(name);
+		return this;
 	}
 
 	@Override
@@ -142,13 +146,15 @@ public class DefaultIdentity implements Identity {
 	}
 
 	@Override
-	public void setTrust(OwnIdentity ownIdentity, Trust trust) {
+	public Identity setTrust(OwnIdentity ownIdentity, Trust trust) {
 		trustCache.put(ownIdentity, trust);
+		return this;
 	}
 
 	@Override
-	public void removeTrust(OwnIdentity ownIdentity) {
+	public Identity removeTrust(OwnIdentity ownIdentity) {
 		trustCache.remove(ownIdentity);
+		return this;
 	}
 
 	//
diff --git a/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultOwnIdentity.java b/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultOwnIdentity.java
index 08b936f..ce34b2a 100644
--- a/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultOwnIdentity.java
+++ b/src/main/java/net/pterodactylus/sone/freenet/wot/DefaultOwnIdentity.java
@@ -54,6 +54,26 @@ public class DefaultOwnIdentity extends DefaultIdentity implements OwnIdentity {
 		return insertUri;
 	}
 
+	@Override
+	public OwnIdentity addContext(String context) {
+		return (OwnIdentity) super.addContext(context);
+	}
+
+	@Override
+	public OwnIdentity removeContext(String context) {
+		return (OwnIdentity) super.removeContext(context);
+	}
+
+	@Override
+	public OwnIdentity setProperty(String name, String value) {
+		return (OwnIdentity) super.setProperty(name, value);
+	}
+
+	@Override
+	public OwnIdentity removeProperty(String name) {
+		return (OwnIdentity) super.removeProperty(name);
+	}
+
 	//
 	// OBJECT METHODS
 	//
diff --git a/src/main/java/net/pterodactylus/sone/freenet/wot/Identity.java b/src/main/java/net/pterodactylus/sone/freenet/wot/Identity.java
index ff87fcd..523a42b 100644
--- a/src/main/java/net/pterodactylus/sone/freenet/wot/Identity.java
+++ b/src/main/java/net/pterodactylus/sone/freenet/wot/Identity.java
@@ -92,7 +92,7 @@ public interface Identity {
 	 * @param context
 	 *            The context to add
 	 */
-	public void addContext(String context);
+	public Identity addContext(String context);
 
 	/**
 	 * Sets all contexts of this identity.
@@ -108,7 +108,7 @@ public interface Identity {
 	 * @param context
 	 *            The context to remove
 	 */
-	public void removeContext(String context);
+	public Identity removeContext(String context);
 
 	/**
 	 * Returns all properties of this identity.
@@ -134,7 +134,7 @@ public interface Identity {
 	 * @param value
 	 *            The value of the property
 	 */
-	public void setProperty(String name, String value);
+	public Identity setProperty(String name, String value);
 
 	/**
 	 * Sets all properties of this identity.
@@ -150,7 +150,7 @@ public interface Identity {
 	 * @param name
 	 *            The name of the property to remove
 	 */
-	public void removeProperty(String name);
+	public Identity removeProperty(String name);
 
 	/**
 	 * Retrieves the trust that this identity receives from the given own
@@ -173,7 +173,7 @@ public interface Identity {
 	 * @param trust
 	 *            The trust given by the given own identity
 	 */
-	public void setTrust(OwnIdentity ownIdentity, Trust trust);
+	public Identity setTrust(OwnIdentity ownIdentity, Trust trust);
 
 	/**
 	 * Removes trust assignment from the given own identity for this identity.
@@ -182,6 +182,6 @@ public interface Identity {
 	 *            The own identity that removed the trust assignment for this
 	 *            identity
 	 */
-	public void removeTrust(OwnIdentity ownIdentity);
+	public Identity removeTrust(OwnIdentity ownIdentity);
 
 }
diff --git a/src/main/java/net/pterodactylus/sone/freenet/wot/OwnIdentity.java b/src/main/java/net/pterodactylus/sone/freenet/wot/OwnIdentity.java
index 6fc7044..da6409b 100644
--- a/src/main/java/net/pterodactylus/sone/freenet/wot/OwnIdentity.java
+++ b/src/main/java/net/pterodactylus/sone/freenet/wot/OwnIdentity.java
@@ -32,4 +32,9 @@ public interface OwnIdentity extends Identity {
 	 */
 	public String getInsertUri();
 
+	public OwnIdentity addContext(String context);
+	public OwnIdentity removeContext(String context);
+	public OwnIdentity setProperty(String name, String value);
+	public OwnIdentity removeProperty(String name);
+
 }
-- 
2.7.4