Add filter for linked images

[Sone.git] / src / main / java / net / pterodactylus / sone / text / SonePart.java

diff --git a/src/main/java/net/pterodactylus/sone/text/SonePart.java b/src/main/java/net/pterodactylus/sone/text/SonePart.java
index 5c6a63d..576eb7b 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/text/SonePart.java
+++ b/src/main/java/net/pterodactylus/sone/text/SonePart.java
@@ -1,5 +1,5 @@
 /*
- * Sone - SoneLinkPart.java - Copyright © 2011 David Roden
+ * Sone - SonePart.java - Copyright © 2011–2016 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -17,7 +17,12 @@
 
 package net.pterodactylus.sone.text;
 
+import java.util.Objects;
+
+import javax.annotation.Nonnull;
+
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.template.SoneAccessor;
 
 /**
  * {@link Part} implementation that stores a reference to a {@link Sone}.
@@ -26,30 +31,20 @@ import net.pterodactylus.sone.data.Sone;
  */
 public class SonePart implements Part {
 
-       /** The referenced {@link Sone}. */
        private final Sone sone;
 
-       /**
-        * Creates a new Sone part.
-        *
-        * @param sone
-        *            The referenced Sone
-        */
-       public SonePart(Sone sone) {
-               this.sone = sone;
+       public SonePart(@Nonnull Sone sone) {
+               this.sone = Objects.requireNonNull(sone);
        }
 
-       //
-       // ACCESSORS
-       //
-
-       /**
-        * Returns the referenced Sone.
-        *
-        * @return The referenced Sone
-        */
+       @Nonnull
        public Sone getSone() {
                return sone;
        }
 
+       @Override
+       public String getText() {
+               return SoneAccessor.getNiceName(sone);
+       }
+
 }