pgml 1.1.1

The official pgml Rust SDK
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
79
80
81
82
83
84
85
86
87
88
89

use crate::types::{GeneralJsonAsyncIterator, GeneralJsonIterator, Json};
use futures::stream::Stream;
use rust_bridge::c::CustomInto;
use std::pin::Pin;

pub type JsonC = std::ffi::c_char;

unsafe impl CustomInto<Json> for *mut JsonC {
    unsafe fn custom_into(self) -> Json {
        let s = std::ffi::CStr::from_ptr(self).to_str().unwrap();
        serde_json::from_str::<serde_json::Value>(s).unwrap().into()
    }
}

unsafe impl CustomInto<*mut JsonC> for Json {
    unsafe fn custom_into(self) -> *mut JsonC {
        let s = serde_json::to_string(&self).unwrap();
        std::ffi::CString::new(s).unwrap().into_raw()
    }
}

pub struct GeneralJsonIteratorC {
    wrapped: *mut std::iter::Peekable<Box<dyn Iterator<Item = Result<Json, anyhow::Error>> + Send>>,
}

unsafe impl CustomInto<*mut GeneralJsonIteratorC> for GeneralJsonIterator {
    unsafe fn custom_into(self) -> *mut GeneralJsonIteratorC {
        Box::into_raw(Box::new(GeneralJsonIteratorC {
            wrapped: Box::into_raw(Box::new(self.0.peekable())),
        }))
    }
}

#[no_mangle]
pub unsafe extern "C" fn pgml_generaljsoniteratorc_done(
    iterator: *mut GeneralJsonIteratorC,
) -> bool {
    let c = Box::leak(Box::from_raw(iterator));
    (*c.wrapped).peek().is_none()
}

#[no_mangle]
pub unsafe extern "C" fn pgml_generaljsoniteratorc_next(
    iterator: *mut GeneralJsonIteratorC,
) -> *mut JsonC {
    let c = Box::leak(Box::from_raw(iterator));
    let b = Box::leak(Box::from_raw(c.wrapped));
    (*b).next().unwrap().unwrap().custom_into()
}

type PeekableStream =
    futures::stream::Peekable<Pin<Box<dyn Stream<Item = Result<Json, anyhow::Error>> + Send>>>;

pub struct GeneralJsonAsyncIteratorC {
    wrapped: *mut PeekableStream,
}

unsafe impl CustomInto<*mut GeneralJsonAsyncIteratorC> for GeneralJsonAsyncIterator {
    unsafe fn custom_into(self) -> *mut GeneralJsonAsyncIteratorC {
        use futures::stream::StreamExt;
        Box::into_raw(Box::new(GeneralJsonAsyncIteratorC {
            wrapped: Box::into_raw(Box::new(self.0.peekable())),
        }))
    }
}

#[no_mangle]
pub unsafe extern "C" fn pgml_generaljsonasynciteratorc_done(
    iterator: *mut GeneralJsonAsyncIteratorC,
) -> bool {
    crate::get_or_set_runtime().block_on(async move {
        let c = Box::leak(Box::from_raw(iterator));
        let s = Box::leak(Box::from_raw(c.wrapped));
        let mut s = Pin::new(s);
        let res = s.as_mut().peek_mut().await;
        res.is_none()
    })
}

#[no_mangle]
pub unsafe extern "C" fn pgml_generaljsonasynciteratorc_next(
    iterator: *mut GeneralJsonAsyncIteratorC,
) -> *mut JsonC {
    crate::get_or_set_runtime().block_on(async move {
        use futures::stream::StreamExt;
        let c = Box::leak(Box::from_raw(iterator));
        (*c.wrapped).next().await.unwrap().unwrap().custom_into()
    })
}