Add elyspace core import export model

2026-05-08 23:05:53 -04:00
parent 784fe5a288
commit 058d377eab
7 changed files with 444 additions and 2 deletions
@@ -0,0 +1,295 @@
+use ely_domain::{
+    ArchivePolicy, BrowserTab, ProfileId, ProfileKind, Space, SpaceId, TabId, UrlText,
+};
+use serde::{Deserialize, Serialize};
+
+use super::BrowserCore;
+use crate::CoreError;
+
+pub const ELYSPACE_SCHEMA_VERSION: u16 = 1;
+pub const ELYSPACE_FILE_EXTENSION: &str = "elyspace";
+
+#[derive(Clone, Copy, Debug, Eq, PartialEq)]
+pub enum SpaceImportProfileMapping {
+    PreserveExisting,
+    UseActiveProfile,
+}
+
+#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
+#[serde(deny_unknown_fields)]
+pub struct ElySpacePackage {
+    version: u16,
+    space: ElySpaceRecord,
+    tabs: Vec<ElySpaceTabRecord>,
+}
+
+impl ElySpacePackage {
+    #[must_use]
+    pub fn version(&self) -> u16 {
+        self.version
+    }
+
+    #[must_use]
+    pub fn space_name(&self) -> &str {
+        &self.space.name
+    }
+
+    #[must_use]
+    pub fn tab_count(&self) -> usize {
+        self.tabs.len()
+    }
+}
+
+#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
+#[serde(deny_unknown_fields)]
+struct ElySpaceRecord {
+    name: String,
+    icon: String,
+    accent_hex: u32,
+    default_profile_id: String,
+    archive_policy: ElySpaceArchivePolicy,
+    sidebar_width_px: u16,
+}
+
+#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
+#[serde(deny_unknown_fields)]
+struct ElySpaceTabRecord {
+    title: String,
+    url: String,
+    profile_id: String,
+    favicon_key: Option<String>,
+    pinned: bool,
+    favorite: bool,
+    sort_key: u64,
+    sync_enabled: bool,
+}
+
+#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
+#[serde(tag = "kind", rename_all = "snake_case")]
+enum ElySpaceArchivePolicy {
+    Manual,
+    IdleDays { days: u16 },
+}
+
+impl BrowserCore {
+    pub fn export_space_package_json(&self, space_id: &SpaceId) -> Result<String, CoreError> {
+        serde_json::to_string_pretty(&self.export_space_package(space_id)?)
+            .map_err(invalid_space_package)
+    }
+
+    pub fn export_space_package(&self, space_id: &SpaceId) -> Result<ElySpacePackage, CoreError> {
+        let space = self
+            .spaces
+            .iter()
+            .find(|space| space.id() == space_id)
+            .ok_or_else(|| CoreError::SpaceNotFound { id: space_id.clone() })?;
+        let tabs = self
+            .tabs
+            .iter()
+            .filter(|tab| tab.space_id() == space_id)
+            .map(ElySpaceTabRecord::from_tab)
+            .collect();
+
+        Ok(ElySpacePackage {
+            version: ELYSPACE_SCHEMA_VERSION,
+            space: ElySpaceRecord::from_space(space),
+            tabs,
+        })
+    }
+
+    pub fn import_space_package_json(
+        &mut self,
+        package_json: &str,
+        profile_mapping: SpaceImportProfileMapping,
+    ) -> Result<SpaceId, CoreError> {
+        let package = serde_json::from_str(package_json).map_err(invalid_space_package)?;
+        self.import_space_package(package, profile_mapping)
+    }
+
+    pub fn import_space_package(
+        &mut self,
+        package: ElySpacePackage,
+        profile_mapping: SpaceImportProfileMapping,
+    ) -> Result<SpaceId, CoreError> {
+        validate_package_version(package.version)?;
+        self.validate_imported_favorites(&package)?;
+        let default_profile_id = self.imported_default_profile_id(&package, profile_mapping)?;
+        let sort_key = self.next_space_sort_key();
+        let mut space = Space::new(
+            package.space.name.clone(),
+            package.space.icon.clone(),
+            package.space.accent_hex,
+            default_profile_id.clone(),
+            sort_key,
+        );
+        space.set_archive_policy(package.space.archive_policy.clone().into());
+        space.set_sidebar_width_px(package.space.sidebar_width_px);
+
+        let space_id = space.id().clone();
+        let tabs = self.imported_tabs(&space_id, &default_profile_id, &package, profile_mapping)?;
+        let active_tab_id = tabs[0].id().clone();
+
+        self.spaces.push(space);
+        self.tabs.extend(tabs);
+        self.active_tabs_by_space.insert(space_id.clone(), active_tab_id.clone());
+        self.select_tab(&active_tab_id)?;
+        Ok(space_id)
+    }
+
+    fn imported_default_profile_id(
+        &self,
+        package: &ElySpacePackage,
+        profile_mapping: SpaceImportProfileMapping,
+    ) -> Result<ProfileId, CoreError> {
+        match profile_mapping {
+            SpaceImportProfileMapping::PreserveExisting => {
+                let profile_id = ProfileId::parse(&package.space.default_profile_id)?;
+                self.standard_profile_id(&profile_id)?;
+                Ok(profile_id)
+            }
+            SpaceImportProfileMapping::UseActiveProfile => self.stable_active_profile_id(),
+        }
+    }
+
+    fn validate_imported_favorites(&self, package: &ElySpacePackage) -> Result<(), CoreError> {
+        let imported_favorites = package.tabs.iter().filter(|tab| tab.favorite).count();
+        let favorite_limit = self.favorite_limit.value();
+        if self.favorites().len() + imported_favorites > favorite_limit {
+            return Err(CoreError::FavoriteLimitReached { limit: favorite_limit });
+        }
+        Ok(())
+    }
+
+    fn imported_tabs(
+        &self,
+        space_id: &SpaceId,
+        default_profile_id: &ProfileId,
+        package: &ElySpacePackage,
+        profile_mapping: SpaceImportProfileMapping,
+    ) -> Result<Vec<BrowserTab>, CoreError> {
+        let mut tabs = package
+            .tabs
+            .iter()
+            .map(|tab| self.imported_tab(space_id, default_profile_id, tab, profile_mapping))
+            .collect::<Result<Vec<_>, _>>()?;
+
+        if tabs.is_empty() {
+            tabs.push(self.build_tab_for(
+                space_id.clone(),
+                default_profile_id.clone(),
+                self.new_tab_url()?,
+            ));
+        }
+        Ok(tabs)
+    }
+
+    fn imported_tab(
+        &self,
+        space_id: &SpaceId,
+        default_profile_id: &ProfileId,
+        record: &ElySpaceTabRecord,
+        profile_mapping: SpaceImportProfileMapping,
+    ) -> Result<BrowserTab, CoreError> {
+        let profile_id = match profile_mapping {
+            SpaceImportProfileMapping::PreserveExisting => {
+                let profile_id = ProfileId::parse(&record.profile_id)?;
+                self.standard_profile_id(&profile_id)?;
+                profile_id
+            }
+            SpaceImportProfileMapping::UseActiveProfile => default_profile_id.clone(),
+        };
+        let mut tab = BrowserTab::new(
+            TabId::new(),
+            space_id.clone(),
+            profile_id,
+            record.title.clone(),
+            UrlText::parse(&record.url)?,
+        )
+        .with_sort_key(record.sort_key);
+        tab.set_pinned(record.pinned);
+        tab.set_favorite(record.favorite);
+        tab.set_sync_enabled(record.sync_enabled);
+        if let Some(favicon_key) = &record.favicon_key {
+            tab.set_favicon_key(favicon_key.clone())?;
+        }
+        Ok(tab)
+    }
+
+    fn standard_profile_id(&self, profile_id: &ProfileId) -> Result<(), CoreError> {
+        let profile = self
+            .profiles
+            .iter()
+            .find(|profile| profile.id() == profile_id)
+            .ok_or_else(|| CoreError::ProfileNotFound { id: profile_id.clone() })?;
+        if profile.kind() == &ProfileKind::Private {
+            return Err(CoreError::PrivateProfileDefaultLocked { id: profile_id.clone() });
+        }
+        Ok(())
+    }
+
+    fn stable_active_profile_id(&self) -> Result<ProfileId, CoreError> {
+        if self.active_profile()?.kind() == &ProfileKind::Private {
+            Ok(self.active_space()?.default_profile_id().clone())
+        } else {
+            Ok(self.active_profile_id.clone())
+        }
+    }
+}
+
+impl ElySpaceRecord {
+    fn from_space(space: &Space) -> Self {
+        Self {
+            name: space.name().to_string(),
+            icon: space.icon().to_string(),
+            accent_hex: space.accent_hex(),
+            default_profile_id: space.default_profile_id().as_str().to_string(),
+            archive_policy: ElySpaceArchivePolicy::from(space.archive_policy()),
+            sidebar_width_px: space.sidebar_width_px(),
+        }
+    }
+}
+
+impl ElySpaceTabRecord {
+    fn from_tab(tab: &BrowserTab) -> Self {
+        Self {
+            title: tab.title().to_string(),
+            url: tab.url().as_str().to_string(),
+            profile_id: tab.profile_id().as_str().to_string(),
+            favicon_key: tab.favicon_key().map(str::to_string),
+            pinned: tab.flags().pinned,
+            favorite: tab.flags().favorite,
+            sort_key: tab.sort_key(),
+            sync_enabled: tab.sync_enabled(),
+        }
+    }
+}
+
+impl From<&ArchivePolicy> for ElySpaceArchivePolicy {
+    fn from(policy: &ArchivePolicy) -> Self {
+        match policy {
+            ArchivePolicy::Manual => Self::Manual,
+            ArchivePolicy::IdleDays(days) => Self::IdleDays { days: *days },
+        }
+    }
+}
+
+impl From<ElySpaceArchivePolicy> for ArchivePolicy {
+    fn from(policy: ElySpaceArchivePolicy) -> Self {
+        match policy {
+            ElySpaceArchivePolicy::Manual => Self::Manual,
+            ElySpaceArchivePolicy::IdleDays { days } => Self::IdleDays(days),
+        }
+    }
+}
+
+fn validate_package_version(version: u16) -> Result<(), CoreError> {
+    if version == ELYSPACE_SCHEMA_VERSION {
+        Ok(())
+    } else {
+        Err(CoreError::InvalidSpacePackage { reason: format!("unsupported version {version}") })
+    }
+}
+
+fn invalid_space_package(error: serde_json::Error) -> CoreError {
+    CoreError::InvalidSpacePackage { reason: error.to_string() }
+}