AppFlowy/frontend/rust-lib/flowy-user/src/entities/user_setting.rs

use std::collections::HashMap;
use std::convert::TryFrom;

use serde::{Deserialize, Serialize};

use flowy_derive::{ProtoBuf, ProtoBuf_Enum};
use flowy_error::FlowyError;
use flowy_server_config::supabase_config::{PostgresConfiguration, SupabaseConfiguration};

#[derive(ProtoBuf, Default, Debug, Clone)]
pub struct UserPreferencesPB {
  #[pb(index = 1)]
  user_id: String,

  #[pb(index = 2)]
  appearance_setting: AppearanceSettingsPB,
}

#[derive(ProtoBuf, Serialize, Deserialize, Debug, Clone)]
pub struct AppearanceSettingsPB {
  #[pb(index = 1)]
  pub theme: String,

  #[pb(index = 2)]
  #[serde(default)]
  pub theme_mode: ThemeModePB,

  #[pb(index = 3)]
  pub font: String,

  #[pb(index = 4)]
  pub monospace_font: String,

  #[pb(index = 5)]
  #[serde(default)]
  pub locale: LocaleSettingsPB,

  #[pb(index = 6)]
  #[serde(default = "DEFAULT_RESET_VALUE")]
  pub reset_to_default: bool,

  #[pb(index = 7)]
  #[serde(default)]
  pub setting_key_value: HashMap<String, String>,

  #[pb(index = 8)]
  #[serde(default)]
  pub is_menu_collapsed: bool,

  #[pb(index = 9)]
  #[serde(default)]
  pub menu_offset: f64,
}

const DEFAULT_RESET_VALUE: fn() -> bool = || APPEARANCE_RESET_AS_DEFAULT;

#[derive(ProtoBuf_Enum, Serialize, Deserialize, Clone, Debug, Default)]
pub enum ThemeModePB {
  Light = 0,
  Dark = 1,
  #[default]
  System = 2,
}

#[derive(ProtoBuf, Serialize, Deserialize, Debug, Clone)]
pub struct LocaleSettingsPB {
  #[pb(index = 1)]
  pub language_code: String,

  #[pb(index = 2)]
  pub country_code: String,
}

impl std::default::Default for LocaleSettingsPB {
  fn default() -> Self {
    Self {
      language_code: "en".to_owned(),
      country_code: "".to_owned(),
    }
  }
}

pub const APPEARANCE_DEFAULT_THEME: &str = "light";
pub const APPEARANCE_DEFAULT_FONT: &str = "Poppins";
pub const APPEARANCE_DEFAULT_MONOSPACE_FONT: &str = "SF Mono";
const APPEARANCE_RESET_AS_DEFAULT: bool = true;
const APPEARANCE_DEFAULT_IS_MENU_COLLAPSED: bool = false;
const APPEARANCE_DEFAULT_MENU_OFFSET: f64 = 0.0;

impl std::default::Default for AppearanceSettingsPB {
  fn default() -> Self {
    AppearanceSettingsPB {
      theme: APPEARANCE_DEFAULT_THEME.to_owned(),
      theme_mode: ThemeModePB::default(),
      font: APPEARANCE_DEFAULT_FONT.to_owned(),
      monospace_font: APPEARANCE_DEFAULT_MONOSPACE_FONT.to_owned(),
      locale: LocaleSettingsPB::default(),
      reset_to_default: APPEARANCE_RESET_AS_DEFAULT,
      setting_key_value: HashMap::default(),
      is_menu_collapsed: APPEARANCE_DEFAULT_IS_MENU_COLLAPSED,
      menu_offset: APPEARANCE_DEFAULT_MENU_OFFSET,
    }
  }
}

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

  #[pb(index = 2)]
  key: String,

  #[pb(index = 3)]
  jwt_secret: String,

  #[pb(index = 4)]
  pub postgres_config: PostgresConfigurationPB,

  #[pb(index = 5)]
  enable_sync: bool,
}

impl TryFrom<SupabaseConfigPB> for SupabaseConfiguration {
  type Error = FlowyError;

  fn try_from(config: SupabaseConfigPB) -> Result<Self, Self::Error> {
    let postgres_config = PostgresConfiguration::try_from(config.postgres_config)?;
    Ok(SupabaseConfiguration {
      url: config.supabase_url,
      key: config.key,
      jwt_secret: config.jwt_secret,
      enable_sync: config.enable_sync,
      postgres_config,
    })
  }
}

impl From<SupabaseConfiguration> for SupabaseConfigPB {
  fn from(value: SupabaseConfiguration) -> Self {
    let postgres_config = PostgresConfigurationPB {
      url: value.postgres_config.url,
      user_name: value.postgres_config.user_name,
      password: value.postgres_config.password,
      port: value.postgres_config.port as u32,
    };
    Self {
      supabase_url: value.url,
      key: value.key,
      jwt_secret: value.jwt_secret,
      postgres_config,
      enable_sync: value.enable_sync,
    }
  }
}

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

  #[pb(index = 2)]
  pub user_name: String,

  #[pb(index = 3)]
  pub password: String,

  #[pb(index = 4)]
  pub port: u32,
}

impl TryFrom<PostgresConfigurationPB> for PostgresConfiguration {
  type Error = FlowyError;

  fn try_from(config: PostgresConfigurationPB) -> Result<Self, Self::Error> {
    Ok(Self {
      url: config.url,
      user_name: config.user_name,
      password: config.password,
      port: config.port as u16,
    })
  }
}