From aebc0ffbee53adb4c53acad761085b31d41ba220 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sun, 1 May 2022 23:58:15 -0400
Subject: [PATCH] add plan details to session endpoint

---
 .../endpoints/users/session_info.rs           | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/crates/service/storyteller_web/src/http_server/endpoints/users/session_info.rs b/crates/service/storyteller_web/src/http_server/endpoints/users/session_info.rs
index b13a984319..71e84840b4 100644
--- a/crates/service/storyteller_web/src/http_server/endpoints/users/session_info.rs
+++ b/crates/service/storyteller_web/src/http_server/endpoints/users/session_info.rs
@@ -17,6 +17,20 @@ use sqlx::error::Error::Database;
 use sqlx::mysql::MySqlDatabaseError;
 use std::sync::Arc;
 
+#[derive(Serialize, Copy, Clone)]
+pub enum FakeYouPlan {
+  Free,
+  Basic,
+  Pro,
+}
+
+#[derive(Serialize, Copy, Clone)]
+pub enum StorytellerStreamPlan {
+  Free,
+  Basic,
+  Pro,
+}
+
 #[derive(Serialize)]
 pub struct UserInfo {
   pub user_token: String,
@@ -24,6 +38,10 @@ pub struct UserInfo {
   pub display_name: String,
   pub email_gravatar_hash: String,
 
+  // Premium plans:
+  pub fakeyou_plan: FakeYouPlan,
+  pub storyteller_stream_plan: StorytellerStreamPlan,
+
   // Usage permissions:
   pub can_use_tts: bool,
   pub can_use_w2l: bool,
@@ -106,6 +124,10 @@ pub async fn session_info_handler(
           display_name: session_data.display_name.to_string(),
           email_gravatar_hash: session_data.email_gravatar_hash.to_string(),
 
+          // Premium plans:
+          fakeyou_plan: FakeYouPlan::Free,
+          storyteller_stream_plan: StorytellerStreamPlan::Free,
+
           // Usage permissions:
           can_use_tts: session_data.can_use_tts,
           can_use_w2l: session_data.can_use_w2l,