mods/download/ModDownload.java

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.mods.unpack.UnpackListener;
import net.oni2.aeinstaller.backend.mods.unpack.Unpacker;
import net.oni2.aeinstaller.backend.network.FileDownloadListener;
import net.oni2.aeinstaller.backend.network.FileDownloader;
import net.oni2.aeinstaller.backend.network.FileDownloader.EState;

/**
 * @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 = mod.getLocalPath();
                try {
                        downloader = new FileDownloader(mod.getFile().getUri_full(),
                                        zipFile);
                        downloader.addListener(this);
                        unpacker = new Unpacker(zipFile, targetFolder, this);
                } catch (IOException e) {
                        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.mods.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;
                }
        }
}
Revision:	634
Committed:	Sat Jan 19 11:58:17 2013 UTC (12 years, 8 months ago) by alloc
Content type:	text/x-java
File size:	4077 byte(s)
Log Message:	AEI2 0.92b:\n- Using zipped Depot cache
#	Content
1	package net.oni2.aeinstaller.backend.mods.download;
2
3	import java.io.File;
4	import java.io.FileNotFoundException;
5	import java.io.IOException;
6	import java.io.PrintWriter;
7
8	import net.oni2.aeinstaller.backend.Paths;
9	import net.oni2.aeinstaller.backend.mods.Mod;
10	import net.oni2.aeinstaller.backend.mods.unpack.UnpackListener;
11	import net.oni2.aeinstaller.backend.mods.unpack.Unpacker;
12	import net.oni2.aeinstaller.backend.network.FileDownloadListener;
13	import net.oni2.aeinstaller.backend.network.FileDownloader;
14	import net.oni2.aeinstaller.backend.network.FileDownloader.EState;
15
16	/**
17	* @author Christian Illy
18	*/
19	public class ModDownload implements FileDownloadListener, UnpackListener {
20	/**
21	* @author Christian Illy
22	*/
23	public enum ModDownloadState {
24	/**
25	* Downloader initialized but not started
26	*/
27	INIT,
28	/**
29	* Download running
30	*/
31	RUNNING,
32	/**
33	* Aborted because of an error
34	*/
35	ERROR,
36	/**
37	* Download interrupted
38	*/
39	INTERRUPTED,
40	/**
41	* Download finished successfully
42	*/
43	DOWNLOADED,
44	/**
45	* Package unzipped successfully
46	*/
47	UNPACKED
48	};
49
50	private Mod mod;
51	private FileDownloader downloader;
52	private Unpacker unpacker;
53	private File zipFile;
54	private File targetFolder;
55	private ModDownloadListener listener;
56	private int size;
57
58	private ModDownloadState state = ModDownloadState.INIT;
59
60	/**
61	* Create a mod download
62	*
63	* @param mod
64	* Mod to download
65	* @param listener
66	* Listener for progress
67	*/
68	public ModDownload(Mod mod, ModDownloadListener listener) {
69	this.mod = mod;
70	this.listener = listener;
71
72	zipFile = new File(Paths.getDownloadPath(),
73	mod.getPackageNumberString() + ".zip");
74	targetFolder = mod.getLocalPath();
75	try {
76	downloader = new FileDownloader(mod.getFile().getUri_full(),
77	zipFile);
78	downloader.addListener(this);
79	unpacker = new Unpacker(zipFile, targetFolder, this);
80	} catch (IOException e) {
81	e.printStackTrace();
82	}
83	}
84
85	/**
86	* @return Size of this download
87	*/
88	public int getSize() {
89	return mod.getZipSize();
90	}
91
92	/**
93	* Start this download
94	*/
95	public void start() {
96	state = ModDownloadState.RUNNING;
97	downloader.start();
98	}
99
100	/**
101	* Abort this download
102	*/
103	public void abort() {
104	switch (state) {
105	case UNPACKED:
106	case INIT:
107	case ERROR:
108	case INTERRUPTED:
109	break;
110	case RUNNING:
111	downloader.stop();
112	break;
113	case DOWNLOADED:
114	unpacker.stop();
115	break;
116	}
117	state = ModDownloadState.INTERRUPTED;
118	}
119
120	/**
121	* @return the mod object handled by this download
122	*/
123	public Mod getMod() {
124	return mod;
125	}
126
127	private void writeTimestamp() {
128	File logFile = new File(targetFolder, "aei.cfg");
129	PrintWriter log = null;
130	try {
131	log = new PrintWriter(logFile);
132	log.println("Timestamp -> " + mod.getFile().getTimestamp());
133	} catch (FileNotFoundException e) {
134	e.printStackTrace();
135	}
136	if (log != null)
137	log.close();
138	}
139
140	@Override
141	public void statusUpdate(FileDownloader source, EState state, int done,
142	int total) {
143	switch (state) {
144	case INIT:
145	break;
146	case RUNNING:
147	listener.modDownloadStatusUpdate(this, this.state, done, total);
148	break;
149	case PAUSED:
150	break;
151	case INTERRUPTED:
152	break;
153	case ERROR:
154	this.state = ModDownloadState.ERROR;
155	listener.modDownloadStatusUpdate(this, this.state, done, total);
156	break;
157	case FINISHED:
158	this.state = ModDownloadState.DOWNLOADED;
159	listener.modDownloadStatusUpdate(this, this.state, done, total);
160	this.size = done;
161	unpacker.start();
162	break;
163	}
164	}
165
166	@Override
167	public void statusUpdate(Unpacker source,
168	net.oni2.aeinstaller.backend.mods.unpack.Unpacker.EState state) {
169	switch (state) {
170	case INIT:
171	break;
172	case RUNNING:
173	break;
174	case INTERRUPTED:
175	this.state = ModDownloadState.INTERRUPTED;
176	listener.modDownloadStatusUpdate(this, this.state, size, size);
177	break;
178	case FINISHED:
179	this.state = ModDownloadState.UNPACKED;
180	writeTimestamp();
181	zipFile.delete();
182	listener.modDownloadStatusUpdate(this, this.state, size, size);
183	break;
184	}
185	}
186	}