use crate::{ entities::view::{CreateViewParams, UpdateViewParams, View}, errors::WorkspaceError, module::WorkspaceDatabase, observable::{send_observable, WorkspaceObservable}, services::{helper::spawn, server::Server}, sql_tables::view::{ViewTable, ViewTableChangeset, ViewTableSql}, }; use crate::{ entities::view::{DeleteViewParams, QueryViewParams}, module::WorkspaceUser, }; use std::sync::Arc; pub(crate) struct ViewController { user: Arc, sql: Arc, server: Server, } impl ViewController { pub(crate) fn new(user: Arc, database: Arc, server: Server) -> Self { let sql = Arc::new(ViewTableSql { database }); Self { user, sql, server } } pub(crate) async fn create_view(&self, params: CreateViewParams) -> Result { let view = self.create_view_on_server(params).await?; let view_table = ViewTable::new(view.clone()); let _ = self.sql.create_view(view_table)?; send_observable(&view.belong_to_id, WorkspaceObservable::AppCreateView); Ok(view) } pub(crate) async fn read_view(&self, params: QueryViewParams) -> Result { let view_table = self.sql.read_view(¶ms.view_id, params.is_trash)?; let view: View = view_table.into(); let _ = self.read_view_on_server(params).await?; Ok(view) } pub(crate) async fn delete_view(&self, view_id: &str) -> Result<(), WorkspaceError> { let view = self.sql.delete_view(view_id)?; let _ = self.delete_view_on_server(view_id).await?; send_observable(&view.belong_to_id, WorkspaceObservable::AppDeleteView); Ok(()) } // belong_to_id will be the app_id or view_id. pub(crate) async fn read_views_belong_to(&self, belong_to_id: &str) -> Result, WorkspaceError> { let views = self .sql .read_views_belong_to(belong_to_id)? .into_iter() .map(|view_table| view_table.into()) .collect::>(); Ok(views) } pub(crate) async fn update_view(&self, params: UpdateViewParams) -> Result<(), WorkspaceError> { let changeset = ViewTableChangeset::new(params.clone()); let view_id = changeset.id.clone(); let _ = self.sql.update_view(changeset)?; let _ = self.update_view_on_server(params).await?; send_observable(&view_id, WorkspaceObservable::ViewUpdated); Ok(()) } } impl ViewController { #[tracing::instrument(skip(self), err)] async fn create_view_on_server(&self, params: CreateViewParams) -> Result { let token = self.user.token()?; let view = self.server.create_view(&token, params).await?; Ok(view) } #[tracing::instrument(skip(self), err)] async fn update_view_on_server(&self, params: UpdateViewParams) -> Result<(), WorkspaceError> { let token = self.user.token()?; let server = self.server.clone(); spawn(async move { match server.update_view(&token, params).await { Ok(_) => {}, Err(e) => { // TODO: retry? log::error!("Update view failed: {:?}", e); }, } }); Ok(()) } #[tracing::instrument(skip(self), err)] async fn delete_view_on_server(&self, view_id: &str) -> Result<(), WorkspaceError> { let token = self.user.token()?; let server = self.server.clone(); let params = DeleteViewParams { view_id: view_id.to_string(), }; spawn(async move { match server.delete_view(&token, params).await { Ok(_) => {}, Err(e) => { // TODO: retry? log::error!("Delete view failed: {:?}", e); }, } }); Ok(()) } #[tracing::instrument(skip(self), err)] async fn read_view_on_server(&self, params: QueryViewParams) -> Result<(), WorkspaceError> { let token = self.user.token()?; let server = self.server.clone(); spawn(async move { match server.read_view(&token, params).await { Ok(_) => {}, Err(e) => { // TODO: retry? log::error!("Read view failed: {:?}", e); }, } }); Ok(()) } }