secstate/Notify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove unused code, minor refactorings

Browse Source
This commit is contained in:
ranfdev
2024-11-21 10:53:22 +01:00
parent 76e3ad9ee1
commit a758ffcd6e
6 changed files with 61 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ntfy-daemon/src/http_client.rs
View File

@ -34,6 +34,7 @@ impl RequestInfo {
#[async_trait]
trait LightHttpClient: Send + Sync {
fn get(&self, url: &str) -> RequestBuilder;
fn post(&self, url: &str) -> RequestBuilder;
async fn execute(&self, request: Request) -> Result<Response>;
}
@ -43,6 +44,10 @@ impl LightHttpClient for Client {
self.get(url)
}
fn post(&self, url: &str) -> RequestBuilder {
self.post(url)
}
async fn execute(&self, request: Request) -> Result<Response> {
Ok(self.execute(request).await?)
}
@ -77,6 +82,10 @@ impl HttpClient {
self.client.get(url)
}
pub fn post(&self, url: &str) -> RequestBuilder {
self.client.post(url)
}
pub async fn execute(&self, request: Request) -> Result<Response> {
self.request_tracker
.push(RequestInfo::from_request(&request))
@ -183,6 +192,10 @@ impl LightHttpClient for NullableClient {
Client::new().get(url)
}
fn post(&self, url: &str) -> RequestBuilder {
Client::new().post(url)
}
async fn execute(&self, request: Request) -> Result<Response> {
time::sleep(Duration::from_millis(1)).await;
let url = request.url().to_string();

7
ntfy-daemon/src/lib.rs
View File

@ -19,10 +19,9 @@ use http_client::HttpClient;
#[derive(Clone)]
pub struct SharedEnv {
db: message_repo::Db,
proxy: Arc<dyn models::NotificationProxy>,
http: reqwest::Client,
nullable_http: HttpClient,
network: Arc<dyn models::NetworkMonitorProxy>,
notifier: Arc<dyn models::NotificationProxy>,
http_client: HttpClient,
network_monitor: Arc<dyn models::NetworkMonitorProxy>,
credentials: credentials::Credentials,
}

64
ntfy-daemon/src/listener.rs
View File

@ -114,35 +114,6 @@ pub struct ListenerActor {
}
impl ListenerActor {
pub fn new(config: ListenerConfig) -> ListenerHandle {
let (event_tx, event_rx) = async_channel::bounded(64);
let (commands_tx, commands_rx) = mpsc::channel(1);
let config_clone = config.clone();
// use a new local set to isolate panics
let local_set = LocalSet::new();
local_set.spawn_local(async move {
let this = Self {
event_tx,
commands_rx: Some(commands_rx),
config: config_clone,
state: ConnectionState::Unitialized,
};
this.run_loop().await;
});
spawn_local(local_set);
ListenerHandle {
events: event_rx,
config,
commands: commands_tx,
listener_actor: Arc::new(RwLock::new(None)),
join_handle: Arc::new(None),
}
}
pub async fn run_loop(mut self) {
let mut commands_rx = self.commands_rx.take().unwrap();
loop {
@ -280,6 +251,35 @@ pub struct ListenerHandle {
}
impl ListenerHandle {
pub fn new(config: ListenerConfig) -> ListenerHandle {
let (event_tx, event_rx) = async_channel::bounded(64);
let (commands_tx, commands_rx) = mpsc::channel(1);
let config_clone = config.clone();
// use a new local set to isolate panics
let local_set = LocalSet::new();
local_set.spawn_local(async move {
let this = ListenerActor {
event_tx,
commands_rx: Some(commands_rx),
config: config_clone,
state: ConnectionState::Unitialized,
};
this.run_loop().await;
});
spawn_local(local_set);
Self {
events: event_rx,
config,
commands: commands_tx,
listener_actor: Arc::new(RwLock::new(None)),
join_handle: Arc::new(None),
}
}
// the response will be sent as an event in self.events
pub async fn request_state(&self) -> ConnectionState {
let (tx, rx) = oneshot::channel();
@ -337,7 +337,7 @@ mod tests {
since: 0,
};
let mut listener = ListenerActor::new(config.clone());
let mut listener = ListenerHandle::new(config.clone());
let items: Vec<_> = listener.events.take(3).collect().await;
dbg!(&items);
@ -383,7 +383,7 @@ mod tests {
since: 0,
};
let mut listener = ListenerActor::new(config.clone());
let mut listener = ListenerHandle::new(config.clone());
let items: Vec<_> = listener.events.take(3).collect().await;
dbg!(&items);
@ -414,7 +414,7 @@ mod tests {
since: 0,
};
let mut listener = ListenerActor::new(config.clone());
let mut listener = ListenerHandle::new(config.clone());
// assert_event_matches!(listener, ListenerEvent::Connected { .. },);
});

15
ntfy-daemon/src/ntfy.rs
View File

@ -249,8 +249,8 @@ impl NtfyActor {
) -> impl Future<Output = anyhow::Result<SubscriptionHandle>> {
let server = sub.server.clone();
let topic = sub.topic.clone();
let listener = ListenerActor::new(ListenerConfig {
http_client: self.env.nullable_http.clone(),
let listener = ListenerHandle::new(ListenerConfig {
http_client: self.env.http_client.clone(),
credentials: self.env.credentials.clone(),
endpoint: server.clone(),
topic: topic.clone(),
@ -378,7 +378,6 @@ impl NtfyHandle {
}
pub fn start(
socket_path: std::path::PathBuf,
dbpath: &str,
notification_proxy: Arc<dyn models::NotificationProxy>,
network_proxy: Arc<dyn models::NetworkMonitorProxy>,
@ -400,10 +399,9 @@ pub fn start(
let env = SharedEnv {
db: Db::connect(&dbpath).unwrap(),
proxy: notification_proxy,
http: build_client().unwrap(),
nullable_http: HttpClient::new(build_client().unwrap()),
network: network_proxy,
notifier: notification_proxy,
http_client: HttpClient::new(build_client().unwrap()),
network_monitor: network_proxy,
credentials,
};
@ -452,9 +450,8 @@ mod tests {
let notification_proxy = Arc::new(NullNotifier::new());
let network_proxy = Arc::new(NullNetworkMonitor::new());
let dbpath = ":memory:";
let socket_path = std::path::PathBuf::from("/tmp/ntfy.sock");
let handle = start(socket_path, dbpath, notification_proxy, network_proxy).unwrap();
let handle = start(dbpath, notification_proxy, network_proxy).unwrap();
let rt = tokio::runtime::Builder::new_current_thread()
.enable_all()

4
ntfy-daemon/src/subscription.rs
View File

@ -182,7 +182,7 @@ impl SubscriptionActor {
async fn publish(&self, msg: String) -> anyhow::Result<()> {
let server = &self.model.server;
let creds = self.env.credentials.get(server);
let mut req = self.env.http.post(server);
let mut req = self.env.http_client.post(server);
if let Some(creds) = creds {
req = req.basic_auth(creds.username, Some(creds.password));
}
@ -212,7 +212,7 @@ impl SubscriptionActor {
if !already_stored {
// Show notification. If this fails, panic
if !{ self.model.muted } {
let notifier = self.env.proxy.clone();
let notifier = self.env.notifier.clone();
let title = { msg.notification_title(&self.model) };