summaryrefslogtreecommitdiff
path: root/org.fox.ttcomics/src/main/java/org/fox/ttcomics2/archive/CbzComicArchive.java
blob: 987ab862dee2602557c00ba8ac295d5a3701ae4f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package org.fox.ttcomics2.archive;

import android.os.Parcel;
import android.os.Parcelable;

import net.lingala.zip4j.core.ZipFile;
import net.lingala.zip4j.exception.ZipException;
import net.lingala.zip4j.model.FileHeader;

import org.fox.ttcomics2.utils.NaturalOrderComparator;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class CbzComicArchive extends ComicArchive {
	private final String TAG = this.getClass().getSimpleName();

    private String m_fileName;
	private ZipFile m_zipFile;
	private ArrayList<String> m_entries = new ArrayList<>();
		
	@Override
	public int getCount() {
		return m_entries.size();
	}
	
	@Override
	public InputStream getItem(int index) throws IOException {
	    try {
            return m_zipFile.getInputStream(m_zipFile.getFileHeader(m_entries.get(index)));
        } catch (ZipException e) {
	        throw new IOException(e.getMessage());
        }
	}

    private void initialize() throws IOException {
	    try {
            m_zipFile = new ZipFile(m_fileName);

            List<FileHeader> fileHeaders = m_zipFile.getFileHeaders();

            for (int i = 0; i < fileHeaders.size(); i++) {
                FileHeader fh = fileHeaders.get(i);

                if (fh != null) {
                    if (!fh.isDirectory() && isValidComic(fh.getFileName())) {
                        m_entries.add(fh.getFileName());
                    }
                }
            }
        } catch (ZipException e) {
	        throw new IOException(e.getMessage());
        }

        Collections.sort(m_entries, new NaturalOrderComparator());
    }

	public CbzComicArchive(String fileName) throws IOException {
        m_fileName = fileName;

        initialize();
	}

    @Override
    public int describeContents() {
        return 0;
    }

    @Override
    public void writeToParcel(Parcel out, int flags) {
        out.writeString(m_fileName);
    }

    public void readFromParcel(Parcel in) {
        m_fileName = in.readString();
    }

    public CbzComicArchive(Parcel in) throws IOException {
        readFromParcel(in);

        initialize();
    }

    @SuppressWarnings("rawtypes")
    public static final Parcelable.Creator CREATOR =
        new Parcelable.Creator() {
            public CbzComicArchive createFromParcel(Parcel in) {
                try {
                    return new CbzComicArchive(in);
                } catch (IOException e) {
                    e.printStackTrace();

                    return null;
                }
            }

            public CbzComicArchive[] newArray(int size) {
                return new CbzComicArchive[size];
            }
        };

}