DMDirc-Plugins/src/com/dmdirc/addons/mediasource_windows/WindowsMediaSourcePlugin.java

/*
 * Copyright (c) 2006-2010 Chris Smith, Shane Mc Cormack, Gregory Holmes
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package com.dmdirc.addons.mediasource_windows;

import com.dmdirc.Main;
import com.dmdirc.addons.nowplaying.MediaSource;
import com.dmdirc.addons.nowplaying.MediaSourceManager;
import com.dmdirc.logger.ErrorLevel;
import com.dmdirc.logger.Logger;
import com.dmdirc.plugins.Plugin;
import com.dmdirc.plugins.PluginInfo;
import com.dmdirc.plugins.PluginManager;
import com.dmdirc.util.resourcemanager.ResourceManager;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.io.File;
import java.io.IOException;

import com.dmdirc.installer.StreamReader;

/**
 * Manages all Windows based media sources.
 */
public class WindowsMediaSourcePlugin extends Plugin implements MediaSourceManager {

    /** Media sources. */
    private final List<MediaSource> sources;

    /** Files dir */
    private static final String filesDir = Main.getConfigDir() + "plugins/windowsmediasource_files/";

    /**
     * Creates a new instance of DcopMediaSourcePlugin.
     */
    public WindowsMediaSourcePlugin() {
        super();
        sources = new ArrayList<MediaSource>();
        sources.add(new DllSource("Winamp", true));
        sources.add(new DllSource("iTunes", false));
    }

    /** {@inheritDoc} */
    @Override
    public List<MediaSource> getSources() {
        return sources;
    }

    /**
     * Get the output from GetMediaInfo.exe for the given player and method
     *
     * @param player Player to ask about
     * @param method Method to call
     * @return a MediaInfoOutput with the results
     */
    protected static MediaInfoOutput getOutput(final String player, final String method) {
        try {
            final Process myProcess = Runtime.getRuntime().exec(new String[]{filesDir + "GetMediaInfo.exe", player, method});
            final StringBuffer data = new StringBuffer();
            new StreamReader(myProcess.getErrorStream()).start();
            new StreamReader(myProcess.getInputStream(), data).start();
            try {
                myProcess.waitFor();
            } catch (InterruptedException e) {
            }

            return new MediaInfoOutput(myProcess.exitValue(), data.toString());
        } catch (SecurityException e) {
        } catch (IOException e) {
        }

        return new MediaInfoOutput(-1, "Error executing GetMediaInfo.exe");
    }

    /**
     * Use the given resource manager to extract files ending with the given suffix
     *
     * @param res ResourceManager
     * @param newDir Directory to extract to.
     * @param suffix Suffix to extract
     */
    private void extractFiles(final ResourceManager res, final File newDir, final String suffix) {
        final Map<String, byte[]> resources = res.getResourcesEndingWithAsBytes(suffix);
        for (Entry<String, byte[]> resource : resources.entrySet()) {
            try {
                final String key = resource.getKey();
                final String resourceName = key.substring(key.lastIndexOf('/'), key.length());

                final File newFile = new File(newDir, resourceName);

                if (!newFile.isDirectory()) {
                    if (newFile.exists()) {
                        newFile.delete();
                    }
                    ResourceManager.getResourceManager().resourceToFile(resource.getValue(), newFile);
                }
            } catch (IOException ex) {
                Logger.userError(ErrorLevel.LOW, "Failed to extract " + suffix + "s for windowsmediasource: " + ex.getMessage(), ex);
            }
        }
    }

    /** {@inheritDoc} */
    @Override
    public void onLoad() {
        // Extract GetMediaInfo.exe and required DLLs
        final PluginInfo pi = PluginManager.getPluginManager().getPluginInfoByName("windowsmediasource");

        // This shouldn't actually happen, but check to make sure.
        if (pi == null) {
            return;
        }

        // Now get the RM
        try {
            final ResourceManager res = pi.getResourceManager();

            // Make sure our files dir exists
            final File newDir = new File(filesDir);
            if (!newDir.exists()) {
                newDir.mkdirs();
            }

            // Now extract the .dlls and .exe
            extractFiles(res, newDir, ".dll");
            extractFiles(res, newDir, ".exe");
        } catch (IOException ioe) {
            Logger.userError(ErrorLevel.LOW, "Unable to open ResourceManager for windowsmediasource: " + ioe.getMessage(), ioe);
        }
    }

    /** {@inheritDoc} */
    @Override
    public void onUnload() { /* Do Nothing */ }

}