mirror of
https://github.com/scsibug/nostr-rs-relay.git
synced 2024-12-22 16:35:51 -05:00
Start of protocol module
This commit is contained in:
parent
010a420465
commit
68a69505ac
87
src/proto.rs
Normal file
87
src/proto.rs
Normal file
|
@ -0,0 +1,87 @@
|
|||
use crate::error::{Error, Result};
|
||||
use crate::event;
|
||||
use log::{debug, info};
|
||||
use uuid::Uuid;
|
||||
|
||||
// A protocol handler/helper. Use one per client.
|
||||
pub struct Proto {
|
||||
client_id: Uuid,
|
||||
}
|
||||
|
||||
impl Proto {
|
||||
pub fn new() -> Self {
|
||||
let p = Proto {
|
||||
client_id: Uuid::new_v4(),
|
||||
};
|
||||
debug!("New client: {:?}", p.client_id);
|
||||
p
|
||||
}
|
||||
|
||||
pub fn process_message(self: &Self, cmd: String) {
|
||||
info!(
|
||||
"Processing message in proto for client: {:?}",
|
||||
self.client_id
|
||||
);
|
||||
// check what kind of message
|
||||
info!("Parse result: {:?}", parse_type(cmd));
|
||||
}
|
||||
}
|
||||
|
||||
// A raw message with the expected type
|
||||
#[derive(PartialEq, Debug)]
|
||||
pub enum NostrRawMessage {
|
||||
Event(String),
|
||||
Req(String),
|
||||
Close(String),
|
||||
}
|
||||
|
||||
// A fully parsed request
|
||||
#[derive(PartialEq, Debug)]
|
||||
pub enum NostrRequest {
|
||||
Event(event::Event),
|
||||
}
|
||||
|
||||
// Wrap the message in the expected request type
|
||||
fn msg_type_wrapper(msg: String) -> Result<NostrRawMessage> {
|
||||
// check prefix.
|
||||
if msg.starts_with(r#"["EVENT","#) {
|
||||
Ok(NostrRawMessage::Event(msg))
|
||||
} else if msg.starts_with(r#"["REQ","#) {
|
||||
Ok(NostrRawMessage::Req(msg))
|
||||
} else if msg.starts_with(r#"["CLOSE","#) {
|
||||
Ok(NostrRawMessage::Close(msg))
|
||||
} else {
|
||||
Err(Error::CommandNotFound)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn parse_type(msg: String) -> Result<NostrRequest> {
|
||||
// turn this raw string into a parsed request
|
||||
let typ = msg_type_wrapper(msg)?;
|
||||
match typ {
|
||||
NostrRawMessage::Event(_) => Err(Error::EventParseFailed),
|
||||
NostrRawMessage::Req(_) => Err(Error::ReqParseFailed),
|
||||
NostrRawMessage::Close(_) => Err(Error::CloseParseFailed),
|
||||
}
|
||||
}
|
||||
|
||||
// Parse the request into a fully deserialized type
|
||||
|
||||
// The protocol-handling process looks something like:
|
||||
// Receive a message (bytes).
|
||||
// Determine the type. We could do this with an untagged deserialization in serde. Or we can peek at the prefix.
|
||||
// Wrap the message string in the client request type (either Event, Req, Close)
|
||||
// For Req/Close, we can fully parse these.
|
||||
|
||||
// For Event, we want to be more cautious.
|
||||
// Before we admit an event, we should reject any duplicates.
|
||||
// duplicates in the datastore will have already been sent out to interested subscribers.
|
||||
// No point in verifying an event that we already have.
|
||||
|
||||
// Event pipeline looks like:
|
||||
// Get message. (./)
|
||||
// Verify it is an event. (./)
|
||||
// Parse into string / number components from JSON.
|
||||
// Perform validation, re-serialize (or can we re-use the original?)
|
||||
// Publish to subscribers.
|
||||
// Push to datastore.
|
Loading…
Reference in New Issue
Block a user