1 files changed, 79 insertions, 4 deletions
diff --git a/crates/vcs_actions/src/actions/local_actions.rs b/crates/vcs_actions/src/actions/local_actions.rs
index 3027218..87eafb8 100644
--- a/crates/vcs_actions/src/actions/local_actions.rs
+++ b/crates/vcs_actions/src/actions/local_actions.rs
@@ -5,7 +5,10 @@ use cfg_file::config::ConfigFile;
 use log::{info, warn};
 use serde::{Deserialize, Serialize};
 use tcp_connection::error::TcpTargetError;
-use vcs_data::data::{local::config::LocalConfig, vault::config::VaultUuid};
+use vcs_data::data::{
+    local::{config::LocalConfig, latest_info::LatestInfo},
+    vault::config::VaultUuid,
+};
 
 use crate::actions::{
     auth_member, check_connection_instance, try_get_local_workspace, try_get_vault,
@@ -26,15 +29,14 @@ pub async fn set_upstream_vault_action(
     ctx: ActionContext,
     upstream: SocketAddr,
 ) -> Result<SetUpstreamVaultActionResult, TcpTargetError> {
-    // Ensure the instance is available
     let instance = check_connection_instance(&ctx)?;
 
-    // Step1: Auth Member
+    // Auth Member
     if let Err(e) = auth_member(&ctx, instance).await {
         return Ok(SetUpstreamVaultActionResult::AuthorizeFailed(e.to_string()));
     }
 
-    // Step2: Direct
+    // Direct
     if ctx.is_proc_on_remote() {
         let vault = try_get_vault(&ctx)?;
         instance
@@ -74,3 +76,76 @@ pub async fn set_upstream_vault_action(
 
     Err(TcpTargetError::NoResult("No result.".to_string()))
 }
+
+#[derive(Serialize, Deserialize)]
+pub enum UpdateToLatestInfoResult {
+    Success,
+
+    // Fail
+    AuthorizeFailed(String),
+}
+
+#[action_gen]
+pub async fn update_to_latest_info_action(
+    ctx: ActionContext,
+    _unused: (),
+) -> Result<UpdateToLatestInfoResult, TcpTargetError> {
+    let instance = check_connection_instance(&ctx)?;
+
+    let member_id = match auth_member(&ctx, instance).await {
+        Ok(id) => id,
+        Err(e) => return Ok(UpdateToLatestInfoResult::AuthorizeFailed(e.to_string())),
+    };
+
+    if ctx.is_proc_on_remote() {
+        let vault = try_get_vault(&ctx)?;
+
+        // Build latest info
+        let mut latest_info = LatestInfo::default();
+
+        // Sheet
+        let mut member_owned = Vec::new();
+        let mut member_visible = Vec::new();
+
+        for sheet in vault.sheets().await? {
+            if sheet.holder() == &member_id {
+                member_owned.push(sheet.name().clone());
+            } else {
+                member_visible.push(sheet.name().clone());
+            }
+        }
+
+        latest_info.my_sheets = member_owned;
+        latest_info.other_sheets = member_visible;
+
+        // RefSheet
+        let ref_sheet_data = vault.sheet(&"ref".to_string()).await?.to_data();
+        latest_info.ref_sheet_content = ref_sheet_data;
+
+        // Members
+        let members = vault.members().await?;
+        latest_info.vault_members = members;
+
+        // Send
+        instance
+            .lock()
+            .await
+            .write_large_msgpack(latest_info, 512 as u16)
+            .await?;
+
+        return Ok(UpdateToLatestInfoResult::Success);
+    }
+
+    if ctx.is_proc_on_local() {
+        let latest_info = instance
+            .lock()
+            .await
+            .read_large_msgpack::<LatestInfo>(512 as u16)
+            .await?;
+        LatestInfo::write(&latest_info).await?;
+
+        return Ok(UpdateToLatestInfoResult::Success);
+    }
+
+    Err(TcpTargetError::NoResult("No result.".to_string()))
+}