AppFlowy/rust-lib/flowy-workspace-infra/src/entities/share/export.rs

use crate::errors::ErrorCode;
use flowy_derive::{ProtoBuf, ProtoBuf_Enum};
use std::convert::TryInto;

#[derive(PartialEq, Debug, ProtoBuf_Enum, Clone)]
pub enum ExportType {
    Text     = 0,
    RichText = 1,
}

impl std::default::Default for ExportType {
    fn default() -> Self { ExportType::Text }
}

impl std::convert::From<i32> for ExportType {
    fn from(val: i32) -> Self {
        match val {
            0 => ExportType::Text,
            1 => ExportType::RichText,
            _ => {
                log::error!("Invalid export type: {}", val);
                ExportType::Text
            },
        }
    }
}

#[derive(Default, ProtoBuf)]
pub struct ExportRequest {
    #[pb(index = 1)]
    pub doc_id: String,

    #[pb(index = 2)]
    pub export_type: ExportType,
}

#[derive(Default)]
pub struct ExportParams {
    pub doc_id: String,
    pub export_type: ExportType,
}

impl TryInto<ExportParams> for ExportRequest {
    type Error = ErrorCode;
    fn try_into(self) -> Result<ExportParams, Self::Error> {
        Ok(ExportParams {
            doc_id: self.doc_id,
            export_type: self.export_type,
        })
    }
}

#[derive(Default, ProtoBuf)]
pub struct ExportData {
    #[pb(index = 1)]
    pub data: String,
}