supabase_config.rs 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. use serde::{Deserialize, Serialize};
  2. use flowy_error::{ErrorCode, FlowyError};
  3. pub const ENABLE_SUPABASE_SYNC: &str = "ENABLE_SUPABASE_SYNC";
  4. pub const SUPABASE_URL: &str = "SUPABASE_URL";
  5. pub const SUPABASE_ANON_KEY: &str = "SUPABASE_ANON_KEY";
  6. pub const SUPABASE_KEY: &str = "SUPABASE_KEY";
  7. pub const SUPABASE_JWT_SECRET: &str = "SUPABASE_JWT_SECRET";
  8. pub const SUPABASE_DB: &str = "SUPABASE_DB";
  9. pub const SUPABASE_DB_USER: &str = "SUPABASE_DB_USER";
  10. pub const SUPABASE_DB_PASSWORD: &str = "SUPABASE_DB_PASSWORD";
  11. pub const SUPABASE_DB_PORT: &str = "SUPABASE_DB_PORT";
  12. #[derive(Debug, Serialize, Deserialize, Clone, Default)]
  13. pub struct SupabaseConfiguration {
  14. /// The url of the supabase server.
  15. pub url: String,
  16. /// The key of the supabase server.
  17. pub key: String,
  18. /// The secret used to sign the JWT tokens.
  19. pub jwt_secret: String,
  20. pub enable_sync: bool,
  21. pub postgres_config: PostgresConfiguration,
  22. }
  23. impl SupabaseConfiguration {
  24. /// Load the configuration from the environment variables.
  25. /// SUPABASE_URL=https://<your-supabase-url>.supabase.co
  26. /// SUPABASE_KEY=<your-supabase-key>
  27. /// SUPABASE_JWT_SECRET=<your-supabase-jwt-secret>
  28. ///
  29. pub fn from_env() -> Result<Self, FlowyError> {
  30. let postgres_config = PostgresConfiguration::from_env()?;
  31. Ok(Self {
  32. enable_sync: std::env::var(ENABLE_SUPABASE_SYNC)
  33. .map(|v| v == "true")
  34. .unwrap_or(false),
  35. url: std::env::var(SUPABASE_URL)
  36. .map_err(|_| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_URL"))?,
  37. key: std::env::var(SUPABASE_KEY)
  38. .map_err(|_| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_KEY"))?,
  39. jwt_secret: std::env::var(SUPABASE_JWT_SECRET).map_err(|_| {
  40. FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_JWT_SECRET")
  41. })?,
  42. postgres_config,
  43. })
  44. }
  45. pub fn write_env(&self) {
  46. if self.enable_sync {
  47. std::env::set_var(ENABLE_SUPABASE_SYNC, "true");
  48. } else {
  49. std::env::set_var(ENABLE_SUPABASE_SYNC, "false");
  50. }
  51. std::env::set_var(SUPABASE_URL, &self.url);
  52. std::env::set_var(SUPABASE_KEY, &self.key);
  53. std::env::set_var(SUPABASE_JWT_SECRET, &self.jwt_secret);
  54. self.postgres_config.write_env();
  55. }
  56. }
  57. #[derive(Debug, Default, Clone, Serialize, Deserialize)]
  58. pub struct PostgresConfiguration {
  59. pub url: String,
  60. pub user_name: String,
  61. pub password: String,
  62. pub port: u16,
  63. }
  64. impl PostgresConfiguration {
  65. pub fn from_env() -> Result<Self, FlowyError> {
  66. let url = std::env::var(SUPABASE_DB)
  67. .map_err(|_| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_DB"))?;
  68. let user_name = std::env::var(SUPABASE_DB_USER)
  69. .map_err(|_| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_DB_USER"))?;
  70. let password = std::env::var(SUPABASE_DB_PASSWORD)
  71. .map_err(|_| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_DB_PASSWORD"))?;
  72. let port = std::env::var(SUPABASE_DB_PORT)
  73. .map_err(|_| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_DB_PORT"))?
  74. .parse::<u16>()
  75. .map_err(|_e| FlowyError::new(ErrorCode::InvalidAuthConfig, "Missing SUPABASE_DB_PORT"))?;
  76. Ok(Self {
  77. url,
  78. user_name,
  79. password,
  80. port,
  81. })
  82. }
  83. pub fn write_env(&self) {
  84. std::env::set_var(SUPABASE_DB, &self.url);
  85. std::env::set_var(SUPABASE_DB_USER, &self.user_name);
  86. std::env::set_var(SUPABASE_DB_PASSWORD, &self.password);
  87. std::env::set_var(SUPABASE_DB_PORT, self.port.to_string());
  88. }
  89. }