imessage-database 4.0.0

Parsers and tools to interact with iMessage SQLite data
Documentation 
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

/*!
Build script for protobuf-generated modules.

Build modes:

- Default build (`cargo build -p imessage-database`):
  validates that generated protobuf Rust files are already present in the repository.
- Regeneration build (`cargo build -p imessage-database --features proto-regen`):
  regenerates protobuf Rust files from the `.proto` sources.

Useful commands:

- `cargo check -p imessage-database`
- `cargo check -p imessage-database --features proto-regen`
*/

#[cfg(not(feature = "proto-regen"))]
use std::fs::exists;

#[cfg(feature = "proto-regen")]
use std::{env, fs::copy, path::PathBuf};

#[cfg(feature = "proto-regen")]
const HANDWRITING_PROTO_INPUT: &str = "src/message_types/handwriting/handwriting.proto";
#[cfg(feature = "proto-regen")]
const HANDWRITING_GENERATED_NAME: &str = "handwriting.rs";
const HANDWRITING_OUTPUT: &str = "src/message_types/handwriting/handwriting_proto.rs";
#[cfg(feature = "proto-regen")]
const DIGITAL_TOUCH_PROTO_INPUT: &str = "src/message_types/digital_touch/digital_touch.proto";
#[cfg(feature = "proto-regen")]
const DIGITAL_TOUCH_GENERATED_NAME: &str = "digital_touch.rs";
const DIGITAL_TOUCH_OUTPUT: &str = "src/message_types/digital_touch/digital_touch_proto.rs";

fn main() {
    #[cfg(feature = "proto-regen")]
    {
        build_proto(
            HANDWRITING_PROTO_INPUT,
            HANDWRITING_GENERATED_NAME,
            HANDWRITING_OUTPUT,
        );
        build_proto(
            DIGITAL_TOUCH_PROTO_INPUT,
            DIGITAL_TOUCH_GENERATED_NAME,
            DIGITAL_TOUCH_OUTPUT,
        );
    }

    #[cfg(not(feature = "proto-regen"))]
    {
        ensure_generated(HANDWRITING_OUTPUT);
        ensure_generated(DIGITAL_TOUCH_OUTPUT);
    }
}

#[cfg(not(feature = "proto-regen"))]
fn ensure_generated(path: &str) {
    if !exists(path).unwrap_or(false) {
        panic!(
            "Missing generated protobuf module `{path}`. Re-generate with `cargo build -p imessage-database --features proto-regen`."
        );
    }
}

#[cfg(feature = "proto-regen")]
fn build_proto(input_proto: &str, generated_name: &str, output_rs: &str) {
    protobuf_codegen::Codegen::new()
        .pure()
        .input(input_proto)
        .include(".")
        .cargo_out_dir("p")
        .run_from_script();

    let mut generated = PathBuf::from(env::var("OUT_DIR").unwrap());
    generated.push("p");
    generated.push(generated_name);
    copy(generated, output_rs).unwrap();
}