librepcb_rust_core/ffi/
zip_writer_ffi.rs

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
//! FFI for
//! [`zip::ZipWriter`](https://docs.rs/zip/0.6.6/zip/write/struct.ZipWriter.html)

use super::cpp_ffi::*;
use std::fs;
use std::fs::File;
use std::io::Write;

/// Proxy type for in-memory and file-based
/// [`zip::ZipWriter`](https://docs.rs/zip/0.6.6/zip/write/struct.ZipWriter.html)
enum Writer {
  /// File-based Zip archive
  File(zip::ZipWriter<File>),
  /// In-memory Zip archive
  Mem(zip::ZipWriter<QByteArrayWriter>),
}

/// Wrapper type for [Writer]
struct ZipWriter(Writer);

/// Create a new [ZipWriter] object writing to a file
#[no_mangle]
extern "C" fn ffi_zipwriter_new_to_file(
  path: &QString,
  err: &mut QString,
) -> *mut ZipWriter {
  let path_str = from_qstring(path);
  let fp = std::path::Path::new(&path_str);
  if let Some(parent) = fp.parent() {
    if !parent.exists() {
      _ = fs::create_dir_all(parent);
    }
  }
  let file = match File::create(fp) {
    Ok(file) => file,
    Err(e) => {
      qstring_set(err, e.to_string().as_str());
      return std::ptr::null_mut();
    }
  };
  let zip = zip::ZipWriter::new(file);
  Box::into_raw(Box::new(ZipWriter(Writer::File(zip))))
}

/// Create a new [ZipWriter] object writing to memory
#[no_mangle]
extern "C" fn ffi_zipwriter_new_to_mem(
  data: &mut QByteArray,
) -> *mut ZipWriter {
  let writer = QByteArrayWriter::new(data);
  let zip = zip::ZipWriter::new(writer);
  Box::into_raw(Box::new(ZipWriter(Writer::Mem(zip))))
}

/// Delete [ZipWriter] object
#[no_mangle]
extern "C" fn ffi_zipwriter_delete(obj: *mut ZipWriter) {
  assert!(!obj.is_null());
  unsafe { drop(Box::from_raw(obj)) };
}

/// Write a file to [ZipWriter]
#[no_mangle]
extern "C" fn ffi_zipwriter_write_file(
  obj: &mut ZipWriter,
  name: &QString,
  data: &QByteArray,
  mode: u32,
  err: &mut QString,
) -> bool {
  let options = zip::write::FileOptions::default()
    .compression_method(zip::CompressionMethod::Deflated)
    .unix_permissions(mode);
  let start_res = match &mut obj.0 {
    Writer::File(zip) => zip.start_file(from_qstring(name), options),
    Writer::Mem(zip) => zip.start_file(from_qstring(name), options),
  };
  if let Err(e) = start_res {
    qstring_set(err, e.to_string().as_str());
    return false;
  }
  let write_res = match &mut obj.0 {
    Writer::File(zip) => zip.write_all(qbytearray_to_slice(data)),
    Writer::Mem(zip) => zip.write_all(qbytearray_to_slice(data)),
  };
  if let Err(e) = write_res {
    qstring_set(err, e.to_string().as_str());
    return false;
  }
  true
}

/// Finish writing to [ZipWriter]
#[no_mangle]
extern "C" fn ffi_zipwriter_finish(
  obj: &mut ZipWriter,
  err: &mut QString,
) -> bool {
  let res = match &mut obj.0 {
    Writer::File(zip) => zip.finish().err(),
    Writer::Mem(zip) => zip.finish().err(),
  };
  if let Some(e) = res {
    qstring_set(err, e.to_string().as_str());
    return false;
  }
  true
}