feat: add PostHog custom survey modal on second app launch

apps/desktop/src-tauri/src/commands.rs

@@ -179,6 +179,31 @@ pub async fn set_recently_opened_sessions<R: tauri::Runtime>(
     app.set_recently_opened_sessions(v)
 }
 
+#[tauri::command]
+#[specta::specta]
+pub async fn increment_app_open_count<R: tauri::Runtime>(
+    app: tauri::AppHandle<R>,
+) -> Result<u32, String> {
+    app.increment_app_open_count()
+}
+
+#[tauri::command]
+#[specta::specta]
+pub async fn get_survey_dismissed<R: tauri::Runtime>(
+    app: tauri::AppHandle<R>,
+) -> Result<bool, String> {
+    app.get_survey_dismissed()
+}
+
+#[tauri::command]
+#[specta::specta]
+pub async fn set_survey_dismissed<R: tauri::Runtime>(
+    app: tauri::AppHandle<R>,
+    v: bool,
+) -> Result<(), String> {
+    app.set_survey_dismissed(v)
+}
+
 #[tauri::command]
 #[specta::specta]
 pub async fn list_plugins<R: tauri::Runtime>(

apps/desktop/src-tauri/src/ext.rs

@@ -17,6 +17,12 @@ pub trait AppExt<R: tauri::Runtime> {
 
     fn get_recently_opened_sessions(&self) -> Result<Option<String>, String>;
     fn set_recently_opened_sessions(&self, v: String) -> Result<(), String>;
+
+    fn get_app_open_count(&self) -> Result<u32, String>;
+    fn increment_app_open_count(&self) -> Result<u32, String>;
+
+    fn get_survey_dismissed(&self) -> Result<bool, String>;
+    fn set_survey_dismissed(&self, v: bool) -> Result<(), String>;
 }
 
 impl<R: tauri::Runtime, T: tauri::Manager<R>> AppExt<R> for T {
@@ -111,4 +117,43 @@ impl<R: tauri::Runtime, T: tauri::Manager<R>> AppExt<R> for T {
             .map_err(|e| e.to_string())?;
         store.save().map_err(|e| e.to_string())
     }
+
+    #[tracing::instrument(skip_all)]
+    fn get_app_open_count(&self) -> Result<u32, String> {
+        let store = self.desktop_store()?;
+        store
+            .get(StoreKey::AppOpenCount)
+            .map(|opt: Option<u32>| opt.unwrap_or(0))
+            .map_err(|e| e.to_string())
+    }
+
+    #[tracing::instrument(skip_all)]
+    fn increment_app_open_count(&self) -> Result<u32, String> {
+        let current = self.get_app_open_count()?;
+        let new_count = current + 1;
+        let store = self.desktop_store()?;
+        store
+            .set(StoreKey::AppOpenCount, new_count)
+            .map_err(|e| e.to_string())?;
+        store.save().map_err(|e| e.to_string())?;
+        Ok(new_count)
+    }
+
+    #[tracing::instrument(skip_all)]
+    fn get_survey_dismissed(&self) -> Result<bool, String> {
+        let store = self.desktop_store()?;
+        store
+            .get(StoreKey::SurveyDismissed)
+            .map(|opt| opt.unwrap_or(false))
+            .map_err(|e| e.to_string())
+    }
+
+    #[tracing::instrument(skip_all)]
+    fn set_survey_dismissed(&self, v: bool) -> Result<(), String> {
+        let store = self.desktop_store()?;
+        store
+            .set(StoreKey::SurveyDismissed, v)
+            .map_err(|e| e.to_string())?;
+        store.save().map_err(|e| e.to_string())
+    }
 }

apps/desktop/src-tauri/src/lib.rs

@@ -357,6 +357,9 @@ fn make_specta_builder<R: tauri::Runtime>() -> tauri_specta::Builder<R> {
             commands::set_pinned_tabs::<tauri::Wry>,
             commands::get_recently_opened_sessions::<tauri::Wry>,
             commands::set_recently_opened_sessions::<tauri::Wry>,
+            commands::increment_app_open_count::<tauri::Wry>,
+            commands::get_survey_dismissed::<tauri::Wry>,
+            commands::set_survey_dismissed::<tauri::Wry>,
             commands::list_plugins::<tauri::Wry>,
         ])
         .error_handling(tauri_specta::ErrorHandlingMode::Result)

apps/desktop/src-tauri/src/store.rs

@@ -8,6 +8,8 @@ pub enum StoreKey {
     TinybaseValues,
     PinnedTabs,
     RecentlyOpenedSessions,
+    AppOpenCount,
+    SurveyDismissed,
 }
 
 impl ScopedStoreKey for StoreKey {}

apps/desktop/src/services/event-listeners.tsx

@@ -14,6 +14,7 @@ import {
   getOrCreateSessionForEventId,
 } from "~/store/tinybase/store/sessions";
 import { useTabs } from "~/store/zustand/tabs";
+import { SurveyModal } from "~/survey/survey-modal";
 
 function useUpdaterEvents() {
   const openNew = useTabs((state) => state.openNew);
@@ -145,5 +146,5 @@ export function EventListeners() {
   useUpdaterEvents();
   useNotificationEvents();
 
-  return null;
+  return <SurveyModal />;
 }