jellyflow-runtime 0.2.0

use std::path::Path;

use jellyflow_core::core::GraphId;
use serde::{Deserialize, Serialize};

use crate::io::config::NodeGraphEditorConfig;
use crate::io::view_state::{NodeGraphPureViewState, NodeGraphViewState};

/// Editor-state file format version.
pub const EDITOR_STATE_FILE_VERSION: u32 = 1;

/// Errors for reading/writing editor-state files.
#[derive(Debug, thiserror::Error)]
pub enum NodeGraphEditorStateFileError {
    /// Read failure.
    #[error("failed to read node graph editor-state file: {path}")]
    Read {
        path: String,
        source: std::io::Error,
    },
    /// JSON parse failure.
    #[error("failed to parse node graph editor-state file JSON: {path}")]
    Parse {
        path: String,
        source: serde_json::Error,
    },
    /// Write failure.
    #[error("failed to write node graph editor-state file: {path}")]
    Write {
        path: String,
        source: std::io::Error,
    },
    /// JSON serialization failure.
    #[error("failed to serialize node graph editor-state JSON: {path}")]
    Serialize {
        path: String,
        source: serde_json::Error,
    },
    /// Wrapper id mismatch.
    #[error("editor-state file wrapper graph_id does not match requested graph_id")]
    InconsistentGraphId,
    /// Unsupported wrapper version.
    #[error("unsupported node graph editor-state version {version}; expected {expected}")]
    UnsupportedVersion { version: u32, expected: u32 },
}

/// Project-scoped editor-state persistence file.
///
/// The graph document is saved separately by `GraphFileV1`; this file owns only user/editor state:
/// pure canvas view state plus persisted editor policy and runtime tuning.
#[derive(Debug, Clone, PartialEq)]
pub struct NodeGraphEditorStateFile {
    /// Graph id.
    pub graph_id: GraphId,
    /// Editor-state schema version.
    pub editor_state_version: u32,
    /// Pure view-state payload.
    pub view_state: NodeGraphViewState,
    /// Persisted editor policy and runtime tuning.
    pub editor_config: NodeGraphEditorConfig,
}

impl NodeGraphEditorStateFile {
    /// Wraps editor state for a graph.
    pub fn new(
        graph_id: GraphId,
        view_state: NodeGraphViewState,
        editor_config: NodeGraphEditorConfig,
    ) -> Self {
        Self {
            graph_id,
            editor_state_version: EDITOR_STATE_FILE_VERSION,
            view_state,
            editor_config,
        }
    }

    /// Loads a JSON file.
    pub fn load_json(
        path: impl AsRef<Path>,
        graph_id: GraphId,
    ) -> Result<Self, NodeGraphEditorStateFileError> {
        let path = path.as_ref();
        let bytes = std::fs::read(path).map_err(|source| NodeGraphEditorStateFileError::Read {
            path: path.display().to_string(),
            source,
        })?;

        let persisted: PersistedNodeGraphEditorStateFile =
            serde_json::from_slice(&bytes).map_err(|source| {
                NodeGraphEditorStateFileError::Parse {
                    path: path.display().to_string(),
                    source,
                }
            })?;
        persisted.validate_for_graph(graph_id)?;
        Ok(persisted.into_editor_state_file())
    }

    /// Loads the JSON file if it exists.
    pub fn load_json_if_exists(
        path: impl AsRef<Path>,
        graph_id: GraphId,
    ) -> Result<Option<Self>, NodeGraphEditorStateFileError> {
        let path = path.as_ref();
        if !path.exists() {
            return Ok(None);
        }
        Self::load_json(path, graph_id).map(Some)
    }

    /// Saves the JSON file (pretty-printed).
    pub fn save_json(&self, path: impl AsRef<Path>) -> Result<(), NodeGraphEditorStateFileError> {
        let path = path.as_ref();
        if let Some(parent) = path.parent() {
            std::fs::create_dir_all(parent).map_err(|source| {
                NodeGraphEditorStateFileError::Write {
                    path: path.display().to_string(),
                    source,
                }
            })?;
        }
        let persisted = PersistedNodeGraphEditorStateFile::from_editor_state_file(self);
        let bytes = serde_json::to_vec_pretty(&persisted).map_err(|source| {
            NodeGraphEditorStateFileError::Serialize {
                path: path.display().to_string(),
                source,
            }
        })?;
        std::fs::write(path, bytes).map_err(|source| NodeGraphEditorStateFileError::Write {
            path: path.display().to_string(),
            source,
        })
    }
}

#[derive(Serialize, Deserialize)]
struct PersistedNodeGraphEditorStateFile {
    graph_id: GraphId,
    editor_state_version: u32,
    view_state: NodeGraphPureViewState,
    #[serde(default, skip_serializing_if = "NodeGraphEditorConfig::is_default")]
    editor_config: NodeGraphEditorConfig,
}

impl PersistedNodeGraphEditorStateFile {
    fn from_editor_state_file(file: &NodeGraphEditorStateFile) -> Self {
        Self {
            graph_id: file.graph_id,
            editor_state_version: EDITOR_STATE_FILE_VERSION,
            view_state: NodeGraphPureViewState::from(&file.view_state),
            editor_config: file.editor_config.clone(),
        }
    }

    fn validate_for_graph(&self, graph_id: GraphId) -> Result<(), NodeGraphEditorStateFileError> {
        if self.graph_id != graph_id {
            return Err(NodeGraphEditorStateFileError::InconsistentGraphId);
        }
        if self.editor_state_version != EDITOR_STATE_FILE_VERSION {
            return Err(NodeGraphEditorStateFileError::UnsupportedVersion {
                version: self.editor_state_version,
                expected: EDITOR_STATE_FILE_VERSION,
            });
        }
        Ok(())
    }

    fn into_editor_state_file(self) -> NodeGraphEditorStateFile {
        let mut view_state = NodeGraphViewState::from(self.view_state);
        view_state.sanitize_viewport();
        NodeGraphEditorStateFile {
            graph_id: self.graph_id,
            editor_state_version: self.editor_state_version,
            view_state,
            editor_config: self.editor_config,
        }
    }
}