@anysphere/file-service 0.0.0-e3fdf62d → 0.0.0-e492c95a

package/src/merkle_tree/local_construction.rs

@@ -1,206 +0,0 @@
-use crate::merkle_tree::{
-  File, MerkleNode, MerkleNodePtr, NodeType, PinnedFuture,
-};
-
-use super::{LocalConstruction, MerkleTree};
-use std::path::PathBuf;
-use std::{collections::HashMap, path::Path, sync::Arc};
-use tonic::async_trait;
-
-#[async_trait]
-impl LocalConstruction for MerkleTree {
-  async fn new(
-    root_directory: Option<String>,
-  ) -> Result<MerkleTree, anyhow::Error> {
-    if let Some(root_directory) = root_directory {
-      let n = MerkleTree::construct_merkle_tree(root_directory).await;
-      return n;
-    }
-
-    Ok(MerkleTree::empty_tree())
-  }
-
-  /// Factory method to construct a merkle tree from a given directory
-  ///
-  /// plan:
-  /// 1. get all files in the directory
-  /// 2. compute hash for each file
-  /// 3. construct merkle tree
-  /// 4. return merkle tree
-  async fn construct_merkle_tree(
-    root_directory: String,
-  ) -> Result<MerkleTree, anyhow::Error> {
-    let path = PathBuf::from(root_directory.clone());
-    if !path.exists() {
-      // FIXME: we should report this via a good logger.
-      panic!("Root directory does not exist!");
-    }
-
-    let root_node = MerkleNode::new(path, None).await;
-    let mut mt = MerkleTree {
-      root: root_node,
-      files: HashMap::new(),
-      root_path: root_directory,
-      cursor: None,
-    };
-
-    // we now iterate over all the nodes and add them to the hashmap
-    // TODO(later): i can make this parallel.
-    fn add_nodes_to_hashmap<'a>(
-      node: &'a MerkleNodePtr,
-      files: &'a mut HashMap<String, File>,
-    ) -> PinnedFuture<'a, ()> {
-      Box::pin(async move {
-        let node_reader = node.read().await;
-        match &node_reader.node_type {
-          NodeType::Branch(n) => {
-            let children = &n.1;
-            files.insert(n.0.clone(), File { node: node.clone() });
-            for child in children {
-              add_nodes_to_hashmap(child, files).await;
-            }
-          }
-          NodeType::File(file_name) => {
-            let f = File { node: node.clone() };
-            files.insert(file_name.clone(), f);
-          }
-          NodeType::ErrorNode(_) => {
-            // do nothing
-          }
-        }
-      })
-    }
-
-    add_nodes_to_hashmap(&mt.root, &mut mt.files).await;
-
-    Ok(mt)
-  }
-
-  async fn update_file(
-    &mut self,
-    file_path: String,
-  ) -> Result<(), anyhow::Error> {
-    // 1. check if the file exists.
-    //   if it does, then update the file
-    // 2. if it does not, then add the file
-
-    let path = Path::new(&file_path);
-
-    if !path.exists() {
-      return Err(anyhow::anyhow!("File does not exist!"));
-    }
-
-    let path_exists_in_tree = self.files.contains_key(&file_path);
-
-    if path_exists_in_tree {
-      // Update the file in two places.
-      let file = self.files.get(&file_path).unwrap();
-
-      let r = self.update_node(file.node.clone()).await;
-
-      if r.is_err() {
-        // FIXME(sualeh): we should report this via a good logger.
-        println!("Could not update node! {}", r.unwrap_err().to_string());
-        return Err(anyhow::anyhow!("Could not update node!"));
-      }
-    } else {
-      // File does not exist in the tree, let's add it.
-      let e = self.attach_new_node_to_tree(file_path.clone()).await;
-
-      if e.is_err() {
-        return Err(anyhow::anyhow!("Could not attach new node to tree!"));
-      }
-    }
-
-    Ok(())
-  }
-
-  async fn delete_file(
-    &mut self,
-    file_path: String,
-  ) -> Result<(), anyhow::Error> {
-    // 1. check if the file exists.
-    //   if it does, then delete the file
-    //
-    // 2. if it does not, then do nothing
-
-    let path = Path::new(&file_path);
-
-    if !path.exists() {
-      return Err(anyhow::anyhow!("File does not exist!"));
-    }
-
-    let path_exists_in_tree = self.files.contains_key(&file_path);
-
-    if !path_exists_in_tree {
-      return Err(anyhow::anyhow!("File does not exist in the tree!"));
-    }
-
-    // Get the parent node
-    let parent_node = self.get_parent_node(&path);
-    if parent_node.is_err() {
-      return Err(anyhow::anyhow!("Could not get parent node! 0"));
-    }
-    let parent_node = parent_node.unwrap();
-    let mut mut_parent = parent_node.write().await;
-
-    // BUG(sualeh): need to actually drop everything that is a child here.
-    // idea: enumerate all nodes that are children of this through a starts with query on the hashtable.
-    // then drop all of them.
-    // in opposite order of length.
-
-    // then remove the node from the parent and you are done
-
-    // Remove the child from the parent node
-    match mut_parent.node_type {
-      NodeType::Branch(ref mut node) => {
-        let children = &mut node.1;
-        let mut found = false;
-        let mut index = 0;
-
-        for (i, child) in children.iter().enumerate() {
-          let child_reader = child.read().await;
-          match &child_reader.node_type {
-            NodeType::File(child_file_path) => {
-              // TODO(sualeh): THERE HAS TO BE A BETTER WAY TO DO THIS.
-              if child_file_path == &file_path {
-                found = true;
-                index = i;
-                break;
-              }
-            }
-            NodeType::Branch(_) => {
-              continue;
-            }
-            NodeType::ErrorNode(_) => {
-              continue;
-            }
-          }
-        }
-
-        if found {
-          // remove the child from the parent.
-          let _ = children.remove(index);
-        }
-      }
-      NodeType::File(_) => {
-        return Err(anyhow::anyhow!(
-          "delete_file: This should not happen! parent was a file ..."
-        ));
-      }
-      NodeType::ErrorNode(_) => {
-        return Err(anyhow::anyhow!(
-          "delete_file: This should not happen! parent was an error node ..."
-        ));
-      }
-    };
-
-    // need to propogate the change up the tree.
-    mut_parent.update_hash_of_ancestor_path();
-
-    // Remove the file from the files map
-    self.files.remove(&file_path);
-
-    Ok(())
-  }
-}