Override Object.toString().

[Sone.git] / src / main / java / net / pterodactylus / sone / fcp / AbstractSoneCommand.java

diff --git a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
index 87a76e2..5b146c8 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
+++ b/src/main/java/net/pterodactylus/sone/fcp/AbstractSoneCommand.java
@@ -44,6 +44,9 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
        /** The Sone core. */
        private final Core core;
 
+       /** Whether this command needs write access. */
+       private final boolean writeAccess;
+
        /**
         * Creates a new abstract Sone FCP command.
         *
@@ -51,7 +54,21 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
         *            The Sone core
         */
        protected AbstractSoneCommand(Core core) {
+               this(core, false);
+       }
+
+       /**
+        * Creates a new abstract Sone FCP command.
+        *
+        * @param core
+        *            The Sone core
+        * @param writeAccess
+        *            {@code true} if this command requires write access,
+        *            {@code false} otherwise
+        */
+       protected AbstractSoneCommand(Core core, boolean writeAccess) {
                this.core = core;
+               this.writeAccess = writeAccess;
        }
 
        //
@@ -67,11 +84,34 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
                return core;
        }
 
+       /**
+        * Returns whether this command requires write access.
+        *
+        * @return {@code true} if this command require write access, {@code false}
+        *         otherwise
+        */
+       public boolean requiresWriteAccess() {
+               return writeAccess;
+       }
+
        //
        // PROTECTED METHODS
        //
 
        /**
+        * Encodes text in a way that makes it possible for the text to be stored in
+        * a {@link SimpleFieldSet}. Backslashes, CR, and LF are prepended with a
+        * backslash.
+        *
+        * @param text
+        *            The text to encode
+        * @return The encoded text
+        */
+       protected String encodeString(String text) {
+               return text.replaceAll("\\\\", "\\\\").replaceAll("\n", "\\\\n").replaceAll("\r", "\\\\r");
+       }
+
+       /**
         * Returns a Sone whose ID is a parameter in the given simple field set.
         *
         * @param simpleFieldSet
@@ -200,7 +240,7 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
                        postBuilder.put(prefix + "Recipient", post.getRecipient().getId());
                }
                postBuilder.put(prefix + "Time", post.getTime());
-               postBuilder.put(prefix + "Text", post.getText());
+               postBuilder.put(prefix + "Text", encodeString(post.getText()));
                postBuilder.put(encodeLikes(core.getLikes(post), prefix + "Likes."));
 
                if (includeReplies) {
@@ -260,7 +300,7 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
                        replyBuilder.put(replyPrefix + "ID", reply.getId());
                        replyBuilder.put(replyPrefix + "Sone", reply.getSone().getId());
                        replyBuilder.put(replyPrefix + "Time", reply.getTime());
-                       replyBuilder.put(replyPrefix + "Text", reply.getText());
+                       replyBuilder.put(replyPrefix + "Text", encodeString(reply.getText()));
                }
 
                return replyBuilder.get();
@@ -290,4 +330,16 @@ public abstract class AbstractSoneCommand extends AbstractCommand {
                return likesBuilder.get();
        }
 
+       //
+       // OBJECT METHODS
+       //
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public String toString() {
+               return getClass().getName() + "[writeAccess=" + writeAccess + "]";
+       }
+
 }