From cf7fd2f615445fff799b212fd5aa9e216c3e4d40 Mon Sep 17 00:00:00 2001
From: sxyazi <sxyazi@gmail.com>
Date: Thu, 12 Sep 2024 18:15:02 +0800
Subject: [PATCH] ..

---
 yazi-plugin/src/elements/gauge.rs |  8 ++------
 yazi-plugin/src/elements/style.rs | 10 +++-------
 2 files changed, 5 insertions(+), 13 deletions(-)

diff --git a/yazi-plugin/src/elements/gauge.rs b/yazi-plugin/src/elements/gauge.rs
index c66570cc..0c1f78d6 100644
--- a/yazi-plugin/src/elements/gauge.rs
+++ b/yazi-plugin/src/elements/gauge.rs
@@ -2,6 +2,7 @@ use mlua::{AnyUserData, ExternalError, Lua, Table, UserData, UserDataMethods, Va
 use ratatui::widgets::Widget;
 
 use super::{RectRef, Renderable, Span};
+use crate::elements::Style;
 
 #[derive(Clone, Default)]
 pub struct Gauge {
@@ -50,12 +51,7 @@ impl UserData for Gauge {
 		});
 
 		methods.add_function("gauge_style", |_, (ud, value): (AnyUserData, Value)| {
-			ud.borrow_mut::<Self>()?.gauge_style = match value {
-				Value::Nil => ratatui::style::Style::default(),
-				Value::Table(tb) => crate::elements::Style::try_from(tb)?.0,
-				Value::UserData(ud) => ud.borrow::<crate::elements::Style>()?.0,
-				_ => return Err("expected a Style or Table or nil".into_lua_err()),
-			};
+			ud.borrow_mut::<Self>()?.gauge_style = Style::try_from(value)?.0;
 			Ok(ud)
 		});
 	}
diff --git a/yazi-plugin/src/elements/style.rs b/yazi-plugin/src/elements/style.rs
index eae714f0..8516b30d 100644
--- a/yazi-plugin/src/elements/style.rs
+++ b/yazi-plugin/src/elements/style.rs
@@ -40,8 +40,8 @@ impl<'a> TryFrom<Value<'a>> for Style {
 	fn try_from(value: Value<'a>) -> Result<Self, Self::Error> {
 		Ok(Self(match value {
 			Value::Nil => Default::default(),
-			Value::Table(tb) => Style::try_from(tb)?.0,
-			Value::UserData(ud) => ud.borrow::<Style>()?.0,
+			Value::Table(tb) => Self::try_from(tb)?.0,
+			Value::UserData(ud) => ud.borrow::<Self>()?.0,
 			_ => Err("expected a Style or Table or nil".into_lua_err())?,
 		}))
 	}
@@ -58,11 +58,7 @@ impl UserData for Style {
 		methods.add_function("patch", |_, (ud, value): (AnyUserData, Value)| {
 			{
 				let mut me = ud.borrow_mut::<Self>()?;
-				me.0 = me.0.patch(match value {
-					Value::Table(tb) => Style::try_from(tb)?.0,
-					Value::UserData(ud) => ud.borrow::<Style>()?.0,
-					_ => return Err("expected a Style or Table".into_lua_err()),
-				});
+				me.0 = me.0.patch(Self::try_from(value)?.0);
 			}
 			Ok(ud)
 		})