nyx_space/io/
watermark.rs

1/*
2    Nyx, blazing fast astrodynamics
3    Copyright (C) 2018-onwards Christopher Rabotin <christopher.rabotin@gmail.com>
4
5    This program is free software: you can redistribute it and/or modify
6    it under the terms of the GNU Affero General Public License as published
7    by the Free Software Foundation, either version 3 of the License, or
8    (at your option) any later version.
9
10    This program is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13    GNU Affero General Public License for more details.
14
15    You should have received a copy of the GNU Affero General Public License
16    along with this program.  If not, see <https://www.gnu.org/licenses/>.
17*/
18
19use std::collections::HashMap;
20
21use hifitime::Epoch;
22use parquet::{
23    basic::{Compression, ZstdLevel},
24    file::properties::WriterProperties,
25    format::KeyValue,
26};
27use shadow_rs::shadow;
28use whoami::{platform, realname, username};
29
30shadow!(build);
31
32/// The parquet writer properties
33pub(crate) fn pq_writer(metadata: Option<HashMap<String, String>>) -> Option<WriterProperties> {
34    let bldr = WriterProperties::builder()
35        .set_compression(Compression::ZSTD(ZstdLevel::try_new(10).unwrap()));
36
37    let mut file_metadata = vec![
38        KeyValue::new("Generated by".to_string(), prj_name_ver()),
39        KeyValue::new(
40            format!("{} License", build::PROJECT_NAME),
41            "AGPL 3.0".to_string(),
42        ),
43        KeyValue::new(
44            "Created by".to_string(),
45            format!("{} ({}) on {}", realname(), username(), platform()),
46        ),
47        KeyValue::new(
48            "Created on".to_string(),
49            format!("{}", Epoch::now().unwrap()),
50        ),
51    ];
52
53    if let Some(custom_md) = metadata {
54        for (k, v) in custom_md {
55            file_metadata.push(KeyValue::new(k, v));
56        }
57    }
58
59    Some(bldr.set_key_value_metadata(Some(file_metadata)).build())
60}
61
62pub(crate) fn prj_name_ver() -> String {
63    format!("Nyx Space v{}", build::PKG_VERSION)
64}
nyx_space/io/watermark.rs

nyx_space/io/
watermark.rs
