c2pa-ml 0.1.0

C2PA manifest embedding for AI/ML model container formats: GGUF, SafeTensors, and ONNX
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

//! WebAssembly bindings, built only for the `wasm32` target and published to
//! npm as `c2pa-ml`.
//!
//! Byte payloads map to/from JavaScript `Uint8Array`; a missing manifest (or an
//! unrecognized format) is thrown as a JS error.

use wasm_bindgen::prelude::*;

use crate::manifest::ManifestSource;

fn js_err(e: crate::Error) -> JsError {
    JsError::new(&e.to_string())
}

/// Embed a C2PA Manifest Store into a model, auto-detecting the format.
#[wasm_bindgen(js_name = embedManifest)]
pub fn embed_manifest(model: &[u8], store: &[u8]) -> Result<Vec<u8>, JsError> {
    crate::embed_manifest(model, &ManifestSource::embedded(store.to_vec())).map_err(js_err)
}

/// Embed a remote manifest URI into a model, auto-detecting the format.
#[wasm_bindgen(js_name = embedManifestRemote)]
pub fn embed_manifest_remote(model: &[u8], uri: &str) -> Result<Vec<u8>, JsError> {
    crate::embed_manifest(model, &ManifestSource::remote(uri)).map_err(js_err)
}

/// Embed both a Manifest Store and a remote URI into a model.
#[wasm_bindgen(js_name = embedManifestBoth)]
pub fn embed_manifest_both(model: &[u8], uri: &str, store: &[u8]) -> Result<Vec<u8>, JsError> {
    crate::embed_manifest(model, &ManifestSource::both(uri, store.to_vec())).map_err(js_err)
}

/// Read the embedded C2PA Manifest Store from a model.
#[wasm_bindgen(js_name = readManifest)]
pub fn read_manifest(model: &[u8]) -> Result<Vec<u8>, JsError> {
    crate::read_manifest(model).map_err(js_err)
}

/// Read the remote manifest URI from a model, if present.
#[wasm_bindgen(js_name = readManifestUri)]
pub fn read_manifest_uri(model: &[u8]) -> Result<Option<String>, JsError> {
    crate::read_manifest_uri(model).map_err(js_err)
}

/// Remove any C2PA metadata from a model.
#[wasm_bindgen(js_name = removeManifest)]
pub fn remove_manifest(model: &[u8]) -> Result<Vec<u8>, JsError> {
    crate::remove_manifest(model).map_err(js_err)
}

/// Detect the container format, returning `"GGUF"`, `"SafeTensors"`, `"ONNX"`,
/// or `undefined`.
#[wasm_bindgen(js_name = detectFormat)]
pub fn detect_format(model: &[u8]) -> Option<String> {
    crate::Format::detect(model).map(|f| f.name().to_string())
}

/// Return the canonical `c2pa.types.model.*` asset type string for the model's
/// format, or `undefined` if the format is unrecognized.
#[wasm_bindgen(js_name = modelType)]
pub fn model_type(model: &[u8]) -> Option<String> {
    crate::Format::detect(model).map(|f| f.model_type().as_str().to_string())
}