source: AE/installer2/src/net/oni2/aeinstaller/backend/mods/download/ModDownload.java@ 613

package net.oni2.aeinstaller.backend.mods.download;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.PrintWriter;

import net.oni2.aeinstaller.backend.Paths;
import net.oni2.aeinstaller.backend.mods.Mod;
import net.oni2.aeinstaller.backend.network.FileDownloadListener;
import net.oni2.aeinstaller.backend.network.FileDownloader;
import net.oni2.aeinstaller.backend.network.FileDownloader.EState;
import net.oni2.aeinstaller.backend.unpack.UnpackListener;
import net.oni2.aeinstaller.backend.unpack.Unpacker;

/**
 * @author Christian Illy
 */
public class ModDownload implements FileDownloadListener, UnpackListener {
        /**
         * @author Christian Illy
         */
        public enum ModDownloadState {
                /**
                 * Downloader initialized but not started
                 */
                INIT,
                /**
                 * Download running
                 */
                RUNNING,
                /**
                 * Aborted because of an error
                 */
                ERROR,
                /**
                 * Download interrupted
                 */
                INTERRUPTED,
                /**
                 * Download finished successfully
                 */
                DOWNLOADED,
                /**
                 * Package unzipped successfully
                 */
                UNPACKED
        };

        private Mod mod;
        private FileDownloader downloader;
        private Unpacker unpacker;
        private File zipFile;
        private File targetFolder;
        private ModDownloadListener listener;
        private int size;

        private ModDownloadState state = ModDownloadState.INIT;

        /**
         * Create a mod download
         * 
         * @param mod
         *            Mod to download
         * @param listener
         *            Listener for progress
         */
        public ModDownload(Mod mod, ModDownloadListener listener) {
                this.mod = mod;
                this.listener = listener;

                zipFile = new File(Paths.getDownloadPath(),
                                mod.getPackageNumberString() + ".zip");
                targetFolder = new File(Paths.getModsPath(),
                                mod.getPackageNumberString());
                try {
                        downloader = new FileDownloader(mod.getFile().getUri_full(),
                                        zipFile.getPath());
                        downloader.addListener(this);
                        unpacker = new Unpacker(zipFile, targetFolder, this);
                } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
        }

        /**
         * @return Size of this download
         */
        public int getSize() {
                return mod.getZipSize();
        }

        /**
         * Start this download
         */
        public void start() {
                state = ModDownloadState.RUNNING;
                downloader.start();
        }

        /**
         * Abort this download
         */
        public void abort() {
                switch (state) {
                        case UNPACKED:
                        case INIT:
                        case ERROR:
                        case INTERRUPTED:
                                break;
                        case RUNNING:
                                downloader.stop();
                                break;
                        case DOWNLOADED:
                                unpacker.stop();
                                break;
                }
                state = ModDownloadState.INTERRUPTED;
        }

        /**
         * @return the mod object handled by this download
         */
        public Mod getMod() {
                return mod;
        }

        private void writeTimestamp() {
                File logFile = new File(targetFolder, "aei.cfg");
                PrintWriter log = null;
                try {
                        log = new PrintWriter(logFile);
                        log.println("Timestamp -> " + mod.getFile().getTimestamp());
                } catch (FileNotFoundException e) {
                        e.printStackTrace();
                }
                if (log != null)
                        log.close();
        }

        @Override
        public void statusUpdate(FileDownloader source, EState state, int done,
                        int total) {
                switch (state) {
                        case INIT:
                                break;
                        case RUNNING:
                                listener.modDownloadStatusUpdate(this, this.state, done, total);
                                break;
                        case PAUSED:
                                break;
                        case INTERRUPTED:
                                break;
                        case ERROR:
                                this.state = ModDownloadState.ERROR;
                                listener.modDownloadStatusUpdate(this, this.state, done, total);
                                break;
                        case FINISHED:
                                this.state = ModDownloadState.DOWNLOADED;
                                listener.modDownloadStatusUpdate(this, this.state, done, total);
                                this.size = done;
                                unpacker.start();
                                break;
                }
        }

        @Override
        public void statusUpdate(Unpacker source,
                        net.oni2.aeinstaller.backend.unpack.Unpacker.EState state) {
                switch (state) {
                        case INIT:
                                break;
                        case RUNNING:
                                break;
                        case INTERRUPTED:
                                this.state = ModDownloadState.INTERRUPTED;
                                listener.modDownloadStatusUpdate(this, this.state, size, size);
                                break;
                        case FINISHED:
                                this.state = ModDownloadState.UNPACKED;
                                writeTimestamp();
                                zipFile.delete();
                                listener.modDownloadStatusUpdate(this, this.state, size, size);
                                break;
                }
        }
}