tokn-convert 0.2.0-rc.3

Request and response conversion pipeline across tokn endpoint formats
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

use eventsource_stream::Event;
use serde_json::Value;

#[derive(Clone, Debug)]
pub struct SseEvent {
  pub event: Option<String>,
  pub data: String,
  pub json: Option<Value>,
}

impl SseEvent {
  pub fn raw(event: Option<String>, data: String) -> Self {
    let payload = data.trim();
    let json = if payload.is_empty() || payload == "[DONE]" {
      None
    } else {
      serde_json::from_str(payload).ok()
    };
    Self { event, data, json }
  }

  pub fn json(event: Option<&str>, value: Value) -> Self {
    let data = serde_json::to_string(&value).unwrap_or_else(|_| "{}".into());
    Self {
      event: event.map(str::to_string),
      data,
      json: Some(value),
    }
  }

  pub fn done() -> Self {
    Self {
      event: None,
      data: "[DONE]".into(),
      json: None,
    }
  }

  pub fn is_done(&self) -> bool {
    self.data.trim() == "[DONE]"
  }
}

impl From<Event> for SseEvent {
  fn from(value: Event) -> Self {
    let event = (!value.event.is_empty()).then_some(value.event);
    Self::raw(event, value.data)
  }
}