Add video entity for the database which is also used for the data exchanged between yew and axum.

Author: Ekkisleif

entity/Cargo.toml

@@ -6,5 +6,11 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-sea-orm = { version = "0.10", features = [ "sqlx-sqlite", "runtime-tokio-native-tls", "macros" ] }
+sea-orm = { version = "0.10", features = ["sqlx-sqlite", "runtime-tokio-native-tls", "macros"], optional = true }
+bincode = { version = "2.0.0-rc.2", optional = true}
 serde = { version = "1.0", features = ["derive"] }
+
+[features]
+user = []
+seaorm = ["dep:sea-orm"]
+video = ["dep:bincode"]

entity/src/lib.rs

@@ -1 +1,4 @@
+#[cfg(feature = "user")]
 pub mod user;
+#[cfg(feature = "video")]
+pub mod video;

entity/src/video.rs

@@ -0,0 +1,24 @@
+use bincode::{Decode, Encode};
+use serde::{Deserialize, Serialize};
+
+#[cfg(feature = "seaorm")]
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, Decode, Encode)] // TODO : remove the Debug
+#[cfg_attr(feature = "seaorm", derive(DeriveEntityModel))]
+#[cfg_attr(feature = "seaorm", sea_orm(table_name = "videos"))]
+pub struct Model {
+    #[cfg_attr(feature = "seaorm", sea_orm(primary_key, auto_increment = false))]
+    pub id: String,
+    pub title: String,
+    pub thumbnail: String,
+    pub author: String,
+    pub duration: String,
+}
+
+#[cfg(feature = "seaorm")]
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {}
+
+#[cfg(feature = "seaorm")]
+impl ActiveModelBehavior for ActiveModel {}

src/lib.rs

@@ -1,47 +1,24 @@
 use anyhow::Result;
-#[cfg(feature = "axum")]
-use axum::extract::ws;
 use bincode::{config, Decode, Encode};
-use gloo_net::websocket;
-use my_youtube_extractor::youtube_info::YtVideoPageInfo;
+use entity::video::Model;
 
 #[derive(Debug, Encode, Decode, Clone)]
-pub enum NetDataAxum {
-    Remove(String),
-    Add(YtVideoPageInfo),
-    Search(Vec<YtVideoPageInfo>),
-    Next,
-}
-
-#[derive(Debug, Encode, Decode)]
-pub enum NetDataYew {
+pub enum NetData {
     Search(String),
+    SearchResult(Vec<Model>),
     Remove(String),
-    Add(YtVideoPageInfo),
+    Add(Model),
     Play,
     Pause,
     Next,
 }
 
-impl NetDataYew {
-    pub fn encode_yew_message(&self) -> Result<websocket::Message> {
-        let encoded = bincode::encode_to_vec(self, config::standard())?;
-        Ok(websocket::Message::Bytes(encoded))
-    }
-
-    pub fn decode_message(bytes: &[u8]) -> Result<NetDataYew> {
-        Ok(bincode::decode_from_slice(bytes, config::standard())?.0)
-    }
-}
-
-impl NetDataAxum {
-    #[cfg(feature = "axum")]
-    pub fn encode_axum_message(&self) -> Result<ws::Message> {
-        let encoded = bincode::encode_to_vec(self, config::standard())?;
-        Ok(ws::Message::Binary(encoded))
+impl NetData {
+    pub fn encode_message(&self) -> Result<Vec<u8>> {
+        Ok(bincode::encode_to_vec(self, config::standard())?)
     }
 
-    pub fn decode_message(bytes: &[u8]) -> Result<NetDataAxum> {
+    pub fn decode_message(bytes: &[u8]) -> Result<NetData> {
         Ok(bincode::decode_from_slice(bytes, config::standard())?.0)
     }
 }