Initial public release of Warp.
Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b
This commit is contained in:
David Stern
@@ -0,0 +1,217 @@
|
||||
//! Remote repository metadata model (client-side).
|
||||
//!
|
||||
//! Holds file tree state for repositories on remote servers. In this initial phase
|
||||
//! there is no syncing or indexing — state is populated externally (e.g. by a future
|
||||
//! remote client model or via test helpers).
|
||||
|
||||
use std::collections::HashMap;
|
||||
use std::sync::Arc;
|
||||
|
||||
use warp_core::HostId;
|
||||
use warpui::ModelContext;
|
||||
|
||||
use crate::file_tree_store::{FileTreeEntry, FileTreeState};
|
||||
use crate::file_tree_update::RepoMetadataUpdate;
|
||||
use crate::local_model::{GetContentsArgs, IndexedRepoState, RepoContent};
|
||||
use crate::repository_identifier::RemoteRepositoryIdentifier;
|
||||
|
||||
use super::local_model::collect_contents_recursive;
|
||||
|
||||
/// Events emitted by the [`RemoteRepoMetadataModel`].
|
||||
#[derive(Debug)]
|
||||
pub enum RemoteRepositoryMetadataEvent {
|
||||
/// A remote repository was added or updated.
|
||||
RepositoryUpdated { id: RemoteRepositoryIdentifier },
|
||||
/// A remote repository was removed.
|
||||
RepositoryRemoved { id: RemoteRepositoryIdentifier },
|
||||
/// The file tree for remote repositories was updated.
|
||||
FileTreeUpdated {
|
||||
ids: Vec<RemoteRepositoryIdentifier>,
|
||||
},
|
||||
/// The file tree entry for a remote repository was updated.
|
||||
FileTreeEntryUpdated { id: RemoteRepositoryIdentifier },
|
||||
}
|
||||
|
||||
/// Client-side model for remote repository metadata.
|
||||
///
|
||||
/// This model holds file tree state for repositories living on remote servers.
|
||||
/// It provides the same read-only query surface as the local model, and write
|
||||
/// methods that will be the integration points for the future remote sync layer.
|
||||
///
|
||||
/// Consumers should access this through the [`RepoMetadataModel`](crate::wrapper_model::RepoMetadataModel)
|
||||
/// wrapper rather than using this type directly.
|
||||
pub struct RemoteRepoMetadataModel {
|
||||
repositories: HashMap<RemoteRepositoryIdentifier, IndexedRepoState>,
|
||||
}
|
||||
|
||||
impl RemoteRepoMetadataModel {
|
||||
pub fn new(_ctx: &mut ModelContext<Self>) -> Self {
|
||||
Self {
|
||||
repositories: HashMap::new(),
|
||||
}
|
||||
}
|
||||
|
||||
// ── Read-only query API ──────────────────────────────────────────
|
||||
|
||||
/// Returns the [`FileTreeState`] for a remote repository, if it is indexed.
|
||||
pub fn get_repository(&self, id: &RemoteRepositoryIdentifier) -> Option<&FileTreeState> {
|
||||
match self.repositories.get(id)? {
|
||||
IndexedRepoState::Indexed(state) => Some(state),
|
||||
IndexedRepoState::Pending | IndexedRepoState::Failed(_) => None,
|
||||
}
|
||||
}
|
||||
|
||||
/// Returns whether the given remote repository is indexed.
|
||||
pub fn has_repository(&self, id: &RemoteRepositoryIdentifier) -> bool {
|
||||
matches!(
|
||||
self.repositories.get(id),
|
||||
Some(IndexedRepoState::Indexed(_))
|
||||
)
|
||||
}
|
||||
|
||||
/// Returns the current [`IndexedRepoState`] for a remote repository.
|
||||
pub fn repository_state(&self, id: &RemoteRepositoryIdentifier) -> Option<&IndexedRepoState> {
|
||||
self.repositories.get(id)
|
||||
}
|
||||
|
||||
/// Returns repository contents for the specified remote repository.
|
||||
pub fn get_repo_contents(
|
||||
&self,
|
||||
id: &RemoteRepositoryIdentifier,
|
||||
args: GetContentsArgs,
|
||||
) -> Option<Vec<RepoContent<'_>>> {
|
||||
let state = match self.repositories.get(id)? {
|
||||
IndexedRepoState::Indexed(state) => state,
|
||||
IndexedRepoState::Pending | IndexedRepoState::Failed(_) => return None,
|
||||
};
|
||||
let mut contents = Vec::new();
|
||||
collect_contents_recursive(
|
||||
&state.entry,
|
||||
state.entry.root_directory(),
|
||||
&mut contents,
|
||||
&args,
|
||||
);
|
||||
Some(contents)
|
||||
}
|
||||
|
||||
/// Returns all tracked remote repository identifiers, including those in
|
||||
/// `Pending` or `Failed` states. Callers that only need indexed repos
|
||||
/// should filter via [`get_repository`](Self::get_repository).
|
||||
pub fn remote_repository_ids(&self) -> impl Iterator<Item = &RemoteRepositoryIdentifier> {
|
||||
self.repositories.keys()
|
||||
}
|
||||
|
||||
// ── Write API (for future sync + test use) ───────────────────────
|
||||
|
||||
/// Inserts or replaces file tree state for a remote repository.
|
||||
pub fn insert_repository(
|
||||
&mut self,
|
||||
id: RemoteRepositoryIdentifier,
|
||||
state: FileTreeState,
|
||||
ctx: &mut ModelContext<Self>,
|
||||
) {
|
||||
self.repositories
|
||||
.insert(id.clone(), IndexedRepoState::Indexed(state));
|
||||
ctx.emit(RemoteRepositoryMetadataEvent::RepositoryUpdated { id });
|
||||
}
|
||||
|
||||
/// Removes a remote repository from tracking.
|
||||
pub fn remove_repository(
|
||||
&mut self,
|
||||
id: &RemoteRepositoryIdentifier,
|
||||
ctx: &mut ModelContext<Self>,
|
||||
) {
|
||||
if self.repositories.remove(id).is_some() {
|
||||
ctx.emit(RemoteRepositoryMetadataEvent::RepositoryRemoved { id: id.clone() });
|
||||
}
|
||||
}
|
||||
|
||||
/// Replaces the file tree entry within an existing remote repository's state.
|
||||
pub fn update_file_tree_entry(
|
||||
&mut self,
|
||||
id: &RemoteRepositoryIdentifier,
|
||||
entry: FileTreeEntry,
|
||||
ctx: &mut ModelContext<Self>,
|
||||
) {
|
||||
if let Some(IndexedRepoState::Indexed(state)) = self.repositories.get_mut(id) {
|
||||
state.entry = entry;
|
||||
ctx.emit(RemoteRepositoryMetadataEvent::FileTreeEntryUpdated { id: id.clone() });
|
||||
}
|
||||
}
|
||||
|
||||
/// Inserts or replaces a remote repository from a snapshot update.
|
||||
///
|
||||
/// Creates a `FileTreeEntry` from the update by starting with an empty
|
||||
/// root and applying the snapshot entries, then wraps it in a
|
||||
/// `FileTreeState`. This is the primary entry point for populating
|
||||
/// remote repo state from server push events.
|
||||
pub fn insert_from_snapshot(
|
||||
&mut self,
|
||||
host_id: HostId,
|
||||
update: &RepoMetadataUpdate,
|
||||
ctx: &mut ModelContext<Self>,
|
||||
) {
|
||||
let mut entry = FileTreeEntry::new_for_directory(Arc::new(update.repo_path.clone()));
|
||||
entry.apply_repo_metadata_update(update);
|
||||
let state = FileTreeState::from_file_tree_entry(entry);
|
||||
let id = RemoteRepositoryIdentifier::new(host_id, update.repo_path.clone());
|
||||
self.insert_repository(id, state, ctx);
|
||||
}
|
||||
|
||||
/// Removes all remote repositories associated with the given host.
|
||||
pub fn remove_repositories_for_host(&mut self, host_id: &HostId, ctx: &mut ModelContext<Self>) {
|
||||
let ids_to_remove: Vec<RemoteRepositoryIdentifier> = self
|
||||
.repositories
|
||||
.keys()
|
||||
.filter(|id| id.host_id == *host_id)
|
||||
.cloned()
|
||||
.collect();
|
||||
for id in ids_to_remove {
|
||||
self.remove_repository(&id, ctx);
|
||||
}
|
||||
}
|
||||
|
||||
/// Applies an incremental update received from the remote server.
|
||||
///
|
||||
/// Looks up the repository by matching `(host_id, repo_path)` against
|
||||
/// tracked [`RemoteRepositoryIdentifier`]s, then delegates to
|
||||
/// [`FileTreeEntry::apply_repo_metadata_update`].
|
||||
pub fn apply_incremental_update(
|
||||
&mut self,
|
||||
host_id: &HostId,
|
||||
update: &RepoMetadataUpdate,
|
||||
ctx: &mut ModelContext<Self>,
|
||||
) {
|
||||
let matching_id = self
|
||||
.repositories
|
||||
.keys()
|
||||
.find(|id| id.host_id == *host_id && id.path == update.repo_path)
|
||||
.cloned();
|
||||
|
||||
let Some(id) = matching_id else {
|
||||
log::warn!(
|
||||
"No remote repository found for incremental update: {}",
|
||||
update.repo_path
|
||||
);
|
||||
return;
|
||||
};
|
||||
|
||||
if let Some(IndexedRepoState::Indexed(state)) = self.repositories.get_mut(&id) {
|
||||
state.entry.apply_repo_metadata_update(update);
|
||||
ctx.emit(RemoteRepositoryMetadataEvent::FileTreeEntryUpdated { id });
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl warpui::Entity for RemoteRepoMetadataModel {
|
||||
type Event = RemoteRepositoryMetadataEvent;
|
||||
}
|
||||
|
||||
#[cfg(any(test, feature = "test-util"))]
|
||||
impl RemoteRepoMetadataModel {
|
||||
/// Insert a repository state directly for testing purposes.
|
||||
pub fn insert_test_state(&mut self, id: RemoteRepositoryIdentifier, state: FileTreeState) {
|
||||
self.repositories
|
||||
.insert(id, IndexedRepoState::Indexed(state));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user