X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsonitus%2Fio%2FOggVorbisIdentifier.java;h=9d384526cec29fff6ff120798beb61695890bdd2;hb=4371ab7e900c5cc47aa4c18e6b125581a3accb2e;hp=d889b976eaa7be01be16d9c812a3b92a9c609f5d;hpb=d0f12db4a8fe02601743e4b23618d44e193fd474;p=sonitus.git

diff --git a/src/main/java/net/pterodactylus/sonitus/io/OggVorbisIdentifier.java b/src/main/java/net/pterodactylus/sonitus/io/OggVorbisIdentifier.java
index d889b97..9d38452 100644
--- a/src/main/java/net/pterodactylus/sonitus/io/OggVorbisIdentifier.java
+++ b/src/main/java/net/pterodactylus/sonitus/io/OggVorbisIdentifier.java
@@ -20,7 +20,9 @@ package net.pterodactylus.sonitus.io;
 import java.io.IOException;
 import java.io.InputStream;
 
-import net.pterodactylus.sonitus.data.Format;
+import net.pterodactylus.sonitus.data.ContentMetadata;
+import net.pterodactylus.sonitus.data.FormatMetadata;
+import net.pterodactylus.sonitus.data.Metadata;
 
 import com.google.common.base.Optional;
 import com.jcraft.jogg.Packet;
@@ -31,7 +33,9 @@ import com.jcraft.jorbis.Comment;
 import com.jcraft.jorbis.Info;
 
 /**
- * Identifies Ogg Vorbis files.
+ * Identifies Ogg Vorbis files. <p> All knowledge used in this class has been
+ * taken from <a href="http://www.jcraft.com/jorbis/tutorial/Tutorial.html">jcraft.com/jorbis/tutorial/Tutorial.html</a>.
+ * </p>
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
@@ -47,16 +51,16 @@ public class OggVorbisIdentifier {
 
 	/**
 	 * Tries to parse the given stream as Ogg Vorbis file and returns a {@link
-	 * Format} describing the stream.
+	 * Metadata} describing the stream.
 	 *
 	 * @param inputStream
 	 * 		The input stream to identify as Ogg Vorbis
-	 * @return The identified format, or {@link com.google.common.base.Optional#absent()}
-	 *         if the stream could not be identified
+	 * @return The identified metadata, or {@link Optional#absent()} if the stream
+	 *         could not be identified
 	 * @throws IOException
 	 * 		if an I/O error occurs
 	 */
-	public static Optional<Format> identify(InputStream inputStream) throws IOException {
+	public static Optional<Metadata> identify(InputStream inputStream) throws IOException {
 
 		/* stuff needed to decode Ogg. */
 		Packet packet = new Packet();
@@ -113,7 +117,40 @@ public class OggVorbisIdentifier {
 			buffer = syncState.data;
 		}
 
-		return Optional.of(new Format(info.channels, info.rate, "Vorbis"));
+		FormatMetadata formatMetadata = new FormatMetadata(info.channels, info.rate, "Vorbis");
+		ContentMetadata contentMetadata = new ContentMetadata("");
+		for (int c = 0; c < comment.comments; ++c) {
+			String field = comment.getComment(c);
+			Optional<String> extractedField = extractField(field, "ARTIST");
+			if (extractedField.isPresent()) {
+				contentMetadata = contentMetadata.artist(extractedField.get());
+				continue;
+			}
+			extractedField = extractField(field, "TITLE");
+			if (extractedField.isPresent()) {
+				contentMetadata = contentMetadata.name(extractedField.get());
+				continue;
+			}
+		}
+		return Optional.of(new Metadata(formatMetadata, contentMetadata));
+	}
+
+	/**
+	 * Extracts the content of the field from the comment if the comment contains
+	 * the given field.
+	 *
+	 * @param comment
+	 * 		The comment to extract the value from
+	 * @param fieldName
+	 * 		The name of the field to extract
+	 * @return The extracted field, or {@link Optional#absent()} if the comment
+	 *         does not contain the given field
+	 */
+	private static Optional<String> extractField(String comment, String fieldName) {
+		if (comment.startsWith(fieldName + "=")) {
+			return Optional.of(comment.substring(fieldName.length() + 1));
+		}
+		return Optional.absent();
 	}
 
 }