Allow method chaining when changing identities.

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 (file)
--- 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 (file)
--- 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 (file)
--- 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 (file)
--- 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);
+
 }