From 87673d7ecce661f3b8c5db949e70640934a55de9 Mon Sep 17 00:00:00 2001
From: MaxVerevkin <maxxverrr@gmail.com>
Date: Thu, 7 Sep 2023 16:31:06 +0300
Subject: some refactoring

---
 src/wm_info_provider/river.rs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'src/wm_info_provider/river.rs')

diff --git a/src/wm_info_provider/river.rs b/src/wm_info_provider/river.rs
index d8e9c3b..6c0993b 100644
--- a/src/wm_info_provider/river.rs
+++ b/src/wm_info_provider/river.rs
@@ -54,12 +54,12 @@ impl RiverInfoProvider {
 }
 
 impl WmInfoProvider for RiverInfoProvider {
-    fn new_ouput(&mut self, conn: &mut Connection<State>, output: WlOutput) {
+    fn new_ouput(&mut self, conn: &mut Connection<State>, output: &Output) {
         let status =
             self.status_manager
-                .get_river_output_status_with_cb(conn, output, output_status_cb);
+                .get_river_output_status_with_cb(conn, output.wl, output_status_cb);
         self.output_statuses.push(OutputStatus {
-            output,
+            output: output.wl,
             status,
             focused_tags: 0,
             urgent_tags: 0,
@@ -68,11 +68,11 @@ impl WmInfoProvider for RiverInfoProvider {
         });
     }
 
-    fn output_removed(&mut self, conn: &mut Connection<State>, output: WlOutput) {
+    fn output_removed(&mut self, conn: &mut Connection<State>, output: &Output) {
         let index = self
             .output_statuses
             .iter()
-            .position(|s| s.output == output)
+            .position(|s| s.output == output.wl)
             .unwrap();
         let output_status = self.output_statuses.swap_remove(index);
         output_status.status.destroy(conn);
-- 
cgit v1.3.1