com.jdom.tvshowdownloader.integration.MockNzbDownloader.java Source code

Introduction

Here is the source code for com.jdom.tvshowdownloader.integration.MockNzbDownloader.java

Source

/** 
 *  Copyright (C) 2012  Just Do One More
 *  
 *  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
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.jdom.tvshowdownloader.integration;

import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;

import com.jdom.mediadownloader.series.SeriesConfiguration;
import com.jdom.mediadownloader.series.domain.SeriesDownload;
import com.jdom.mediadownloader.series.download.DownloadedNzbMover;
import com.jdom.mediadownloader.series.download.NzbAdder;
import com.jdom.mediadownloader.series.download.SabnzbdNzbDownloader;
import com.jdom.mediadownloader.series.services.SeriesDasFactory;
import com.jdom.mediadownloader.services.UrlDownloadService;

public class MockNzbDownloader extends SabnzbdNzbDownloader implements ApplicationContextAware {

    public static ApplicationContext context;

    public MockNzbDownloader(SeriesDasFactory dasFactory, UrlDownloadService urlDownloadService,
            DownloadedNzbMover downloadedNzbMover, NzbAdder nzbAdder) {
        super(dasFactory, urlDownloadService, downloadedNzbMover, nzbAdder);
    }

    @Override
    public void downloadNzb(SeriesDownload download) {
        super.downloadNzb(download);

        String[] nzbs = SeriesConfiguration.NZB_QUEUE_DIRECTORY.list(new FilenameFilter() {
            @Override
            public boolean accept(File dir, String name) {
                return name.endsWith(".nzb");
            }
        });
        if (nzbs != null) {
            for (String string : nzbs) {
                File file = new File(SeriesConfiguration.NZB_QUEUE_DIRECTORY, string);
                File destinationFolder = new File(SeriesConfiguration.NZB_DOWNLOADED_DIRECTORY,
                        file.getName().replaceAll(".nzb", ""));
                try {
                    org.apache.commons.io.FileUtils.moveFileToDirectory(file, destinationFolder, true);
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
        }

    }

    @Override
    public void setApplicationContext(ApplicationContext arg0) throws BeansException {
        MockNzbDownloader.context = arg0;
    }

}