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
109
110
111
112
113
114
115
116
117
118
use std::collections::HashMap;
use chrono::{NaiveDateTime};
use reqwest::blocking::Client as HttpClient;
use reqwest::header::CONTENT_TYPE;
use serde::{Serialize};
extern crate serde_json;
const API_ENDPOINT: &str = "https://app.posthog.com/capture/";
pub fn client<C: Into<ClientOptions>>(options: C) -> Client {
Client {
options: options.into(),
client: HttpClient::new(),
}
}
#[derive(Debug)]
pub enum Error {
Connection(String)
}
pub struct ClientOptions {
api_endpoint: String,
api_key: String,
}
impl From<&str> for ClientOptions {
fn from(api_key: &str) -> Self {
ClientOptions {
api_endpoint: API_ENDPOINT.to_string(),
api_key: api_key.to_string(),
}
}
}
pub struct Client {
options: ClientOptions,
client: HttpClient,
}
impl Client {
pub fn capture(&self, event: Event) -> Result<(), Error> {
let inner_event = InnerEvent::new(event, self.options.api_key.clone());
let _res = self.client.post(self.options.api_endpoint.clone())
.header(CONTENT_TYPE, "application/json")
.body(serde_json::to_string(&inner_event).expect("unwrap here is safe"))
.send()
.map_err(|e| Error::Connection(e.to_string()))?;
Ok(())
}
pub fn capture_batch(&self, events: Vec<Event>) -> Result<(), Error> {
for event in events {
self.capture(event)?;
}
Ok(())
}
}
#[derive(Serialize)]
struct InnerEvent {
api_key: String,
event: String,
properties: Properties,
timestamp: Option<NaiveDateTime>,
}
impl InnerEvent {
fn new(event: Event, api_key: String) -> Self {
Self {
api_key,
event: event.event,
properties: event.properties,
timestamp: event.timestamp,
}
}
}
pub struct Event {
event: String,
properties: Properties,
timestamp: Option<NaiveDateTime>,
}
#[derive(Serialize)]
pub struct Properties {
distinct_id: String,
props: HashMap<String, String>,
}
#[cfg(test)]
pub mod tests {
use super::*;
use chrono::{Utc};
#[test]
fn get_client() {
let client = crate::client(env!("POSTHOG_API_KEY"));
let mut props = HashMap::new();
props.insert("key1".to_string(), "value1".to_string());
props.insert("key2".to_string(), "value2".to_string());
let event = Event {
event: "test".to_string(),
properties: Properties { distinct_id: "1234".to_string(), props },
timestamp: Some(Utc::now().naive_utc()),
};
let res = client.capture(event).unwrap();
}
}