mirror of
https://github.com/denoland/deno.git
synced 2024-12-22 07:14:47 -05:00
127 lines
3.1 KiB
Rust
127 lines
3.1 KiB
Rust
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
|
|
|
|
mod in_memory_broadcast_channel;
|
|
|
|
pub use in_memory_broadcast_channel::InMemoryBroadcastChannel;
|
|
pub use in_memory_broadcast_channel::InMemoryBroadcastChannelResource;
|
|
|
|
use std::cell::RefCell;
|
|
use std::path::PathBuf;
|
|
use std::rc::Rc;
|
|
|
|
use async_trait::async_trait;
|
|
use deno_core::error::AnyError;
|
|
use deno_core::op2;
|
|
use deno_core::JsBuffer;
|
|
use deno_core::OpState;
|
|
use deno_core::Resource;
|
|
use deno_core::ResourceId;
|
|
|
|
pub const UNSTABLE_FEATURE_NAME: &str = "broadcast-channel";
|
|
|
|
#[async_trait]
|
|
pub trait BroadcastChannel: Clone {
|
|
type Resource: Resource;
|
|
|
|
fn subscribe(&self) -> Result<Self::Resource, AnyError>;
|
|
|
|
fn unsubscribe(&self, resource: &Self::Resource) -> Result<(), AnyError>;
|
|
|
|
async fn send(
|
|
&self,
|
|
resource: &Self::Resource,
|
|
name: String,
|
|
data: Vec<u8>,
|
|
) -> Result<(), AnyError>;
|
|
|
|
async fn recv(
|
|
&self,
|
|
resource: &Self::Resource,
|
|
) -> Result<Option<Message>, AnyError>;
|
|
}
|
|
|
|
pub type Message = (String, Vec<u8>);
|
|
|
|
#[op2(fast)]
|
|
#[smi]
|
|
pub fn op_broadcast_subscribe<BC>(
|
|
state: &mut OpState,
|
|
) -> Result<ResourceId, AnyError>
|
|
where
|
|
BC: BroadcastChannel + 'static,
|
|
{
|
|
// TODO(bartlomieju): replace with `state.feature_checker.check_or_exit`
|
|
// once we phase out `check_or_exit_with_legacy_fallback`
|
|
state.feature_checker.check_or_exit_with_legacy_fallback(
|
|
UNSTABLE_FEATURE_NAME,
|
|
"BroadcastChannel",
|
|
);
|
|
let bc = state.borrow::<BC>();
|
|
let resource = bc.subscribe()?;
|
|
Ok(state.resource_table.add(resource))
|
|
}
|
|
|
|
#[op2(fast)]
|
|
pub fn op_broadcast_unsubscribe<BC>(
|
|
state: &mut OpState,
|
|
#[smi] rid: ResourceId,
|
|
) -> Result<(), AnyError>
|
|
where
|
|
BC: BroadcastChannel + 'static,
|
|
{
|
|
let resource = state.resource_table.get::<BC::Resource>(rid)?;
|
|
let bc = state.borrow::<BC>();
|
|
bc.unsubscribe(&resource)
|
|
}
|
|
|
|
#[op2(async)]
|
|
pub async fn op_broadcast_send<BC>(
|
|
state: Rc<RefCell<OpState>>,
|
|
#[smi] rid: ResourceId,
|
|
#[string] name: String,
|
|
#[buffer] buf: JsBuffer,
|
|
) -> Result<(), AnyError>
|
|
where
|
|
BC: BroadcastChannel + 'static,
|
|
{
|
|
let resource = state.borrow().resource_table.get::<BC::Resource>(rid)?;
|
|
let bc = state.borrow().borrow::<BC>().clone();
|
|
bc.send(&resource, name, buf.to_vec()).await
|
|
}
|
|
|
|
#[op2(async)]
|
|
#[serde]
|
|
pub async fn op_broadcast_recv<BC>(
|
|
state: Rc<RefCell<OpState>>,
|
|
#[smi] rid: ResourceId,
|
|
) -> Result<Option<Message>, AnyError>
|
|
where
|
|
BC: BroadcastChannel + 'static,
|
|
{
|
|
let resource = state.borrow().resource_table.get::<BC::Resource>(rid)?;
|
|
let bc = state.borrow().borrow::<BC>().clone();
|
|
bc.recv(&resource).await
|
|
}
|
|
|
|
deno_core::extension!(deno_broadcast_channel,
|
|
deps = [ deno_webidl, deno_web ],
|
|
parameters = [BC: BroadcastChannel],
|
|
ops = [
|
|
op_broadcast_subscribe<BC>,
|
|
op_broadcast_unsubscribe<BC>,
|
|
op_broadcast_send<BC>,
|
|
op_broadcast_recv<BC>,
|
|
],
|
|
esm = [ "01_broadcast_channel.js" ],
|
|
options = {
|
|
bc: BC,
|
|
},
|
|
state = |state, options| {
|
|
state.put(options.bc);
|
|
},
|
|
);
|
|
|
|
pub fn get_declaration() -> PathBuf {
|
|
PathBuf::from(env!("CARGO_MANIFEST_DIR"))
|
|
.join("lib.deno_broadcast_channel.d.ts")
|
|
}
|