feat: dalle2 (#122)

src-tauri/src/app/cmd.rs

@@ -1,16 +1,23 @@
 use crate::{
+    app::window,
     conf::{ChatConfJson, GITHUB_PROMPTS_CSV_URL},
-    utils::{self, exists},
+    utils,
 };
 use log::info;
 use std::{collections::HashMap, fs, path::PathBuf};
 use tauri::{api, command, AppHandle, Manager};
+use walkdir::WalkDir;
 
 #[command]
 pub fn drag_window(app: AppHandle) {
     app.get_window("core").unwrap().start_dragging().unwrap();
 }
 
+#[command]
+pub fn dalle2_window(app: AppHandle, query: String) {
+    window::dalle2_window(&app.app_handle(), query);
+}
+
 #[command]
 pub fn fullscreen(app: AppHandle) {
     let win = app.get_window("core").unwrap();
@@ -122,9 +129,6 @@ pub fn window_reload(app: AppHandle, label: &str) {
         .unwrap();
 }
 
-use utils::chat_root;
-use walkdir::WalkDir;
-
 #[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
 pub struct ModelRecord {
     pub cmd: String,
@@ -137,7 +141,7 @@ pub struct ModelRecord {
 #[command]
 pub fn cmd_list() -> Vec<ModelRecord> {
     let mut list = vec![];
-    for entry in WalkDir::new(chat_root().join("cache_model"))
+    for entry in WalkDir::new(utils::chat_root().join("cache_model"))
         .into_iter()
         .filter_map(|e| e.ok())
     {
@@ -177,11 +181,13 @@ pub async fn sync_prompts(app: AppHandle, time: u64) -> Option<Vec<ModelRecord>>
 
         let data2 = data.clone();
 
-        let model = chat_root().join("chat.model.json");
-        let model_cmd = chat_root().join("chat.model.cmd.json");
-        let chatgpt_prompts = chat_root().join("cache_model").join("chatgpt_prompts.json");
+        let model = utils::chat_root().join("chat.model.json");
+        let model_cmd = utils::chat_root().join("chat.model.cmd.json");
+        let chatgpt_prompts = utils::chat_root()
+            .join("cache_model")
+            .join("chatgpt_prompts.json");
 
-        if !exists(&model) {
+        if !utils::exists(&model) {
             fs::write(
                 &model,
                 serde_json::json!({

src-tauri/src/app/menu.rs

@@ -261,6 +261,7 @@ pub fn menu_handler(event: WindowMenuEvent<tauri::Wry>) {
         "reload" => win.eval("window.location.reload()").unwrap(),
         "go_back" => win.eval("window.history.go(-1)").unwrap(),
         "go_forward" => win.eval("window.history.go(1)").unwrap(),
+        // core: document.querySelector('main .overflow-y-auto')
         "scroll_top" => win
             .eval(
                 r#"window.scroll({
@@ -308,10 +309,7 @@ pub fn tray_menu() -> SystemTray {
                     "hide_dock_icon".to_string(),
                     "Hide Dock Icon",
                 ))
-                .add_item(CustomMenuItem::new(
-                    "show_core".to_string(),
-                    "Show ChatGPT",
-                ))
+                .add_item(CustomMenuItem::new("show_core".to_string(), "Show ChatGPT"))
                 .add_native_item(SystemTrayMenuItem::Separator)
                 .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")),
         )
@@ -322,10 +320,7 @@ pub fn tray_menu() -> SystemTray {
                     "control_center".to_string(),
                     "Control Center",
                 ))
-                .add_item(CustomMenuItem::new(
-                    "show_core".to_string(),
-                    "Show ChatGPT",
-                ))
+                .add_item(CustomMenuItem::new("show_core".to_string(), "Show ChatGPT"))
                 .add_native_item(SystemTrayMenuItem::Separator)
                 .add_item(CustomMenuItem::new("quit".to_string(), "Quit ChatGPT")),
         )

src-tauri/src/app/setup.rs

@@ -61,9 +61,12 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
                 .always_on_top(chat_conf.stay_on_top)
                 .title_bar_style(ChatConfJson::titlebar())
                 .initialization_script(&utils::user_script())
-                .initialization_script(include_str!("../assets/html2canvas.js"))
-                .initialization_script(include_str!("../assets/jspdf.js"))
+                .initialization_script(include_str!("../vendors/floating-ui-core.js"))
+                .initialization_script(include_str!("../vendors/floating-ui-dom.js"))
+                .initialization_script(include_str!("../vendors/html2canvas.js"))
+                .initialization_script(include_str!("../vendors/jspdf.js"))
                 .initialization_script(include_str!("../assets/core.js"))
+                .initialization_script(include_str!("../assets/dalle2.core.js"))
                 .initialization_script(include_str!("../assets/export.js"))
                 .initialization_script(include_str!("../assets/cmd.js"))
                 .user_agent(&chat_conf.ua_window)
@@ -79,9 +82,12 @@ pub fn init(app: &mut App) -> std::result::Result<(), Box<dyn std::error::Error>
                 .theme(theme)
                 .always_on_top(chat_conf.stay_on_top)
                 .initialization_script(&utils::user_script())
-                .initialization_script(include_str!("../assets/html2canvas.js"))
-                .initialization_script(include_str!("../assets/jspdf.js"))
+                .initialization_script(include_str!("../vendors/floating-ui-core.js"))
+                .initialization_script(include_str!("../vendors/floating-ui-dom.js"))
+                .initialization_script(include_str!("../vendors/html2canvas.js"))
+                .initialization_script(include_str!("../vendors/jspdf.js"))
                 .initialization_script(include_str!("../assets/core.js"))
+                .initialization_script(include_str!("../assets/dalle2.core.js"))
                 .initialization_script(include_str!("../assets/export.js"))
                 .initialization_script(include_str!("../assets/cmd.js"))
                 .user_agent(&chat_conf.ua_window)

src-tauri/src/app/window.rs

@@ -1,4 +1,5 @@
 use crate::{conf, utils};
+use std::time::SystemTime;
 use tauri::{utils::config::WindowUrl, window::WindowBuilder};
 
 pub fn tray_window(handle: &tauri::AppHandle) {
@@ -26,6 +27,33 @@ pub fn tray_window(handle: &tauri::AppHandle) {
     });
 }
 
+pub fn dalle2_window(handle: &tauri::AppHandle, query: String) {
+    let timestamp = SystemTime::now()
+        .duration_since(SystemTime::UNIX_EPOCH)
+        .unwrap()
+        .as_secs();
+    let theme = conf::ChatConfJson::theme();
+    let app = handle.clone();
+
+    tauri::async_runtime::spawn(async move {
+        WindowBuilder::new(&app, format!("dalle2_{}", timestamp), WindowUrl::App("https://labs.openai.com".into()))
+            .title("ChatGPT & DALL·E 2")
+            .resizable(true)
+            .fullscreen(false)
+            .inner_size(800.0, 600.0)
+            .always_on_top(false)
+            .theme(theme)
+            .initialization_script(include_str!("../assets/core.js"))
+            .initialization_script(&format!(
+                "window.addEventListener('DOMContentLoaded', function() {{\nwindow.__CHATGPT_QUERY__='{}';\n}})",
+                query
+            ))
+            .initialization_script(include_str!("../assets/dalle2.js"))
+            .build()
+            .unwrap();
+    });
+}
+
 pub fn control_window(handle: &tauri::AppHandle) {
     let app = handle.clone();
     tauri::async_runtime::spawn(async move {