1mod compression;
2
3use std::fs::File;
4use std::io::{Error, ErrorKind, Read, Seek};
5
6use crate::contents::enums;
7
8fn find_subsequence(source: &[u8], target: &[u8]) -> Option<usize> {
9 source
10 .windows(target.len())
11 .position(|window| window == target)
12}
13
14pub fn get_file_type(file_path: &std::path::PathBuf) -> Result<enums::FileType, std::io::Error> {
15 struct CompressMagic {
16 magic_number: &'static [u8],
17 length: usize,
18 file_type: enums::FileType,
19 }
20
21 let compress_magic_startswith_list = [
22 CompressMagic {
23 magic_number: b"BZh",
24 length: 3,
25 file_type: enums::FileType::Bz2,
26 },
27 CompressMagic {
28 magic_number: &[0x1f, 0x8b],
29 length: 2,
30 file_type: enums::FileType::Gz,
31 },
32 CompressMagic {
33 magic_number: &[0x50, 0x4b, 0x03, 0x04],
34 length: 4,
35 file_type: enums::FileType::Zip,
36 },
37 CompressMagic {
38 magic_number: &[0x37, 0x7a, 0xbc, 0xaf],
39 length: 4,
40 file_type: enums::FileType::SevenZ,
41 },
42 CompressMagic {
43 magic_number: &[0xfd, 0x37, 0x7a, 0x58],
44 length: 4,
45 file_type: enums::FileType::Xz,
46 },
47 CompressMagic {
48 magic_number: &[0x28, 0xb5, 0x2f, 0xfd],
49 length: 4,
50 file_type: enums::FileType::Zst,
51 },
52 CompressMagic {
53 magic_number: &[0x04, 0x22, 0x4d, 0x18],
54 length: 4,
55 file_type: enums::FileType::Lz4,
56 },
57 ];
58
59 let compress_magic_include_list = [CompressMagic {
60 magic_number: &[0x75, 0x73, 0x74, 0x61, 0x72],
61 length: 5,
62 file_type: enums::FileType::Tar,
63 }];
64
65 let mut startswith_buffer = [0u8; 4];
66 let mut file = File::open(file_path).expect("File open failed");
67 match file.read_exact(&mut startswith_buffer) {
68 Ok(_) => (),
69 Err(_) => return Err(Error::from(ErrorKind::Unsupported)),
70 };
71
72 for compress_magic in compress_magic_startswith_list.iter() {
73 if startswith_buffer.get(..compress_magic.length).unwrap() == compress_magic.magic_number {
74 return Ok(compress_magic.file_type);
75 }
76 }
77
78 file.rewind().expect("Seek 0 failed");
79 let mut include_vec = Vec::new();
80 file.read_to_end(&mut include_vec)
81 .expect("Read file failed");
82
83 for compress_magic in compress_magic_include_list.iter() {
84 if find_subsequence(&include_vec, compress_magic.magic_number).is_some() {
85 return Ok(compress_magic.file_type);
86 }
87 }
88
89 Err(Error::from(ErrorKind::Unsupported))
90}
91
92pub fn compress(
93 file_type: enums::FileType,
94 src_path: &std::path::Path,
95 dst_path: &std::path::Path,
96) {
97 match file_type {
98 enums::FileType::Zip => {
99 compression::zip::compress(src_path, dst_path);
100 }
101 enums::FileType::Tar => {
102 compression::tar::compress(src_path, dst_path);
103 }
104 enums::FileType::Bz2 => {
105 compression::bz2::compress(src_path, dst_path);
106 }
107 enums::FileType::Gz => {
108 compression::gz::compress(src_path, dst_path);
109 }
110 enums::FileType::Tarbz2 => {
111 compression::tar_bz2::compress(src_path, dst_path);
112 }
113 enums::FileType::Targz => {
114 compression::tar_gz::compress(src_path, dst_path);
115 }
116 enums::FileType::SevenZ => {
117 compression::sevenz::compress(src_path, dst_path);
118 }
119 enums::FileType::Xz => {
120 compression::xz::compress(src_path, dst_path);
121 }
122 enums::FileType::Tarxz => {
123 compression::tar_xz::compress(src_path, dst_path);
124 }
125 enums::FileType::Zst => {
126 compression::zst::compress(src_path, dst_path);
127 }
128 enums::FileType::Tarzst => {
129 compression::tar_zst::compress(src_path, dst_path);
130 }
131 enums::FileType::Lz4 => {
132 compression::lz4::compress(src_path, dst_path);
133 }
134 enums::FileType::Tarlz4 => {
135 compression::tar_lz4::compress(src_path, dst_path);
136 }
137 }
138}
139
140pub fn decompress(
141 file_type: enums::FileType,
142 src_path: &std::path::Path,
143 dst_path: &std::path::Path,
144) {
145 match file_type {
146 enums::FileType::Zip => {
147 compression::zip::decompress(src_path, dst_path);
148 }
149 enums::FileType::Tar => {
150 compression::tar::decompress(src_path, dst_path);
151 }
152 enums::FileType::Tarbz2 => {
153 compression::tar_bz2::decompress(src_path, dst_path);
154 }
155 enums::FileType::Targz => {
156 compression::tar_gz::decompress(src_path, dst_path);
157 }
158 enums::FileType::Bz2 => {
159 compression::bz2::decompress(src_path, dst_path);
160 }
161 enums::FileType::Gz => {
162 compression::gz::decompress(src_path, dst_path);
163 }
164 enums::FileType::SevenZ => {
165 compression::sevenz::decompress(src_path, dst_path);
166 }
167 enums::FileType::Xz => {
168 compression::xz::decompress(src_path, dst_path);
169 }
170 enums::FileType::Tarxz => {
171 compression::tar_xz::decompress(src_path, dst_path);
172 }
173 enums::FileType::Zst => {
174 compression::zst::decompress(src_path, dst_path);
175 }
176 enums::FileType::Tarzst => {
177 compression::tar_zst::decompress(src_path, dst_path);
178 }
179 enums::FileType::Lz4 => {
180 compression::lz4::decompress(src_path, dst_path);
181 }
182 enums::FileType::Tarlz4 => {
183 compression::tar_lz4::decompress(src_path, dst_path);
184 }
185 }
186}