-
Notifications
You must be signed in to change notification settings - Fork 43
/
zipfile.monkey2
181 lines (119 loc) · 3.17 KB
/
zipfile.monkey2
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
Namespace std.zipfile
Using miniz
#rem monkeydoc The ZipFile class.
#end
Class ZipFile
#rem monkeydoc Array of all files contained in the zip.
#end
Property Files:String[]()
Return _files
End
#rem monkeydoc Closes the zip.
#end
Method Close()
If Not _data Return
libc.free( _zip )
_files=Null
_sizes=Null
_data.Discard()
_data=Null
End
#rem monkeydoc Checks if a file is contained in the zip.
#end
Method Contains:Bool( file:String )
Return FindFile( file )<>-1
End
#rem moneydoc Extracts a file from the zip into a databuffer.
#end
Method ExtractData:DataBuffer( file:String )
Local i:=FindFile( file )
If i=-1 Return Null
Local size:=_sizes[i]
Local buf:=New DataBuffer( size )
If mz_zip_reader_extract_to_mem( _zip,i,buf.Data,size,0 ) Return buf
buf.Discard()
Return Null
End
#rem monkeydoc @hidden
#end
Method FindFile:Int( file:String )
For Local i:=0 Until _files.Length
If file=_files[i] Return i
Next
Return -1
End
#rem monkeydoc @hidden
#end
Method Extract:Bool( dir:String,prefix:String="" )
If Not dir.EndsWith( "/" ) dir+="/"
Local result:=True
For Local i:=0 Until _files.Length
Local name:=_files[i]
If Not name Continue
If prefix And Not name.StartsWith( prefix ) Continue
Local path:=dir+name
If mz_zip_reader_is_file_a_directory( _zip,i )
filesystem.CreateDir( path )
Continue
Endif
Local size:=_sizes[i]
Local buf:=New DataBuffer( size )
If mz_zip_reader_extract_to_mem( _zip,i,buf.Data,size,0 )
filesystem.CreateDir( ExtractDir( path ) )
buf.Save( path )
Else
result=False
Endif
buf.Discard()
Next
Return result
End
Function Open:ZipFile( path:String )
Local data:=DataBuffer.Load( path )
If Not data Return Null
Local zip:mz_zip_archive Ptr
zip=Cast<mz_zip_archive Ptr>( libc.malloc( sizeof( zip[0] ) ) )
memset( zip,0,sizeof( zip[0] ) )
If Not mz_zip_reader_init_mem( zip,data.Data,data.Length,0 )
free( zip )
data.Discard()
Return Null
Endif
Return New ZipFile( data,zip )
End
Private
Field _data:DataBuffer
Field _zip:mz_zip_archive Ptr
Field _files:String[]
Field _sizes:Int[]
Method New( data:DataBuffer,zip:mz_zip_archive Ptr )
_data=data
_zip=zip
Local nfiles:=mz_zip_reader_get_num_files( _zip )
_files=New String[nfiles]
_sizes=New Int[nfiles]
For Local i:=0 Until nfiles
Local stat:mz_zip_archive_file_stat
If Not mz_zip_reader_file_stat( zip,i,Varptr stat ) Continue
_files[i]=String.FromCString( stat.m_filename )
_sizes[i]=stat.m_uncomp_size
Next
End
End
#rem monkeydoc Extracts a zip file to a directory.
@param zipFile File path of zipfile to extract.
@param dstDir Directory to extract zip file to.
@return True if successful.
#end
Function ExtractZip:Bool( zipFile:String,dstDir:String )
Return ExtractZip( zipFile,dstDir,"" )
End
#rem monkeydoc @hidden
#end
Function ExtractZip:Bool( zipFile:String,dstDir:String,prefix:String )
Local zip:=ZipFile.Open( zipFile )
If Not zip Return False
Local result:=zip.Extract( dstDir,prefix )
zip.Close()
Return result
End