X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsonitus%2Fdata%2Fsink%2FIcecast2Sink.java;h=5487cc48fb5b34784f706dbd50ef4939122f2b76;hb=f554e51504f7e39e658a84d79c041e0b0d8a6393;hp=f3c60396ee16b2a345416cf17ff89607f3d6de96;hpb=6a38dc83de243eeef3dc1fc8ae1daf285fe933fc;p=sonitus.git diff --git a/src/main/java/net/pterodactylus/sonitus/data/sink/Icecast2Sink.java b/src/main/java/net/pterodactylus/sonitus/data/sink/Icecast2Sink.java index f3c6039..5487cc4 100644 --- a/src/main/java/net/pterodactylus/sonitus/data/sink/Icecast2Sink.java +++ b/src/main/java/net/pterodactylus/sonitus/data/sink/Icecast2Sink.java @@ -17,28 +17,33 @@ package net.pterodactylus.sonitus.data.sink; -import static com.google.common.base.Preconditions.*; - import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.io.UnsupportedEncodingException; import java.net.Socket; +import java.net.URLEncoder; +import java.util.Arrays; +import java.util.Collections; +import java.util.List; +import java.util.logging.Level; import java.util.logging.Logger; -import net.pterodactylus.sonitus.data.ConnectException; -import net.pterodactylus.sonitus.data.Connection; -import net.pterodactylus.sonitus.data.Format; +import net.pterodactylus.sonitus.data.Controller; +import net.pterodactylus.sonitus.data.Metadata; import net.pterodactylus.sonitus.data.Sink; -import net.pterodactylus.sonitus.data.Source; import net.pterodactylus.sonitus.io.InputStreamDrainer; +import com.google.common.base.Function; +import com.google.common.base.Joiner; +import com.google.common.base.Optional; +import com.google.common.collect.FluentIterable; import com.google.common.io.BaseEncoding; import com.google.common.io.Closeables; /** - * {@link Sink} implementation that delivers all incoming data to an Icecast2 - * server. + * {@link net.pterodactylus.sonitus.data.Sink} implementation that delivers all + * incoming data to an Icecast2 server. * * @author David ‘Bombe’ Roden */ @@ -71,6 +76,11 @@ public class Icecast2Sink implements Sink { /** Whether to publish the server. */ private final boolean publishServer; + private OutputStream socketOutputStream; + + /** The current metadata. */ + private Metadata metadata; + /** * Creates a new Icecast2 sink. * @@ -104,57 +114,109 @@ public class Icecast2Sink implements Sink { } // + // CONTROLLED METHODS + // + + @Override + public String name() { + return String.format("icecast://%s:%d/%s", server, port, mountPoint); + } + + @Override + public Metadata metadata() { + return metadata; + } + + @Override + public List> controllers() { + return Collections.emptyList(); + } + + // // SINK METHODS // @Override - public void connect(Source source) throws ConnectException { - checkNotNull(source, "source must not be null"); + public void open(Metadata metadata) throws IOException { + logger.info(String.format("Connecting to %s:%d...", server, port)); + Socket socket = new Socket(server, port); + logger.info("Connected."); + socketOutputStream = socket.getOutputStream(); + InputStream socketInputStream = socket.getInputStream(); + + sendLine(socketOutputStream, String.format("SOURCE /%s ICE/1.0", mountPoint)); + sendLine(socketOutputStream, String.format("Authorization: Basic %s", generatePassword(password))); + sendLine(socketOutputStream, String.format("Content-Type: %s", getContentType(metadata))); + sendLine(socketOutputStream, String.format("ICE-Name: %s", serverName)); + sendLine(socketOutputStream, String.format("ICE-Description: %s", serverDescription)); + sendLine(socketOutputStream, String.format("ICE-Genre: %s", genre)); + sendLine(socketOutputStream, String.format("ICE-Public: %d", publishServer ? 1 : 0)); + sendLine(socketOutputStream, ""); + socketOutputStream.flush(); + + new Thread(new InputStreamDrainer(socketInputStream)).start(); + + metadataUpdated(metadata); + } + @Override + public void close() { try { - logger.info(String.format("Icecast2Sink: Connecting to %s:%d...", server, port)); - final Socket socket = new Socket(server, port); - logger.info("Icecast2Sink: Connected."); - final OutputStream socketOutputStream = socket.getOutputStream(); - final InputStream socketInputStream = socket.getInputStream(); - - sendLine(socketOutputStream, String.format("SOURCE /%s ICE/1.0", mountPoint)); - sendLine(socketOutputStream, String.format("Authorization: Basic %s", generatePassword(password))); - sendLine(socketOutputStream, String.format("Content-Type: %s", getContentType(source.format()))); - sendLine(socketOutputStream, String.format("ICE-Name: %s", serverName)); - sendLine(socketOutputStream, String.format("ICE-Description: %s", serverDescription)); - sendLine(socketOutputStream, String.format("ICE-Genre: %s", genre)); - sendLine(socketOutputStream, String.format("ICE-Public: %d", publishServer ? 1 : 0)); - sendLine(socketOutputStream, ""); - socketOutputStream.flush(); - - new Thread(new InputStreamDrainer(socketInputStream)).start(); - new Thread(new Connection(source) { - - private long counter; - - @Override - protected int bufferSize() { - return 4096; - } + Closeables.close(socketOutputStream, true); + } catch (IOException e) { + /* will never throw. */ + } + } + + @Override + public void metadataUpdated(final Metadata metadata) { + this.metadata = metadata; + new Thread(new Runnable() { + + @Override + public void run() { + String metadataString = String.format("%s (%s)", Joiner.on(" - ").skipNulls().join(FluentIterable.from(Arrays.asList(metadata.artist(), metadata.name())).transform(new Function, Object>() { + + @Override + public Object apply(Optional input) { + return input.orNull(); + } + })), "Sonitus"); + logger.info(String.format("Updating metadata to %s", metadataString)); - @Override - protected void feed(byte[] buffer) throws IOException { - socketOutputStream.write(buffer); + Socket socket = null; + OutputStream socketOutputStream = null; + try { + socket = new Socket(server, port); + socketOutputStream = socket.getOutputStream(); + + sendLine(socketOutputStream, String.format("GET /admin/metadata?pass=%s&mode=updinfo&mount=/%s&song=%s HTTP/1.0", password, mountPoint, URLEncoder.encode(metadataString, "UTF-8"))); + sendLine(socketOutputStream, String.format("Authorization: Basic %s", generatePassword(password))); + sendLine(socketOutputStream, String.format("User-Agent: Mozilla/Sonitus")); + sendLine(socketOutputStream, ""); socketOutputStream.flush(); - counter += buffer.length; - logger.finest(String.format("Wrote %d Bytes.", counter)); - } - @Override - protected void finish() throws IOException { - Closeables.close(socketOutputStream, true); - Closeables.close(socket, true); + new InputStreamDrainer(socket.getInputStream()).run(); + } catch (IOException ioe1) { + logger.log(Level.WARNING, "Could not update metadata!", ioe1); + } finally { + try { + Closeables.close(socketOutputStream, true); + if (socket != null) { + socket.close(); + } + } catch (IOException ioe1) { + /* ignore. */ + } } - }).start(); - } catch (IOException ioe1) { - throw new ConnectException(ioe1); - } + } + }).start(); + } + + @Override + public void process(byte[] buffer) throws IOException { + socketOutputStream.write(buffer); + socketOutputStream.flush(); } // @@ -169,7 +231,7 @@ public class Icecast2Sink implements Sink { * The output stream to send the line to * @param line * The line to send - * @throws IOException + * @throws java.io.IOException * if an I/O error occurs */ private static void sendLine(OutputStream outputStream, String line) throws IOException { @@ -183,7 +245,7 @@ public class Icecast2Sink implements Sink { * @param password * The password to encode * @return The encoded password - * @throws UnsupportedEncodingException + * @throws java.io.UnsupportedEncodingException * if the UTF-8 encoding is not supported (which can never happen) */ private static String generatePassword(String password) throws UnsupportedEncodingException { @@ -191,24 +253,34 @@ public class Icecast2Sink implements Sink { } /** - * Returns a MIME type for the given format. Currently only Vorbis, MP3, and - * PCM formats are recognized. + * Returns a MIME type for the given metadata. Currently only Vorbis, MP3, PCM, + * Ogg Vorbis, Opus, and FLAC formats are recognized. * - * @param format - * The format to get a MIME type for - * @return The MIME type of the format + * @param metadata + * The metadata to get a MIME type for + * @return The MIME type of the metadata */ - private static String getContentType(Format format) { - switch (format.encoding().toUpperCase()) { - case "VORBIS": - return "audio/ogg"; - case "MP3": - return "audio/mpeg"; - case "PCM": - return "audio/vnd.wave"; - default: - return "application/octet-stream"; + private static String getContentType(Metadata metadata) { + String encoding = metadata.encoding(); + if ("Vorbis".equalsIgnoreCase(encoding)) { + return "audio/ogg"; + } + if ("MP3".equalsIgnoreCase(encoding)) { + return "audio/mpeg"; + } + if ("PCM".equalsIgnoreCase(encoding)) { + return "audio/vnd.wave"; + } + if ("Vorbis".equalsIgnoreCase(encoding)) { + return "application/ogg"; + } + if ("Opus".equalsIgnoreCase(encoding)) { + return "audio/ogg; codecs=opus"; + } + if ("FLAC".equalsIgnoreCase(encoding)) { + return "audio/flac"; } + return "application/octet-stream"; } }