Graphite/editor/src/document/overlays_message_handler.rs

use crate::message_prelude::*;

use graphene::document::Document as GrapheneDocument;
use graphene::layers::style::ViewMode;

#[derive(Debug, Clone, Default)]
pub struct OverlaysMessageHandler {
	pub overlays_graphene_document: GrapheneDocument,
}

impl MessageHandler<OverlaysMessage, bool> for OverlaysMessageHandler {
	#[remain::check]
	fn process_action(&mut self, message: OverlaysMessage, overlays_visible: bool, responses: &mut VecDeque<Message>) {
		use OverlaysMessage::*;

		#[remain::sorted]
		match message {
			// Sub-messages
			#[remain::unsorted]
			DispatchOperation(operation) => match self.overlays_graphene_document.handle_operation(&operation) {
				Ok(_) => responses.push_back(OverlaysMessage::Rerender.into()),
				Err(e) => log::error!("OverlaysError: {:?}", e),
			},

			// Messages
			ClearAllOverlays => todo!(),
			Rerender =>
			// Render overlays
			{
				responses.push_back(
					FrontendMessage::UpdateDocumentOverlays {
						svg: if overlays_visible {
							self.overlays_graphene_document.render_root(ViewMode::Normal)
						} else {
							String::from("")
						},
					}
					.into(),
				)
			}
		}
	}

	fn actions(&self) -> ActionList {
		actions!(OverlaysMessageDiscriminant; ClearAllOverlays)
	}
}