Compare commits
11 Commits
bt-new-mod
...
webapp-sty
| Author | SHA1 | Date |
|---|---|---|
|
|
8227e33a97 | |
|
|
f41a2b2564 | |
|
|
ba4be8aa6b | |
|
|
93b664d871 | |
|
|
92e6f438c3 | |
|
|
df707bf8e1 | |
|
|
780e4494e9 | |
|
|
c3844acdad | |
|
|
8456e2fd06 | |
|
|
95f8f659a5 | |
|
|
3c5038c159 |
|
|
@ -44,6 +44,7 @@ explicitly authorizes a cache cleanup.
|
|||
- Prefer `use` imports over inline fully-qualified paths; only qualify inline for true one-offs or std collisions
|
||||
- TypeScript with Nx, React, Vite, Zustand, and Three.js
|
||||
- Use two spaces for indentation
|
||||
- Each Tauri command belongs in its own file module, named after the command. Keep shared state, types, and helpers in separate support modules; see the [Storyteller command conventions](crates/desktop/artcraft/src/services/storyteller/commands/AGENTS.md).
|
||||
|
||||
### File Layout
|
||||
|
||||
|
|
|
|||
|
|
@ -262,6 +262,7 @@ version = "0.0.1"
|
|||
dependencies = [
|
||||
"anyhow",
|
||||
"artcraft_api_defs",
|
||||
"artcraft_client_identity",
|
||||
"backoff",
|
||||
"chrono",
|
||||
"cloudflare_errors",
|
||||
|
|
@ -282,6 +283,13 @@ dependencies = [
|
|||
"uuid_utils",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "artcraft_client_identity"
|
||||
version = "0.0.1"
|
||||
dependencies = [
|
||||
"url",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "artcraft_router"
|
||||
version = "0.0.1"
|
||||
|
|
@ -7018,6 +7026,7 @@ dependencies = [
|
|||
name = "tauri-plugin-http"
|
||||
version = "2.4.3"
|
||||
dependencies = [
|
||||
"artcraft_client_identity",
|
||||
"bytes",
|
||||
"cookie_store 0.21.1",
|
||||
"data-url",
|
||||
|
|
|
|||
|
|
@ -16,6 +16,7 @@ members = [
|
|||
"crates/api_clients/worldlabs_api_client",
|
||||
"crates/api_clients/worldlabs_consumer_client",
|
||||
"crates/desktop/artcraft",
|
||||
"crates/lib/artcraft_client_identity",
|
||||
"crates/lib/browser_emulation",
|
||||
"crates/lib/build_metadata",
|
||||
"crates/lib/caching/memory_store",
|
||||
|
|
@ -49,6 +50,7 @@ publish = false
|
|||
[workspace.dependencies]
|
||||
artcraft_api_defs = { path = "crates/api_clients/artcraft/artcraft_api_defs" }
|
||||
artcraft_client = { path = "crates/api_clients/artcraft/artcraft_client" }
|
||||
artcraft_client_identity = { path = "crates/lib/artcraft_client_identity" }
|
||||
artcraft_router = { path = "crates/api_clients/artcraft/artcraft_router" }
|
||||
browser_emulation = { path = "crates/lib/browser_emulation" }
|
||||
cookie_store_wrapper = { path = "crates/lib/cookie_store_wrapper" }
|
||||
|
|
|
|||
|
|
@ -9,21 +9,67 @@ To set up the ArtCraft development environment, install the following:
|
|||
|
||||
1. [Install Rust](https://doc.rust-lang.org/cargo/getting-started/installation.html).
|
||||
2. [Install npm](https://nodejs.org/en/download) or [nvm](https://github.com/nvm-sh/nvm). (Node version `v24.13.0` works at time of writing.)
|
||||
3. [Install nx](https://nx.dev/docs/getting-started/installation). (Nx version `v22.4.5` works at time of writing.)
|
||||
3. The frontend's Nx and Vite dependencies are installed locally with npm; a global Nx installation is not required for the combined Unix launcher.
|
||||
4. [Install Tauri CLI](https://v2.tauri.app/reference/cli/). (Version `tauri-cli 2.10.0` works at time of writing.)
|
||||
|
||||
An easy way to get started with running the app in development is to run the two commands (in separate terminals):
|
||||
|
||||
**Mac and Linux Development**
|
||||
|
||||
```bash
|
||||
# Run the frontend dev server
|
||||
./script/artcraft/unix_frontend_dev.sh
|
||||
# Start the frontend and Rust/Tauri together in one terminal
|
||||
./script/artcraft/unix_dev.sh
|
||||
|
||||
# Run the Tauri Rust application
|
||||
./script/artcraft/unix_rust_dev.sh
|
||||
# Optionally start the free-port search at another port
|
||||
ARTCRAFT_DEV_PORT=6200 ./script/artcraft/unix_dev.sh
|
||||
```
|
||||
|
||||
Use the combined launcher instead of starting the two legacy dev scripts. It
|
||||
works from any working directory when invoked by its path and installs frontend
|
||||
dependencies if the local Vite installation is missing. Node.js 20+ is required.
|
||||
|
||||
The launcher binds Vite to `127.0.0.1`, starting at port **5193** and trying
|
||||
successive ports until one is available. It leaves existing listeners alone.
|
||||
Only after the real frontend socket is bound does it start Tauri with that exact
|
||||
`devUrl`. HTTP and the HMR websocket share the selected port; Rust communicates
|
||||
through native Tauri IPC and needs no separate HTTP port. Configuration overrides
|
||||
are passed in memory, without changing the checked-in Tauri config.
|
||||
|
||||
The native HTTP bridge normalizes loopback development origins for first-party
|
||||
API requests so an automatically selected port does not break login or media
|
||||
loading. See the [desktop session regression checks](../docs/desktop-session-regression.md)
|
||||
when changing ports, the launcher, or authentication.
|
||||
|
||||
JavaScript, TypeScript, React, and CSS changes use Vite hot reload; compatible
|
||||
React component edits use Fast Refresh. Changes that cannot be hot-replaced
|
||||
reload the page. Rust changes use `cargo tauri dev`'s automatic **rebuild and app
|
||||
restart**, including dependent workspace crates. SQLx metadata and SQLite
|
||||
migrations are also watched. Rust reload is a process restart, so it does not
|
||||
preserve unsaved in-memory app state. See [Tauri's watch behavior](https://v2.tauri.app/develop/#reacting-to-source-code-changes).
|
||||
|
||||
After startup, a Vite configuration restart must reuse the chosen port or report
|
||||
an error; it cannot silently switch to another port while Tauri uses an old URL.
|
||||
Ctrl-C stops this session's frontend, Rust watcher/compiler, and desktop app.
|
||||
If the Rust watcher exits, the launcher closes Vite and propagates the exit code.
|
||||
It never finds or terminates another app by name or port.
|
||||
|
||||
The launcher retains the existing Unix defaults for `VITE_ENVIRONMENT_TYPE`,
|
||||
`SQLX_OFFLINE`, `RUSTFLAGS`, and the Linux WebKit workarounds. Explicit environment
|
||||
overrides are preserved. It uses the normal Artcraft app data and Cargo target
|
||||
directory; port separation does not create an isolated account/data profile, and
|
||||
concurrent Rust builds can wait on Cargo's build lock.
|
||||
|
||||
To verify the launcher without compiling Rust or opening a desktop window:
|
||||
|
||||
```bash
|
||||
node --test frontend/scripts/unix-dev.test.mjs
|
||||
```
|
||||
|
||||
The tests use real Vite listeners and HMR, with a test-owned fake Cargo process
|
||||
tree. They cover occupied ports, simultaneous launches, React edits, shutdown,
|
||||
Rust-process failure, and watcher arguments. To verify native Rust reload, run
|
||||
the launcher, wait for the desktop window, make a Rust source edit, and check for
|
||||
a rebuild/restart in the same terminal. Frontend edits should update the webview
|
||||
without restarting the Rust process.
|
||||
|
||||
**Windows Development**
|
||||
|
||||
```powershell
|
||||
|
|
|
|||
|
|
@ -6,6 +6,7 @@ use enums::common::generation::common_bitrate::CommonBitrate;
|
|||
use enums::common::generation::common_quality::CommonQuality;
|
||||
use enums::common::generation::common_resolution::CommonResolution;
|
||||
use enums::common::generation::common_video_model::CommonVideoModel;
|
||||
use enums::common::generation::common_video_output_format::CommonVideoOutputFormat;
|
||||
use tokens::tokens::characters::CharacterToken;
|
||||
use tokens::tokens::media_files::MediaFileToken;
|
||||
|
||||
|
|
@ -55,6 +56,11 @@ pub struct OmniGenVideoCostAndGenerateRequest {
|
|||
/// Not all models support this; models that don't simply ignore it.
|
||||
pub bitrate: Option<CommonBitrate>,
|
||||
|
||||
/// Output container for Seedance 2.5 (including Ultra). Other models ignore it.
|
||||
/// Use "mp4" or "mov" (QuickTime). Omitted or null defaults to MP4.
|
||||
#[serde(rename = "output_format", skip_serializing_if = "Option::is_none")]
|
||||
pub maybe_output_format: Option<CommonVideoOutputFormat>,
|
||||
|
||||
/// The quality to use.
|
||||
pub quality: Option<CommonQuality>,
|
||||
|
||||
|
|
|
|||
|
|
@ -2,6 +2,7 @@ use enums::common::generation::common_aspect_ratio::CommonAspectRatio;
|
|||
use enums::common::generation::common_quality::CommonQuality;
|
||||
use enums::common::generation::common_resolution::CommonResolution;
|
||||
use enums::common::generation::common_video_model::CommonVideoModel;
|
||||
use enums::common::generation::common_video_output_format::CommonVideoOutputFormat;
|
||||
use enums::common::generation::model_creator::ModelCreator;
|
||||
use enums::common::generation_provider::GenerationProvider;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
|
|
@ -156,6 +157,12 @@ pub struct OmniGenVideoModelDetails {
|
|||
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub bitrate_default: Option<CommonBitrate>,
|
||||
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub output_format_options: Option<Vec<CommonVideoOutputFormat>>,
|
||||
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub output_format_default: Option<CommonVideoOutputFormat>,
|
||||
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub quality_options: Option<Vec<CommonQuality>>,
|
||||
|
|
@ -229,6 +236,8 @@ impl Default for OmniGenVideoModelDetails {
|
|||
resolution_default: None,
|
||||
bitrate_options: None,
|
||||
bitrate_default: None,
|
||||
output_format_options: None,
|
||||
output_format_default: None,
|
||||
quality_options: None,
|
||||
default_quality: None,
|
||||
duration_seconds_min: None,
|
||||
|
|
|
|||
|
|
@ -0,0 +1,73 @@
|
|||
use chrono::{DateTime, Utc};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use utoipa::ToSchema;
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct CreateLoginChallengeRequest {}
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct ReviewLoginChallengeRequest {
|
||||
pub approval_token: String,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct DecideLoginChallengeRequest {
|
||||
pub approval_token: String,
|
||||
pub approve: bool,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct PollLoginChallengeRequest {
|
||||
pub device_token: String,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct CreateLoginChallengeResponse {
|
||||
pub success: bool,
|
||||
pub device_token: String,
|
||||
pub verification_url: String,
|
||||
pub confirmation_code: String,
|
||||
pub expires_at: DateTime<Utc>,
|
||||
pub poll_interval_seconds: u32,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct ReviewLoginChallengeResponse {
|
||||
pub success: bool,
|
||||
pub status: LoginChallengeState,
|
||||
pub maybe_failure_type: Option<LoginChallengeFailure>,
|
||||
pub confirmation_code: String,
|
||||
pub requesting_ip: String,
|
||||
pub expires_at: DateTime<Utc>,
|
||||
pub username: String,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, ToSchema)]
|
||||
pub struct LoginChallengeResponse {
|
||||
pub success: bool,
|
||||
pub status: LoginChallengeState,
|
||||
pub maybe_failure_type: Option<LoginChallengeFailure>,
|
||||
/// Present only to the device after the redemption transaction commits.
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub maybe_signed_session: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Copy, Debug, Eq, PartialEq, Deserialize, Serialize, ToSchema)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum LoginChallengeState {
|
||||
Pending,
|
||||
Approved,
|
||||
Redeemed,
|
||||
Failed,
|
||||
#[serde(other)]
|
||||
Unknown,
|
||||
}
|
||||
|
||||
#[derive(Clone, Copy, Debug, Eq, PartialEq, Deserialize, Serialize, ToSchema)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum LoginChallengeFailure {
|
||||
UserDeclined,
|
||||
Expired,
|
||||
#[serde(other)]
|
||||
Unknown,
|
||||
}
|
||||
|
|
@ -2,4 +2,5 @@ pub mod change_password;
|
|||
pub mod edit_email;
|
||||
pub mod edit_username;
|
||||
pub mod login;
|
||||
pub mod session_info;
|
||||
pub mod session_info;
|
||||
pub mod login_challenges;
|
||||
|
|
|
|||
|
|
@ -1,11 +1,12 @@
|
|||
use crate::common::responses::user_details_light::UserDetailsLight;
|
||||
use enums::by_table::users::user_feature_flag::UserFeatureFlag;
|
||||
use serde_derive::Serialize;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
use serde::{Deserialize as DeserializeTrait, Deserializer};
|
||||
use std::collections::BTreeSet;
|
||||
use tokens::tokens::users::UserToken;
|
||||
use utoipa::ToSchema;
|
||||
|
||||
#[derive(Serialize, Copy, Clone, ToSchema)]
|
||||
#[derive(Serialize, Deserialize, Copy, Clone, ToSchema)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum FakeYouPlan {
|
||||
Free,
|
||||
|
|
@ -14,7 +15,7 @@ pub enum FakeYouPlan {
|
|||
Pro,
|
||||
}
|
||||
|
||||
#[derive(Serialize, Copy, Clone, ToSchema)]
|
||||
#[derive(Serialize, Deserialize, Copy, Clone, ToSchema)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum StorytellerStreamPlan {
|
||||
Free,
|
||||
|
|
@ -23,7 +24,7 @@ pub enum StorytellerStreamPlan {
|
|||
Pro,
|
||||
}
|
||||
|
||||
#[derive(Serialize, ToSchema)]
|
||||
#[derive(Serialize, Deserialize, ToSchema)]
|
||||
pub struct SessionUserInfo {
|
||||
pub core_info: UserDetailsLight,
|
||||
|
||||
|
|
@ -41,6 +42,7 @@ pub struct SessionUserInfo {
|
|||
/// Collection of feature / rollout flags
|
||||
/// This is the proper place to detect if a user has access to some rollout (non-paywall) feature.
|
||||
/// NB: The BTreeSet maintains order so React doesn't introduce re-render state bugs when order changes
|
||||
#[serde(default, deserialize_with = "deserialize_known_feature_flags")]
|
||||
pub maybe_feature_flags: BTreeSet<UserFeatureFlag>,
|
||||
|
||||
// Premium plans:
|
||||
|
|
@ -76,7 +78,7 @@ pub struct SessionUserInfo {
|
|||
pub can_delete_users: bool,
|
||||
}
|
||||
|
||||
#[derive(Serialize, ToSchema)]
|
||||
#[derive(Serialize, Deserialize, ToSchema)]
|
||||
pub struct SessionOnboardingState {
|
||||
/// If true, the user hasn't set their email.
|
||||
pub email_not_set: bool,
|
||||
|
|
@ -91,9 +93,18 @@ pub struct SessionOnboardingState {
|
|||
pub username_not_customized: bool,
|
||||
}
|
||||
|
||||
#[derive(Serialize, ToSchema)]
|
||||
#[derive(Serialize, Deserialize, ToSchema)]
|
||||
pub struct SessionInfoSuccessResponse {
|
||||
pub success: bool,
|
||||
pub logged_in: bool,
|
||||
pub user: Option<SessionUserInfo>,
|
||||
}
|
||||
|
||||
// Server rollouts may introduce flags before a desktop release understands them.
|
||||
// Ignore only unknown flag names; malformed values and authentication fields
|
||||
// still fail decoding. Unknown flags never grant a client-side capability.
|
||||
fn deserialize_known_feature_flags<'de, D>(deserializer: D) -> Result<BTreeSet<UserFeatureFlag>, D::Error>
|
||||
where D: Deserializer<'de> {
|
||||
let names = Vec::<String>::deserialize(deserializer)?;
|
||||
Ok(names.into_iter().filter_map(|name| UserFeatureFlag::from_str(&name).ok()).collect())
|
||||
}
|
||||
|
|
|
|||
|
|
@ -0,0 +1,27 @@
|
|||
# artcraft_client
|
||||
|
||||
Rust HTTP endpoint bindings used by the ArtCraft Tauri desktop app.
|
||||
|
||||
## Endpoint layout
|
||||
|
||||
- Each HTTP endpoint binding must live in its own descriptively named `.rs` file.
|
||||
This applies to both free functions and methods implemented on a shared client.
|
||||
Do not combine several routes into a catch-all file such as `login_challenges.rs`
|
||||
or `password_auth.rs`.
|
||||
- Group related bindings in a directory module. Keep `mod.rs` for module declarations;
|
||||
it must not contain endpoint implementations. For example:
|
||||
`endpoints/users/login_challenges/create_login_challenge.rs` and
|
||||
`endpoints/users/login_challenges/poll_login_challenge.rs`.
|
||||
- Keep each endpoint's route constant, request construction, and endpoint-specific
|
||||
request/response types in its file. Keep shared wire types in `artcraft_api_defs`
|
||||
or a dedicated shared type module; do not duplicate them across endpoint files.
|
||||
- Put shared HTTP transport, client setup, and common error handling in a separate
|
||||
utility module. A transport module must not accumulate endpoint bindings.
|
||||
- Update callers to import types from their new modules when splitting bindings.
|
||||
Preserve request methods, paths, headers, error behavior, and cookie handling.
|
||||
|
||||
## Validation
|
||||
|
||||
Run `cargo check --offline -p artcraft_client` for client changes. For native login
|
||||
bindings, also run `SQLX_OFFLINE=true cargo test --offline -p artcraft --lib
|
||||
login_bridge_tests`; these tests use local HTTP fixtures and temporary cookie jars.
|
||||
|
|
@ -19,6 +19,7 @@ path = "src/lib.rs"
|
|||
|
||||
# Internal
|
||||
artcraft_api_defs.workspace = true
|
||||
artcraft_client_identity.workspace = true
|
||||
cloudflare_errors.workspace = true
|
||||
enums.workspace = true
|
||||
errors.workspace = true
|
||||
|
|
|
|||
|
|
@ -29,7 +29,7 @@ pub async fn get_media_file(api_host: &ApiHost, media_file_token: &MediaFileToke
|
|||
|
||||
debug!("Requesting {:?}", &url);
|
||||
|
||||
let response = http_get_anonymous(url).await?;
|
||||
let response = http_get_anonymous(api_host, url).await?;
|
||||
let response = filter_bad_response(response).await?;
|
||||
let response_body = &response.text().await
|
||||
.map_err(|err| StorytellerError::Api(ApiError::from(err)))?;
|
||||
|
|
|
|||
|
|
@ -3,7 +3,8 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use crate::utils::http_get_anonymous::http_get_anonymous;
|
||||
use chrono::{DateTime, Utc};
|
||||
|
|
@ -11,7 +12,6 @@ use enums::common::generation_provider::GenerationProvider;
|
|||
use uuid_utils::uuid::generate_random_uuid;
|
||||
use log::debug;
|
||||
use reqwest::multipart::{Form, Part};
|
||||
use reqwest::Client;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
use std::path::Path;
|
||||
use tokens::tokens::batch_generations::BatchGenerationToken;
|
||||
|
|
@ -56,8 +56,7 @@ pub async fn legacy_upload_media_file_from_file<P: AsRef<Path>>(
|
|||
|
||||
//let mut file = File::open(path)?;
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(args.api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
|
|
@ -86,7 +85,6 @@ pub async fn legacy_upload_media_file_from_file<P: AsRef<Path>>(
|
|||
}
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = args.maybe_creds {
|
||||
|
|
|
|||
|
|
@ -3,7 +3,8 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use crate::utils::http_get_anonymous::http_get_anonymous;
|
||||
use enums::common::generation_provider::GenerationProvider;
|
||||
|
|
@ -11,7 +12,6 @@ use chrono::{DateTime, Utc};
|
|||
use uuid_utils::uuid::generate_random_uuid;
|
||||
use log::debug;
|
||||
use reqwest::multipart::{Form, Part};
|
||||
use reqwest::Client;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
use std::fs::File;
|
||||
use std::io::Read;
|
||||
|
|
@ -72,8 +72,7 @@ pub async fn upload_image_media_file_from_bytes(
|
|||
|
||||
debug!("Requesting {:?}", &url);
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(args.api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
|
|
@ -95,7 +94,6 @@ pub async fn upload_image_media_file_from_bytes(
|
|||
}
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = args.maybe_creds {
|
||||
|
|
|
|||
|
|
@ -3,7 +3,8 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use crate::utils::http_get_anonymous::http_get_anonymous;
|
||||
use chrono::{DateTime, Utc};
|
||||
|
|
@ -11,7 +12,6 @@ use enums::common::generation_provider::GenerationProvider;
|
|||
use uuid_utils::uuid::generate_random_uuid;
|
||||
use log::debug;
|
||||
use reqwest::multipart::{Form, Part};
|
||||
use reqwest::Client;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
use std::path::Path;
|
||||
use tokens::tokens::batch_generations::BatchGenerationToken;
|
||||
|
|
@ -65,8 +65,7 @@ pub async fn upload_image_media_file_from_file_with_idempotency_token<P: AsRef<P
|
|||
|
||||
//let mut file = File::open(path)?;
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(args.api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
|
|
@ -95,7 +94,6 @@ pub async fn upload_image_media_file_from_file_with_idempotency_token<P: AsRef<P
|
|||
}
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = args.maybe_creds {
|
||||
|
|
|
|||
|
|
@ -1,7 +1,8 @@
|
|||
use crate::credentials::storyteller_credential_set::StorytellerCredentialSet;
|
||||
use crate::error::api_error::ApiError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use crate::utils::http_get_anonymous::http_get_anonymous;
|
||||
use enums::common::generation_provider::GenerationProvider;
|
||||
|
|
@ -9,7 +10,6 @@ use chrono::{DateTime, Utc};
|
|||
use uuid_utils::uuid::generate_random_uuid;
|
||||
use log::debug;
|
||||
use reqwest::multipart::{Form, Part};
|
||||
use reqwest::Client;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
use std::path::Path;
|
||||
use tokens::tokens::batch_generations::BatchGenerationToken;
|
||||
|
|
@ -32,8 +32,7 @@ pub async fn upload_new_engine_asset_from_file<P: AsRef<Path>>(
|
|||
|
||||
debug!("Requesting {:?}", &url);
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(api_host)
|
||||
.build()?;
|
||||
|
||||
let file_bytes = std::fs::read(path.as_ref())?;
|
||||
|
|
@ -49,7 +48,6 @@ pub async fn upload_new_engine_asset_from_file<P: AsRef<Path>>(
|
|||
}
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = maybe_creds {
|
||||
|
|
|
|||
|
|
@ -1,7 +1,8 @@
|
|||
use crate::credentials::storyteller_credential_set::StorytellerCredentialSet;
|
||||
use crate::error::api_error::ApiError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use crate::utils::http_get_anonymous::http_get_anonymous;
|
||||
use enums::common::generation_provider::GenerationProvider;
|
||||
|
|
@ -9,7 +10,6 @@ use chrono::{DateTime, Utc};
|
|||
use uuid_utils::uuid::generate_random_uuid;
|
||||
use log::debug;
|
||||
use reqwest::multipart::{Form, Part};
|
||||
use reqwest::Client;
|
||||
use serde_derive::{Deserialize, Serialize};
|
||||
use std::path::Path;
|
||||
use tokens::tokens::batch_generations::BatchGenerationToken;
|
||||
|
|
@ -58,8 +58,7 @@ pub async fn upload_video_media_file_from_file<P: AsRef<Path>>(
|
|||
|
||||
debug!("Requesting {:?}", &url);
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(args.api_host)
|
||||
.build()?;
|
||||
|
||||
let file_bytes = std::fs::read(args.path.as_ref())?;
|
||||
|
|
@ -78,7 +77,6 @@ pub async fn upload_video_media_file_from_file<P: AsRef<Path>>(
|
|||
}
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = args.maybe_creds {
|
||||
|
|
|
|||
|
|
@ -7,3 +7,4 @@ pub mod media_files;
|
|||
pub mod prompts;
|
||||
pub mod stripe_artcraft;
|
||||
pub mod subscriptions;
|
||||
pub mod users;
|
||||
|
|
|
|||
|
|
@ -158,6 +158,10 @@ pub struct OmniGenVideoModelDetails {
|
|||
#[serde(default)]
|
||||
pub bitrate_default: Option<ApiClientBitrate>,
|
||||
#[serde(default)]
|
||||
pub output_format_options: Option<Vec<String>>,
|
||||
#[serde(default)]
|
||||
pub output_format_default: Option<String>,
|
||||
#[serde(default)]
|
||||
pub quality_options: Option<Vec<ApiClientQuality>>,
|
||||
#[serde(default)]
|
||||
pub default_quality: Option<ApiClientQuality>,
|
||||
|
|
|
|||
|
|
@ -0,0 +1,16 @@
|
|||
use artcraft_api_defs::users::login_challenges::{CreateLoginChallengeRequest, CreateLoginChallengeResponse};
|
||||
use reqwest::Method;
|
||||
|
||||
use crate::utils::login_challenge_client::{LoginChallengeClient, LoginChallengeClientError};
|
||||
|
||||
pub const CREATE_PATH: &str = "/v1/login_challenges/create";
|
||||
|
||||
impl LoginChallengeClient {
|
||||
pub async fn create(&self) -> Result<CreateLoginChallengeResponse, LoginChallengeClientError> {
|
||||
let request = self
|
||||
.request(Method::POST, CREATE_PATH)
|
||||
.json(&CreateLoginChallengeRequest {});
|
||||
let (result, _) = self.send(request, CREATE_PATH).await?;
|
||||
Ok(result)
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,2 @@
|
|||
pub mod create_login_challenge;
|
||||
pub mod poll_login_challenge;
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
use artcraft_api_defs::users::login_challenges::{LoginChallengeResponse, PollLoginChallengeRequest};
|
||||
use reqwest::Method;
|
||||
|
||||
use crate::utils::login_challenge_client::{LoginChallengeClient, LoginChallengeClientError};
|
||||
|
||||
pub const POLL_PATH: &str = "/v1/login_challenges/poll";
|
||||
|
||||
// Deliberately no Debug/Serialize: the wire result contains a bearer credential.
|
||||
pub struct PollLoginChallengeResult {
|
||||
pub response: LoginChallengeResponse,
|
||||
pub session_set_cookie: Option<String>,
|
||||
}
|
||||
|
||||
impl LoginChallengeClient {
|
||||
pub async fn poll(
|
||||
&self,
|
||||
device_token: &str,
|
||||
) -> Result<PollLoginChallengeResult, LoginChallengeClientError> {
|
||||
let request = self
|
||||
.request(Method::POST, POLL_PATH)
|
||||
.json(&PollLoginChallengeRequest {
|
||||
device_token: device_token.to_owned(),
|
||||
});
|
||||
let (response, session_set_cookie) = self.send(request, POLL_PATH).await?;
|
||||
Ok(PollLoginChallengeResult {
|
||||
response,
|
||||
session_set_cookie,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,5 @@
|
|||
pub mod login_challenges;
|
||||
pub mod password_auth_response;
|
||||
pub mod password_login;
|
||||
pub mod password_signup;
|
||||
pub mod session_info;
|
||||
|
|
@ -0,0 +1,8 @@
|
|||
use serde::Deserialize;
|
||||
|
||||
// No Debug: the response contains a bearer credential.
|
||||
#[derive(Deserialize)]
|
||||
pub struct PasswordAuthResponse {
|
||||
pub success: bool,
|
||||
pub signed_session: Option<String>,
|
||||
}
|
||||
|
|
@ -0,0 +1,20 @@
|
|||
use reqwest::Method;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use crate::utils::login_challenge_client::{LoginChallengeClient, LoginChallengeClientError};
|
||||
use super::password_auth_response::PasswordAuthResponse;
|
||||
|
||||
pub const LOGIN_PATH: &str = "/v1/login";
|
||||
|
||||
// No Debug: the request contains credentials.
|
||||
#[derive(Deserialize, Serialize)]
|
||||
pub struct PasswordLoginRequest {
|
||||
pub username_or_email: String,
|
||||
pub password: String,
|
||||
}
|
||||
|
||||
impl LoginChallengeClient {
|
||||
pub async fn password_login(&self, request: &PasswordLoginRequest) -> Result<(PasswordAuthResponse, Option<String>), LoginChallengeClientError> {
|
||||
self.send(self.request(Method::POST, LOGIN_PATH).json(request), LOGIN_PATH).await
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,23 @@
|
|||
use reqwest::Method;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use crate::utils::login_challenge_client::{LoginChallengeClient, LoginChallengeClientError};
|
||||
use super::password_auth_response::PasswordAuthResponse;
|
||||
|
||||
pub const SIGNUP_PATH: &str = "/v1/create_account";
|
||||
|
||||
// No Debug: the request contains credentials.
|
||||
#[derive(Deserialize, Serialize)]
|
||||
pub struct PasswordSignupRequest {
|
||||
pub username: String,
|
||||
pub email_address: String,
|
||||
pub password: String,
|
||||
pub password_confirmation: String,
|
||||
pub signup_source: String,
|
||||
}
|
||||
|
||||
impl LoginChallengeClient {
|
||||
pub async fn password_signup(&self, request: &PasswordSignupRequest) -> Result<(PasswordAuthResponse, Option<String>), LoginChallengeClientError> {
|
||||
self.send(self.request(Method::POST, SIGNUP_PATH).json(request), SIGNUP_PATH).await
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,19 @@
|
|||
use artcraft_api_defs::users::session_info::SessionInfoSuccessResponse;
|
||||
use reqwest::Method;
|
||||
|
||||
use crate::utils::login_challenge_client::{LoginChallengeClient, LoginChallengeClientError};
|
||||
|
||||
pub const SESSION_PATH: &str = "/v1/session";
|
||||
|
||||
impl LoginChallengeClient {
|
||||
pub async fn session(
|
||||
&self,
|
||||
signed_session: &str,
|
||||
) -> Result<SessionInfoSuccessResponse, LoginChallengeClientError> {
|
||||
let request = self
|
||||
.request(Method::GET, SESSION_PATH)
|
||||
.header("Cookie", format!("session={signed_session}"));
|
||||
let (result, _) = self.send(request, SESSION_PATH).await?;
|
||||
Ok(result)
|
||||
}
|
||||
}
|
||||
|
|
@ -1,4 +1,6 @@
|
|||
|
||||
use artcraft_client_identity::origins::{ARTCRAFT_DESKTOP_DEVELOPMENT_ORIGIN, ARTCRAFT_DESKTOP_ORIGIN};
|
||||
|
||||
const HTTP_SCHEME: &str = "http";
|
||||
|
||||
const HTTPS_SCHEME: &str = "https";
|
||||
|
|
@ -27,6 +29,15 @@ impl ApiHost {
|
|||
}
|
||||
}
|
||||
|
||||
/// The `Origin` the desktop client declares to this API.
|
||||
pub fn request_origin(&self) -> &'static str {
|
||||
match self {
|
||||
ApiHost::Storyteller => ARTCRAFT_DESKTOP_ORIGIN,
|
||||
ApiHost::FakeYou => ARTCRAFT_DESKTOP_ORIGIN,
|
||||
ApiHost::Localhost { .. } => ARTCRAFT_DESKTOP_DEVELOPMENT_ORIGIN,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn scheme(&self) -> &'static str {
|
||||
match self {
|
||||
ApiHost::Storyteller => HTTPS_SCHEME,
|
||||
|
|
|
|||
|
|
@ -3,10 +3,10 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use log::{debug, info};
|
||||
use reqwest::Client;
|
||||
use serde::de::DeserializeOwned;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
|
|
@ -21,13 +21,11 @@ pub async fn basic_json_delete_request<Req: Serialize, Res: DeserializeOwned>(
|
|||
|
||||
info!("DELETE {:?}", &url);
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
let mut request_builder = client.delete(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON)
|
||||
.header("Content-Type", APPLICATION_JSON);
|
||||
|
||||
|
|
|
|||
|
|
@ -3,10 +3,10 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use log::{debug, info};
|
||||
use reqwest::Client;
|
||||
use serde::de::DeserializeOwned;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
|
|
@ -20,13 +20,11 @@ pub async fn basic_json_get_request<Res: DeserializeOwned>(
|
|||
|
||||
debug!("Requesting {:?}", &url);
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
let mut request_builder = client.get(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = maybe_creds {
|
||||
|
|
|
|||
|
|
@ -3,10 +3,10 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use log::{debug, info};
|
||||
use reqwest::Client;
|
||||
use serde::de::DeserializeOwned;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
|
|
@ -30,13 +30,11 @@ pub async fn basic_json_post_request<Req: Serialize, Res: DeserializeOwned>(
|
|||
info!("Requesting {:?}", &url);
|
||||
}
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON)
|
||||
.header("Content-Type", APPLICATION_JSON);
|
||||
|
||||
|
|
|
|||
|
|
@ -3,11 +3,11 @@ use crate::error::api_error::ApiError;
|
|||
use crate::error::client_error::ClientError;
|
||||
use crate::error::storyteller_error::StorytellerError;
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::constants::{APPLICATION_JSON, USER_AGENT};
|
||||
use crate::utils::constants::APPLICATION_JSON;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
use crate::utils::filter_bad_response::filter_bad_response;
|
||||
use errors::AnyhowResult;
|
||||
use log::{debug, info};
|
||||
use reqwest::Client;
|
||||
use serde::de::DeserializeOwned;
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::collections::HashMap;
|
||||
|
|
@ -26,13 +26,11 @@ pub async fn basic_query_string_post_request<Res: DeserializeOwned>(
|
|||
|
||||
debug!("Requesting {:?}", &url);
|
||||
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
let client = storyteller_client_builder(api_host)
|
||||
.build()
|
||||
.map_err(|err| StorytellerError::Client(ClientError::from(err)))?;
|
||||
|
||||
let mut request_builder = client.post(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", APPLICATION_JSON);
|
||||
|
||||
if let Some(creds) = maybe_creds {
|
||||
|
|
|
|||
|
|
@ -1,4 +1 @@
|
|||
|
||||
pub const USER_AGENT: &str = "storyteller-client/1.0";
|
||||
|
||||
pub const APPLICATION_JSON : &str = "application/json";
|
||||
|
|
|
|||
|
|
@ -1,14 +1,12 @@
|
|||
use crate::error::api_error::ApiError;
|
||||
use reqwest::Client;
|
||||
const USER_AGENT: &str = "storyteller-client/1.0";
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
|
||||
pub async fn http_get_anonymous(url: String) -> Result<reqwest::Response, ApiError> {
|
||||
let client = Client::builder()
|
||||
.gzip(true)
|
||||
pub async fn http_get_anonymous(api_host: &ApiHost, url: String) -> Result<reqwest::Response, ApiError> {
|
||||
let client = storyteller_client_builder(api_host)
|
||||
.build()?;
|
||||
|
||||
let response = client.get(url)
|
||||
.header("User-Agent", USER_AGENT)
|
||||
.header("Accept", "application/json")
|
||||
//.header("Accept-Encoding", "gzip, deflate, br")
|
||||
.send()
|
||||
|
|
|
|||
|
|
@ -0,0 +1,150 @@
|
|||
use std::time::Duration;
|
||||
|
||||
use log::{info, warn};
|
||||
use reqwest::header::SET_COOKIE;
|
||||
use reqwest::{Client, Method, RequestBuilder};
|
||||
use serde::de::DeserializeOwned;
|
||||
use url::Url;
|
||||
|
||||
use crate::utils::api_host::ApiHost;
|
||||
use crate::utils::storyteller_client_builder::storyteller_client_builder;
|
||||
|
||||
/// Native transport for the session bridge. No request/response bodies, cookies,
|
||||
/// or challenge credentials are logged, even at debug level. Never follows redirects.
|
||||
#[derive(Clone)]
|
||||
pub struct LoginChallengeClient {
|
||||
client: Client,
|
||||
api_host: ApiHost,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct LoginChallengeClientError {
|
||||
pub status: Option<u16>,
|
||||
pub message: &'static str,
|
||||
}
|
||||
|
||||
impl LoginChallengeClient {
|
||||
pub fn new(api_host: &ApiHost) -> Result<Self, LoginChallengeClientError> {
|
||||
let mut builder = storyteller_client_builder(api_host);
|
||||
if matches!(api_host, ApiHost::Localhost { .. }) {
|
||||
builder = builder.no_proxy();
|
||||
}
|
||||
let client = builder
|
||||
.timeout(Duration::from_secs(15))
|
||||
.redirect(reqwest::redirect::Policy::none())
|
||||
.build()
|
||||
.map_err(|_| LoginChallengeClientError::invalid("Unable to initialize login client"))?;
|
||||
Ok(Self {
|
||||
client,
|
||||
api_host: api_host.clone(),
|
||||
})
|
||||
}
|
||||
|
||||
pub fn api_origin(&self) -> String {
|
||||
self.api_host.to_api_hostname_and_scheme()
|
||||
}
|
||||
|
||||
pub fn api_url(&self) -> Url {
|
||||
Url::parse(&format!("{}/", self.api_origin())).expect("ApiHost must produce a valid origin")
|
||||
}
|
||||
|
||||
pub fn allows_verification_url(&self, value: &str) -> bool {
|
||||
let Ok(url) = Url::parse(value) else {
|
||||
return false;
|
||||
};
|
||||
let origin_allowed = match self.api_host {
|
||||
ApiHost::Localhost { .. } => ["http://localhost:4200", "http://127.0.0.1:4200", "http://localhost:4201", "http://127.0.0.1:4201"]
|
||||
.contains(&url.origin().ascii_serialization().as_str()),
|
||||
ApiHost::Storyteller => ["https://app.getartcraft.com", "https://getartcraft.com", "https://www.getartcraft.com"]
|
||||
.contains(&url.origin().ascii_serialization().as_str()),
|
||||
ApiHost::FakeYou => false,
|
||||
};
|
||||
origin_allowed
|
||||
&& url.username().is_empty()
|
||||
&& url.password().is_none()
|
||||
&& url.path() == "/login/desktop"
|
||||
&& url.query().is_none()
|
||||
&& url
|
||||
.fragment()
|
||||
.and_then(|f| f.strip_prefix("approval_token="))
|
||||
.map(|s| {
|
||||
s.len() == 43
|
||||
&& s
|
||||
.bytes()
|
||||
.all(|b| b.is_ascii_alphanumeric() || b == b'_' || b == b'-')
|
||||
})
|
||||
.unwrap_or(false)
|
||||
}
|
||||
|
||||
pub(crate) fn request(&self, method: Method, path: &str) -> RequestBuilder {
|
||||
self
|
||||
.client
|
||||
.request(method, format!("{}{path}", self.api_origin()))
|
||||
.header("Accept", "application/json")
|
||||
}
|
||||
|
||||
pub(crate) async fn send<T: DeserializeOwned>(
|
||||
&self,
|
||||
request: RequestBuilder,
|
||||
path: &str,
|
||||
) -> Result<(T, Option<String>), LoginChallengeClientError> {
|
||||
let origin = self.api_origin();
|
||||
info!("Website login request: origin={} path={}", origin, path);
|
||||
let response = request.send().await.map_err(|_| {
|
||||
warn!(
|
||||
"Website login network failure: origin={} path={}",
|
||||
origin, path
|
||||
);
|
||||
LoginChallengeClientError::invalid("Unable to reach the login server")
|
||||
})?;
|
||||
let status = response.status();
|
||||
if !status.is_success() {
|
||||
warn!(
|
||||
"Website login HTTP failure: origin={} path={} status={}",
|
||||
origin,
|
||||
path,
|
||||
status.as_u16()
|
||||
);
|
||||
return Err(LoginChallengeClientError {
|
||||
status: Some(status.as_u16()),
|
||||
message: "Login server rejected the request",
|
||||
});
|
||||
}
|
||||
let session_cookie = response
|
||||
.headers()
|
||||
.get_all(SET_COOKIE)
|
||||
.iter()
|
||||
.filter_map(|v| v.to_str().ok())
|
||||
.find(|v| {
|
||||
cookie::Cookie::parse(*v)
|
||||
.map(|c| c.name() == "session")
|
||||
.unwrap_or(false)
|
||||
})
|
||||
.map(str::to_owned);
|
||||
let body = response.json().await.map_err(|_| {
|
||||
warn!(
|
||||
"Website login invalid response: origin={} path={}",
|
||||
origin, path
|
||||
);
|
||||
LoginChallengeClientError::invalid("Invalid response from login server")
|
||||
})?;
|
||||
Ok((body, session_cookie))
|
||||
}
|
||||
}
|
||||
|
||||
impl LoginChallengeClientError {
|
||||
pub fn invalid(message: &'static str) -> Self {
|
||||
Self {
|
||||
status: None,
|
||||
message,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl std::fmt::Display for LoginChallengeClientError {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
write!(f, "{} (HTTP {:?})", self.message, self.status)
|
||||
}
|
||||
}
|
||||
|
||||
impl std::error::Error for LoginChallengeClientError {}
|
||||
|
|
@ -4,6 +4,8 @@ pub mod basic_json_post_request;
|
|||
pub (crate) mod basic_query_string_post_request;
|
||||
pub (crate) mod filter_bad_response;
|
||||
pub (crate) mod http_get_anonymous;
|
||||
pub (crate) mod storyteller_client_builder;
|
||||
pub mod api_host;
|
||||
pub mod constants;
|
||||
pub mod status_codes;
|
||||
pub mod login_challenge_client;
|
||||
|
|
|
|||
|
|
@ -0,0 +1,17 @@
|
|||
use artcraft_client_identity::user_agents::ARTCRAFT_DESKTOP_USER_AGENT;
|
||||
use reqwest::header::{HeaderMap, HeaderValue, ORIGIN};
|
||||
use reqwest::{Client, ClientBuilder};
|
||||
|
||||
use crate::utils::api_host::ApiHost;
|
||||
|
||||
/// Every request to our API is built from this so the desktop client identifies itself
|
||||
/// consistently (`User-Agent` and `Origin`). Only use it for Storyteller / ArtCraft APIs.
|
||||
pub(crate) fn storyteller_client_builder(api_host: &ApiHost) -> ClientBuilder {
|
||||
let mut headers = HeaderMap::new();
|
||||
headers.insert(ORIGIN, HeaderValue::from_static(api_host.request_origin()));
|
||||
|
||||
Client::builder()
|
||||
.user_agent(ARTCRAFT_DESKTOP_USER_AGENT)
|
||||
.default_headers(headers)
|
||||
.gzip(true)
|
||||
}
|
||||
|
|
@ -1,10 +1,20 @@
|
|||
use serde_derive::{Deserialize, Serialize};
|
||||
|
||||
/// Common quality levels you can specify when enqueuing a generation.
|
||||
/// Not every model will use this; models that don't simply ignore it.
|
||||
/// Not every model will use this; models that don't simply ignore it, and models that only
|
||||
/// offer a subset of tiers clamp to the nearest tier they support.
|
||||
///
|
||||
/// Serialized names mirror the server's `CommonQuality` so the two convert via serde.
|
||||
#[derive(Copy, Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum RouterQuality {
|
||||
/// Let the model choose. Models without a native "auto" use their default tier.
|
||||
Auto,
|
||||
/// Highest tier (GPT Image 2.5 `max`).
|
||||
Max,
|
||||
/// Between High and Max (GPT Image 2.5 `xhigh`).
|
||||
#[serde(rename = "xhigh")]
|
||||
XHigh,
|
||||
High,
|
||||
Medium,
|
||||
Low,
|
||||
|
|
|
|||
|
|
@ -151,8 +151,11 @@ fn to_resolution_enum(r: RouterResolution) -> CommonResolutionEnum {
|
|||
|
||||
fn to_quality_enum(q: RouterQuality) -> CommonQualityEnum {
|
||||
match q {
|
||||
RouterQuality::Low => CommonQualityEnum::Low,
|
||||
RouterQuality::Medium => CommonQualityEnum::Medium,
|
||||
RouterQuality::Auto => CommonQualityEnum::Auto,
|
||||
RouterQuality::Max => CommonQualityEnum::Max,
|
||||
RouterQuality::XHigh => CommonQualityEnum::XHigh,
|
||||
RouterQuality::High => CommonQualityEnum::High,
|
||||
RouterQuality::Medium => CommonQualityEnum::Medium,
|
||||
RouterQuality::Low => CommonQualityEnum::Low,
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -36,7 +36,7 @@ impl ArtcraftGptImage1CostState {
|
|||
}
|
||||
|
||||
pub fn estimate_cost(&self) -> ImageGenerationCostEstimate {
|
||||
let quality = self.quality.unwrap_or(CommonQualityEnum::High);
|
||||
let quality = clamp_to_supported_quality(self.quality);
|
||||
let is_square = is_square_or_none(self.aspect_ratio);
|
||||
|
||||
let output_cost_per_image: u64 = match (quality, is_square) {
|
||||
|
|
@ -45,6 +45,8 @@ impl ArtcraftGptImage1CostState {
|
|||
(CommonQualityEnum::Medium, false) => 7,
|
||||
(CommonQualityEnum::High, true) => 17,
|
||||
(CommonQualityEnum::High, false) => 25,
|
||||
// Clamped away above.
|
||||
(CommonQualityEnum::Auto | CommonQualityEnum::XHigh | CommonQualityEnum::Max, _) => unreachable!(),
|
||||
};
|
||||
|
||||
let output_cost = output_cost_per_image * self.num_images as u64;
|
||||
|
|
@ -63,6 +65,20 @@ impl ArtcraftGptImage1CostState {
|
|||
}
|
||||
}
|
||||
|
||||
/// GPT Image 1 only offers low / medium / high. `auto` prices as the default (high) and the
|
||||
/// higher GPT Image 2.5 tiers clamp down to high, matching the Fal request planner.
|
||||
fn clamp_to_supported_quality(quality: Option<CommonQualityEnum>) -> CommonQualityEnum {
|
||||
match quality {
|
||||
Some(CommonQualityEnum::Low) => CommonQualityEnum::Low,
|
||||
Some(CommonQualityEnum::Medium) => CommonQualityEnum::Medium,
|
||||
Some(CommonQualityEnum::High)
|
||||
| Some(CommonQualityEnum::XHigh)
|
||||
| Some(CommonQualityEnum::Max)
|
||||
| Some(CommonQualityEnum::Auto)
|
||||
| None => CommonQualityEnum::High,
|
||||
}
|
||||
}
|
||||
|
||||
fn is_square_or_none(aspect_ratio: Option<CommonAspectRatioEnum>) -> bool {
|
||||
use CommonAspectRatioEnum as Ar;
|
||||
matches!(
|
||||
|
|
|
|||
|
|
@ -37,7 +37,7 @@ impl ArtcraftGptImage1p5CostState {
|
|||
}
|
||||
|
||||
pub fn estimate_cost(&self) -> ImageGenerationCostEstimate {
|
||||
let quality = self.quality.unwrap_or(CommonQualityEnum::High);
|
||||
let quality = clamp_to_supported_quality(self.quality);
|
||||
let size = size_bucket(self.aspect_ratio);
|
||||
|
||||
let cost_per_image: u64 = match (quality, size) {
|
||||
|
|
@ -50,6 +50,8 @@ impl ArtcraftGptImage1p5CostState {
|
|||
|
||||
(CommonQualityEnum::High, SizeBucket::Square) => 14,
|
||||
(CommonQualityEnum::High, _) => 20,
|
||||
// Clamped away above.
|
||||
(CommonQualityEnum::Auto | CommonQualityEnum::XHigh | CommonQualityEnum::Max, _) => unreachable!(),
|
||||
};
|
||||
|
||||
let cost_in_usd_cents = cost_per_image * self.num_images as u64;
|
||||
|
|
@ -65,6 +67,20 @@ impl ArtcraftGptImage1p5CostState {
|
|||
}
|
||||
}
|
||||
|
||||
/// GPT Image 1.5 only offers low / medium / high. `auto` prices as the default (high) and the
|
||||
/// higher GPT Image 2.5 tiers clamp down to high, matching the Fal request planner.
|
||||
fn clamp_to_supported_quality(quality: Option<CommonQualityEnum>) -> CommonQualityEnum {
|
||||
match quality {
|
||||
Some(CommonQualityEnum::Low) => CommonQualityEnum::Low,
|
||||
Some(CommonQualityEnum::Medium) => CommonQualityEnum::Medium,
|
||||
Some(CommonQualityEnum::High)
|
||||
| Some(CommonQualityEnum::XHigh)
|
||||
| Some(CommonQualityEnum::Max)
|
||||
| Some(CommonQualityEnum::Auto)
|
||||
| None => CommonQualityEnum::High,
|
||||
}
|
||||
}
|
||||
|
||||
fn size_bucket(aspect_ratio: Option<CommonAspectRatioEnum>) -> SizeBucket {
|
||||
use CommonAspectRatioEnum as Ar;
|
||||
match aspect_ratio {
|
||||
|
|
|
|||
|
|
@ -42,7 +42,7 @@ impl ArtcraftGptImage2CostState {
|
|||
}
|
||||
|
||||
pub fn estimate_cost(&self) -> ImageGenerationCostEstimate {
|
||||
let quality = self.quality.unwrap_or(CommonQualityEnum::High);
|
||||
let quality = clamp_to_supported_quality(self.quality);
|
||||
let size = size_bucket(self.aspect_ratio);
|
||||
|
||||
let cost_per_image: u64 = match (quality, size) {
|
||||
|
|
@ -62,6 +62,8 @@ impl ArtcraftGptImage2CostState {
|
|||
| (CommonQualityEnum::High, SizeBucket::Portrait4x3) => 15,
|
||||
(CommonQualityEnum::High, SizeBucket::Landscape16x9)
|
||||
| (CommonQualityEnum::High, SizeBucket::Portrait16x9) => 16,
|
||||
// Clamped away above.
|
||||
(CommonQualityEnum::Auto | CommonQualityEnum::XHigh | CommonQualityEnum::Max, _) => unreachable!(),
|
||||
};
|
||||
|
||||
let cost_in_usd_cents = cost_per_image * self.num_images as u64;
|
||||
|
|
@ -77,6 +79,20 @@ impl ArtcraftGptImage2CostState {
|
|||
}
|
||||
}
|
||||
|
||||
/// GPT Image 2 only offers low / medium / high. `auto` prices as the default (high) and the
|
||||
/// higher GPT Image 2.5 tiers clamp down to high, matching the Fal request planner.
|
||||
fn clamp_to_supported_quality(quality: Option<CommonQualityEnum>) -> CommonQualityEnum {
|
||||
match quality {
|
||||
Some(CommonQualityEnum::Low) => CommonQualityEnum::Low,
|
||||
Some(CommonQualityEnum::Medium) => CommonQualityEnum::Medium,
|
||||
Some(CommonQualityEnum::High)
|
||||
| Some(CommonQualityEnum::XHigh)
|
||||
| Some(CommonQualityEnum::Max)
|
||||
| Some(CommonQualityEnum::Auto)
|
||||
| None => CommonQualityEnum::High,
|
||||
}
|
||||
}
|
||||
|
||||
/// Mirrors v1's `plan_image_size` for GPT Image 2 — see the v1 file
|
||||
/// `plan_generate_image_artcraft_gpt_image_2.rs::plan_image_size`.
|
||||
fn size_bucket(aspect_ratio: Option<CommonAspectRatioEnum>) -> SizeBucket {
|
||||
|
|
|
|||
|
|
@ -8,25 +8,27 @@ use fal_client::requests::api::image::common::gpt_image_2p5::gpt_image_2p5_image
|
|||
use crate::generate::generate_image::image_generation_cost_estimate::ImageGenerationCostEstimate;
|
||||
|
||||
/// Artcraft price per output image, in hundredths of a US cent, by output dimensions and
|
||||
/// quality (`[low, medium, high]`). Rows are stored landscape-first; portrait orientations of
|
||||
/// the same dimensions price identically. Dimensions come from the aspect ratio preset and
|
||||
/// resolution tier the same way the Fal request builder derives them.
|
||||
const OUTPUT_IMAGE_PRICE_TABLE: &[(u32, u32, [u64; 3])] = &[
|
||||
/// quality (`[low, medium, high, xhigh, max]`). Each cell is the `fal_client` GPT Image 2.5
|
||||
/// estimate for that size and tier with a 10% markup, rounded up. Rows are stored
|
||||
/// landscape-first; portrait orientations of the same dimensions price identically.
|
||||
/// Dimensions come from the aspect ratio preset and resolution tier the same way the Fal
|
||||
/// request builder derives them.
|
||||
const OUTPUT_IMAGE_PRICE_TABLE: &[(u32, u32, [u64; 5])] = &[
|
||||
// 1:1
|
||||
(1024, 1024, [ 65, 152, 582]),
|
||||
(2048, 2048, [103, 239, 919]),
|
||||
(2880, 2880, [151, 352, 1357]),
|
||||
(1024, 1024, [ 65, 152, 582, 1034, 2327]),
|
||||
(2048, 2048, [103, 239, 919, 1633, 3672]),
|
||||
(2880, 2880, [151, 352, 1357, 2412, 5426]),
|
||||
// 4:3
|
||||
(1024, 768, [ 46, 105, 400]),
|
||||
(2048, 1536, [ 73, 170, 652]),
|
||||
(3072, 2304, [120, 279, 1073]),
|
||||
(3312, 2480, [134, 311, 1194]),
|
||||
(1024, 768, [ 46, 105, 400, 710, 1597]),
|
||||
(2048, 1536, [ 73, 170, 652, 1159, 2605]),
|
||||
(3072, 2304, [120, 279, 1073, 1906, 4287]),
|
||||
(3312, 2480, [134, 311, 1194, 2122, 4773]),
|
||||
// 16:9
|
||||
(1088, 608, [ 32, 75, 285]),
|
||||
(1920, 1080, [ 49, 114, 436]),
|
||||
(2048, 1152, [ 53, 121, 467]),
|
||||
(3072, 1728, [ 87, 204, 783]),
|
||||
(3840, 2160, [124, 286, 1102]),
|
||||
(1088, 608, [ 32, 75, 285, 506, 1139]),
|
||||
(1920, 1080, [ 49, 114, 436, 775, 1743]),
|
||||
(2048, 1152, [ 53, 121, 467, 830, 1865]),
|
||||
(3072, 1728, [ 87, 204, 783, 1390, 3127]),
|
||||
(3840, 2160, [124, 286, 1102, 1957, 4404]),
|
||||
];
|
||||
|
||||
/// Artcraft price per reference image on an edit request, in hundredths of a US cent.
|
||||
|
|
@ -55,7 +57,8 @@ impl ArtcraftGptImage2p5CostInputs {
|
|||
}
|
||||
|
||||
/// Per-image price rounded up to whole cents, times the batch size. Quality defaults to
|
||||
/// high. Without an aspect ratio, text-to-image prices as landscape 4:3 and edit as auto,
|
||||
/// high, and `auto` is priced as high (the documented default the model falls back to).
|
||||
/// Without an aspect ratio, text-to-image prices as landscape 4:3 and edit as auto,
|
||||
/// matching the Fal defaults the request would fall back to.
|
||||
pub fn estimate_cost(&self) -> ImageGenerationCostEstimate {
|
||||
let quality = self.quality.unwrap_or(CommonQualityEnum::High);
|
||||
|
|
@ -91,11 +94,13 @@ fn output_image_price(quality: CommonQualityEnum, width: u32, height: u32) -> u6
|
|||
match quality {
|
||||
CommonQualityEnum::Low => prices[0],
|
||||
CommonQualityEnum::Medium => prices[1],
|
||||
CommonQualityEnum::High => prices[2],
|
||||
CommonQualityEnum::High | CommonQualityEnum::Auto => prices[2],
|
||||
CommonQualityEnum::XHigh => prices[3],
|
||||
CommonQualityEnum::Max => prices[4],
|
||||
}
|
||||
}
|
||||
|
||||
fn lookup_price_row(width: u32, height: u32) -> Option<[u64; 3]> {
|
||||
fn lookup_price_row(width: u32, height: u32) -> Option<[u64; 5]> {
|
||||
let long_edge = width.max(height);
|
||||
let short_edge = width.min(height);
|
||||
OUTPUT_IMAGE_PRICE_TABLE.iter()
|
||||
|
|
@ -176,6 +181,18 @@ mod tests {
|
|||
(Some(Ar::Auto4k), Some(R::OneK), [2, 3, 12]),
|
||||
];
|
||||
|
||||
// (aspect_ratio, resolution, [xhigh, max]) — cents per image, text-to-image
|
||||
const TEXT_GRID_UPPER_TIERS: &[(Option<Ar>, Option<R>, [u64; 2])] = &[
|
||||
(None, None, [ 8, 16]), // landscape 4:3 default
|
||||
(Some(Ar::Square), None, [11, 24]),
|
||||
(Some(Ar::Square), Some(R::TwoK), [17, 37]),
|
||||
(Some(Ar::Square), Some(R::FourK), [25, 55]),
|
||||
(Some(Ar::WideSixteenByNine), None, [ 8, 18]),
|
||||
(Some(Ar::WideSixteenByNine), Some(R::OneK), [ 6, 12]),
|
||||
(Some(Ar::TallNineBySixteen), Some(R::FourK), [20, 45]),
|
||||
(Some(Ar::Auto), None, [20, 45]),
|
||||
];
|
||||
|
||||
mod text_to_image_tests {
|
||||
use super::*;
|
||||
|
||||
|
|
@ -189,11 +206,32 @@ mod tests {
|
|||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn upper_tier_grid() {
|
||||
for &(aspect_ratio, resolution, expected) in TEXT_GRID_UPPER_TIERS {
|
||||
for (quality, expected) in [Q::XHigh, Q::Max].into_iter().zip(expected) {
|
||||
let actual = cents(make_inputs(0, Some(quality), aspect_ratio, resolution, 1));
|
||||
assert_eq!(actual, expected, "{aspect_ratio:?} {resolution:?} {quality:?}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn quality_defaults_to_high() {
|
||||
assert_eq!(cents(make_inputs(0, None, Some(Ar::Square), None, 1)), 6);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn auto_prices_as_high() {
|
||||
for &(aspect_ratio, resolution, _) in TEXT_GRID {
|
||||
assert_eq!(
|
||||
cents(make_inputs(0, Some(Q::Auto), aspect_ratio, resolution, 1)),
|
||||
cents(make_inputs(0, Some(Q::High), aspect_ratio, resolution, 1)),
|
||||
"{aspect_ratio:?} {resolution:?}",
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn batch_multiplies_per_image_cents() {
|
||||
for &(aspect_ratio, resolution, _) in TEXT_GRID {
|
||||
|
|
@ -265,7 +303,7 @@ mod tests {
|
|||
#[test]
|
||||
fn prices_rise_with_quality_and_size() {
|
||||
for &(_, _, prices) in OUTPUT_IMAGE_PRICE_TABLE {
|
||||
assert!(prices[0] < prices[1] && prices[1] < prices[2], "{prices:?}");
|
||||
assert!(prices.windows(2).all(|pair| pair[0] < pair[1]), "{prices:?}");
|
||||
}
|
||||
assert!(output_image_price(Q::High, 1024, 1024) < output_image_price(Q::High, 2048, 2048));
|
||||
assert!(output_image_price(Q::High, 2048, 2048) < output_image_price(Q::High, 2880, 2880));
|
||||
|
|
|
|||
|
|
@ -100,11 +100,17 @@ fn plan_num_images(
|
|||
}
|
||||
}
|
||||
|
||||
/// This endpoint only offers low / medium / high. `auto` falls back to Fal's default (`high`)
|
||||
/// and the higher GPT Image 2.5 tiers (`xhigh`, `max`) clamp down to `high`.
|
||||
fn plan_quality(quality: Option<RouterQuality>) -> PlannedQuality {
|
||||
match quality {
|
||||
Some(RouterQuality::Low) => PlannedQuality::Low,
|
||||
Some(RouterQuality::Medium) => PlannedQuality::Medium,
|
||||
Some(RouterQuality::High) | None => PlannedQuality::High,
|
||||
Some(RouterQuality::High)
|
||||
| Some(RouterQuality::XHigh)
|
||||
| Some(RouterQuality::Max)
|
||||
| Some(RouterQuality::Auto)
|
||||
| None => PlannedQuality::High,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -337,6 +343,10 @@ mod tests {
|
|||
(Some(RouterQuality::Low), "Some(Low)"),
|
||||
(Some(RouterQuality::Medium), "Some(Medium)"),
|
||||
(Some(RouterQuality::High), "Some(High)"),
|
||||
// Tiers this endpoint lacks clamp to high.
|
||||
(Some(RouterQuality::Auto), "Some(High)"),
|
||||
(Some(RouterQuality::XHigh), "Some(High)"),
|
||||
(Some(RouterQuality::Max), "Some(High)"),
|
||||
];
|
||||
|
||||
for (quality, expected) in cases {
|
||||
|
|
|
|||
|
|
@ -100,11 +100,17 @@ fn plan_num_images(
|
|||
}
|
||||
}
|
||||
|
||||
/// This endpoint only offers low / medium / high. `auto` falls back to Fal's default (`high`)
|
||||
/// and the higher GPT Image 2.5 tiers (`xhigh`, `max`) clamp down to `high`.
|
||||
fn plan_quality(quality: Option<RouterQuality>) -> PlannedQuality {
|
||||
match quality {
|
||||
Some(RouterQuality::Low) => PlannedQuality::Low,
|
||||
Some(RouterQuality::Medium) => PlannedQuality::Medium,
|
||||
Some(RouterQuality::High) | None => PlannedQuality::High,
|
||||
Some(RouterQuality::High)
|
||||
| Some(RouterQuality::XHigh)
|
||||
| Some(RouterQuality::Max)
|
||||
| Some(RouterQuality::Auto)
|
||||
| None => PlannedQuality::High,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -337,6 +343,10 @@ mod tests {
|
|||
(Some(RouterQuality::Low), "Some(Low)"),
|
||||
(Some(RouterQuality::Medium), "Some(Medium)"),
|
||||
(Some(RouterQuality::High), "Some(High)"),
|
||||
// Tiers this endpoint lacks clamp to high.
|
||||
(Some(RouterQuality::Auto), "Some(High)"),
|
||||
(Some(RouterQuality::XHigh), "Some(High)"),
|
||||
(Some(RouterQuality::Max), "Some(High)"),
|
||||
];
|
||||
|
||||
for (quality, expected) in cases {
|
||||
|
|
|
|||
|
|
@ -106,11 +106,17 @@ fn plan_num_images(
|
|||
}
|
||||
}
|
||||
|
||||
/// This endpoint only offers low / medium / high. `auto` falls back to Fal's default (`high`)
|
||||
/// and the higher GPT Image 2.5 tiers (`xhigh`, `max`) clamp down to `high`.
|
||||
fn plan_quality(quality: Option<RouterQuality>) -> PlannedQuality {
|
||||
match quality {
|
||||
Some(RouterQuality::Low) => PlannedQuality::Low,
|
||||
Some(RouterQuality::Medium) => PlannedQuality::Medium,
|
||||
Some(RouterQuality::High) | None => PlannedQuality::High,
|
||||
Some(RouterQuality::High)
|
||||
| Some(RouterQuality::XHigh)
|
||||
| Some(RouterQuality::Max)
|
||||
| Some(RouterQuality::Auto)
|
||||
| None => PlannedQuality::High,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -362,6 +368,10 @@ mod tests {
|
|||
(Some(RouterQuality::Low), "Some(Low)"),
|
||||
(Some(RouterQuality::Medium), "Some(Medium)"),
|
||||
(Some(RouterQuality::High), "Some(High)"),
|
||||
// Tiers this endpoint lacks clamp to high.
|
||||
(Some(RouterQuality::Auto), "Some(High)"),
|
||||
(Some(RouterQuality::XHigh), "Some(High)"),
|
||||
(Some(RouterQuality::Max), "Some(High)"),
|
||||
];
|
||||
|
||||
for (quality, expected) in cases {
|
||||
|
|
|
|||
|
|
@ -82,13 +82,16 @@ fn plan_num_images(
|
|||
}
|
||||
}
|
||||
|
||||
/// The router only exposes low / medium / high; GPT Image 2.5's `xhigh` and `max` tiers are
|
||||
/// not reachable through it. Fal's default (`high`) applies when unspecified.
|
||||
/// GPT Image 2.5 supports every router tier natively. Fal's default (`high`) applies when
|
||||
/// unspecified.
|
||||
fn plan_quality(quality: Option<RouterQuality>) -> GptImage2p5Quality {
|
||||
match quality {
|
||||
Some(RouterQuality::Low) => GptImage2p5Quality::Low,
|
||||
Some(RouterQuality::Medium) => GptImage2p5Quality::Medium,
|
||||
Some(RouterQuality::Auto) => GptImage2p5Quality::Auto,
|
||||
Some(RouterQuality::Max) => GptImage2p5Quality::Max,
|
||||
Some(RouterQuality::XHigh) => GptImage2p5Quality::XHigh,
|
||||
Some(RouterQuality::High) | None => GptImage2p5Quality::High,
|
||||
Some(RouterQuality::Medium) => GptImage2p5Quality::Medium,
|
||||
Some(RouterQuality::Low) => GptImage2p5Quality::Low,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -220,9 +223,12 @@ mod tests {
|
|||
fn quality_maps_exhaustively_for_text_and_edit() {
|
||||
let cases = [
|
||||
(None, "Some(High)"),
|
||||
(Some(RouterQuality::Low), "Some(Low)"),
|
||||
(Some(RouterQuality::Medium), "Some(Medium)"),
|
||||
(Some(RouterQuality::Auto), "Some(Auto)"),
|
||||
(Some(RouterQuality::Max), "Some(Max)"),
|
||||
(Some(RouterQuality::XHigh), "Some(XHigh)"),
|
||||
(Some(RouterQuality::High), "Some(High)"),
|
||||
(Some(RouterQuality::Medium), "Some(Medium)"),
|
||||
(Some(RouterQuality::Low), "Some(Low)"),
|
||||
];
|
||||
for (quality, expected) in cases {
|
||||
let text = unwrap_t2i(plan_gpt_image_2p5_params(&GenerateImageRequestBuilder { quality, ..base_builder() }));
|
||||
|
|
|
|||
|
|
@ -53,6 +53,9 @@ mod tests {
|
|||
(Some(Q::Low), Some(Ar::Square), None, 1, 1),
|
||||
(Some(Q::Medium), Some(Ar::Square), None, 1, 2),
|
||||
(Some(Q::High), Some(Ar::Square), None, 1, 6),
|
||||
(Some(Q::Auto), Some(Ar::Square), None, 1, 6), // auto is priced as high
|
||||
(Some(Q::XHigh), Some(Ar::Square), None, 1, 10),
|
||||
(Some(Q::Max), Some(Ar::Square), None, 1, 22),
|
||||
(Some(Q::High), Some(Ar::Square), None, 4, 24),
|
||||
(Some(Q::High), Some(Ar::SquareHd), None, 1, 9),
|
||||
(Some(Q::High), Some(Ar::Square), Some(R::TwoK), 1, 9),
|
||||
|
|
|
|||
|
|
@ -53,6 +53,9 @@ mod tests {
|
|||
(Some(Q::Low), Some(Ar::Square), None, 1, 1),
|
||||
(Some(Q::Medium), Some(Ar::Square), None, 1, 2),
|
||||
(Some(Q::High), Some(Ar::Square), None, 1, 6),
|
||||
(Some(Q::Auto), Some(Ar::Square), None, 1, 6), // auto is priced as high
|
||||
(Some(Q::XHigh), Some(Ar::Square), None, 1, 10),
|
||||
(Some(Q::Max), Some(Ar::Square), None, 1, 22),
|
||||
(Some(Q::High), Some(Ar::Square), None, 4, 24),
|
||||
(Some(Q::High), Some(Ar::SquareHd), None, 1, 9),
|
||||
(Some(Q::High), Some(Ar::Square), Some(R::TwoK), 1, 9),
|
||||
|
|
|
|||
|
|
@ -122,10 +122,14 @@ pub(crate) fn plan_quality(quality: Option<RouterQuality>) -> Option<GenerateMid
|
|||
// Kinovi pricing — pass through whatever the caller asked for, and let
|
||||
// `None` ride Midjourney's server-side default.
|
||||
match quality {
|
||||
None => None,
|
||||
// `auto` has no Midjourney equivalent; leave it to the provider default.
|
||||
None | Some(RouterQuality::Auto) => None,
|
||||
Some(RouterQuality::Low) => Some(GenerateMidjourneyV7Quality::Quarter),
|
||||
Some(RouterQuality::Medium) => Some(GenerateMidjourneyV7Quality::Half),
|
||||
Some(RouterQuality::High) => Some(GenerateMidjourneyV7Quality::Full),
|
||||
// Midjourney tops out at full quality; the higher GPT Image tiers clamp to it.
|
||||
Some(RouterQuality::High)
|
||||
| Some(RouterQuality::XHigh)
|
||||
| Some(RouterQuality::Max) => Some(GenerateMidjourneyV7Quality::Full),
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -110,10 +110,14 @@ pub(crate) fn plan_aspect_ratio(
|
|||
|
||||
pub(crate) fn plan_quality(quality: Option<RouterQuality>) -> Option<GenerateMidjourneyV7NijiQuality> {
|
||||
match quality {
|
||||
None => None,
|
||||
// `auto` has no Midjourney equivalent; leave it to the provider default.
|
||||
None | Some(RouterQuality::Auto) => None,
|
||||
Some(RouterQuality::Low) => Some(GenerateMidjourneyV7NijiQuality::Quarter),
|
||||
Some(RouterQuality::Medium) => Some(GenerateMidjourneyV7NijiQuality::Half),
|
||||
Some(RouterQuality::High) => Some(GenerateMidjourneyV7NijiQuality::Full),
|
||||
// Midjourney tops out at full quality; the higher GPT Image tiers clamp to it.
|
||||
Some(RouterQuality::High)
|
||||
| Some(RouterQuality::XHigh)
|
||||
| Some(RouterQuality::Max) => Some(GenerateMidjourneyV7NijiQuality::Full),
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -110,10 +110,14 @@ pub(crate) fn plan_aspect_ratio(
|
|||
|
||||
pub(crate) fn plan_quality(quality: Option<RouterQuality>) -> Option<GenerateMidjourneyV8Quality> {
|
||||
match quality {
|
||||
None => None,
|
||||
// `auto` has no Midjourney equivalent; leave it to the provider default.
|
||||
None | Some(RouterQuality::Auto) => None,
|
||||
Some(RouterQuality::Low) => Some(GenerateMidjourneyV8Quality::Quarter),
|
||||
Some(RouterQuality::Medium) => Some(GenerateMidjourneyV8Quality::Half),
|
||||
Some(RouterQuality::High) => Some(GenerateMidjourneyV8Quality::Full),
|
||||
// Midjourney tops out at full quality; the higher GPT Image tiers clamp to it.
|
||||
Some(RouterQuality::High)
|
||||
| Some(RouterQuality::XHigh)
|
||||
| Some(RouterQuality::Max) => Some(GenerateMidjourneyV8Quality::Full),
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -77,6 +77,7 @@ pub fn build_artcraft_omni_video_request(
|
|||
resolution,
|
||||
aspect_ratio,
|
||||
bitrate,
|
||||
maybe_output_format: None,
|
||||
duration_seconds: duration_seconds.map(|d| d as u16),
|
||||
video_batch_count: Some(batch_count),
|
||||
negative_prompt: None,
|
||||
|
|
|
|||
|
|
@ -59,6 +59,7 @@ mod tests {
|
|||
resolution: None,
|
||||
aspect_ratio: None,
|
||||
bitrate: None,
|
||||
maybe_output_format: None,
|
||||
quality: None,
|
||||
duration_seconds: Some(5),
|
||||
video_batch_count: Some(1),
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
# Don't restart the dev server on Rust changes
|
||||
crates/
|
||||
Cargo.toml
|
||||
Cargo.lock
|
||||
# Rust sources and Cargo manifests must remain watched for `cargo tauri dev`.
|
||||
# Vite owns frontend reloads; build artifacts must not trigger Rust rebuilds.
|
||||
target/
|
||||
node_modules/
|
||||
dist/
|
||||
|
|
|
|||
|
|
@ -5,6 +5,10 @@ use artcraft_router::api::router_quality::RouterQuality;
|
|||
#[derive(Copy, Clone, Debug, Serialize, Deserialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum CommonQuality {
|
||||
Auto,
|
||||
Max,
|
||||
#[serde(rename = "xhigh")]
|
||||
XHigh,
|
||||
High,
|
||||
Medium,
|
||||
Low,
|
||||
|
|
@ -13,6 +17,9 @@ pub enum CommonQuality {
|
|||
impl CommonQuality {
|
||||
pub fn to_artcraft_router_type(&self) -> RouterQuality {
|
||||
match self {
|
||||
CommonQuality::Auto => RouterQuality::Auto,
|
||||
CommonQuality::Max => RouterQuality::Max,
|
||||
CommonQuality::XHigh => RouterQuality::XHigh,
|
||||
CommonQuality::High => RouterQuality::High,
|
||||
CommonQuality::Medium => RouterQuality::Medium,
|
||||
CommonQuality::Low => RouterQuality::Low,
|
||||
|
|
|
|||
|
|
@ -30,6 +30,9 @@ pub fn convert_aspect_ratio(value: EnumsAspectRatio) -> RouterAspectRatio {
|
|||
|
||||
pub fn convert_quality(value: EnumsQuality) -> RouterQuality {
|
||||
match value {
|
||||
EnumsQuality::Auto => RouterQuality::Auto,
|
||||
EnumsQuality::Max => RouterQuality::Max,
|
||||
EnumsQuality::XHigh => RouterQuality::XHigh,
|
||||
EnumsQuality::High => RouterQuality::High,
|
||||
EnumsQuality::Medium => RouterQuality::Medium,
|
||||
EnumsQuality::Low => RouterQuality::Low,
|
||||
|
|
|
|||
|
|
@ -552,3 +552,25 @@ impl From<OmniGenImageModelDetails> for ListImageModelsModelDetails {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod catalog_parity_tests {
|
||||
use super::*;
|
||||
use serde_json::json;
|
||||
|
||||
#[test]
|
||||
fn gpt_image_2p5_quality_tiers_survive_the_desktop_listing_boundary() {
|
||||
for model in ["gpt_image_2p5_flare", "gpt_image_2p5_sunburst"] {
|
||||
let source = json!({
|
||||
"model": model,
|
||||
"quality_options": ["auto", "max", "xhigh", "high", "medium", "low", "future_quality"],
|
||||
"default_quality": "high",
|
||||
"resolution_options": ["one_k", "two_k", "three_k", "four_k"],
|
||||
"resolution_default": "one_k"
|
||||
});
|
||||
let client: OmniGenImageModelDetails = serde_json::from_value(source.clone()).unwrap();
|
||||
let desktop = serde_json::to_value(ListImageModelsModelDetails::from(client)).unwrap();
|
||||
assert_eq!(desktop, source);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -597,6 +597,10 @@ pub struct ListVideoModelsModelDetails {
|
|||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub bitrate_default: Option<ListVideoModelsBitrate>,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub output_format_options: Option<Vec<String>>,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub output_format_default: Option<String>,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub quality_options: Option<Vec<ListVideoModelsQuality>>,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub default_quality: Option<ListVideoModelsQuality>,
|
||||
|
|
@ -655,6 +659,8 @@ impl From<OmniGenVideoModelDetails> for ListVideoModelsModelDetails {
|
|||
resolution_default: v.resolution_default.map(Into::into),
|
||||
bitrate_options: v.bitrate_options.map(|items| items.into_iter().map(Into::into).collect()),
|
||||
bitrate_default: v.bitrate_default.map(Into::into),
|
||||
output_format_options: v.output_format_options,
|
||||
output_format_default: v.output_format_default,
|
||||
quality_options: v.quality_options.map(|items| items.into_iter().map(Into::into).collect()),
|
||||
default_quality: v.default_quality.map(Into::into),
|
||||
duration_seconds_min: v.duration_seconds_min,
|
||||
|
|
@ -670,3 +676,26 @@ impl From<OmniGenVideoModelDetails> for ListVideoModelsModelDetails {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod catalog_parity_tests {
|
||||
use super::*;
|
||||
use serde_json::json;
|
||||
|
||||
#[test]
|
||||
fn wan_and_output_formats_survive_the_desktop_listing_boundary() {
|
||||
for model in ["wan_3p0", "wan_3p0_prime", "seedance_2p5", "seedance_2p5_u", "future_video"] {
|
||||
let source = json!({
|
||||
"model": model,
|
||||
"output_format_options": ["mp4", "mov", "future_container"],
|
||||
"output_format_default": "mp4",
|
||||
"batch_size_options": [1, 2, 3, 4],
|
||||
"duration_seconds_max": 30,
|
||||
"video_references_max": 5
|
||||
});
|
||||
let client: OmniGenVideoModelDetails = serde_json::from_value(source.clone()).unwrap();
|
||||
let desktop = serde_json::to_value(ListVideoModelsModelDetails::from(client)).unwrap();
|
||||
assert_eq!(desktop, source);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -123,6 +123,31 @@ mod tests {
|
|||
use super::*;
|
||||
use serde_json::json;
|
||||
|
||||
#[test]
|
||||
fn catalog_parity_new_models_and_options_reach_the_api_unchanged() {
|
||||
for model in ["gpt_image_2p5_flare", "gpt_image_2p5_sunburst"] {
|
||||
for quality in ["auto", "max", "xhigh", "high", "medium", "low"] {
|
||||
let source = json!({"model": model, "quality": quality, "resolution": "four_k"});
|
||||
let request: OmniRequest = serde_json::from_value(source.clone()).unwrap();
|
||||
assert!(request.uses_artcraft());
|
||||
assert!(!request.uses_legacy_image_endpoint());
|
||||
assert_eq!(Value::Object(request.api_fields(Modality::Image)), source);
|
||||
}
|
||||
}
|
||||
for model in ["wan_3p0", "wan_3p0_prime", "seedance_2p5", "seedance_2p5_u"] {
|
||||
let source = json!({
|
||||
"model": model, "duration_seconds": 15, "resolution": "ten_eighty_p",
|
||||
"output_format": "mov", "generate_audio": true,
|
||||
"reference_image_media_tokens": ["mf_image"],
|
||||
"reference_video_media_tokens": ["mf_video"],
|
||||
"reference_audio_media_tokens": ["mf_audio"]
|
||||
});
|
||||
let request: OmniRequest = serde_json::from_value(source.clone()).unwrap();
|
||||
assert!(request.uses_artcraft());
|
||||
assert_eq!(Value::Object(request.api_fields(Modality::Video)), source);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn future_model_and_options_survive_without_desktop_metadata() {
|
||||
let request: OmniRequest = serde_json::from_value(json!({
|
||||
|
|
|
|||
|
|
@ -1,10 +1,10 @@
|
|||
use crate::core::state::app_env_configs::app_env_configs::AppEnvConfigs;
|
||||
use crate::core::state::app_startup_time::AppStartupTime;
|
||||
use crate::services::storyteller::state::storyteller_credential_manager::StorytellerCredentialManager;
|
||||
use anyhow::anyhow;
|
||||
use chrono::TimeDelta;
|
||||
use errors::AnyhowResult;
|
||||
use log::{error, info};
|
||||
use once_cell::sync::Lazy;
|
||||
use reqwest::Url;
|
||||
use artcraft_client::credentials::storyteller_avt_cookie::StorytellerAvtCookie;
|
||||
use artcraft_client::credentials::storyteller_credential_set::StorytellerCredentialSet;
|
||||
|
|
@ -13,24 +13,13 @@ use tauri::{AppHandle, Manager};
|
|||
use tauri_plugin_http::reqwest_cookie_store::CookieStore;
|
||||
use tauri_plugin_http::Http;
|
||||
|
||||
const STORYTELLER_ROOT_COOKIE_URL_STR: &str = "https://api.storyteller.ai/";
|
||||
|
||||
static STORYTELLER_ROOT_COOKIE_URL: Lazy<Url> = Lazy::new(|| {
|
||||
Url::parse(STORYTELLER_ROOT_COOKIE_URL_STR).expect("URL should parse")
|
||||
});
|
||||
|
||||
const AVT_COOKIE_NAME : &str = "visitor";
|
||||
const SESSION_COOKIE_NAME : &str = "session";
|
||||
const AVT_COOKIE_NAME: &str = "visitor";
|
||||
const SESSION_COOKIE_NAME: &str = "session";
|
||||
|
||||
/// There's some kind of race condition that we need to wait for before inspecting cookies.
|
||||
const RACE_CONDITION_WAIT_TIME: TimeDelta = TimeDelta::milliseconds(5000);
|
||||
|
||||
pub async fn persist_storyteller_cookies_task(
|
||||
app: &AppHandle,
|
||||
storyteller_credential_manager: &StorytellerCredentialManager,
|
||||
app_startup_time: &AppStartupTime,
|
||||
) -> AnyhowResult<()> {
|
||||
|
||||
pub async fn persist_storyteller_cookies_task(app: &AppHandle, storyteller_credential_manager: &StorytellerCredentialManager, app_startup_time: &AppStartupTime) -> AnyhowResult<()> {
|
||||
//if app_startup_time.time_delta_since() < RACE_CONDITION_WAIT_TIME {
|
||||
// // NB: There's an issue when the "main window thread" inquires about the
|
||||
// // webview cookies shortly after app startup. When it attempts to dump the
|
||||
|
|
@ -41,33 +30,30 @@ pub async fn persist_storyteller_cookies_task(
|
|||
//}
|
||||
|
||||
let maybe_http = app.try_state::<Http>();
|
||||
|
||||
|
||||
let http = match maybe_http {
|
||||
None => {
|
||||
error!("No HTTP plugin found");
|
||||
return Err(anyhow!("No HTTP plugin found"));
|
||||
}
|
||||
},
|
||||
Some(http) => http,
|
||||
};
|
||||
|
||||
|
||||
match http.cookies_jar.store.lock() {
|
||||
Err(err) => {
|
||||
error!("Failed to lock cookie jar: {:?}", err);
|
||||
return Err(anyhow!("Failed to lock cookie jar"));
|
||||
}
|
||||
},
|
||||
Ok(cookie_jar) => {
|
||||
sync_tauri_credentials(&cookie_jar, storyteller_credential_manager)?;
|
||||
}
|
||||
sync_tauri_credentials(&cookie_jar, storyteller_credential_manager, &storyteller_cookie_url(app)?)?;
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn sync_tauri_credentials(
|
||||
cookie_store: &CookieStore,
|
||||
storyteller_credential_manager: &StorytellerCredentialManager,
|
||||
) -> AnyhowResult<()> {
|
||||
let current_http_plugin_credentials = get_credentials_from_cookie_store(cookie_store)?;
|
||||
pub(crate) fn sync_tauri_credentials(cookie_store: &CookieStore, storyteller_credential_manager: &StorytellerCredentialManager, api_url: &Url) -> AnyhowResult<()> {
|
||||
let current_http_plugin_credentials = get_credentials_from_cookie_store(cookie_store, api_url)?;
|
||||
|
||||
let mut replace_credentials = true;
|
||||
|
||||
|
|
@ -78,31 +64,80 @@ fn sync_tauri_credentials(
|
|||
replace_credentials = false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
if replace_credentials {
|
||||
info!("Syncing ArtCraft credentials ...");
|
||||
storyteller_credential_manager.set_credentials(¤t_http_plugin_credentials)?;
|
||||
// NB: tauri-plugin-http stores the credentials on disk, so we can defer to that for now.
|
||||
//storyteller_credential_manager.persist_all_to_disk()?;
|
||||
}
|
||||
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn get_credentials_from_cookie_store(cookie_jar: &CookieStore) -> AnyhowResult<StorytellerCredentialSet> {
|
||||
pub fn get_credentials_from_cookie_store(cookie_jar: &CookieStore, api_url: &Url) -> AnyhowResult<StorytellerCredentialSet> {
|
||||
let mut avt_cookie = None;
|
||||
let mut session_cookie = None;
|
||||
|
||||
for cookie in cookie_jar.iter_unexpired() {
|
||||
if let Some(avt) = StorytellerAvtCookie::maybe_from_cookie(&cookie) {
|
||||
avt_cookie = Some(avt);
|
||||
} else if let Some(session) = StorytellerSessionCookie::maybe_from_cookie(&cookie) {
|
||||
session_cookie = Some(session);
|
||||
// Only cookies applicable to our API origin can become ArtCraft credentials.
|
||||
for (name, value) in cookie_jar.get_request_values(api_url) {
|
||||
match name {
|
||||
AVT_COOKIE_NAME => avt_cookie = Some(StorytellerAvtCookie::new(value.to_owned())),
|
||||
SESSION_COOKIE_NAME => session_cookie = Some(StorytellerSessionCookie::new(value.to_owned())),
|
||||
_ => {},
|
||||
}
|
||||
}
|
||||
|
||||
Ok(StorytellerCredentialSet::initialize(
|
||||
avt_cookie,
|
||||
session_cookie,
|
||||
))
|
||||
Ok(StorytellerCredentialSet::initialize(avt_cookie, session_cookie))
|
||||
}
|
||||
|
||||
pub(crate) fn storyteller_cookie_url(app: &AppHandle) -> AnyhowResult<Url> {
|
||||
let config = app.try_state::<AppEnvConfigs>().ok_or_else(|| anyhow!("API configuration unavailable"))?;
|
||||
Ok(Url::parse(&format!("{}/", config.storyteller_host.to_api_hostname_and_scheme()))?)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod login_bridge_tests {
|
||||
use super::*;
|
||||
use crate::core::state::data_dir::app_data_root::AppDataRoot;
|
||||
use tauri_plugin_http::reqwest_cookie_store::RawCookie;
|
||||
|
||||
#[test]
|
||||
fn approved_response_cookie_reaches_native_credentials() {
|
||||
let directory = tempfile::tempdir().unwrap();
|
||||
let root = AppDataRoot::create_existing(directory.path()).unwrap();
|
||||
let manager = StorytellerCredentialManager::initialize_empty(&root);
|
||||
let mut jar = CookieStore::default();
|
||||
let url = Url::parse("https://api.storyteller.ai/v1/login_challenges/poll").unwrap();
|
||||
jar.store_response_cookies([RawCookie::parse("session=signed_backend_cookie; Secure; HttpOnly; Path=/; Max-Age=3600").unwrap().into_owned(), RawCookie::parse("visitor=existing_visitor; Secure; Path=/; Max-Age=3600").unwrap().into_owned()].into_iter(), &url);
|
||||
sync_tauri_credentials(&jar, &manager, &url).unwrap();
|
||||
let credentials = manager.get_credentials_required().unwrap();
|
||||
assert_eq!(credentials.session.unwrap().as_str(), "signed_backend_cookie");
|
||||
assert_eq!(credentials.avt.unwrap().as_str(), "existing_visitor");
|
||||
// Retrying the same successful poll is idempotent for native credentials too.
|
||||
sync_tauri_credentials(&jar, &manager, &url).unwrap();
|
||||
assert_eq!(manager.get_credentials_required().unwrap().session.unwrap().as_str(), "signed_backend_cookie");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unrelated_and_expired_cookies_cannot_become_artcraft_sessions() {
|
||||
let url = Url::parse("https://api.storyteller.ai/").unwrap();
|
||||
let mut jar = CookieStore::default();
|
||||
jar.store_response_cookies([RawCookie::parse("session=unrelated; Path=/; Max-Age=3600").unwrap().into_owned()].into_iter(), &Url::parse("https://unrelated.example/").unwrap());
|
||||
assert!(get_credentials_from_cookie_store(&jar, &url).unwrap().session.is_none());
|
||||
jar.store_response_cookies([RawCookie::parse("session=expired; Secure; Path=/; Max-Age=0").unwrap().into_owned()].into_iter(), &url);
|
||||
assert!(get_credentials_from_cookie_store(&jar, &url).unwrap().session.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn configured_local_api_does_not_import_production_cookies() {
|
||||
let production = Url::parse("https://api.storyteller.ai/").unwrap();
|
||||
let local = Url::parse("http://localhost:12345/").unwrap();
|
||||
let mut jar = CookieStore::default();
|
||||
jar.store_response_cookies([RawCookie::parse("session=production; Secure; Path=/").unwrap().into_owned()].into_iter(), &production);
|
||||
assert!(get_credentials_from_cookie_store(&jar, &local).unwrap().session.is_none());
|
||||
jar.store_response_cookies([RawCookie::parse("session=development; Path=/").unwrap().into_owned()].into_iter(), &local);
|
||||
assert_eq!(get_credentials_from_cookie_store(&jar, &local).unwrap().session.unwrap().as_str(), "development");
|
||||
assert_eq!(get_credentials_from_cookie_store(&jar, &production).unwrap().session.unwrap().as_str(), "production");
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -63,6 +63,13 @@ use crate::services::sora::commands::sora_logout_command::sora_logout_command;
|
|||
use crate::services::sora::state::sora_credential_manager::SoraCredentialManager;
|
||||
use crate::services::sora::state::sora_task_queue::SoraTaskQueue;
|
||||
use crate::services::sora::threads::sora_task_polling::sora_task_polling_thread::sora_task_polling_thread;
|
||||
use crate::services::storyteller::commands::login_bridge::DesktopLoginBridgeState;
|
||||
use crate::services::storyteller::commands::storyteller_cancel_login_challenge_command::storyteller_cancel_login_challenge_command;
|
||||
use crate::services::storyteller::commands::storyteller_create_login_challenge_command::storyteller_create_login_challenge_command;
|
||||
use crate::services::storyteller::commands::storyteller_get_login_session_command::storyteller_get_login_session_command;
|
||||
use crate::services::storyteller::commands::storyteller_password_login_command::storyteller_password_login_command;
|
||||
use crate::services::storyteller::commands::storyteller_password_signup_command::storyteller_password_signup_command;
|
||||
use crate::services::storyteller::commands::storyteller_poll_login_challenge_command::storyteller_poll_login_challenge_command;
|
||||
use crate::services::storyteller::commands::storyteller_get_credits_command::storyteller_get_credits_command;
|
||||
use crate::services::storyteller::commands::storyteller_get_subscription_command::storyteller_get_subscription_command;
|
||||
use crate::services::storyteller::commands::storyteller_purge_credentials_command::storyteller_purge_credentials_command;
|
||||
|
|
@ -201,6 +208,7 @@ pub fn run() {
|
|||
})
|
||||
.manage(app_data_root)
|
||||
.manage(app_env_configs)
|
||||
.manage(DesktopLoginBridgeState::default())
|
||||
.manage(app_preferences)
|
||||
.manage(artcraft_platform_info)
|
||||
.manage(artcraft_usage_tracker)
|
||||
|
|
@ -259,6 +267,12 @@ pub fn run() {
|
|||
set_provider_order_command,
|
||||
sora_get_credential_info_command,
|
||||
sora_logout_command,
|
||||
storyteller_get_login_session_command,
|
||||
storyteller_password_login_command,
|
||||
storyteller_password_signup_command,
|
||||
storyteller_create_login_challenge_command,
|
||||
storyteller_poll_login_challenge_command,
|
||||
storyteller_cancel_login_challenge_command,
|
||||
storyteller_get_credits_command,
|
||||
storyteller_get_subscription_command,
|
||||
storyteller_open_credits_purchase_command,
|
||||
|
|
|
|||
|
|
@ -0,0 +1,17 @@
|
|||
# Tauri Commands
|
||||
|
||||
- Put each `#[tauri::command]` function in its own `.rs` file module named after
|
||||
that command, such as `storyteller_create_login_challenge_command.rs`. Do not
|
||||
collect multiple commands in a catch-all `*_commands.rs` file.
|
||||
- Keep `mod.rs` for module declarations. Put shared state, response/error types,
|
||||
and implementation helpers in separate support modules, with the narrowest
|
||||
visibility needed by their callers.
|
||||
- Update the imports used by `tauri::generate_handler!` when moving commands.
|
||||
Preserve IPC command names and arguments unless changing that contract is
|
||||
explicitly part of the task.
|
||||
- Authentication HTTP requests run in Rust through `artcraft_client` endpoint
|
||||
bindings. The frontend invokes Tauri commands; it does not call the API
|
||||
directly, and desktop commands do not connect to the backend MySQL database.
|
||||
- For login bridge changes, run the existing native fixture tests with
|
||||
`SQLX_OFFLINE=true cargo test --offline -p artcraft --lib login_bridge_tests`.
|
||||
These tests use loopback HTTP servers and temporary cookie stores.
|
||||
|
|
@ -0,0 +1,340 @@
|
|||
use std::collections::HashMap;
|
||||
use std::sync::Arc;
|
||||
|
||||
use artcraft_api_defs::users::login_challenges::{LoginChallengeFailure, LoginChallengeState};
|
||||
use artcraft_api_defs::users::session_info::SessionUserInfo;
|
||||
use artcraft_client::endpoints::users::password_login::PasswordLoginRequest;
|
||||
use artcraft_client::endpoints::users::password_signup::PasswordSignupRequest;
|
||||
use artcraft_client::utils::api_host::ApiHost;
|
||||
use artcraft_client::utils::login_challenge_client::{
|
||||
LoginChallengeClient, LoginChallengeClientError,
|
||||
};
|
||||
use chrono::{DateTime, Utc};
|
||||
use log::{info, warn};
|
||||
use serde::Serialize;
|
||||
use tauri::{AppHandle, Manager};
|
||||
use tauri_plugin_http::reqwest_cookie_store::{CookieStoreMutex, RawCookie};
|
||||
use tauri_plugin_http::Http;
|
||||
use tokio::sync::Mutex;
|
||||
use uuid_utils::uuid::generate_random_uuid;
|
||||
|
||||
use crate::core::threads::main_window_thread::persist_storyteller_cookies_task::get_credentials_from_cookie_store;
|
||||
use crate::services::storyteller::state::storyteller_credential_manager::StorytellerCredentialManager;
|
||||
|
||||
const MAX_PENDING_CHALLENGES: usize = 8;
|
||||
|
||||
/// All bearer credentials stay in Rust. The UI holds only a local lookup handle.
|
||||
#[derive(Default)]
|
||||
pub struct DesktopLoginBridgeState {
|
||||
pending: Mutex<HashMap<String, PendingChallenge>>,
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
struct PendingChallenge {
|
||||
device_token: String,
|
||||
expires_at: DateTime<Utc>,
|
||||
client: LoginChallengeClient,
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
pub struct DesktopLoginChallenge {
|
||||
pub challenge_id: String,
|
||||
pub verification_url: String,
|
||||
pub confirmation_code: String,
|
||||
pub expires_at: DateTime<Utc>,
|
||||
pub poll_interval_seconds: u32,
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
pub struct DesktopLoginOutcome {
|
||||
pub status: LoginChallengeState,
|
||||
pub maybe_failure_type: Option<LoginChallengeFailure>,
|
||||
pub maybe_user: Option<SessionUserInfo>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Serialize)]
|
||||
pub struct DesktopLoginError {
|
||||
pub status: Option<u16>,
|
||||
pub message: String,
|
||||
pub retryable: bool,
|
||||
}
|
||||
|
||||
impl DesktopLoginError {
|
||||
fn local(message: &str) -> Self {
|
||||
Self {
|
||||
status: None,
|
||||
message: message.to_owned(),
|
||||
retryable: false,
|
||||
}
|
||||
}
|
||||
|
||||
pub(super) fn from_client(origin: &str, error: LoginChallengeClientError) -> Self {
|
||||
warn!(
|
||||
"Website login failed: origin={} status={:?} reason={}",
|
||||
origin, error.status, error.message
|
||||
);
|
||||
Self {
|
||||
retryable: error
|
||||
.status
|
||||
.map(|s| s == 429 || s >= 500)
|
||||
.unwrap_or(error.message == "Unable to reach the login server"),
|
||||
status: error.status,
|
||||
message: format!("{}: {} (HTTP {:?})", origin, error.message, error.status),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub(super) async fn current_login_session(
|
||||
client: &LoginChallengeClient,
|
||||
jar: &Arc<CookieStoreMutex>,
|
||||
) -> Result<Option<SessionUserInfo>, DesktopLoginError> {
|
||||
let credentials = {
|
||||
let store = jar
|
||||
.store
|
||||
.lock()
|
||||
.map_err(|_| DesktopLoginError::local("HTTP cookie store unavailable"))?;
|
||||
get_credentials_from_cookie_store(&store, &client.api_url())
|
||||
.map_err(|_| DesktopLoginError::local("Unable to read session cookie"))?
|
||||
};
|
||||
let Some(signed) = credentials.session else {
|
||||
return Ok(None);
|
||||
};
|
||||
let session = match client.session(signed.as_str()).await {
|
||||
Ok(session) => session,
|
||||
Err(error) if error.status == Some(401) => return Ok(None),
|
||||
Err(error) => return Err(DesktopLoginError::from_client(&client.api_origin(), error)),
|
||||
};
|
||||
Ok(
|
||||
session
|
||||
.user
|
||||
.filter(|_| session.success && session.logged_in),
|
||||
)
|
||||
}
|
||||
|
||||
pub(super) async fn password_auth(
|
||||
host: &ApiHost,
|
||||
manager: &StorytellerCredentialManager,
|
||||
jar: &Arc<CookieStoreMutex>,
|
||||
login: Option<PasswordLoginRequest>,
|
||||
signup: Option<PasswordSignupRequest>,
|
||||
) -> Result<SessionUserInfo, DesktopLoginError> {
|
||||
let client = LoginChallengeClient::new(host)
|
||||
.map_err(|e| DesktopLoginError::from_client(&host.to_api_hostname_and_scheme(), e))?;
|
||||
let response = match (login, signup) {
|
||||
(Some(request), None) => client.password_login(&request).await,
|
||||
(None, Some(request)) => client.password_signup(&request).await,
|
||||
_ => return Err(DesktopLoginError::local("Invalid authentication request")),
|
||||
}
|
||||
.map_err(|e| DesktopLoginError::from_client(&client.api_origin(), e))?;
|
||||
let (response, cookie) = response;
|
||||
if !response.success {
|
||||
return Err(DesktopLoginError::local(
|
||||
"Sign-in failed. Check your account details and try again.",
|
||||
));
|
||||
}
|
||||
let signed = response
|
||||
.signed_session
|
||||
.ok_or_else(|| DesktopLoginError::local("Login server did not issue a session"))?;
|
||||
let cookie =
|
||||
cookie.ok_or_else(|| DesktopLoginError::local("Login server did not set a session cookie"))?;
|
||||
let (cookie, user) = verify_session(&client, &signed, cookie).await?;
|
||||
install_session_cookie(&client, &signed, cookie, manager, jar)?;
|
||||
persist_session_cookie(jar).await?;
|
||||
Ok(user)
|
||||
}
|
||||
|
||||
pub(super) fn http_cookie_jar(app: &AppHandle) -> Result<Arc<CookieStoreMutex>, DesktopLoginError> {
|
||||
Ok(
|
||||
app
|
||||
.try_state::<Http>()
|
||||
.ok_or_else(|| DesktopLoginError::local("HTTP cookie store unavailable"))?
|
||||
.cookies_jar
|
||||
.clone(),
|
||||
)
|
||||
}
|
||||
|
||||
pub(super) async fn begin_challenge(
|
||||
host: &ApiHost,
|
||||
state: &DesktopLoginBridgeState,
|
||||
) -> Result<DesktopLoginChallenge, DesktopLoginError> {
|
||||
let client = LoginChallengeClient::new(host)
|
||||
.map_err(|e| DesktopLoginError::from_client(&host.to_api_hostname_and_scheme(), e))?;
|
||||
let created = client
|
||||
.create()
|
||||
.await
|
||||
.map_err(|e| DesktopLoginError::from_client(&client.api_origin(), e))?;
|
||||
if !created.success
|
||||
|| !client.allows_verification_url(&created.verification_url)
|
||||
|| created.device_token.len() != 43
|
||||
|| created.expires_at <= Utc::now()
|
||||
{
|
||||
return Err(DesktopLoginError::local(
|
||||
"Invalid login challenge or website/API environment mismatch",
|
||||
));
|
||||
}
|
||||
let mut pending = state.pending.lock().await;
|
||||
pending.retain(|_, value| value.expires_at > Utc::now());
|
||||
if pending.len() >= MAX_PENDING_CHALLENGES {
|
||||
return Err(DesktopLoginError::local("Too many pending login attempts"));
|
||||
}
|
||||
let challenge_id = generate_random_uuid();
|
||||
pending.insert(
|
||||
challenge_id.clone(),
|
||||
PendingChallenge {
|
||||
device_token: created.device_token,
|
||||
expires_at: created.expires_at,
|
||||
client,
|
||||
},
|
||||
);
|
||||
Ok(DesktopLoginChallenge {
|
||||
challenge_id,
|
||||
verification_url: created.verification_url,
|
||||
confirmation_code: created.confirmation_code,
|
||||
expires_at: created.expires_at,
|
||||
poll_interval_seconds: created.poll_interval_seconds.max(5),
|
||||
})
|
||||
}
|
||||
|
||||
pub(super) async fn poll_challenge(
|
||||
state: &DesktopLoginBridgeState,
|
||||
manager: &StorytellerCredentialManager,
|
||||
jar: &Arc<CookieStoreMutex>,
|
||||
challenge_id: &str,
|
||||
) -> Result<DesktopLoginOutcome, DesktopLoginError> {
|
||||
let challenge = {
|
||||
let mut pending = state.pending.lock().await;
|
||||
let challenge = pending.get(challenge_id).cloned().ok_or_else(|| {
|
||||
DesktopLoginError::local("Login attempt was cancelled or is no longer available")
|
||||
})?;
|
||||
if challenge.expires_at <= Utc::now() {
|
||||
pending.remove(challenge_id);
|
||||
return Ok(expired());
|
||||
}
|
||||
challenge
|
||||
};
|
||||
let origin = challenge.client.api_origin();
|
||||
let polled = challenge
|
||||
.client
|
||||
.poll(&challenge.device_token)
|
||||
.await
|
||||
.map_err(|e| DesktopLoginError::from_client(&origin, e))?;
|
||||
if !polled.response.success || polled.response.status == LoginChallengeState::Unknown {
|
||||
return Err(DesktopLoginError::local("Unrecognized login response"));
|
||||
}
|
||||
if polled.response.status != LoginChallengeState::Redeemed {
|
||||
if polled.response.status == LoginChallengeState::Failed {
|
||||
state.pending.lock().await.remove(challenge_id);
|
||||
}
|
||||
return Ok(DesktopLoginOutcome {
|
||||
status: polled.response.status,
|
||||
maybe_failure_type: polled.response.maybe_failure_type,
|
||||
maybe_user: None,
|
||||
});
|
||||
}
|
||||
let signed = polled
|
||||
.response
|
||||
.maybe_signed_session
|
||||
.ok_or_else(|| DesktopLoginError::local("Login server did not issue a session"))?;
|
||||
let set_cookie = polled
|
||||
.session_set_cookie
|
||||
.ok_or_else(|| DesktopLoginError::local("Login server did not set a session cookie"))?;
|
||||
let (cookie, user) = verify_session(&challenge.client, &signed, set_cookie).await?;
|
||||
{
|
||||
let pending = state.pending.lock().await;
|
||||
// A cancellation during either HTTP request must prevent local installation.
|
||||
if !pending.contains_key(challenge_id) {
|
||||
return Err(DesktopLoginError::local("Login attempt was cancelled"));
|
||||
}
|
||||
if challenge.expires_at <= Utc::now() {
|
||||
return Ok(expired());
|
||||
}
|
||||
install_session_cookie(&challenge.client, &signed, cookie, manager, jar)?;
|
||||
}
|
||||
// Keep the challenge for idempotent IPC retries; it expires on its original deadline.
|
||||
persist_session_cookie(jar).await?;
|
||||
info!("Website login complete: origin={}", origin);
|
||||
Ok(DesktopLoginOutcome {
|
||||
status: LoginChallengeState::Redeemed,
|
||||
maybe_failure_type: None,
|
||||
maybe_user: Some(user),
|
||||
})
|
||||
}
|
||||
|
||||
pub(super) async fn cancel_challenge(state: &DesktopLoginBridgeState, challenge_id: &str) {
|
||||
state.pending.lock().await.remove(challenge_id);
|
||||
}
|
||||
|
||||
async fn verify_session(
|
||||
client: &LoginChallengeClient,
|
||||
signed: &str,
|
||||
set_cookie: String,
|
||||
) -> Result<(RawCookie<'static>, SessionUserInfo), DesktopLoginError> {
|
||||
let cookie = RawCookie::parse(set_cookie)
|
||||
.map_err(|_| DesktopLoginError::local("Invalid session cookie"))?
|
||||
.into_owned();
|
||||
if cookie.name() != "session" || cookie.value() != signed {
|
||||
return Err(DesktopLoginError::local(
|
||||
"Session cookie did not match the redeemed session",
|
||||
));
|
||||
}
|
||||
// Validate on the SAME native API host before publishing any new credentials.
|
||||
let session = client
|
||||
.session(&signed)
|
||||
.await
|
||||
.map_err(|e| DesktopLoginError::from_client(&client.api_origin(), e))?;
|
||||
let user = session
|
||||
.user
|
||||
.filter(|_| session.success && session.logged_in)
|
||||
.ok_or_else(|| DesktopLoginError::local("The redeemed session could not authenticate"))?;
|
||||
Ok((cookie, user))
|
||||
}
|
||||
|
||||
fn install_session_cookie(
|
||||
client: &LoginChallengeClient,
|
||||
signed: &str,
|
||||
cookie: RawCookie<'static>,
|
||||
manager: &StorytellerCredentialManager,
|
||||
jar: &Arc<CookieStoreMutex>,
|
||||
) -> Result<(), DesktopLoginError> {
|
||||
let mut store = jar
|
||||
.store
|
||||
.lock()
|
||||
.map_err(|_| DesktopLoginError::local("HTTP cookie store unavailable"))?;
|
||||
let mut candidate = store.clone();
|
||||
candidate.store_response_cookies([cookie].into_iter(), &client.api_url());
|
||||
let credentials = get_credentials_from_cookie_store(&candidate, &client.api_url())
|
||||
.map_err(|_| DesktopLoginError::local("Unable to read session cookie"))?;
|
||||
if credentials.session.as_ref().map(|c| c.as_str()) != Some(signed) {
|
||||
return Err(DesktopLoginError::local(
|
||||
"Session cookie does not apply to the configured API host",
|
||||
));
|
||||
}
|
||||
manager
|
||||
.set_credentials(&credentials)
|
||||
.map_err(|_| DesktopLoginError::local("Unable to store native login credentials"))?;
|
||||
*store = candidate;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn persist_session_cookie(jar: &Arc<CookieStoreMutex>) -> Result<(), DesktopLoginError> {
|
||||
let saved = jar
|
||||
.request_save()
|
||||
.map_err(|_| DesktopLoginError::local("Unable to persist login cookie"))?;
|
||||
tokio::task::spawn_blocking(move || saved.recv())
|
||||
.await
|
||||
.map_err(|_| DesktopLoginError::local("Unable to persist login cookie"))?
|
||||
.map_err(|_| DesktopLoginError::local("Unable to persist login cookie"))?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn expired() -> DesktopLoginOutcome {
|
||||
DesktopLoginOutcome {
|
||||
status: LoginChallengeState::Failed,
|
||||
maybe_failure_type: Some(LoginChallengeFailure::Expired),
|
||||
maybe_user: None,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
#[path = "storyteller_login_bridge_tests.rs"]
|
||||
mod login_bridge_tests;
|
||||
|
|
@ -2,4 +2,11 @@ pub mod storyteller_purge_credentials_command;
|
|||
pub mod storyteller_get_credits_command;
|
||||
pub mod storyteller_get_subscription_command;
|
||||
pub mod stripe_checkout;
|
||||
pub mod stripe_customer_portal;
|
||||
pub mod stripe_customer_portal;
|
||||
pub(crate) mod login_bridge;
|
||||
pub mod storyteller_cancel_login_challenge_command;
|
||||
pub mod storyteller_create_login_challenge_command;
|
||||
pub mod storyteller_get_login_session_command;
|
||||
pub mod storyteller_password_login_command;
|
||||
pub mod storyteller_password_signup_command;
|
||||
pub mod storyteller_poll_login_challenge_command;
|
||||
|
|
|
|||
|
|
@ -0,0 +1,14 @@
|
|||
use tauri::State;
|
||||
|
||||
use crate::services::storyteller::commands::login_bridge::{
|
||||
cancel_challenge, DesktopLoginBridgeState, DesktopLoginError,
|
||||
};
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn storyteller_cancel_login_challenge_command(
|
||||
state: State<'_, DesktopLoginBridgeState>,
|
||||
challenge_id: String,
|
||||
) -> Result<(), DesktopLoginError> {
|
||||
cancel_challenge(&state, &challenge_id).await;
|
||||
Ok(())
|
||||
}
|
||||
|
|
@ -0,0 +1,14 @@
|
|||
use tauri::State;
|
||||
|
||||
use crate::core::state::app_env_configs::app_env_configs::AppEnvConfigs;
|
||||
use crate::services::storyteller::commands::login_bridge::{
|
||||
begin_challenge, DesktopLoginBridgeState, DesktopLoginChallenge, DesktopLoginError,
|
||||
};
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn storyteller_create_login_challenge_command(
|
||||
config: State<'_, AppEnvConfigs>,
|
||||
state: State<'_, DesktopLoginBridgeState>,
|
||||
) -> Result<DesktopLoginChallenge, DesktopLoginError> {
|
||||
begin_challenge(&config.storyteller_host, &state).await
|
||||
}
|
||||
|
|
@ -0,0 +1,19 @@
|
|||
use artcraft_api_defs::users::session_info::SessionUserInfo;
|
||||
use artcraft_client::utils::login_challenge_client::LoginChallengeClient;
|
||||
use tauri::{AppHandle, State};
|
||||
|
||||
use crate::core::state::app_env_configs::app_env_configs::AppEnvConfigs;
|
||||
use crate::services::storyteller::commands::login_bridge::{
|
||||
current_login_session, http_cookie_jar, DesktopLoginError,
|
||||
};
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn storyteller_get_login_session_command(
|
||||
app: AppHandle,
|
||||
config: State<'_, AppEnvConfigs>,
|
||||
) -> Result<Option<SessionUserInfo>, DesktopLoginError> {
|
||||
let client = LoginChallengeClient::new(&config.storyteller_host).map_err(|error| {
|
||||
DesktopLoginError::from_client(&config.storyteller_host.to_api_hostname_and_scheme(), error)
|
||||
})?;
|
||||
current_login_session(&client, &http_cookie_jar(&app)?).await
|
||||
}
|
||||
|
|
@ -0,0 +1,436 @@
|
|||
//! Commands' native implementation + real ArtCraft bindings + loopback HTTP.
|
||||
//! All listeners bind to an ephemeral local port; no production host is contacted.
|
||||
use super::*;
|
||||
use crate::core::state::data_dir::app_data_root::AppDataRoot;
|
||||
use artcraft_client::credentials::storyteller_avt_cookie::StorytellerAvtCookie;
|
||||
use artcraft_client::credentials::storyteller_credential_set::StorytellerCredentialSet;
|
||||
use serde_json::{json, Value};
|
||||
use std::io::{Read, Write};
|
||||
use std::net::TcpListener;
|
||||
use std::thread;
|
||||
use std::time::Duration;
|
||||
use tauri_plugin_http::reqwest_cookie_store::CookieStore;
|
||||
|
||||
const SIGNED: &str = "signed_native_session_fixture";
|
||||
const DEVICE: &str = "DDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDD";
|
||||
|
||||
#[tokio::test]
|
||||
async fn approved_session_is_verified_installed_persisted_and_retried_entirely_in_rust() {
|
||||
let server = server(vec![
|
||||
create(),
|
||||
pending(),
|
||||
redeemed(),
|
||||
session(),
|
||||
redeemed(),
|
||||
session(),
|
||||
]);
|
||||
let h = harness();
|
||||
// Native installation must preserve visitor cookies and avoid importing other origins.
|
||||
h.jar.store.lock().unwrap().store_response_cookies(
|
||||
[RawCookie::parse("visitor=existing; Path=/; Max-Age=3600")
|
||||
.unwrap()
|
||||
.into_owned()]
|
||||
.into_iter(),
|
||||
&server_url(&server),
|
||||
);
|
||||
h.manager
|
||||
.set_credentials(&StorytellerCredentialSet::initialize_with_just_avt(
|
||||
StorytellerAvtCookie::new("existing".into()),
|
||||
))
|
||||
.unwrap();
|
||||
let created = begin_challenge(&server.host, &h.state).await.unwrap();
|
||||
let public = serde_json::to_string(&created).unwrap();
|
||||
assert!(!public.contains(DEVICE));
|
||||
assert!(!public.contains("device_token"));
|
||||
assert!(h
|
||||
.manager
|
||||
.get_credentials_required()
|
||||
.unwrap()
|
||||
.session
|
||||
.is_none());
|
||||
let wait = poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id)
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(wait.status, LoginChallengeState::Pending);
|
||||
assert!(h
|
||||
.manager
|
||||
.get_credentials_required()
|
||||
.unwrap()
|
||||
.session
|
||||
.is_none());
|
||||
for _ in 0..2 {
|
||||
let result = poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id)
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(result.status, LoginChallengeState::Redeemed);
|
||||
assert_eq!(result.maybe_user.as_ref().unwrap().username, "google_user");
|
||||
// This was the real failure: use_qt exists on the server, not this desktop.
|
||||
let flags = &result.maybe_user.as_ref().unwrap().maybe_feature_flags;
|
||||
assert_eq!(flags.len(), 1);
|
||||
assert_eq!(flags.first().unwrap().to_str(), "studio");
|
||||
assert!(!serde_json::to_string(&result).unwrap().contains(SIGNED));
|
||||
let credentials = h.manager.get_credentials_required().unwrap();
|
||||
assert_eq!(credentials.session.unwrap().as_str(), SIGNED);
|
||||
assert_eq!(credentials.avt.unwrap().as_str(), "existing");
|
||||
assert!(std::fs::read_to_string(&h.jar.path)
|
||||
.unwrap()
|
||||
.contains(SIGNED));
|
||||
}
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn rejection_and_timeout_never_install_a_session() {
|
||||
for expired in [false, true] {
|
||||
let mut steps = vec![create()];
|
||||
if !expired {
|
||||
steps.push(step(
|
||||
"/v1/login_challenges/poll",
|
||||
json!({"success":true,"status":"failed","maybe_failure_type":"user_declined"}),
|
||||
));
|
||||
}
|
||||
let server = server(steps);
|
||||
let h = harness();
|
||||
let created = begin_challenge(&server.host, &h.state).await.unwrap();
|
||||
if expired {
|
||||
h.state
|
||||
.pending
|
||||
.lock()
|
||||
.await
|
||||
.get_mut(&created.challenge_id)
|
||||
.unwrap()
|
||||
.expires_at = Utc::now();
|
||||
}
|
||||
let result = poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id)
|
||||
.await
|
||||
.unwrap();
|
||||
assert_eq!(result.status, LoginChallengeState::Failed);
|
||||
assert_eq!(
|
||||
result.maybe_failure_type,
|
||||
Some(if expired {
|
||||
LoginChallengeFailure::Expired
|
||||
} else {
|
||||
LoginChallengeFailure::UserDeclined
|
||||
})
|
||||
);
|
||||
assert!(h.manager.get_credentials().unwrap().is_none());
|
||||
assert!(h
|
||||
.jar
|
||||
.store
|
||||
.lock()
|
||||
.unwrap()
|
||||
.iter_unexpired()
|
||||
.next()
|
||||
.is_none());
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invalid_downstream_session_is_not_installed_and_error_has_host_without_credentials() {
|
||||
let mut invalid = session();
|
||||
invalid.status = 401;
|
||||
invalid.body = json!({"echo": SIGNED});
|
||||
let server = server(vec![create(), redeemed(), invalid]);
|
||||
let h = harness();
|
||||
let created = begin_challenge(&server.host, &h.state).await.unwrap();
|
||||
let error = poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id)
|
||||
.await
|
||||
.err()
|
||||
.unwrap();
|
||||
assert_eq!(error.status, Some(401));
|
||||
assert!(!error.retryable);
|
||||
assert!(error
|
||||
.message
|
||||
.contains(&server.host.to_api_hostname_and_scheme()));
|
||||
assert!(!format!("{error:?}").contains(SIGNED));
|
||||
assert!(h.manager.get_credentials().unwrap().is_none());
|
||||
assert!(h
|
||||
.jar
|
||||
.store
|
||||
.lock()
|
||||
.unwrap()
|
||||
.iter_unexpired()
|
||||
.next()
|
||||
.is_none());
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn cancelling_during_http_prevents_late_cookie_installation() {
|
||||
let mut delayed = redeemed();
|
||||
delayed.delay = Duration::from_millis(150);
|
||||
let server = server(vec![create(), delayed, session()]);
|
||||
let h = harness();
|
||||
let created = begin_challenge(&server.host, &h.state).await.unwrap();
|
||||
let (result, _) = tokio::join!(
|
||||
poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id),
|
||||
async {
|
||||
tokio::time::sleep(Duration::from_millis(30)).await;
|
||||
cancel_challenge(&h.state, &created.challenge_id).await;
|
||||
}
|
||||
);
|
||||
assert!(result.is_err());
|
||||
assert!(h.manager.get_credentials().unwrap().is_none());
|
||||
assert!(h
|
||||
.jar
|
||||
.store
|
||||
.lock()
|
||||
.unwrap()
|
||||
.iter_unexpired()
|
||||
.next()
|
||||
.is_none());
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn unknown_states_and_redirects_fail_closed() {
|
||||
for redirect in [false, true] {
|
||||
let mut result = step(
|
||||
"/v1/login_challenges/poll",
|
||||
json!({"success":true,"status":"future_state","maybe_signed_session":SIGNED}),
|
||||
);
|
||||
if redirect {
|
||||
result.status = 302;
|
||||
result.headers = "Location: https://must-never-be-contacted.invalid/\r\n".into();
|
||||
}
|
||||
let server = server(vec![create(), result]);
|
||||
let h = harness();
|
||||
let created = begin_challenge(&server.host, &h.state).await.unwrap();
|
||||
let error = poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id)
|
||||
.await
|
||||
.err()
|
||||
.unwrap();
|
||||
assert!(!error.retryable);
|
||||
if redirect {
|
||||
assert_eq!(error.status, Some(302));
|
||||
}
|
||||
assert!(h.manager.get_credentials().unwrap().is_none());
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn website_origin_must_match_native_api_environment() {
|
||||
let local = LoginChallengeClient::new(&ApiHost::Localhost { port: 12345 }).unwrap();
|
||||
let prod = LoginChallengeClient::new(&ApiHost::Storyteller).unwrap();
|
||||
let fragment = format!("#approval_token={}", "A".repeat(43));
|
||||
assert!(local.allows_verification_url(&format!("http://localhost:4201/login/desktop{fragment}")));
|
||||
assert!(!local.allows_verification_url(&format!(
|
||||
"https://app.getartcraft.com/login/desktop{fragment}"
|
||||
)));
|
||||
assert!(prod.allows_verification_url(&format!(
|
||||
"https://app.getartcraft.com/login/desktop{fragment}"
|
||||
)));
|
||||
assert!(!prod.allows_verification_url(&format!("http://localhost:4201/login/desktop{fragment}")));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn malformed_session_data_still_cannot_install_credentials() {
|
||||
for (field, value) in [("maybe_feature_flags", json!([42])), ("username", Value::Null)] {
|
||||
let mut invalid = session();
|
||||
invalid.body["user"][field] = value;
|
||||
let server = server(vec![create(), redeemed(), invalid]);
|
||||
let h = harness();
|
||||
let created = begin_challenge(&server.host, &h.state).await.unwrap();
|
||||
assert!(poll_challenge(&h.state, &h.manager, &h.jar, &created.challenge_id).await.is_err());
|
||||
assert!(h.manager.get_credentials().unwrap().is_none());
|
||||
assert!(h.jar.store.lock().unwrap().iter_unexpired().next().is_none());
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn password_login_and_signup_verify_persist_and_recheck_via_native_api() {
|
||||
for signup in [false, true] {
|
||||
let mut auth = redeemed();
|
||||
auth.path = if signup { "/v1/create_account" } else { "/v1/login" };
|
||||
auth.body = json!({"success":true,"signed_session":SIGNED});
|
||||
let server = server(vec![auth, session(), session()]);
|
||||
let h = harness();
|
||||
let (login, signup) = if signup {
|
||||
(None, Some(PasswordSignupRequest { username: "google_user".into(), email_address: "user@example.test".into(), password: "fixture_password".into(), password_confirmation: "fixture_password".into(), signup_source: "artcraft".into() }))
|
||||
} else {
|
||||
(Some(PasswordLoginRequest { username_or_email: "google_user".into(), password: "fixture_password".into() }), None)
|
||||
};
|
||||
let user = password_auth(&server.host, &h.manager, &h.jar, login, signup).await.unwrap();
|
||||
assert_eq!(user.username, "google_user");
|
||||
assert_eq!(h.manager.get_credentials_required().unwrap().session.unwrap().as_str(), SIGNED);
|
||||
assert!(std::fs::read_to_string(&h.jar.path).unwrap().contains(SIGNED));
|
||||
let client = LoginChallengeClient::new(&server.host).unwrap();
|
||||
assert_eq!(current_login_session(&client, &h.jar).await.unwrap().unwrap().username, "google_user");
|
||||
server.thread.join().unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn empty_native_cookie_jar_needs_no_session_http_request() {
|
||||
let h = harness();
|
||||
let client = LoginChallengeClient::new(&ApiHost::Localhost { port: 1 }).unwrap();
|
||||
assert!(current_login_session(&client, &h.jar).await.unwrap().is_none());
|
||||
}
|
||||
|
||||
struct Harness {
|
||||
_directory: tempfile::TempDir,
|
||||
state: DesktopLoginBridgeState,
|
||||
manager: StorytellerCredentialManager,
|
||||
jar: Arc<CookieStoreMutex>,
|
||||
}
|
||||
|
||||
struct Server {
|
||||
host: ApiHost,
|
||||
thread: thread::JoinHandle<()>,
|
||||
}
|
||||
|
||||
struct Step {
|
||||
path: &'static str,
|
||||
status: u16,
|
||||
body: Value,
|
||||
headers: String,
|
||||
delay: Duration,
|
||||
}
|
||||
|
||||
fn harness() -> Harness {
|
||||
let directory = tempfile::tempdir().unwrap();
|
||||
let root = AppDataRoot::create_existing(directory.path()).unwrap();
|
||||
let manager = StorytellerCredentialManager::initialize_empty(&root);
|
||||
let jar = Arc::new(CookieStoreMutex::new(
|
||||
directory.path().join("cookies.json"),
|
||||
CookieStore::default(),
|
||||
));
|
||||
Harness {
|
||||
_directory: directory,
|
||||
state: DesktopLoginBridgeState::default(),
|
||||
manager,
|
||||
jar,
|
||||
}
|
||||
}
|
||||
|
||||
fn server_url(server: &Server) -> reqwest::Url {
|
||||
reqwest::Url::parse(&format!("{}/", server.host.to_api_hostname_and_scheme())).unwrap()
|
||||
}
|
||||
|
||||
fn server(steps: Vec<Step>) -> Server {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").unwrap();
|
||||
let port = listener.local_addr().unwrap().port();
|
||||
let thread = thread::spawn(move || {
|
||||
for step in steps {
|
||||
let (mut stream, _) = listener.accept().unwrap();
|
||||
stream
|
||||
.set_read_timeout(Some(Duration::from_secs(10)))
|
||||
.unwrap();
|
||||
let mut data = Vec::new();
|
||||
let mut buffer = [0; 4096];
|
||||
loop {
|
||||
let count = stream.read(&mut buffer).unwrap();
|
||||
assert!(count > 0);
|
||||
data.extend_from_slice(&buffer[..count]);
|
||||
if let Some(end) = data.windows(4).position(|v| v == b"\r\n\r\n") {
|
||||
let headers = String::from_utf8_lossy(&data[..end]).to_lowercase();
|
||||
let length: usize = headers
|
||||
.lines()
|
||||
.find_map(|s| s.strip_prefix("content-length:"))
|
||||
.map(|s| s.trim().parse().unwrap())
|
||||
.unwrap_or(0);
|
||||
if data.len() >= end + 4 + length {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
let request = String::from_utf8(data).unwrap();
|
||||
assert_eq!(request.split_whitespace().nth(1), Some(step.path));
|
||||
assert!(request
|
||||
.to_lowercase()
|
||||
.contains(&format!("host: localhost:{port}")));
|
||||
if step.path == "/v1/session" {
|
||||
assert!(request.starts_with("GET "));
|
||||
assert!(request.contains(&format!("session={SIGNED}")));
|
||||
} else {
|
||||
assert!(request.starts_with("POST "));
|
||||
assert!(!request.to_lowercase().contains("cookie:"));
|
||||
if step.path.ends_with("/poll") {
|
||||
let body: Value =
|
||||
serde_json::from_str(request.split_once("\r\n\r\n").unwrap().1).unwrap();
|
||||
assert_eq!(body, json!({"device_token":DEVICE}));
|
||||
}
|
||||
}
|
||||
thread::sleep(step.delay);
|
||||
let body = step.body.to_string();
|
||||
write!(stream, "HTTP/1.1 {} Test\r\nContent-Type: application/json\r\nContent-Length: {}\r\nConnection: close\r\n{}\r\n{}", step.status, body.len(), step.headers, body).unwrap();
|
||||
}
|
||||
});
|
||||
Server {
|
||||
host: ApiHost::Localhost { port: port.into() },
|
||||
thread,
|
||||
}
|
||||
}
|
||||
|
||||
fn step(path: &'static str, body: Value) -> Step {
|
||||
Step {
|
||||
path,
|
||||
body,
|
||||
status: 200,
|
||||
headers: String::new(),
|
||||
delay: Duration::ZERO,
|
||||
}
|
||||
}
|
||||
|
||||
fn create() -> Step {
|
||||
step(
|
||||
"/v1/login_challenges/create",
|
||||
json!({"success":true,"device_token":DEVICE,"verification_url":format!("http://localhost:4201/login/desktop#approval_token={}","A".repeat(43)),"confirmation_code":"WDJBMJHT","expires_at":Utc::now()+chrono::Duration::minutes(20),"poll_interval_seconds":5}),
|
||||
)
|
||||
}
|
||||
|
||||
fn pending() -> Step {
|
||||
step(
|
||||
"/v1/login_challenges/poll",
|
||||
json!({"success":true,"status":"pending","maybe_failure_type":null}),
|
||||
)
|
||||
}
|
||||
|
||||
fn redeemed() -> Step {
|
||||
let mut response = step(
|
||||
"/v1/login_challenges/poll",
|
||||
json!({"success":true,"status":"redeemed","maybe_failure_type":null,"maybe_signed_session":SIGNED}),
|
||||
);
|
||||
response.headers = format!("Set-Cookie: session={SIGNED}; HttpOnly; Path=/; Max-Age=3600\r\n");
|
||||
response
|
||||
}
|
||||
|
||||
fn session() -> Step {
|
||||
let mut user = json!({
|
||||
"user_token":"u_native_test", "username":"google_user", "display_name":"Google User", "email_gravatar_hash":"fixture",
|
||||
"core_info":{"user_token":"u_native_test","username":"google_user","display_name":"Google User","gravatar_hash":"fixture","default_avatar":{"image_index":1,"color_index":1}},
|
||||
"onboarding":{"email_not_set":false,"email_not_confirmed":false,"password_not_set":true,"username_not_customized":false},
|
||||
"maybe_feature_flags":["studio", "use_qt", "future_server_flag"],"fakeyou_plan":"free","storyteller_stream_plan":"free"
|
||||
});
|
||||
for field in [
|
||||
"can_access_studio",
|
||||
"can_use_tts",
|
||||
"can_use_w2l",
|
||||
"can_delete_own_tts_results",
|
||||
"can_delete_own_w2l_results",
|
||||
"can_delete_own_account",
|
||||
"can_upload_tts_models",
|
||||
"can_upload_w2l_templates",
|
||||
"can_delete_own_tts_models",
|
||||
"can_delete_own_w2l_templates",
|
||||
"can_approve_w2l_templates",
|
||||
"can_edit_other_users_profiles",
|
||||
"can_edit_other_users_tts_models",
|
||||
"can_edit_other_users_w2l_templates",
|
||||
"can_delete_other_users_tts_models",
|
||||
"can_delete_other_users_w2l_templates",
|
||||
"can_delete_other_users_tts_results",
|
||||
"can_delete_other_users_w2l_results",
|
||||
"can_ban_users",
|
||||
"can_delete_users",
|
||||
] {
|
||||
user[field] = json!(false);
|
||||
}
|
||||
step(
|
||||
"/v1/session",
|
||||
json!({"success":true,"logged_in":true,"user":user}),
|
||||
)
|
||||
}
|
||||
|
|
@ -0,0 +1,26 @@
|
|||
use artcraft_api_defs::users::session_info::SessionUserInfo;
|
||||
use artcraft_client::endpoints::users::password_login::PasswordLoginRequest;
|
||||
use tauri::{AppHandle, State};
|
||||
|
||||
use crate::core::state::app_env_configs::app_env_configs::AppEnvConfigs;
|
||||
use crate::services::storyteller::commands::login_bridge::{
|
||||
http_cookie_jar, password_auth, DesktopLoginError,
|
||||
};
|
||||
use crate::services::storyteller::state::storyteller_credential_manager::StorytellerCredentialManager;
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn storyteller_password_login_command(
|
||||
app: AppHandle,
|
||||
config: State<'_, AppEnvConfigs>,
|
||||
manager: State<'_, StorytellerCredentialManager>,
|
||||
request: PasswordLoginRequest,
|
||||
) -> Result<SessionUserInfo, DesktopLoginError> {
|
||||
password_auth(
|
||||
&config.storyteller_host,
|
||||
&manager,
|
||||
&http_cookie_jar(&app)?,
|
||||
Some(request),
|
||||
None,
|
||||
)
|
||||
.await
|
||||
}
|
||||
|
|
@ -0,0 +1,26 @@
|
|||
use artcraft_api_defs::users::session_info::SessionUserInfo;
|
||||
use artcraft_client::endpoints::users::password_signup::PasswordSignupRequest;
|
||||
use tauri::{AppHandle, State};
|
||||
|
||||
use crate::core::state::app_env_configs::app_env_configs::AppEnvConfigs;
|
||||
use crate::services::storyteller::commands::login_bridge::{
|
||||
http_cookie_jar, password_auth, DesktopLoginError,
|
||||
};
|
||||
use crate::services::storyteller::state::storyteller_credential_manager::StorytellerCredentialManager;
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn storyteller_password_signup_command(
|
||||
app: AppHandle,
|
||||
config: State<'_, AppEnvConfigs>,
|
||||
manager: State<'_, StorytellerCredentialManager>,
|
||||
request: PasswordSignupRequest,
|
||||
) -> Result<SessionUserInfo, DesktopLoginError> {
|
||||
password_auth(
|
||||
&config.storyteller_host,
|
||||
&manager,
|
||||
&http_cookie_jar(&app)?,
|
||||
None,
|
||||
Some(request),
|
||||
)
|
||||
.await
|
||||
}
|
||||
|
|
@ -0,0 +1,17 @@
|
|||
use tauri::{AppHandle, State};
|
||||
|
||||
use crate::services::storyteller::commands::login_bridge::{
|
||||
http_cookie_jar, poll_challenge, DesktopLoginBridgeState, DesktopLoginError, DesktopLoginOutcome,
|
||||
};
|
||||
use crate::services::storyteller::state::storyteller_credential_manager::StorytellerCredentialManager;
|
||||
|
||||
#[tauri::command]
|
||||
pub async fn storyteller_poll_login_challenge_command(
|
||||
app: AppHandle,
|
||||
state: State<'_, DesktopLoginBridgeState>,
|
||||
manager: State<'_, StorytellerCredentialManager>,
|
||||
challenge_id: String,
|
||||
) -> Result<DesktopLoginOutcome, DesktopLoginError> {
|
||||
let jar = http_cookie_jar(&app)?;
|
||||
poll_challenge(&state, &manager, &jar, &challenge_id).await
|
||||
}
|
||||
|
|
@ -3,9 +3,10 @@ use crate::core::state::artcraft_platform_info::ArtcraftPlatformInfo;
|
|||
use crate::core::state::artcraft_usage_tracker::artcraft_usage_tracker::ArtcraftUsageTracker;
|
||||
use crate::services::storyteller::state::storyteller_credential_manager::StorytellerCredentialManager;
|
||||
use anyhow::anyhow;
|
||||
use artcraft_api_defs::analytics::log_active_user::LogAppActiveUserRequest;
|
||||
use artcraft_api_defs::analytics::log_active_user::{LogAppActiveUserRequest, LOG_ACTIVE_USER_V2_PATH};
|
||||
use artcraft_client::endpoints::analytics::log_active_user_v2::log_active_user_v2;
|
||||
use artcraft_client::error::api_error::ApiError;
|
||||
use artcraft_client::credentials::storyteller_credential_set::StorytellerCredentialSet;
|
||||
use artcraft_client::error::storyteller_error::StorytellerError;
|
||||
use errors::AnyhowResult;
|
||||
use log::{debug, error, info};
|
||||
|
|
@ -38,7 +39,7 @@ pub async fn storyteller_activity_thread(
|
|||
&app_session_token,
|
||||
).await;
|
||||
if let Err(err) = res {
|
||||
error!("An error occurred: {:?}", err);
|
||||
error!("Activity request failed: origin={} path={} error={:?}", app_env_configs.storyteller_host.to_api_hostname_and_scheme(), LOG_ACTIVE_USER_V2_PATH, err);
|
||||
}
|
||||
// NB: Sleep if an error occurs.
|
||||
tokio::time::sleep(std::time::Duration::from_millis(ERROR_SLEEP_MILLIS)).await;
|
||||
|
|
@ -62,7 +63,7 @@ async fn polling_loop(
|
|||
continue;
|
||||
}
|
||||
Some(creds) => {
|
||||
if creds.is_empty() {
|
||||
if !has_activity_session(&creds) {
|
||||
tokio::time::sleep(std::time::Duration::from_millis(5_000)).await;
|
||||
continue;
|
||||
}
|
||||
|
|
@ -123,7 +124,31 @@ async fn polling_loop(
|
|||
};
|
||||
|
||||
// Wait at least a minute, no matter what the server tells us.
|
||||
let wait_millis = std::cmp::min(wait_millis, 60_000);
|
||||
let wait_millis = std::cmp::max(wait_millis, 60_000);
|
||||
tokio::time::sleep(std::time::Duration::from_millis(wait_millis)).await;
|
||||
}
|
||||
}
|
||||
|
||||
// Anonymous visitor tracking is not authentication for active_user_v2.
|
||||
fn has_activity_session(creds: &StorytellerCredentialSet) -> bool {
|
||||
creds.session.as_ref().map(|session| !session.as_str().is_empty()).unwrap_or(false)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod login_bridge_tests {
|
||||
use super::has_activity_session;
|
||||
use artcraft_client::credentials::storyteller_credential_set::StorytellerCredentialSet;
|
||||
use artcraft_client::credentials::storyteller_avt_cookie::StorytellerAvtCookie;
|
||||
use artcraft_client::credentials::storyteller_session_cookie::StorytellerSessionCookie;
|
||||
|
||||
#[test]
|
||||
fn anonymous_visitor_cookie_does_not_authorize_activity_requests() {
|
||||
assert!(!has_activity_session(&StorytellerCredentialSet::empty()));
|
||||
let visitor_only = StorytellerCredentialSet::initialize_with_just_avt(StorytellerAvtCookie::new("visitor_fixture".into()));
|
||||
assert!(!has_activity_session(&visitor_only));
|
||||
let empty_session = StorytellerCredentialSet::initialize_with_just_cookie(StorytellerSessionCookie::new(String::new()));
|
||||
assert!(!has_activity_session(&empty_session));
|
||||
let signed_in = StorytellerCredentialSet::initialize_with_just_cookie(StorytellerSessionCookie::new("signed_session_fixture".into()));
|
||||
assert!(has_activity_session(&signed_in));
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
|
||||
/// The version of the ArtCraft desktop app reported in the UI and via Analytics.
|
||||
/// This should match the CI builds.
|
||||
pub const ARTCRAFT_VERSION: &str = "0.39.0";
|
||||
pub const ARTCRAFT_VERSION: &str = "0.41.0";
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
{
|
||||
"$schema": "https://schema.tauri.app/config/2.0.0-rc",
|
||||
"productName": "ArtCraft",
|
||||
"version": "0.39.0",
|
||||
"version": "0.41.0",
|
||||
"identifier": "ai.artcraft.app",
|
||||
"build": {
|
||||
"frontendDist": "../../../frontend/apps/artcraft/dist",
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
{
|
||||
"$schema": "https://schema.tauri.app/config/2.0.0-rc",
|
||||
"productName": "ArtCraft",
|
||||
"version": "0.39.0",
|
||||
"version": "0.41.0",
|
||||
"identifier": "ai.artcraft.app",
|
||||
"build": {
|
||||
"frontendDist": "../../../frontend/apps/artcraft/dist",
|
||||
|
|
|
|||
|
|
@ -0,0 +1,17 @@
|
|||
[package]
|
||||
name = "artcraft_client_identity"
|
||||
edition = "2024"
|
||||
version = "0.0.1"
|
||||
publish = false
|
||||
|
||||
[lib]
|
||||
name = "artcraft_client_identity"
|
||||
path = "src/lib.rs"
|
||||
|
||||
[dependencies]
|
||||
|
||||
# External
|
||||
url.workspace = true
|
||||
|
||||
[dev-dependencies]
|
||||
# (None)
|
||||
|
|
@ -0,0 +1,166 @@
|
|||
use url::{Host, Url};
|
||||
|
||||
use crate::origins::{ARTCRAFT_DESKTOP_DEVELOPMENT_ORIGIN, ARTCRAFT_DESKTOP_ORIGIN};
|
||||
use crate::third_party_provider::{ThirdPartyProvider, THIRD_PARTY_PROVIDERS};
|
||||
|
||||
/// Hostnames of our deployed APIs (HTTPS only).
|
||||
const ARTCRAFT_API_HOSTNAMES: [&str; 3] = [
|
||||
"api.storyteller.ai",
|
||||
"api.getartcraft.com",
|
||||
"api.fakeyou.com",
|
||||
];
|
||||
|
||||
/// Domains we own. Subdomains match too (CDNs, uploads, websites).
|
||||
const ARTCRAFT_DOMAINS: [&str; 4] = [
|
||||
"storyteller.ai",
|
||||
"getartcraft.com",
|
||||
"fakeyou.com",
|
||||
"artcraft.ai",
|
||||
];
|
||||
|
||||
/// Who a request is going to, which decides how the desktop app identifies itself.
|
||||
#[derive(Debug, PartialEq, Eq)]
|
||||
pub enum Destination {
|
||||
/// Our HTTP API, deployed or running locally, and the `Origin` to declare to it.
|
||||
ArtcraftApi { origin: &'static str },
|
||||
/// Our other hosts: CDNs, uploads, websites.
|
||||
ArtcraftHost,
|
||||
/// A known third-party provider.
|
||||
ThirdParty(&'static ThirdPartyProvider),
|
||||
/// Anything else.
|
||||
Unknown,
|
||||
}
|
||||
|
||||
impl Destination {
|
||||
pub fn classify(destination: &Url) -> Self {
|
||||
if is_local_development_api(destination) {
|
||||
return Self::ArtcraftApi { origin: ARTCRAFT_DESKTOP_DEVELOPMENT_ORIGIN };
|
||||
}
|
||||
let Some(host) = destination.host_str() else {
|
||||
return Self::Unknown;
|
||||
};
|
||||
if destination.scheme() == "https" && ARTCRAFT_API_HOSTNAMES.contains(&host) {
|
||||
return Self::ArtcraftApi { origin: ARTCRAFT_DESKTOP_ORIGIN };
|
||||
}
|
||||
if ARTCRAFT_DOMAINS.iter().any(|domain| is_domain_or_subdomain(host, domain)) {
|
||||
return Self::ArtcraftHost;
|
||||
}
|
||||
THIRD_PARTY_PROVIDERS
|
||||
.iter()
|
||||
.find(|provider| provider.domains.iter().any(|domain| is_domain_or_subdomain(host, domain)))
|
||||
.map_or(Self::Unknown, Self::ThirdParty)
|
||||
}
|
||||
}
|
||||
|
||||
fn is_local_development_api(destination: &Url) -> bool {
|
||||
if !matches!(destination.scheme(), "http" | "https") {
|
||||
return false;
|
||||
}
|
||||
match destination.host() {
|
||||
Some(Host::Domain("localhost")) => true,
|
||||
Some(Host::Ipv4(address)) => address.is_loopback(),
|
||||
Some(Host::Ipv6(address)) => address.is_loopback(),
|
||||
_ => false,
|
||||
}
|
||||
}
|
||||
|
||||
fn is_domain_or_subdomain(host: &str, domain: &str) -> bool {
|
||||
host == domain
|
||||
|| host
|
||||
.strip_suffix(domain)
|
||||
.is_some_and(|prefix| prefix.ends_with('.'))
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
mod artcraft {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn deployed_apis_get_desktop_origin() {
|
||||
for destination in [
|
||||
"https://api.storyteller.ai/v1/session",
|
||||
"https://api.getartcraft.com/v1/session",
|
||||
"https://api.fakeyou.com/v1/session",
|
||||
] {
|
||||
assert_eq!(classify(destination), Destination::ArtcraftApi { origin: "https://desktop.getartcraft.com" });
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn local_apis_get_development_origin() {
|
||||
for destination in [
|
||||
"http://localhost:12345/v1/session",
|
||||
"https://localhost:12345/v1/session",
|
||||
"http://127.0.0.1:12345/v1/session",
|
||||
"http://[::1]:12345/v1/session",
|
||||
] {
|
||||
assert_eq!(classify(destination), Destination::ArtcraftApi { origin: "http://localhost" });
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn other_owned_hosts_are_not_apis() {
|
||||
for destination in [
|
||||
"https://cdn-2.fakeyou.com/media/a/b.png",
|
||||
"https://cdn.storyteller.ai/media/a/b.png",
|
||||
"https://style.storyteller.ai/",
|
||||
"https://app.getartcraft.com/",
|
||||
"https://getartcraft.com/",
|
||||
"http://api.storyteller.ai/v1/session",
|
||||
] {
|
||||
assert_eq!(classify(destination), Destination::ArtcraftHost, "{destination}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
mod third_party {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn providers_get_their_own_origins() {
|
||||
for (destination, name, origin) in [
|
||||
("https://grok.com/rest/app-chat/conversations/new", "Grok", "https://grok.com"),
|
||||
("https://assets.grok.com/users/a/b.mp4", "Grok", "https://grok.com"),
|
||||
("https://imagine-public.x.ai/a.png", "Grok", "https://grok.com"),
|
||||
("https://www.midjourney.com/api/submit-jobs", "Midjourney", "https://www.midjourney.com"),
|
||||
("https://cdn.midjourney.com/a/0_0.png", "Midjourney", "https://www.midjourney.com"),
|
||||
("https://sora.chatgpt.com/backend/nf/create", "Sora", "https://sora.chatgpt.com"),
|
||||
("https://videos.openai.com/a.mp4", "Sora", "https://sora.chatgpt.com"),
|
||||
("https://chatgpt.com/backend-api/sentinel", "ChatGPT", "https://chatgpt.com"),
|
||||
("https://api.worldlabs.ai/api/v1/worlds", "World Labs", "https://marble.worldlabs.ai"),
|
||||
("https://static.kinovi.ai/a.png", "Kinovi", "https://kinovi.ai"),
|
||||
("https://static.seedance2-pro.com/materials/a.png", "Seedance 2 Pro", "https://seedance2-pro.com"),
|
||||
("https://v3b.fal.media/files/a.png", "fal", "https://fal.ai"),
|
||||
("https://console.gmicloud.ai/api", "GMI Cloud", "https://console.gmicloud.ai"),
|
||||
] {
|
||||
let Destination::ThirdParty(provider) = classify(destination) else {
|
||||
panic!("{destination} is not a third party");
|
||||
};
|
||||
assert_eq!((provider.name, provider.origin), (name, origin));
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn lookalikes_and_others_are_unknown() {
|
||||
for destination in [
|
||||
"https://api.storyteller.ai.example.com/v1/session",
|
||||
"https://notstoryteller.ai/",
|
||||
"https://evilgrok.com/",
|
||||
"https://grok.com.example.com/",
|
||||
"https://www.gravatar.com/avatar/a",
|
||||
"https://storage.googleapis.com/bucket/a.png",
|
||||
"http://192.168.1.2:12345/v1/session",
|
||||
"tauri://localhost",
|
||||
] {
|
||||
assert_eq!(classify(destination), Destination::Unknown, "{destination}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn classify(destination: &str) -> Destination {
|
||||
Destination::classify(&Url::parse(destination).unwrap())
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,12 @@
|
|||
//! artcraft_client_identity
|
||||
//!
|
||||
//! How the ArtCraft desktop app identifies itself (`Origin`, `User-Agent`) to each destination.
|
||||
//!
|
||||
//! Our own APIs get the declared ArtCraft desktop identity. Known third-party providers get an
|
||||
//! identity close to their own website, and never ours. The native third-party Rust clients
|
||||
//! (Grok, Midjourney, Sora, etc.) set their own headers and do not use this crate.
|
||||
|
||||
pub mod destination;
|
||||
pub mod origins;
|
||||
pub mod third_party_provider;
|
||||
pub mod user_agents;
|
||||
|
|
@ -0,0 +1,11 @@
|
|||
/// `Origin` sent to our deployed APIs.
|
||||
/// Keep in sync with `actix_cors_configs/src/configs/artcraft_desktop.rs` in artcraft-services.
|
||||
pub const ARTCRAFT_DESKTOP_ORIGIN: &str = "https://desktop.getartcraft.com";
|
||||
|
||||
/// `Origin` sent to a locally running API (eg. `http://localhost:12345`).
|
||||
/// Development CORS accepts any `localhost` origin.
|
||||
pub const ARTCRAFT_DESKTOP_DEVELOPMENT_ORIGIN: &str = "http://localhost";
|
||||
|
||||
/// `Origin` packaged builds historically sent everywhere. Now only sent to our non-API hosts
|
||||
/// (CDNs, websites), where it is left unchanged.
|
||||
pub const LEGACY_DESKTOP_ORIGIN: &str = "https://studio.storyteller.ai";
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
/// A third-party provider, and the `Origin` of its own website that we present to its hosts.
|
||||
#[derive(Debug, PartialEq, Eq)]
|
||||
pub struct ThirdPartyProvider {
|
||||
pub name: &'static str,
|
||||
/// Registrable domains; subdomains match too.
|
||||
pub domains: &'static [&'static str],
|
||||
pub origin: &'static str,
|
||||
}
|
||||
|
||||
/// Every third party that gets an invented origin. Origins match what each provider's native
|
||||
/// Rust client already sends. First match wins, so more specific entries go first.
|
||||
pub const THIRD_PARTY_PROVIDERS: [ThirdPartyProvider; 9] = [
|
||||
ThirdPartyProvider { name: "Grok", domains: &["grok.com", "x.ai"], origin: "https://grok.com" },
|
||||
ThirdPartyProvider { name: "Midjourney", domains: &["midjourney.com"], origin: "https://www.midjourney.com" },
|
||||
ThirdPartyProvider {
|
||||
name: "Sora",
|
||||
domains: &["sora.chatgpt.com", "sora.com", "openai.com", "oaistatic.com"],
|
||||
origin: "https://sora.chatgpt.com",
|
||||
},
|
||||
ThirdPartyProvider { name: "ChatGPT", domains: &["chatgpt.com"], origin: "https://chatgpt.com" },
|
||||
ThirdPartyProvider {
|
||||
name: "World Labs",
|
||||
domains: &["worldlabs.ai", "wlt-ai.art"],
|
||||
origin: "https://marble.worldlabs.ai",
|
||||
},
|
||||
ThirdPartyProvider { name: "Kinovi", domains: &["kinovi.ai"], origin: "https://kinovi.ai" },
|
||||
ThirdPartyProvider { name: "Seedance 2 Pro", domains: &["seedance2-pro.com"], origin: "https://seedance2-pro.com" },
|
||||
ThirdPartyProvider { name: "fal", domains: &["fal.ai", "fal.run", "fal.media"], origin: "https://fal.ai" },
|
||||
ThirdPartyProvider { name: "GMI Cloud", domains: &["gmicloud.ai"], origin: "https://console.gmicloud.ai" },
|
||||
];
|
||||
|
|
@ -0,0 +1,23 @@
|
|||
/// `User-Agent` sent to our own APIs and hosts.
|
||||
///
|
||||
/// TODO: Change to "artcraft-desktop/1.0" once the backend services can parse it.
|
||||
pub const ARTCRAFT_DESKTOP_USER_AGENT: &str = "storyteller-client/1.0";
|
||||
|
||||
/// `User-Agent` sent to third parties: the default of the OS webview Tauri runs on
|
||||
/// (WKWebView on macOS, WebView2 on Windows, WebKitGTK on Linux).
|
||||
///
|
||||
/// NB: These are hardcoded snapshots and WILL go out of date as the OS webviews update
|
||||
/// (especially WebView2's Chrome/Edge version, which auto-updates on Windows). Rust can't read
|
||||
/// the webview's real `navigator.userAgent`, so refresh these periodically by checking
|
||||
/// `navigator.userAgent` in the desktop app's devtools on each OS.
|
||||
#[cfg(target_os = "macos")]
|
||||
pub const WEBVIEW_USER_AGENT: &str =
|
||||
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko)";
|
||||
|
||||
#[cfg(target_os = "windows")]
|
||||
pub const WEBVIEW_USER_AGENT: &str =
|
||||
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/140.0.0.0 Safari/537.36 Edg/140.0.0.0";
|
||||
|
||||
#[cfg(not(any(target_os = "macos", target_os = "windows")))]
|
||||
pub const WEBVIEW_USER_AGENT: &str =
|
||||
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/605.1.15 (KHTML, like Gecko)";
|
||||
|
|
@ -192,6 +192,12 @@ pub enum InferenceModelType {
|
|||
Kling3p0Pro,
|
||||
#[serde(rename = "happy_horse_1p0")]
|
||||
HappyHorse1p0,
|
||||
|
||||
#[serde(rename = "wan_3p0")]
|
||||
Wan3p0,
|
||||
|
||||
#[serde(rename = "wan_3p0_prime")]
|
||||
Wan3p0Prime,
|
||||
#[serde(rename = "minimax_h3")]
|
||||
MinimaxH3,
|
||||
#[serde(rename = "minimax_h3_turbo")]
|
||||
|
|
@ -394,6 +400,8 @@ impl InferenceModelType {
|
|||
Self::Kling3p0Standard => "kling_3p0_standard",
|
||||
Self::Kling3p0Pro => "kling_3p0_pro",
|
||||
Self::HappyHorse1p0 => "happy_horse_1p0",
|
||||
Self::Wan3p0 => "wan_3p0",
|
||||
Self::Wan3p0Prime => "wan_3p0_prime",
|
||||
Self::MinimaxH3 => "minimax_h3",
|
||||
Self::MinimaxH3Turbo => "minimax_h3_turbo",
|
||||
Self::MinimaxH3Ultra => "minimax_h3_ultra",
|
||||
|
|
@ -534,6 +542,8 @@ impl InferenceModelType {
|
|||
"kling_3p0_standard" => Ok(Self::Kling3p0Standard),
|
||||
"kling_3p0_pro" => Ok(Self::Kling3p0Pro),
|
||||
"happy_horse_1p0" => Ok(Self::HappyHorse1p0),
|
||||
"wan_3p0" => Ok(Self::Wan3p0),
|
||||
"wan_3p0_prime" => Ok(Self::Wan3p0Prime),
|
||||
"minimax_h3" => Ok(Self::MinimaxH3),
|
||||
"minimax_h3_turbo" => Ok(Self::MinimaxH3Turbo),
|
||||
"minimax_h3_ultra" => Ok(Self::MinimaxH3Ultra),
|
||||
|
|
@ -678,6 +688,8 @@ impl InferenceModelType {
|
|||
Self::Kling3p0Standard,
|
||||
Self::Kling3p0Pro,
|
||||
Self::HappyHorse1p0,
|
||||
Self::Wan3p0,
|
||||
Self::Wan3p0Prime,
|
||||
Self::MinimaxH3,
|
||||
Self::MinimaxH3Turbo,
|
||||
Self::MinimaxH3Ultra,
|
||||
|
|
@ -805,6 +817,8 @@ impl InferenceModelType {
|
|||
CommonModelType::Kling3p0Standard => Self::Kling3p0Standard,
|
||||
CommonModelType::Kling3p0Pro => Self::Kling3p0Pro,
|
||||
CommonModelType::HappyHorse1p0 => Self::HappyHorse1p0,
|
||||
CommonModelType::Wan3p0 => Self::Wan3p0,
|
||||
CommonModelType::Wan3p0Prime => Self::Wan3p0Prime,
|
||||
CommonModelType::MinimaxH3 => Self::MinimaxH3,
|
||||
CommonModelType::MinimaxH3Turbo => Self::MinimaxH3Turbo,
|
||||
CommonModelType::MinimaxH3Ultra => Self::MinimaxH3Ultra,
|
||||
|
|
@ -954,6 +968,8 @@ mod tests {
|
|||
assert_serialization(InferenceModelType::Kling3p0Standard, "kling_3p0_standard");
|
||||
assert_serialization(InferenceModelType::Kling3p0Pro, "kling_3p0_pro");
|
||||
assert_serialization(InferenceModelType::HappyHorse1p0, "happy_horse_1p0");
|
||||
assert_serialization(InferenceModelType::Wan3p0, "wan_3p0");
|
||||
assert_serialization(InferenceModelType::Wan3p0Prime, "wan_3p0_prime");
|
||||
assert_serialization(InferenceModelType::MinimaxH3, "minimax_h3");
|
||||
assert_serialization(InferenceModelType::MinimaxH3Turbo, "minimax_h3_turbo");
|
||||
assert_serialization(InferenceModelType::MinimaxH3Ultra, "minimax_h3_ultra");
|
||||
|
|
@ -1090,6 +1106,8 @@ mod tests {
|
|||
assert_eq!(InferenceModelType::Kling3p0Standard.to_str(), "kling_3p0_standard");
|
||||
assert_eq!(InferenceModelType::Kling3p0Pro.to_str(), "kling_3p0_pro");
|
||||
assert_eq!(InferenceModelType::HappyHorse1p0.to_str(), "happy_horse_1p0");
|
||||
assert_eq!(InferenceModelType::Wan3p0.to_str(), "wan_3p0");
|
||||
assert_eq!(InferenceModelType::Wan3p0Prime.to_str(), "wan_3p0_prime");
|
||||
assert_eq!(InferenceModelType::MinimaxH3.to_str(), "minimax_h3");
|
||||
assert_eq!(InferenceModelType::MinimaxH3Turbo.to_str(), "minimax_h3_turbo");
|
||||
assert_eq!(InferenceModelType::MinimaxH3Ultra.to_str(), "minimax_h3_ultra");
|
||||
|
|
@ -1228,6 +1246,8 @@ mod tests {
|
|||
assert_eq!(InferenceModelType::from_str("kling_3p0_standard").unwrap(), InferenceModelType::Kling3p0Standard);
|
||||
assert_eq!(InferenceModelType::from_str("kling_3p0_pro").unwrap(), InferenceModelType::Kling3p0Pro);
|
||||
assert_eq!(InferenceModelType::from_str("happy_horse_1p0").unwrap(), InferenceModelType::HappyHorse1p0);
|
||||
assert_eq!(InferenceModelType::from_str("wan_3p0").unwrap(), InferenceModelType::Wan3p0);
|
||||
assert_eq!(InferenceModelType::from_str("wan_3p0_prime").unwrap(), InferenceModelType::Wan3p0Prime);
|
||||
assert_eq!(InferenceModelType::from_str("minimax_h3").unwrap(), InferenceModelType::MinimaxH3);
|
||||
assert_eq!(InferenceModelType::from_str("minimax_h3_turbo").unwrap(), InferenceModelType::MinimaxH3Turbo);
|
||||
assert_eq!(InferenceModelType::from_str("minimax_h3_ultra").unwrap(), InferenceModelType::MinimaxH3Ultra);
|
||||
|
|
|
|||
|
|
@ -163,6 +163,12 @@ pub enum CommonModelType {
|
|||
Kling3p0Pro,
|
||||
#[serde(rename = "happy_horse_1p0")]
|
||||
HappyHorse1p0,
|
||||
|
||||
#[serde(rename = "wan_3p0")]
|
||||
Wan3p0,
|
||||
|
||||
#[serde(rename = "wan_3p0_prime")]
|
||||
Wan3p0Prime,
|
||||
#[serde(rename = "minimax_h3")]
|
||||
MinimaxH3,
|
||||
#[serde(rename = "minimax_h3_turbo")]
|
||||
|
|
@ -349,6 +355,8 @@ impl CommonModelType {
|
|||
Self::Kling3p0Standard => "kling_3p0_standard",
|
||||
Self::Kling3p0Pro => "kling_3p0_pro",
|
||||
Self::HappyHorse1p0 => "happy_horse_1p0",
|
||||
Self::Wan3p0 => "wan_3p0",
|
||||
Self::Wan3p0Prime => "wan_3p0_prime",
|
||||
Self::MinimaxH3 => "minimax_h3",
|
||||
Self::MinimaxH3Turbo => "minimax_h3_turbo",
|
||||
Self::MinimaxH3Ultra => "minimax_h3_ultra",
|
||||
|
|
@ -472,6 +480,8 @@ impl CommonModelType {
|
|||
"kling_3p0_standard" => Ok(Self::Kling3p0Standard),
|
||||
"kling_3p0_pro" => Ok(Self::Kling3p0Pro),
|
||||
"happy_horse_1p0" => Ok(Self::HappyHorse1p0),
|
||||
"wan_3p0" => Ok(Self::Wan3p0),
|
||||
"wan_3p0_prime" => Ok(Self::Wan3p0Prime),
|
||||
"minimax_h3" => Ok(Self::MinimaxH3),
|
||||
"minimax_h3_turbo" => Ok(Self::MinimaxH3Turbo),
|
||||
"minimax_h3_ultra" => Ok(Self::MinimaxH3Ultra),
|
||||
|
|
@ -599,6 +609,8 @@ impl CommonModelType {
|
|||
Self::Kling3p0Standard,
|
||||
Self::Kling3p0Pro,
|
||||
Self::HappyHorse1p0,
|
||||
Self::Wan3p0,
|
||||
Self::Wan3p0Prime,
|
||||
Self::MinimaxH3,
|
||||
Self::MinimaxH3Turbo,
|
||||
Self::MinimaxH3Ultra,
|
||||
|
|
@ -723,6 +735,8 @@ impl CommonModelType {
|
|||
Self::Kling3p0Standard => CommonModelClass::Video,
|
||||
Self::Kling3p0Pro => CommonModelClass::Video,
|
||||
Self::HappyHorse1p0 => CommonModelClass::Video,
|
||||
Self::Wan3p0 => CommonModelClass::Video,
|
||||
Self::Wan3p0Prime => CommonModelClass::Video,
|
||||
Self::MinimaxH3 => CommonModelClass::Video,
|
||||
Self::MinimaxH3Turbo => CommonModelClass::Video,
|
||||
Self::MinimaxH3Ultra => CommonModelClass::Video,
|
||||
|
|
@ -856,6 +870,8 @@ mod tests {
|
|||
assert_serialization(CommonModelType::Kling3p0Standard, "kling_3p0_standard");
|
||||
assert_serialization(CommonModelType::Kling3p0Pro, "kling_3p0_pro");
|
||||
assert_serialization(CommonModelType::HappyHorse1p0, "happy_horse_1p0");
|
||||
assert_serialization(CommonModelType::Wan3p0, "wan_3p0");
|
||||
assert_serialization(CommonModelType::Wan3p0Prime, "wan_3p0_prime");
|
||||
assert_serialization(CommonModelType::MinimaxH3, "minimax_h3");
|
||||
assert_serialization(CommonModelType::MinimaxH3Turbo, "minimax_h3_turbo");
|
||||
assert_serialization(CommonModelType::MinimaxH3Ultra, "minimax_h3_ultra");
|
||||
|
|
@ -974,6 +990,8 @@ mod tests {
|
|||
assert_eq!(CommonModelType::Kling3p0Standard.to_str(), "kling_3p0_standard");
|
||||
assert_eq!(CommonModelType::Kling3p0Pro.to_str(), "kling_3p0_pro");
|
||||
assert_eq!(CommonModelType::HappyHorse1p0.to_str(), "happy_horse_1p0");
|
||||
assert_eq!(CommonModelType::Wan3p0.to_str(), "wan_3p0");
|
||||
assert_eq!(CommonModelType::Wan3p0Prime.to_str(), "wan_3p0_prime");
|
||||
assert_eq!(CommonModelType::MinimaxH3.to_str(), "minimax_h3");
|
||||
assert_eq!(CommonModelType::MinimaxH3Turbo.to_str(), "minimax_h3_turbo");
|
||||
assert_eq!(CommonModelType::MinimaxH3Ultra.to_str(), "minimax_h3_ultra");
|
||||
|
|
@ -1094,6 +1112,8 @@ mod tests {
|
|||
assert_eq!(CommonModelType::from_str("kling_3p0_standard").unwrap(), CommonModelType::Kling3p0Standard);
|
||||
assert_eq!(CommonModelType::from_str("kling_3p0_pro").unwrap(), CommonModelType::Kling3p0Pro);
|
||||
assert_eq!(CommonModelType::from_str("happy_horse_1p0").unwrap(), CommonModelType::HappyHorse1p0);
|
||||
assert_eq!(CommonModelType::from_str("wan_3p0").unwrap(), CommonModelType::Wan3p0);
|
||||
assert_eq!(CommonModelType::from_str("wan_3p0_prime").unwrap(), CommonModelType::Wan3p0Prime);
|
||||
assert_eq!(CommonModelType::from_str("minimax_h3").unwrap(), CommonModelType::MinimaxH3);
|
||||
assert_eq!(CommonModelType::from_str("minimax_h3_turbo").unwrap(), CommonModelType::MinimaxH3Turbo);
|
||||
assert_eq!(CommonModelType::from_str("minimax_h3_ultra").unwrap(), CommonModelType::MinimaxH3Ultra);
|
||||
|
|
@ -1158,7 +1178,7 @@ mod tests {
|
|||
#[test]
|
||||
fn all_variants() {
|
||||
let mut variants = CommonModelType::all_variants();
|
||||
assert_eq!(variants.len(), 108);
|
||||
assert_eq!(variants.len(), 110);
|
||||
// Image models
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::Flux1Dev));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::Flux1Schnell));
|
||||
|
|
@ -1214,6 +1234,8 @@ mod tests {
|
|||
assert_eq!(variants.pop_first(), Some(CommonModelType::Kling3p0Standard));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::Kling3p0Pro));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::HappyHorse1p0));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::Wan3p0));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::Wan3p0Prime));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::MinimaxH3));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::MinimaxH3Turbo));
|
||||
assert_eq!(variants.pop_first(), Some(CommonModelType::MinimaxH3Ultra));
|
||||
|
|
|
|||
|
|
@ -13,11 +13,22 @@ pub const MAX_LENGTH: usize = 16;
|
|||
|
||||
/// Common quality levels for generation.
|
||||
///
|
||||
/// Declared from "let the model decide" through the highest tier down to the lowest, so the
|
||||
/// derived `Ord` (and `all_variants()`) reads top-down. Not every model supports every tier;
|
||||
/// the router clamps unsupported tiers to the nearest one the model offers.
|
||||
///
|
||||
/// NB: Keep the max serialized length to 16 characters.
|
||||
#[cfg_attr(test, derive(EnumIter, EnumCount))]
|
||||
#[derive(Clone, Copy, PartialEq, Eq, Hash, Ord, PartialOrd, Serialize, Deserialize, ToSchema)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum CommonQuality {
|
||||
/// Let the model choose. Models without a native "auto" treat this as their default tier.
|
||||
Auto,
|
||||
/// Highest tier (GPT Image 2.5 `max`).
|
||||
Max,
|
||||
/// Between High and Max (GPT Image 2.5 `xhigh`).
|
||||
#[serde(rename = "xhigh")]
|
||||
XHigh,
|
||||
High,
|
||||
Medium,
|
||||
Low,
|
||||
|
|
@ -30,6 +41,9 @@ impl_mysql_from_row!(CommonQuality);
|
|||
impl CommonQuality {
|
||||
pub fn to_str(&self) -> &'static str {
|
||||
match self {
|
||||
Self::Auto => "auto",
|
||||
Self::Max => "max",
|
||||
Self::XHigh => "xhigh",
|
||||
Self::High => "high",
|
||||
Self::Medium => "medium",
|
||||
Self::Low => "low",
|
||||
|
|
@ -38,6 +52,9 @@ impl CommonQuality {
|
|||
|
||||
pub fn from_str(value: &str) -> Result<Self, EnumError> {
|
||||
match value {
|
||||
"auto" => Ok(Self::Auto),
|
||||
"max" => Ok(Self::Max),
|
||||
"xhigh" => Ok(Self::XHigh),
|
||||
"high" => Ok(Self::High),
|
||||
"medium" => Ok(Self::Medium),
|
||||
"low" => Ok(Self::Low),
|
||||
|
|
@ -47,6 +64,9 @@ impl CommonQuality {
|
|||
|
||||
pub fn all_variants() -> BTreeSet<Self> {
|
||||
BTreeSet::from([
|
||||
Self::Auto,
|
||||
Self::Max,
|
||||
Self::XHigh,
|
||||
Self::High,
|
||||
Self::Medium,
|
||||
Self::Low,
|
||||
|
|
@ -66,6 +86,9 @@ mod tests {
|
|||
|
||||
#[test]
|
||||
fn test_serialization() {
|
||||
assert_serialization(CommonQuality::Auto, "auto");
|
||||
assert_serialization(CommonQuality::Max, "max");
|
||||
assert_serialization(CommonQuality::XHigh, "xhigh");
|
||||
assert_serialization(CommonQuality::High, "high");
|
||||
assert_serialization(CommonQuality::Medium, "medium");
|
||||
assert_serialization(CommonQuality::Low, "low");
|
||||
|
|
@ -73,6 +96,9 @@ mod tests {
|
|||
|
||||
#[test]
|
||||
fn to_str() {
|
||||
assert_eq!(CommonQuality::Auto.to_str(), "auto");
|
||||
assert_eq!(CommonQuality::Max.to_str(), "max");
|
||||
assert_eq!(CommonQuality::XHigh.to_str(), "xhigh");
|
||||
assert_eq!(CommonQuality::High.to_str(), "high");
|
||||
assert_eq!(CommonQuality::Medium.to_str(), "medium");
|
||||
assert_eq!(CommonQuality::Low.to_str(), "low");
|
||||
|
|
@ -80,6 +106,9 @@ mod tests {
|
|||
|
||||
#[test]
|
||||
fn from_str() {
|
||||
assert_eq!(CommonQuality::from_str("auto").unwrap(), CommonQuality::Auto);
|
||||
assert_eq!(CommonQuality::from_str("max").unwrap(), CommonQuality::Max);
|
||||
assert_eq!(CommonQuality::from_str("xhigh").unwrap(), CommonQuality::XHigh);
|
||||
assert_eq!(CommonQuality::from_str("high").unwrap(), CommonQuality::High);
|
||||
assert_eq!(CommonQuality::from_str("medium").unwrap(), CommonQuality::Medium);
|
||||
assert_eq!(CommonQuality::from_str("low").unwrap(), CommonQuality::Low);
|
||||
|
|
@ -99,7 +128,10 @@ mod tests {
|
|||
#[test]
|
||||
fn all_variants() {
|
||||
let mut variants = CommonQuality::all_variants();
|
||||
assert_eq!(variants.len(), 3);
|
||||
assert_eq!(variants.len(), 6);
|
||||
assert_eq!(variants.pop_first(), Some(CommonQuality::Auto));
|
||||
assert_eq!(variants.pop_first(), Some(CommonQuality::Max));
|
||||
assert_eq!(variants.pop_first(), Some(CommonQuality::XHigh));
|
||||
assert_eq!(variants.pop_first(), Some(CommonQuality::High));
|
||||
assert_eq!(variants.pop_first(), Some(CommonQuality::Medium));
|
||||
assert_eq!(variants.pop_first(), Some(CommonQuality::Low));
|
||||
|
|
|
|||
|
|
@ -51,6 +51,12 @@ pub enum CommonVideoModel {
|
|||
#[serde(rename = "happy_horse_1p0")]
|
||||
HappyHorse1p0,
|
||||
|
||||
#[serde(rename = "wan_3p0")]
|
||||
Wan3p0,
|
||||
|
||||
#[serde(rename = "wan_3p0_prime")]
|
||||
Wan3p0Prime,
|
||||
|
||||
#[serde(rename = "minimax_h3")]
|
||||
MinimaxH3,
|
||||
|
||||
|
|
@ -168,6 +174,8 @@ impl CommonVideoModel {
|
|||
Self::Kling3p0Standard => CommonModelType::Kling3p0Standard,
|
||||
Self::Kling3p0Pro => CommonModelType::Kling3p0Pro,
|
||||
Self::HappyHorse1p0 => CommonModelType::HappyHorse1p0,
|
||||
Self::Wan3p0 => CommonModelType::Wan3p0,
|
||||
Self::Wan3p0Prime => CommonModelType::Wan3p0Prime,
|
||||
Self::MinimaxH3 => CommonModelType::MinimaxH3,
|
||||
Self::MinimaxH3Turbo => CommonModelType::MinimaxH3Turbo,
|
||||
Self::MinimaxH3Ultra => CommonModelType::MinimaxH3Ultra,
|
||||
|
|
@ -224,6 +232,8 @@ mod tests {
|
|||
assert_serialization(CommonVideoModel::Kling3p0Standard, "kling_3p0_standard");
|
||||
assert_serialization(CommonVideoModel::Kling3p0Pro, "kling_3p0_pro");
|
||||
assert_serialization(CommonVideoModel::HappyHorse1p0, "happy_horse_1p0");
|
||||
assert_serialization(CommonVideoModel::Wan3p0, "wan_3p0");
|
||||
assert_serialization(CommonVideoModel::Wan3p0Prime, "wan_3p0_prime");
|
||||
assert_serialization(CommonVideoModel::MinimaxH3, "minimax_h3");
|
||||
assert_serialization(CommonVideoModel::MinimaxH3Turbo, "minimax_h3_turbo");
|
||||
assert_serialization(CommonVideoModel::MinimaxH3Ultra, "minimax_h3_ultra");
|
||||
|
|
@ -273,6 +283,8 @@ mod tests {
|
|||
("kling_3p0_standard", CommonVideoModel::Kling3p0Standard),
|
||||
("kling_3p0_pro", CommonVideoModel::Kling3p0Pro),
|
||||
("happy_horse_1p0", CommonVideoModel::HappyHorse1p0),
|
||||
("wan_3p0", CommonVideoModel::Wan3p0),
|
||||
("wan_3p0_prime", CommonVideoModel::Wan3p0Prime),
|
||||
("minimax_h3", CommonVideoModel::MinimaxH3),
|
||||
("minimax_h3_turbo", CommonVideoModel::MinimaxH3Turbo),
|
||||
("minimax_h3_ultra", CommonVideoModel::MinimaxH3Ultra),
|
||||
|
|
@ -329,6 +341,8 @@ mod tests {
|
|||
CommonVideoModel::Kling3p0Standard,
|
||||
CommonVideoModel::Kling3p0Pro,
|
||||
CommonVideoModel::HappyHorse1p0,
|
||||
CommonVideoModel::Wan3p0,
|
||||
CommonVideoModel::Wan3p0Prime,
|
||||
CommonVideoModel::MinimaxH3,
|
||||
CommonVideoModel::MinimaxH3Turbo,
|
||||
CommonVideoModel::MinimaxH3Ultra,
|
||||
|
|
@ -384,6 +398,8 @@ mod tests {
|
|||
(CommonVideoModel::Kling3p0Standard, CommonModelType::Kling3p0Standard),
|
||||
(CommonVideoModel::Kling3p0Pro, CommonModelType::Kling3p0Pro),
|
||||
(CommonVideoModel::HappyHorse1p0, CommonModelType::HappyHorse1p0),
|
||||
(CommonVideoModel::Wan3p0, CommonModelType::Wan3p0),
|
||||
(CommonVideoModel::Wan3p0Prime, CommonModelType::Wan3p0Prime),
|
||||
(CommonVideoModel::MinimaxH3, CommonModelType::MinimaxH3),
|
||||
(CommonVideoModel::MinimaxH3Turbo, CommonModelType::MinimaxH3Turbo),
|
||||
(CommonVideoModel::MinimaxH3Ultra, CommonModelType::MinimaxH3Ultra),
|
||||
|
|
|
|||
|
|
@ -0,0 +1,162 @@
|
|||
use std::collections::BTreeSet;
|
||||
|
||||
#[cfg(test)]
|
||||
use strum::EnumCount;
|
||||
#[cfg(test)]
|
||||
use strum::EnumIter;
|
||||
use utoipa::ToSchema;
|
||||
|
||||
use crate::error::enum_error::EnumError;
|
||||
|
||||
/// Maximum serialized string length for database storage.
|
||||
pub const MAX_LENGTH: usize = 16;
|
||||
|
||||
/// Output video containers for generation. Unset options retain the provider default.
|
||||
/// Models without output-format support ignore this option.
|
||||
///
|
||||
/// NB: Keep the max serialized length to 16 characters.
|
||||
#[cfg_attr(test, derive(EnumIter, EnumCount))]
|
||||
#[derive(Clone, Copy, PartialEq, Eq, Hash, Ord, PartialOrd, Serialize, Deserialize, ToSchema)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub enum CommonVideoOutputFormat {
|
||||
#[serde(rename = "mp4")]
|
||||
Mp4,
|
||||
#[serde(rename = "mov")]
|
||||
Mov,
|
||||
}
|
||||
|
||||
impl_enum_display_and_debug_using_to_str!(CommonVideoOutputFormat);
|
||||
impl_mysql_enum_coders!(CommonVideoOutputFormat);
|
||||
impl_mysql_from_row!(CommonVideoOutputFormat);
|
||||
|
||||
impl CommonVideoOutputFormat {
|
||||
pub fn to_str(&self) -> &'static str {
|
||||
match self {
|
||||
Self::Mp4 => "mp4",
|
||||
Self::Mov => "mov",
|
||||
}
|
||||
}
|
||||
|
||||
pub fn from_str(value: &str) -> Result<Self, EnumError> {
|
||||
match value {
|
||||
"mp4" => Ok(Self::Mp4),
|
||||
"mov" => Ok(Self::Mov),
|
||||
_ => Err(EnumError::CouldNotConvertFromString(value.to_string())),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn all_variants() -> BTreeSet<Self> {
|
||||
BTreeSet::from([
|
||||
Self::Mp4,
|
||||
Self::Mov,
|
||||
])
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use crate::common::generation::common_video_output_format::CommonVideoOutputFormat;
|
||||
use crate::common::generation::common_video_output_format::MAX_LENGTH;
|
||||
use crate::test_helpers::assert_serialization;
|
||||
|
||||
mod explicit_checks {
|
||||
use super::*;
|
||||
use crate::error::enum_error::EnumError;
|
||||
|
||||
#[test]
|
||||
fn test_serialization() {
|
||||
assert_serialization(CommonVideoOutputFormat::Mp4, "mp4");
|
||||
assert_serialization(CommonVideoOutputFormat::Mov, "mov");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn to_str() {
|
||||
assert_eq!(CommonVideoOutputFormat::Mp4.to_str(), "mp4");
|
||||
assert_eq!(CommonVideoOutputFormat::Mov.to_str(), "mov");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_str() {
|
||||
assert_eq!(CommonVideoOutputFormat::from_str("mp4").unwrap(), CommonVideoOutputFormat::Mp4);
|
||||
assert_eq!(CommonVideoOutputFormat::from_str("mov").unwrap(), CommonVideoOutputFormat::Mov);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_str_err() {
|
||||
let result = CommonVideoOutputFormat::from_str("invalid");
|
||||
assert!(result.is_err());
|
||||
if let Err(EnumError::CouldNotConvertFromString(value)) = result {
|
||||
assert_eq!(value, "invalid");
|
||||
} else {
|
||||
panic!("Expected EnumError::CouldNotConvertFromString");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn all_variants() {
|
||||
let mut variants = CommonVideoOutputFormat::all_variants();
|
||||
assert_eq!(variants.len(), 2);
|
||||
assert_eq!(variants.pop_first(), Some(CommonVideoOutputFormat::Mp4));
|
||||
assert_eq!(variants.pop_first(), Some(CommonVideoOutputFormat::Mov));
|
||||
assert_eq!(variants.pop_first(), None);
|
||||
}
|
||||
}
|
||||
|
||||
mod mechanical_checks {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn variant_length() {
|
||||
use strum::IntoEnumIterator;
|
||||
assert_eq!(CommonVideoOutputFormat::all_variants().len(), CommonVideoOutputFormat::iter().len());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn round_trip() {
|
||||
for variant in CommonVideoOutputFormat::all_variants() {
|
||||
assert_eq!(variant, CommonVideoOutputFormat::from_str(variant.to_str()).unwrap());
|
||||
assert_eq!(variant, CommonVideoOutputFormat::from_str(&format!("{}", variant)).unwrap());
|
||||
assert_eq!(variant, CommonVideoOutputFormat::from_str(&format!("{:?}", variant)).unwrap());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn serialized_length_ok_for_database() {
|
||||
for variant in CommonVideoOutputFormat::all_variants() {
|
||||
let serialized = variant.to_str();
|
||||
assert!(serialized.len() > 0, "variant {:?} is too short", variant);
|
||||
assert!(serialized.len() <= MAX_LENGTH, "variant {:?} is too long via to_str()", variant);
|
||||
}
|
||||
for variant in CommonVideoOutputFormat::all_variants() {
|
||||
let json = serde_json::to_string(&variant).unwrap().replace('"', "");
|
||||
assert!(json.len() <= MAX_LENGTH, "variant {:?} is too long via JSON: {:?}", variant, json);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn serialized_names_must_not_contain_dots() {
|
||||
for variant in CommonVideoOutputFormat::all_variants() {
|
||||
let to_str_value = variant.to_str();
|
||||
assert!(!to_str_value.contains('.'), "to_str() for {:?} contains a dot: {:?}", variant, to_str_value);
|
||||
|
||||
let json_value = serde_json::to_string(&variant).unwrap().replace('"', "");
|
||||
assert!(!json_value.contains('.'), "JSON serialization for {:?} contains a dot: {:?}", variant, json_value);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn serialized_names_must_only_contain_lowercase_alphanumeric_and_underscore() {
|
||||
let valid_pattern = regex::Regex::new(r"^[a-z0-9_]+$").unwrap();
|
||||
|
||||
for variant in CommonVideoOutputFormat::all_variants() {
|
||||
let to_str_value = variant.to_str();
|
||||
assert!(valid_pattern.is_match(to_str_value),
|
||||
"to_str() for {:?} contains invalid characters: {:?} (only a-z, 0-9, _ allowed)", variant, to_str_value);
|
||||
|
||||
let json_value = serde_json::to_string(&variant).unwrap().replace('"', "");
|
||||
assert!(valid_pattern.is_match(&json_value),
|
||||
"JSON serialization for {:?} contains invalid characters: {:?} (only a-z, 0-9, _ allowed)", variant, json_value);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -15,3 +15,4 @@ pub mod model_creator;
|
|||
pub mod common_resolution;
|
||||
pub mod common_splat_model;
|
||||
pub mod common_video_model;
|
||||
pub mod common_video_output_format;
|
||||
|
|
|
|||
|
|
@ -36,6 +36,7 @@ urlpattern = "0.3"
|
|||
regex = "1"
|
||||
|
||||
[dependencies]
|
||||
artcraft_client_identity = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
tauri = { workspace = true }
|
||||
|
|
|
|||
|
|
@ -0,0 +1,207 @@
|
|||
use artcraft_client_identity::destination::Destination;
|
||||
use artcraft_client_identity::origins::LEGACY_DESKTOP_ORIGIN;
|
||||
use artcraft_client_identity::user_agents::{ARTCRAFT_DESKTOP_USER_AGENT, WEBVIEW_USER_AGENT};
|
||||
use http::{header, HeaderMap, HeaderValue};
|
||||
use url::{Host, Url};
|
||||
|
||||
/// The `User-Agent` to send when the frontend didn't set one.
|
||||
pub(crate) fn artcraft_default_user_agent(destination: &Url) -> &'static str {
|
||||
match Destination::classify(destination) {
|
||||
Destination::ArtcraftApi { .. } | Destination::ArtcraftHost => ARTCRAFT_DESKTOP_USER_AGENT,
|
||||
Destination::ThirdParty(_) | Destination::Unknown => WEBVIEW_USER_AGENT,
|
||||
}
|
||||
}
|
||||
|
||||
/// Replace the webview's own `Origin` (packaged `tauri://localhost` etc., or a Vite loopback
|
||||
/// origin at any port) with the identity chosen for the destination. The native HTTP bridge
|
||||
/// shares the desktop cookie jar, so its identity must not depend on the webview.
|
||||
pub(crate) fn normalize_artcraft_origin(destination: &Url, headers: &mut HeaderMap) {
|
||||
let origin = headers.get(header::ORIGIN).and_then(|value| value.to_str().ok());
|
||||
// Packaged-app webview origins on macOS and Windows.
|
||||
let packaged_origin = headers.get(header::ORIGIN).is_none()
|
||||
|| matches!(origin, Some("null" | "http://tauri.localhost" | "tauri://localhost"));
|
||||
// Development webview origins (Vite on any loopback port).
|
||||
let webview_origin = packaged_origin || origin.is_some_and(is_loopback_origin);
|
||||
|
||||
let replacement = match Destination::classify(destination) {
|
||||
// Our APIs (deployed or local) see the declared desktop identity.
|
||||
Destination::ArtcraftApi { origin } if webview_origin => origin,
|
||||
// Known third parties see an origin close to their own website.
|
||||
Destination::ThirdParty(provider) if webview_origin => provider.origin,
|
||||
// Our CDNs and websites keep the historical packaged-app handling.
|
||||
Destination::ArtcraftHost if packaged_origin => LEGACY_DESKTOP_ORIGIN,
|
||||
// Unknown hosts and ordinary website origins are left untouched.
|
||||
_ => return,
|
||||
};
|
||||
|
||||
// HeaderMap is a multimap; insert replaces all existing Origin values.
|
||||
headers.insert(header::ORIGIN, HeaderValue::from_static(replacement));
|
||||
}
|
||||
|
||||
fn is_loopback_origin(origin: &str) -> bool {
|
||||
let Ok(url) = Url::parse(origin) else {
|
||||
return false;
|
||||
};
|
||||
if !matches!(url.scheme(), "http" | "https") {
|
||||
return false;
|
||||
}
|
||||
match url.host() {
|
||||
Some(Host::Domain("localhost")) => true,
|
||||
Some(Host::Ipv4(address)) => address.is_loopback(),
|
||||
Some(Host::Ipv6(address)) => address.is_loopback(),
|
||||
_ => false,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
const API: &str = "https://api.storyteller.ai/v1/session";
|
||||
const LOCAL_API: &str = "http://localhost:12345/v1/session";
|
||||
const CDN: &str = "https://cdn-2.fakeyou.com/media/a/b.png";
|
||||
const GCS: &str = "https://storage.googleapis.com/bucket/a.png";
|
||||
const GROK: &str = "https://grok.com/rest/app-chat/conversations/new";
|
||||
const MIDJOURNEY: &str = "https://www.midjourney.com/api/submit-jobs";
|
||||
const UNKNOWN: &str = "https://www.gravatar.com/avatar/a";
|
||||
|
||||
const DEV_WEBVIEW_ORIGIN: &str = "http://127.0.0.1:5193";
|
||||
const PACKAGED_ORIGINS: [Option<&str>; 4] =
|
||||
[None, Some("null"), Some("http://tauri.localhost"), Some("tauri://localhost")];
|
||||
|
||||
mod development {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn normalizes_dynamic_ports_and_loopback_hosts() {
|
||||
for origin in [
|
||||
"http://127.0.0.1:5193",
|
||||
"http://127.0.0.1:5194",
|
||||
"http://127.0.0.1:65535",
|
||||
"http://localhost:5173",
|
||||
"http://[::1]:5193",
|
||||
] {
|
||||
assert_eq!(normalized(API, Some(origin)), "https://desktop.getartcraft.com");
|
||||
assert_eq!(normalized(LOCAL_API, Some(origin)), "http://localhost");
|
||||
assert_eq!(normalized(GROK, Some(origin)), "https://grok.com");
|
||||
}
|
||||
assert_eq!(
|
||||
normalized("https://api.fakeyou.com/v1/session", Some(DEV_WEBVIEW_ORIGIN)),
|
||||
"https://desktop.getartcraft.com"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn leaves_cdns_and_unknown_destinations_unchanged() {
|
||||
for destination in [
|
||||
CDN,
|
||||
GCS,
|
||||
UNKNOWN,
|
||||
"https://api.storyteller.ai.example.com/v1/session",
|
||||
"http://api.storyteller.ai/v1/session",
|
||||
] {
|
||||
assert_eq!(normalized(destination, Some(DEV_WEBVIEW_ORIGIN)), DEV_WEBVIEW_ORIGIN, "{destination}");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn leaves_non_webview_origins_unchanged() {
|
||||
for origin in [
|
||||
"https://app.getartcraft.com",
|
||||
"https://example.com",
|
||||
"http://localhost.example.com:5193",
|
||||
"http://192.168.1.2:5193",
|
||||
"not a URL",
|
||||
"",
|
||||
] {
|
||||
assert_eq!(normalized(API, Some(origin)), origin);
|
||||
assert_eq!(normalized(GROK, Some(origin)), origin);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
mod packaged {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn our_apis_get_desktop_identity() {
|
||||
for origin in PACKAGED_ORIGINS {
|
||||
assert_eq!(normalized(API, origin), "https://desktop.getartcraft.com");
|
||||
assert_eq!(normalized(LOCAL_API, origin), "http://localhost");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn third_parties_get_their_own_origin() {
|
||||
for origin in PACKAGED_ORIGINS {
|
||||
assert_eq!(normalized(GROK, origin), "https://grok.com");
|
||||
assert_eq!(normalized(MIDJOURNEY, origin), "https://www.midjourney.com");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn cdns_keep_legacy_origin() {
|
||||
for origin in PACKAGED_ORIGINS {
|
||||
assert_eq!(normalized(CDN, origin), "https://studio.storyteller.ai");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unknown_destinations_never_get_our_origin() {
|
||||
for destination in [UNKNOWN, GCS] {
|
||||
assert_eq!(normalized(destination, Some("tauri://localhost")), "tauri://localhost");
|
||||
assert_eq!(normalized(destination, Some("null")), "null");
|
||||
assert_eq!(normalized_or_missing(destination, None), None);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn replaces_duplicate_origin_values_without_changing_session_headers() {
|
||||
let mut headers = HeaderMap::new();
|
||||
headers.append(header::ORIGIN, HeaderValue::from_static(DEV_WEBVIEW_ORIGIN));
|
||||
headers.append(header::ORIGIN, HeaderValue::from_static(DEV_WEBVIEW_ORIGIN));
|
||||
headers.insert(header::COOKIE, HeaderValue::from_static("session=fixture"));
|
||||
normalize_artcraft_origin(&Url::parse(API).unwrap(), &mut headers);
|
||||
assert_eq!(headers.get_all(header::ORIGIN).iter().count(), 1);
|
||||
assert_eq!(headers[header::ORIGIN], "https://desktop.getartcraft.com");
|
||||
assert_eq!(headers[header::COOKIE], "session=fixture");
|
||||
}
|
||||
}
|
||||
|
||||
mod user_agent {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn ours_get_artcraft_user_agent() {
|
||||
for destination in [API, LOCAL_API, CDN] {
|
||||
assert_eq!(user_agent(destination), "storyteller-client/1.0");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn third_parties_get_webview_user_agent() {
|
||||
for destination in [GROK, MIDJOURNEY, GCS, UNKNOWN] {
|
||||
let user_agent = user_agent(destination);
|
||||
assert!(user_agent.starts_with("Mozilla/5.0 ("), "{user_agent}");
|
||||
assert!(!user_agent.contains("storyteller"), "{user_agent}");
|
||||
}
|
||||
}
|
||||
|
||||
fn user_agent(destination: &str) -> &'static str {
|
||||
artcraft_default_user_agent(&Url::parse(destination).unwrap())
|
||||
}
|
||||
}
|
||||
|
||||
fn normalized(destination: &str, origin: Option<&str>) -> String {
|
||||
normalized_or_missing(destination, origin).unwrap()
|
||||
}
|
||||
|
||||
fn normalized_or_missing(destination: &str, origin: Option<&str>) -> Option<String> {
|
||||
let mut headers = HeaderMap::new();
|
||||
if let Some(origin) = origin {
|
||||
headers.insert(header::ORIGIN, HeaderValue::from_str(origin).unwrap());
|
||||
}
|
||||
normalize_artcraft_origin(&Url::parse(destination).unwrap(), &mut headers);
|
||||
headers.get(header::ORIGIN).map(|value| value.to_str().unwrap().to_owned())
|
||||
}
|
||||
}
|
||||
|
|
@ -16,12 +16,11 @@ use tauri::{
|
|||
use tokio::sync::oneshot::{channel, Receiver, Sender};
|
||||
|
||||
use crate::{
|
||||
artcraft_identity::{artcraft_default_user_agent, normalize_artcraft_origin},
|
||||
scope::{Entry, Scope},
|
||||
Error, Http, Result,
|
||||
};
|
||||
|
||||
const HTTP_USER_AGENT: &str = concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"),);
|
||||
|
||||
struct ReqwestResponse(reqwest::Response);
|
||||
impl tauri::Resource for ReqwestResponse {}
|
||||
|
||||
|
|
@ -269,7 +268,7 @@ pub async fn fetch<R: Runtime>(
|
|||
builder = builder.cookie_provider(state.cookies_jar.clone());
|
||||
}
|
||||
|
||||
let mut request = builder.build()?.request(method.clone(), url);
|
||||
let mut request = builder.build()?.request(method.clone(), url.clone());
|
||||
|
||||
// POST and PUT requests should always have a 0 length content-length,
|
||||
// if there is no body. https://fetch.spec.whatwg.org/#http-network-or-cache-fetch
|
||||
|
|
@ -284,7 +283,7 @@ pub async fn fetch<R: Runtime>(
|
|||
}
|
||||
|
||||
if !headers.contains_key(header::USER_AGENT) {
|
||||
headers.append(header::USER_AGENT, HeaderValue::from_str(HTTP_USER_AGENT)?);
|
||||
headers.append(header::USER_AGENT, HeaderValue::from_static(artcraft_default_user_agent(&url)));
|
||||
}
|
||||
|
||||
// ensure we have an Origin header set
|
||||
|
|
@ -305,26 +304,7 @@ pub async fn fetch<R: Runtime>(
|
|||
headers.remove(header::ORIGIN);
|
||||
};
|
||||
|
||||
// NB(bt): In production builds, the origin seems to be "null". This is what our server CORS filter sees:
|
||||
// [2025-05-01T23:42:35Z WARN actix_cors_configs::configs::fakeyou] Invalid origin: "null"
|
||||
// [2025-05-01T23:42:35Z WARN actix_cors_configs::util::netlify_branch_domain_matches] Invalid origin: "null"
|
||||
// [2025-05-01T23:42:35Z WARN actix_cors_configs::util::netlify_branch_domain_matches] Invalid origin: "null"
|
||||
// ...
|
||||
// [2025-05-01T23:42:35Z WARN actix_cors_configs::util::netlify_branch_domain_matches] Invalid origin: "null"
|
||||
// [2025-05-01T23:42:35Z INFO actix_web::middleware::logger] [storyteller-web-85f9b4974-ddvq7] IP=[redact]
|
||||
// 939322e1095ee5e4-IAD "POST /v1/login HTTP/1.1" 400 42 "-" "tauri-plugin-http/2.4.3" 0.000076
|
||||
let mut maybe_origin_header = headers.get(header::ORIGIN);
|
||||
|
||||
let replace_origin_header = maybe_origin_header.is_none()
|
||||
|| maybe_origin_header == Some(&HeaderValue::from_static("null")) // Above error (Mac)
|
||||
|| maybe_origin_header == Some(&HeaderValue::from_static("http://tauri.localhost")) // Windows
|
||||
|| maybe_origin_header == Some(&HeaderValue::from_static("tauri://localhost")); // Mac plausibly causes this
|
||||
|
||||
if replace_origin_header {
|
||||
// TODO(bt): Make this a sentinel origin, eg. sentinel.storyteller.ai.
|
||||
// NB: HeadersMap is a multimap, but this should ensure we just have one value.
|
||||
headers.insert(header::ORIGIN, HeaderValue::from_static("https://studio.storyteller.ai"));
|
||||
}
|
||||
normalize_artcraft_origin(&url, &mut headers);
|
||||
|
||||
if let Some(data) = data {
|
||||
request = request.body(data);
|
||||
|
|
|
|||
|
|
@ -12,6 +12,8 @@ use tauri::{
|
|||
|
||||
pub use error::{Error, Result};
|
||||
|
||||
mod artcraft_identity;
|
||||
|
||||
// NB(bt): We're making everything public so we can access it.
|
||||
pub mod commands;
|
||||
pub mod error;
|
||||
|
|
|
|||
|
|
@ -0,0 +1,189 @@
|
|||
# Desktop session regression — September 27, 2026
|
||||
|
||||
The combined development launcher (`92e6f438c3`, following the performance
|
||||
change `df707bf8e1`) moved Vite from `localhost:5173` to an available port on
|
||||
`127.0.0.1`. The vendored HTTP plugin forwarded that webview Origin to the
|
||||
production API. The API rejected it with **400: Origin is not allowed to make
|
||||
this request**.
|
||||
|
||||
The native login bridge and credits commands use a different HTTP transport and
|
||||
continued working. Consequently the login modal correctly recognized the native
|
||||
session, but Accounts, Library, and page media requests failed. Accounts treated
|
||||
the error as logged out; its Log In button rechecked the valid native session
|
||||
and immediately closed the login modal again.
|
||||
|
||||
The fix extends the existing packaged-desktop Origin normalization to loopback
|
||||
development origins, at any port, for HTTPS requests to `api.storyteller.ai` and
|
||||
`api.fakeyou.com`. Other development request destinations and non-loopback
|
||||
website origins retain their headers. Cookie handling and API authorization are
|
||||
unchanged. A frontend reload alone cannot update this fix: Rust must rebuild.
|
||||
|
||||
## Dedicated desktop origin and rollout
|
||||
|
||||
The desktop app now declares itself per destination. Both transports read the
|
||||
values from one place, `crates/lib/artcraft_client_identity`
|
||||
(`Destination::classify`, origins, user agents, third-party provider table):
|
||||
|
||||
| Destination | Origin sent | User-Agent |
|
||||
|--------------------------------------------------------|-------------------------------------|--------------------------|
|
||||
| Our APIs (`api.storyteller.ai`, `api.fakeyou.com`, …) | `https://desktop.getartcraft.com` | `storyteller-client/1.0` |
|
||||
| Local API (`localhost`, `127.0.0.1`, `[::1]`) | `http://localhost` | `storyteller-client/1.0` |
|
||||
| Our CDNs / sites (`*.storyteller.ai`, `*.fakeyou.com`) | unchanged (`studio.storyteller.ai`) | `storyteller-client/1.0` |
|
||||
| Named third parties (`THIRD_PARTY_PROVIDERS`) | the provider's own site | OS webview |
|
||||
| Any other host (incl. `storage.googleapis.com`) | webview origin, untouched | OS webview |
|
||||
|
||||
- The native Rust client (`artcraft_client`) sends the Origin and User-Agent on
|
||||
every API request via `storyteller_client_builder`; it previously sent no
|
||||
Origin.
|
||||
- The "OS webview" User-Agents are hardcoded snapshots in
|
||||
`artcraft_client_identity/src/user_agents.rs` and go out of date as the
|
||||
webviews update (notably WebView2's Edge version on Windows). Refresh them
|
||||
from `navigator.userAgent` in the desktop app's devtools on each OS.
|
||||
- The vendored HTTP plugin only replaces the webview's own origin (packaged
|
||||
`tauri://localhost` etc., or a loopback Vite origin), and only sets the
|
||||
User-Agent when the frontend didn't.
|
||||
- Native third-party Rust clients (Grok, Midjourney, Sora, World Labs, Kinovi)
|
||||
set their own headers and are not affected.
|
||||
- Older desktop releases keep working: the server still allows the studio and
|
||||
Tauri origins, and requests without an Origin.
|
||||
|
||||
Keep the deployed origin in sync with
|
||||
`../artcraft-services/crates/lib/actix_cors_configs/src/configs/artcraft_desktop.rs`.
|
||||
Development servers accept any `localhost` origin.
|
||||
|
||||
Deploy the server allowlist to the APIs **before running or releasing a desktop
|
||||
build with the new origin**. Otherwise the deployed server will reject desktop
|
||||
requests until its allowlist is updated. The server retains the studio and
|
||||
legacy Tauri origins so existing desktop releases continue working.
|
||||
|
||||
Check both repositories:
|
||||
|
||||
```sh
|
||||
# From artcraft/
|
||||
cargo test --offline -p artcraft_client_identity -p tauri-plugin-http --lib
|
||||
|
||||
# From artcraft-services/
|
||||
cargo test --offline -p actix_cors_configs --lib
|
||||
```
|
||||
|
||||
The server tests verify credentialed GET/POST requests and OPTIONS preflights in
|
||||
production and development, rejection of lookalike domains/HTTP/alternate ports,
|
||||
and compatibility with previous desktop origins. After server deployment, repeat
|
||||
the native checks below using the updated app.
|
||||
|
||||
## Observed native results
|
||||
|
||||
These observations were captured with the earlier studio origin, before the
|
||||
dedicated desktop origin migration. They do not establish that production has
|
||||
deployed the new allowlist.
|
||||
|
||||
Checked in the real macOS Tauri dev app at `http://127.0.0.1:5193`, using its
|
||||
existing session. Temporary probes were removed after verification. No cookies,
|
||||
tokens, passwords, account identifiers, or media contents are recorded here.
|
||||
|
||||
| Check | Before | After |
|
||||
|--------------------------------|---------------------------------------|----------------------------|
|
||||
| Frontend HTTP `/v1/session` | 400, Origin rejected | 200 |
|
||||
| Shared `UsersApi.GetSession()` | `success: false` | Success, logged in |
|
||||
| Native vs. frontend identity | Frontend could not resolve an account | Same account |
|
||||
| Shared media API, page size 1 | Not measured separately | Success, one item returned |
|
||||
|
||||
## Automated checks
|
||||
|
||||
From the repository root:
|
||||
|
||||
```sh
|
||||
cargo test --offline -p tauri-plugin-http artcraft_origin --lib
|
||||
./script/artcraft/unix_dev.sh
|
||||
```
|
||||
|
||||
In a second terminal, substitute the URL printed by the launcher:
|
||||
|
||||
```sh
|
||||
cd frontend
|
||||
npm run test:desktop-session -- http://127.0.0.1:5193
|
||||
```
|
||||
|
||||
The Rust tests cover the default port, alternate/collision ports, IPv4, IPv6,
|
||||
localhost, packaged origins, duplicate headers, session-header preservation,
|
||||
and destinations/origins that must remain unchanged.
|
||||
|
||||
The browser check uses the real frontend with an isolated Chrome profile and
|
||||
mocked IPC. It verifies matching credits and Accounts state, a visible Library
|
||||
image, media after opening the deferred image page, video/audio page rendering,
|
||||
and logout followed by sign-in. It blocks remote browser traffic and never uses
|
||||
your real account or native credentials. It requires installed Google Chrome.
|
||||
It can also run against a locally served production frontend build.
|
||||
|
||||
These browser mocks cannot validate native transport or server CORS. The
|
||||
performance fixture deliberately rejects HTTP requests and is **not** an
|
||||
authentication acceptance test. Run the native check below when changing the
|
||||
launcher, HTTP bridge, or authentication plumbing.
|
||||
|
||||
Validation on September 27: all 12 HTTP-plugin tests, 20 login-modal tests, and
|
||||
10 app tests passed. The browser check passed against both the Vite dev server
|
||||
and a fresh production build. To repeat the production check, serve a build in
|
||||
one terminal, then pass its printed URL to `test:desktop-session` in another:
|
||||
|
||||
```sh
|
||||
cd frontend/apps/artcraft
|
||||
npm run build -- --outDir /tmp/artcraft-session-build
|
||||
npm run preview -- --outDir /tmp/artcraft-session-build --host 127.0.0.1 --port 6215
|
||||
```
|
||||
|
||||
## Repeat the real native check
|
||||
|
||||
1. Start the combined launcher, with an existing signed-in test session. Verify
|
||||
Accounts displays that account, My Library loads its media, and Create Image,
|
||||
Create Video, and Create Audio load their content.
|
||||
2. Right-click inside the desktop webview, choose **Inspect Element**, and open
|
||||
**Console**. Run this read-only probe. It prints statuses/counts, not credentials
|
||||
or account/media contents:
|
||||
|
||||
```js
|
||||
void (async () => {
|
||||
const invoke = window.__TAURI_INTERNALS__.invoke;
|
||||
const host = (await invoke("get_app_info_command")).payload.storyteller_host;
|
||||
const options = { method: "GET", credentials: "include" };
|
||||
const started = performance.now();
|
||||
const [native, response] = await Promise.all([
|
||||
invoke("storyteller_get_login_session_command"),
|
||||
window.FetchProxy(`${host}/v1/session`, { ...options }),
|
||||
]);
|
||||
if (!response.ok) {
|
||||
console.log({ origin: location.origin, sessionStatus: response.status });
|
||||
return;
|
||||
}
|
||||
const session = await response.json();
|
||||
const result = {
|
||||
origin: location.origin,
|
||||
sessionStatus: response.status,
|
||||
nativeLoggedIn: !!native,
|
||||
frontendLoggedIn: session.logged_in,
|
||||
sameAccount: native?.user_token === session.user?.user_token,
|
||||
sessionElapsedMs: performance.now() - started,
|
||||
};
|
||||
if (session.logged_in && session.user) {
|
||||
const username = encodeURIComponent(session.user.username);
|
||||
const media = await window.FetchProxy(
|
||||
`${host}/v1/media_files/list/user/${username}?page_size=1`, { ...options },
|
||||
);
|
||||
result.mediaStatus = media.status;
|
||||
if (media.ok) result.mediaCount = (await media.json()).results?.length;
|
||||
}
|
||||
console.log(result);
|
||||
})().catch((error) => console.error("Desktop session probe failed", String(error)));
|
||||
```
|
||||
|
||||
3. Repeat with a changed port (`ARTCRAFT_DEV_PORT=6200`) and with the starting
|
||||
port occupied. For example, keep this listener running in another terminal
|
||||
before starting the launcher; stop it with Ctrl-C afterward:
|
||||
|
||||
```sh
|
||||
node -e 'require("node:net").createServer().listen(5193, "127.0.0.1")'
|
||||
```
|
||||
|
||||
The launcher should select a later port, and both session transports should
|
||||
still agree. An empty account may legitimately return zero media; it should
|
||||
still return HTTP 200. Verify a genuinely signed-out session shows the login
|
||||
screen using the isolated browser check, or a disposable account/profile.
|
||||
|
|
@ -0,0 +1,224 @@
|
|||
# Desktop frontend performance experiments
|
||||
|
||||
Measured September 26, 2026 on an Apple M4 Pro, macOS arm64, Chrome
|
||||
153.0.8010.53, Node 24.13.0. These are production frontend measurements in
|
||||
headless Chrome with mocked Tauri services. They do **not** measure native
|
||||
process launch, WKWebView/WebView2, Rust, disk startup, real account loading,
|
||||
or generation latency.
|
||||
|
||||
These timings are not authentication acceptance tests: the benchmark mocks the
|
||||
native session and rejects frontend HTTP requests. Before accepting a desktop
|
||||
change, also run the [session and Library checks](desktop-session-regression.md),
|
||||
including real Tauri HTTP requests when changing the launcher or bridge.
|
||||
|
||||
## Changes and results
|
||||
|
||||
Seven samples per version; table values are medians. Each sample uses a fresh
|
||||
browser context at 1440 × 1000 with device scale 1 and browser HTTP cache
|
||||
disabled. The browser process stays alive between samples. A reload in the
|
||||
same context is measured separately. There is no CPU/network throttling.
|
||||
External requests, including analytics, fonts, and media, are blocked in both
|
||||
versions. The login fixture is signed in; the bundled model overlay is used.
|
||||
|
||||
| Measurement | Before | After | Reduction |
|
||||
|-----------------------------|-----------|-----------|-----------|
|
||||
| Startup to frame | 566.40 ms | 350.60 ms | 38.1% |
|
||||
| Refresh to frame | 248.60 ms | 198.10 ms | 20.3% |
|
||||
| Startup JavaScript | 9.05 MB | 6.47 MB | 28.6% |
|
||||
| Startup long tasks (sum) | 336.00 ms | 184.00 ms | 45.2% |
|
||||
| First drawing page to frame | 120.90 ms | 118.20 ms | 2.2% |
|
||||
| Drawing to home: dispatch | 23.40 ms | 9.00 ms | 61.5% |
|
||||
| Drawing to home: frame | 29.50 ms | 23.00 ms | 22.0% |
|
||||
| Home to drawing: dispatch | 42.10 ms | 18.80 ms | 55.3% |
|
||||
| Home to drawing: frame | 105.70 ms | 86.50 ms | 18.2% |
|
||||
|
||||
The 2.2% first-open difference is small enough to treat as unchanged. These
|
||||
local measurements are evidence of reduced frontend work, not predictions
|
||||
of identical percentage gains on every user's machine. Raw samples, ranges,
|
||||
environment, fixture sizes, browser errors, and build provenance are in
|
||||
[before.json](performance/2026-09-26-before.json) and
|
||||
[after.json](performance/2026-09-26-after.json).
|
||||
|
||||
Three changes produced these results:
|
||||
|
||||
1. `MainApp.tsx` imported every editor at launch. Fifteen pages now use
|
||||
`React.lazy` behind a Suspense boundary below the persistent shell. Their
|
||||
code loads when selected; the top bar stays available during loading.
|
||||
The default Apps page remains eager. Drawing also remains eager: its
|
||||
shared code is already part of the shell, and experimentally deferring
|
||||
its small host adapter made first-open latency worse for little saving.
|
||||
First-open latency for the other deferred editors is a remaining tradeoff
|
||||
to measure, especially on slower disks/CPUs. All 15 deferred modules were
|
||||
imported successfully in the browser, but this is not an end-to-end test
|
||||
of every editor's features.
|
||||
2. `TabState.ts` exported the drawing scene to JSON when leaving and imported
|
||||
it when returning. The Zustand scene store already survives page unmount.
|
||||
With four imported images, the old path performed four FileReader reads
|
||||
on exit and four more on return, allocated base64 copies, rebuilt image
|
||||
nodes, cleared the selection, and appended an undo-history entry. Tab
|
||||
changes now keep the live store. The measured round trip performs **zero
|
||||
image-file reads**, preserves the selected node, and keeps history length
|
||||
at one instead of growing to two. Explicit file export/import remains
|
||||
available. Tests cover image identity, model metadata, selection, redo,
|
||||
gallery reset, rapid navigation, and portable file export.
|
||||
3. The shell ran `getGPUTier()` at startup and stored its result in an unread
|
||||
React state value. Removing it eliminates **one startup WebGL context
|
||||
request** and its associated detection work. The real 3D renderer still
|
||||
creates its context when used. This experiment reports combined timing
|
||||
gains, not an isolated timing attribution for each individual change.
|
||||
|
||||
## Repeat the experiment
|
||||
|
||||
From the repository root:
|
||||
|
||||
```sh
|
||||
cd frontend
|
||||
npm ci
|
||||
npm run perf:desktop -- --output /tmp/artcraft-perf-current --smoke-imports
|
||||
```
|
||||
|
||||
The default browser is an installed Google Chrome (`--channel chrome`). The
|
||||
runner requires permission to launch a browser and bind a loopback server.
|
||||
It never uses your regular browser profile, account, or native app data.
|
||||
`@playwright/test` is pinned as a development dependency. Artifacts are written
|
||||
outside the repository by default. Use a dedicated output directory: its
|
||||
`dist` subdirectory is rebuilt by Vite.
|
||||
|
||||
For this particular two-module change, reproduce the original implementation
|
||||
without replacing local source files:
|
||||
|
||||
```sh
|
||||
npm run perf:desktop -- --output /tmp/artcraft-perf-before \
|
||||
--baseline-ref 780e4494e995163fed1d677b88c3cf03b3fca63c
|
||||
npm run perf:desktop -- --output /tmp/artcraft-perf-after --smoke-imports
|
||||
node tools/performance/compare.mjs \
|
||||
/tmp/artcraft-perf-before/results.json \
|
||||
/tmp/artcraft-perf-after/results.json
|
||||
```
|
||||
|
||||
`--baseline-ref` overlays **only** `MainApp.tsx` and `pages/Stores/TabState.ts`
|
||||
from that Git revision in the benchmark build. All other source files and
|
||||
dependencies remain current. It reproduces the two production files changed
|
||||
in this experiment; it is not a general historical-checkout benchmark. For
|
||||
future changes, capture a normal baseline **before** editing, use a new output
|
||||
directory for the changed version, and keep the runner, fixture, dependencies,
|
||||
browser, and machine identical. Run sequentially while the machine is idle;
|
||||
do not build or run other CPU-heavy tasks during samples. Repeat in reverse
|
||||
order if results are close or noisy.
|
||||
|
||||
Useful options:
|
||||
|
||||
```sh
|
||||
# Build in a restricted shell, then run where browser/server use is allowed.
|
||||
npm run perf:desktop -- --output /tmp/artcraft-perf-current --build-only
|
||||
npm run perf:desktop -- --output /tmp/artcraft-perf-current --skip-build
|
||||
|
||||
# More samples; seven is the default.
|
||||
npm run perf:desktop -- --output /tmp/artcraft-perf-current --runs 15
|
||||
|
||||
# Correctness checks for the navigation change.
|
||||
npx vitest run --config apps/artcraft/vitest.config.ts
|
||||
```
|
||||
|
||||
Do not use `--skip-build` after changing app code; it intentionally measures
|
||||
the existing artifact. `build-info.json` records the revision, changed-file
|
||||
names, baseline overlay, and build time. The benchmark plugin alone exposes
|
||||
the stores needed to seed the scene. It is not part of the normal Vite/Tauri
|
||||
build, and no mock IPC or benchmark global is added to the shipped app.
|
||||
|
||||
The runner builds from the app's working directory, matching its normal
|
||||
build command. This matters because Tailwind resolves its config and content
|
||||
relative to that directory. It rejects a run without working Tailwind styles
|
||||
or with uncaught browser errors. `drawing.png` provides a visual check after
|
||||
the measured operations. The initial unstyled exploratory runs were discarded.
|
||||
|
||||
## What the measurements mean
|
||||
|
||||
The fixture creates four deterministic 1024 × 1024 PNG image nodes and 80
|
||||
strokes of 200 coordinates each, with one selected image. On the recorded
|
||||
browser each PNG is 2,356,706 bytes. Fixture construction, image decoding,
|
||||
and a one-second settling period are outside the tab-transition timings.
|
||||
Each sample follows the same sequence:
|
||||
|
||||
1. Navigate to the production app, wait for the Apps heading and for the
|
||||
signed-in login overlay to disappear, then wait two animation frames.
|
||||
2. Reload that page and repeat the same readiness checks.
|
||||
3. Seed the scene while on Apps, open drawing, and wait for its canvas and
|
||||
two animation frames. Allow a second for its background bake to settle.
|
||||
4. Switch to Apps and then back to drawing, measuring each separately.
|
||||
5. After a settling period, inspect node count, selection, history, and IPC
|
||||
counters. Save a screenshot on the first run. Optionally evaluate all
|
||||
deferred page modules after the final timed operation.
|
||||
|
||||
`readyMs` is `performance.now()` after readiness checks and two animation
|
||||
frames; its origin is navigation start. It includes browser loading and
|
||||
automation readiness-observation overhead. `jsBytes` sums decoded resource
|
||||
sizes of loaded JavaScript; it is uncompressed bytes, not all packaged files.
|
||||
`longTaskMs` sums observed startup tasks longer than 50 ms, not total CPU time
|
||||
or total blocking time. Long Tasks support is browser-dependent.
|
||||
|
||||
`stateReadyMs` measures the awaited `setActiveTab()` action, including React
|
||||
work synchronously flushed by its store subscribers. `paintMs` continues
|
||||
until the destination heading/canvas exists and two animation frames have
|
||||
passed. This is a **rendering-opportunity proxy**, not proof of GPU presentation
|
||||
or of every asynchronous image decode completing. In particular, the old
|
||||
return path could still be decoding images after the canvas first appeared.
|
||||
Tab measurements also create `artcraft:tab:APPS` and `artcraft:tab:2D` User
|
||||
Timing entries in the benchmark page.
|
||||
|
||||
The runner calls the same store action used by navigation UI. It excludes
|
||||
pointer-event dispatch, menu animations, and physical click-to-screen latency.
|
||||
It does not automate paid generations, sign in to a real account, or measure
|
||||
remote services. Save the raw JSON as well as the summary; a single minimum
|
||||
time is not a reliable comparison.
|
||||
|
||||
`dist/performance-bundle.json` ranks the largest retained modules per chunk
|
||||
and records chunk imports. Module `renderedBytes` is Rollup's intermediate
|
||||
module size; it is useful for attribution but should not be summed as final
|
||||
minified/gzipped size. Use actual resource bytes for the startup total.
|
||||
|
||||
## Remaining investigation
|
||||
|
||||
- The shell still loads about 6.47 MB of JavaScript. Shared imports through
|
||||
drawing, scene, gallery, and settings entry points need a deeper dependency
|
||||
audit. Separate lightweight store/type imports from editor code, then use
|
||||
the bundle report and the same startup experiment to verify what actually
|
||||
leaves the initial graph. The report identifies Spark, Three.js core and
|
||||
renderer, the inline FBX-to-GLB conversion worker, PostHog, and Mediabunny
|
||||
among the largest retained modules. Their presence is measured; the runtime
|
||||
benefit of deferring each individually has not yet been measured.
|
||||
- Returning to the populated canvas still takes about 86 ms. The editor
|
||||
remount, Konva drawing, image/transformer setup, and composite bake need a
|
||||
timeline profile before choosing the next change. Keep scene restoration,
|
||||
undo, selection, and background generation behavior in correctness checks.
|
||||
- `TaskQueue.tsx` polls every five seconds and rebuilds task arrays even when
|
||||
the popover is closed. The benchmark uses an empty queue, so it does not
|
||||
establish the cost for a large real queue. A fixed, representative task
|
||||
fixture plus idle CPU/render counts is the next useful experiment there.
|
||||
- `useStageSnapshot` looks expensive in isolation, but its only call is
|
||||
commented out in `PaintSurface.tsx`. It was excluded as a current bottleneck.
|
||||
|
||||
Validation completed: normal production Vite build, all 10 app tests and 20
|
||||
login tests, all 15 deferred page-module imports, visual canvas inspection,
|
||||
and a check that the normal build contains no benchmark globals or fixtures.
|
||||
The broad TypeScript check remains blocked by existing workspace declaration/
|
||||
project-reference errors and Remix/Deno types. Comparing diagnostics with the
|
||||
two original modules showed the same diagnostic messages (some import traces
|
||||
and reporting locations changed); this is not a clean typecheck claim.
|
||||
|
||||
## Validate in the native app
|
||||
|
||||
Before claiming native launch or interaction gains, run matching release
|
||||
Tauri builds on the same machine and WebView version. Keep account, window
|
||||
size, network conditions, and canvas artwork fixed. Record the Web Inspector
|
||||
timeline for launch/reload, opening a tool for the first time, drawing → Apps,
|
||||
and Apps → drawing. Repeat at least seven times and retain the traces with
|
||||
build IDs and medians. Separate native process launch from page reload; do
|
||||
not mix dev/HMR traces with production measurements.
|
||||
|
||||
For canvas tests, create a blank 1024-square canvas, import the same four
|
||||
images, add the same strokes, select an image, and wait for background work
|
||||
before switching. Check image content, selection, undo/redo, and opening a new
|
||||
gallery image afterward. Record actual click-to-visible-frame latency in
|
||||
the inspector as well as JavaScript tasks. The browser results above justify
|
||||
the code changes but do not replace that native validation.
|
||||
|
|
@ -0,0 +1,527 @@
|
|||
{
|
||||
"timestamp": "2026-09-27T00:32:54.524Z",
|
||||
"build": {
|
||||
"builtAt": "2026-09-27T00:30:35.760Z",
|
||||
"revision": "780e4494e995163fed1d677b88c3cf03b3fca63c",
|
||||
"baselineOverlay": null,
|
||||
"changedFiles": [
|
||||
"frontend/apps/artcraft/app/src/pages/MainApp.tsx",
|
||||
"frontend/apps/artcraft/app/src/pages/Stores/TabState.ts",
|
||||
"frontend/package-lock.json",
|
||||
"frontend/package.json"
|
||||
]
|
||||
},
|
||||
"environment": {
|
||||
"platform": "darwin",
|
||||
"arch": "arm64",
|
||||
"cpu": "Apple M4 Pro",
|
||||
"browser": "153.0.8010.53",
|
||||
"node": "v24.13.0",
|
||||
"runs": 7,
|
||||
"viewport": "1440x1000@1",
|
||||
"ipc": "mocked",
|
||||
"network": "local assets only, browser cache disabled"
|
||||
},
|
||||
"summary": {
|
||||
"startup": {
|
||||
"readyMs": {
|
||||
"median": 350.60000002384186,
|
||||
"min": 348.7000000476837,
|
||||
"max": 457.39999997615814
|
||||
},
|
||||
"jsBytes": {
|
||||
"median": 6466606,
|
||||
"min": 6466606,
|
||||
"max": 6466606
|
||||
},
|
||||
"longTaskMs": {
|
||||
"median": 184,
|
||||
"min": 179,
|
||||
"max": 284
|
||||
},
|
||||
"webglContexts": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
}
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": {
|
||||
"median": 198.10000002384186,
|
||||
"min": 197.10000002384186,
|
||||
"max": 198.80000007152557
|
||||
},
|
||||
"jsBytes": {
|
||||
"median": 6466606,
|
||||
"min": 6466606,
|
||||
"max": 6466606
|
||||
},
|
||||
"longTaskMs": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
},
|
||||
"webglContexts": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
}
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": {
|
||||
"median": 38.300000071525574,
|
||||
"min": 37.60000002384186,
|
||||
"max": 40.299999952316284
|
||||
},
|
||||
"paintMs": {
|
||||
"median": 118.19999992847443,
|
||||
"min": 115.30000007152557,
|
||||
"max": 120
|
||||
},
|
||||
"fileReads": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
}
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": {
|
||||
"median": 9,
|
||||
"min": 7.5,
|
||||
"max": 11
|
||||
},
|
||||
"paintMs": {
|
||||
"median": 23,
|
||||
"min": 19.600000023841858,
|
||||
"max": 26.600000023841858
|
||||
},
|
||||
"fileReads": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
}
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": {
|
||||
"median": 18.800000071525574,
|
||||
"min": 15.799999952316284,
|
||||
"max": 20.700000047683716
|
||||
},
|
||||
"paintMs": {
|
||||
"median": 86.5,
|
||||
"min": 78.79999995231628,
|
||||
"max": 90.10000002384186
|
||||
},
|
||||
"fileReads": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
}
|
||||
}
|
||||
},
|
||||
"samples": [
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 457.39999997615814,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 284,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 197.80000007152557,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 38.800000071525574,
|
||||
"paintMs": 119.5,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 7.600000023841858,
|
||||
"paintMs": 20.399999976158142,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 17.799999952316284,
|
||||
"paintMs": 87.79999995231628,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 365.7999999523163,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 185,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 197.10000002384186,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 37.60000002384186,
|
||||
"paintMs": 116,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 7.5,
|
||||
"paintMs": 26.600000023841858,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 15.799999952316284,
|
||||
"paintMs": 78.79999995231628,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 366.59999990463257,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 193,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 198.39999997615814,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 40.299999952316284,
|
||||
"paintMs": 118.19999992847443,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 9,
|
||||
"paintMs": 23,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 19.699999928474426,
|
||||
"paintMs": 86.5,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 348.7000000476837,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 184,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 197.89999997615814,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 37.700000047683716,
|
||||
"paintMs": 115.60000002384186,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 9.300000071525574,
|
||||
"paintMs": 26.40000009536743,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 18.700000047683716,
|
||||
"paintMs": 84.90000009536743,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 349.60000002384186,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 179,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 198.10000002384186,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 38.300000071525574,
|
||||
"paintMs": 118.5,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 11,
|
||||
"paintMs": 19.600000023841858,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 20.700000047683716,
|
||||
"paintMs": 90.10000002384186,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 350.60000002384186,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 179,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 198.20000004768372,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 39.60000002384186,
|
||||
"paintMs": 120,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 11,
|
||||
"paintMs": 22.199999928474426,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 20,
|
||||
"paintMs": 87,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 349.89999997615814,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 180,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 198.80000007152557,
|
||||
"jsBytes": 6466606,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 0
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 38,
|
||||
"paintMs": 115.30000007152557,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 8.200000047683716,
|
||||
"paintMs": 24.899999976158142,
|
||||
"fileReads": 0
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 18.800000071525574,
|
||||
"paintMs": 84.5,
|
||||
"fileReads": 0
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 1,
|
||||
"historyLength": 1,
|
||||
"commands": {
|
||||
"plugin:event|listen": 25,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
}
|
||||
],
|
||||
"importedPages": [
|
||||
"src/pages/PageAngles/index.ts",
|
||||
"src/pages/PageAudio/CreateAudio.tsx",
|
||||
"src/pages/PageBackgroundChange/index.ts",
|
||||
"src/pages/PageImage/TextToImage.tsx",
|
||||
"src/pages/PageImageTo3DObject/index.ts",
|
||||
"src/pages/PageImageTo3DWorld/index.ts",
|
||||
"src/pages/PageImageWatermarkRemover/index.ts",
|
||||
"src/pages/PageMoodboard/index.ts",
|
||||
"src/pages/PageRemoveBackground/index.ts",
|
||||
"src/pages/PageScene/index.tsx",
|
||||
"src/pages/PageStoryboard/index.ts",
|
||||
"src/pages/PageVideo/ImageToVideo.tsx",
|
||||
"src/pages/PageVideoEditor/index.tsx",
|
||||
"src/pages/PageVideoFrameExtractor/index.ts",
|
||||
"src/pages/PageVideoWatermarkRemover/index.ts"
|
||||
],
|
||||
"errors": []
|
||||
}
|
||||
|
|
@ -0,0 +1,510 @@
|
|||
{
|
||||
"timestamp": "2026-09-27T00:30:21.217Z",
|
||||
"build": {
|
||||
"builtAt": "2026-09-27T00:28:39.209Z",
|
||||
"revision": "780e4494e995163fed1d677b88c3cf03b3fca63c",
|
||||
"baselineOverlay": "HEAD",
|
||||
"changedFiles": [
|
||||
"frontend/apps/artcraft/app/src/pages/MainApp.tsx",
|
||||
"frontend/apps/artcraft/app/src/pages/Stores/TabState.ts",
|
||||
"frontend/package-lock.json",
|
||||
"frontend/package.json"
|
||||
]
|
||||
},
|
||||
"environment": {
|
||||
"platform": "darwin",
|
||||
"arch": "arm64",
|
||||
"cpu": "Apple M4 Pro",
|
||||
"browser": "153.0.8010.53",
|
||||
"node": "v24.13.0",
|
||||
"runs": 7,
|
||||
"viewport": "1440x1000@1",
|
||||
"ipc": "mocked",
|
||||
"network": "local assets only, browser cache disabled"
|
||||
},
|
||||
"summary": {
|
||||
"startup": {
|
||||
"readyMs": {
|
||||
"median": 566.3999999761581,
|
||||
"min": 532.3999999761581,
|
||||
"max": 847.7999999523163
|
||||
},
|
||||
"jsBytes": {
|
||||
"median": 9052991,
|
||||
"min": 9052991,
|
||||
"max": 9052991
|
||||
},
|
||||
"longTaskMs": {
|
||||
"median": 336,
|
||||
"min": 314,
|
||||
"max": 407
|
||||
},
|
||||
"webglContexts": {
|
||||
"median": 1,
|
||||
"min": 1,
|
||||
"max": 1
|
||||
}
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": {
|
||||
"median": 248.60000002384186,
|
||||
"min": 247.59999990463257,
|
||||
"max": 351.1999999284744
|
||||
},
|
||||
"jsBytes": {
|
||||
"median": 9052991,
|
||||
"min": 9052991,
|
||||
"max": 9052991
|
||||
},
|
||||
"longTaskMs": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
},
|
||||
"webglContexts": {
|
||||
"median": 1,
|
||||
"min": 1,
|
||||
"max": 1
|
||||
}
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": {
|
||||
"median": 40.40000009536743,
|
||||
"min": 37.299999952316284,
|
||||
"max": 43.299999952316284
|
||||
},
|
||||
"paintMs": {
|
||||
"median": 120.90000009536743,
|
||||
"min": 117.89999997615814,
|
||||
"max": 126.79999995231628
|
||||
},
|
||||
"fileReads": {
|
||||
"median": 0,
|
||||
"min": 0,
|
||||
"max": 0
|
||||
}
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": {
|
||||
"median": 23.399999976158142,
|
||||
"min": 20.100000023841858,
|
||||
"max": 26
|
||||
},
|
||||
"paintMs": {
|
||||
"median": 29.5,
|
||||
"min": 21.200000047683716,
|
||||
"max": 35.5
|
||||
},
|
||||
"fileReads": {
|
||||
"median": 4,
|
||||
"min": 4,
|
||||
"max": 4
|
||||
}
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": {
|
||||
"median": 42.10000002384186,
|
||||
"min": 36.59999990463257,
|
||||
"max": 45.200000047683716
|
||||
},
|
||||
"paintMs": {
|
||||
"median": 105.69999992847443,
|
||||
"min": 100.89999997615814,
|
||||
"max": 109.20000004768372
|
||||
},
|
||||
"fileReads": {
|
||||
"median": 4,
|
||||
"min": 4,
|
||||
"max": 4
|
||||
}
|
||||
}
|
||||
},
|
||||
"samples": [
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 847.7999999523163,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 407,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 351.1999999284744,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 43.299999952316284,
|
||||
"paintMs": 126.79999995231628,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 25.100000023841858,
|
||||
"paintMs": 35.5,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 36.59999990463257,
|
||||
"paintMs": 100.89999997615814,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 539.3999999761581,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 317,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 264.60000002384186,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 39.39999997615814,
|
||||
"paintMs": 120.20000004768372,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 20.100000023841858,
|
||||
"paintMs": 21.200000047683716,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 41.60000002384186,
|
||||
"paintMs": 104.60000002384186,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 561.9000000953674,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 329,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 248.19999992847443,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 40.40000009536743,
|
||||
"paintMs": 120.90000009536743,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 23.399999976158142,
|
||||
"paintMs": 33.39999997615814,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 37.299999952316284,
|
||||
"paintMs": 101.10000002384186,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 569.5,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 336,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 248,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 38,
|
||||
"paintMs": 119.30000007152557,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 26,
|
||||
"paintMs": 34.799999952316284,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 45.200000047683716,
|
||||
"paintMs": 109.20000004768372,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 567.5,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 339,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 248.60000002384186,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 40.89999997615814,
|
||||
"paintMs": 122.10000002384186,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 23.200000047683716,
|
||||
"paintMs": 29.5,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 42.10000002384186,
|
||||
"paintMs": 105.69999992847443,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 566.3999999761581,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 336,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 281.39999997615814,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 43.09999990463257,
|
||||
"paintMs": 124.69999992847443,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 23.399999976158142,
|
||||
"paintMs": 26.199999928474426,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 42.799999952316284,
|
||||
"paintMs": 107.19999992847443,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"startup": {
|
||||
"readyMs": 532.3999999761581,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 314,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"refresh": {
|
||||
"readyMs": 247.59999990463257,
|
||||
"jsBytes": 9052991,
|
||||
"longTaskMs": 0,
|
||||
"webglContexts": 1
|
||||
},
|
||||
"firstDraw": {
|
||||
"stateReadyMs": 37.299999952316284,
|
||||
"paintMs": 117.89999997615814,
|
||||
"fileReads": 0
|
||||
},
|
||||
"leaveDraw": {
|
||||
"stateReadyMs": 21.5,
|
||||
"paintMs": 22.899999976158142,
|
||||
"fileReads": 4
|
||||
},
|
||||
"returnDraw": {
|
||||
"stateReadyMs": 44.09999990463257,
|
||||
"paintMs": 108.39999997615814,
|
||||
"fileReads": 4
|
||||
},
|
||||
"fixture": {
|
||||
"images": 4,
|
||||
"imageSize": 1024,
|
||||
"imageBytesEach": 2356706,
|
||||
"strokes": 80,
|
||||
"coordinatesPerStroke": 200
|
||||
},
|
||||
"afterReturn": {
|
||||
"nodeCount": 84,
|
||||
"selectedCount": 0,
|
||||
"historyLength": 2,
|
||||
"commands": {
|
||||
"plugin:event|listen": 26,
|
||||
"get_app_info_command": 1,
|
||||
"list_image_models_command": 1,
|
||||
"list_video_models_command": 1,
|
||||
"get_task_queue_command": 1,
|
||||
"estimate_image_cost_command": 5,
|
||||
"storyteller_get_login_session_command": 1,
|
||||
"plugin:http|fetch": 5,
|
||||
"storyteller_get_credits_command": 1,
|
||||
"storyteller_get_subscription_command": 1,
|
||||
"plugin:window|is_maximized": 1,
|
||||
"plugin:event|unlisten": 2
|
||||
}
|
||||
}
|
||||
}
|
||||
],
|
||||
"errors": []
|
||||
}
|
||||
|
|
@ -0,0 +1,80 @@
|
|||
# Website login from the desktop app
|
||||
|
||||
The React login/signup dialog calls Tauri commands only, including its startup
|
||||
session check and password forms. Website and QR actions appear only in login
|
||||
mode. Tauri never connects to MySQL; the desktop uses Rust HTTP API bindings. Creating/polling login challenges and
|
||||
verifying `/v1/session` use `artcraft_client::endpoints::users::login_challenges`.
|
||||
The Rust command uses `AppEnvConfigs.storyteller_host` for every request, keeps the
|
||||
private device token in native memory, and returns only a local challenge handle,
|
||||
approval URL/code/deadline, status, and the verified user to JavaScript. The signed
|
||||
session cookie never crosses IPC or gets written to JavaScript local storage.
|
||||
|
||||
After redemption, Rust verifies the signed session against the same API host,
|
||||
installs the server's cookie into the native HTTP jar and credential manager, and
|
||||
persists the jar. Closing the dialog removes the local handle; a cancellation
|
||||
while HTTP is in flight prevents installing the late response. Retries retrieve
|
||||
the same server session. Neither creation nor website approval pre-creates one.
|
||||
|
||||
## Session response compatibility
|
||||
|
||||
A server feature flag added after a desktop release must not prevent login. The
|
||||
session decoder keeps known feature flags and ignores unknown flag names, such
|
||||
as `use_qt` on older clients. Malformed flag values and required authentication
|
||||
fields still fail validation. Tests cover a real Rust HTTP session response with
|
||||
both known and future flags, followed by native cookie installation/persistence.
|
||||
A late startup session check cannot reset an active or completed website login.
|
||||
|
||||
## Diagnosing 401 responses
|
||||
|
||||
`storyteller_activity_thread` reports errors from `/v1/analytics/active_user_v2`.
|
||||
This is the background activity ping, not the login-challenge endpoint. It requires
|
||||
a user session. An anonymous `visitor` cookie alone is now skipped. Expired or
|
||||
revoked sessions can still produce a 401 and require signing in again.
|
||||
|
||||
Activity failures log the configured API origin (including port), endpoint, and
|
||||
error. Native website-login requests/failures log origin, endpoint, and HTTP
|
||||
status. Request bodies, device tokens, approval URLs, and signed cookies are not
|
||||
logged by the new client, including at debug level.
|
||||
|
||||
## API environment
|
||||
|
||||
`~/Artcraft/settings/env_configs.json` controls the native API host:
|
||||
|
||||
```json
|
||||
{"storyteller_host":"localhost","storyteller_port":12345}
|
||||
```
|
||||
|
||||
Local mode uses `http://localhost:12345` and a local approval website at
|
||||
`http://localhost:4201/login/desktop`. Run storyteller-web with its migrations
|
||||
applied, and the services webapp with `USE_LOCAL_API=1`. The local website must
|
||||
also use the local API. A challenge in a local database cannot be approved on the
|
||||
production website, even when that browser already has a production session.
|
||||
|
||||
For the production API and production website, use `"storyteller_host":"production"`
|
||||
(or omit the override), then restart ArtCraft. API host configuration is not
|
||||
silently changed by login. Local and production approval URLs cannot be mixed.
|
||||
Localhost development URLs are for the same computer; mobile QR testing needs
|
||||
a shared/deployed environment.
|
||||
|
||||
## Tests
|
||||
|
||||
From the desktop repository:
|
||||
|
||||
```sh
|
||||
SQLX_OFFLINE=true cargo test --offline -p artcraft --lib login_bridge_tests
|
||||
cd frontend
|
||||
npm exec vitest -- run --config libs/components/login-modal/vite.config.ts src/lib/DesktopLoginBridge.spec.tsx src/lib/login-modal.spec.tsx
|
||||
```
|
||||
|
||||
Native tests use only ephemeral loopback HTTP servers and temporary cookie stores.
|
||||
They exercise the real API bindings and the command implementation, including
|
||||
cookie verification/persistence, rejection, expiry, retries, in-flight cancellation,
|
||||
origin separation, no redirects, redacted errors, visitor-only telemetry, future
|
||||
feature flags, malformed session data, and password login/signup/session checks.
|
||||
UI tests prohibit JavaScript HTTP and runtime API imports. They verify the IPC
|
||||
contract, native browser opening, QR/code display, status, backoff, cleanup,
|
||||
login-only button placement, native password forms, and stale session responses.
|
||||
|
||||
The services repository also has real MySQL integration tests for approval,
|
||||
redemption, rejection, concurrency, and downstream authentication, using isolated
|
||||
local databases. No test contacts a production API or database.
|
||||
|
|
@ -31,7 +31,7 @@
|
|||
<link rel="icon" type="image/x-icon" href="/favicon.ico" />
|
||||
<link rel="preconnect" href="https://fonts.googleapis.com" />
|
||||
<link rel="preconnect" href="https://fonts.gstatic.com" crossorigin />
|
||||
<link href="https://fonts.googleapis.com/css2?family=Outfit:wght@100..900&family=Source+Sans+3:ital,wght@0,200..900;1,200..900&display=swap"
|
||||
<link href="https://fonts.googleapis.com/css2?family=Archivo:wdth,wght@62..125,100..900&family=Instrument+Serif:ital@0;1&family=Geist+Mono:wght@300..800&display=swap"
|
||||
rel="stylesheet" />
|
||||
<!-- <link rel="stylesheet" href="/src/styles.css" /> -->
|
||||
</head>
|
||||
|
|
|
|||
|
|
@ -191,11 +191,11 @@ export function GlobalFileDropHandler() {
|
|||
return (
|
||||
<>
|
||||
{isDragging && modalType === null && (
|
||||
<div className="pointer-events-none fixed inset-0 z-[9999] flex items-center justify-center bg-black/40">
|
||||
<div className="flex flex-col items-center gap-3 rounded-2xl border-2 border-dashed border-white/60 bg-black/30 px-16 py-12 text-white backdrop-blur-sm">
|
||||
<Maximize2Icon className="text-4xl opacity-80" />
|
||||
<div className="text-xl font-semibold">Drop to Upload</div>
|
||||
<div className="text-sm opacity-60">GLB, PNG, JPG, JPEG, SPZ</div>
|
||||
<div className="pointer-events-none fixed inset-0 z-[9999] flex items-center justify-center bg-black/70">
|
||||
<div className="flex flex-col items-center gap-3 border border-dashed border-white/60 bg-ui-panel px-16 py-12 text-white">
|
||||
<Maximize2Icon className="text-3xl opacity-60" />
|
||||
<div className="font-display text-2xl tracking-tight">Drop to Upload</div>
|
||||
<div className="hud-label text-white/50">GLB, PNG, JPG, JPEG, SPZ</div>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
|
|
|||
|
|
@ -36,6 +36,7 @@ import {
|
|||
useCostBreakdownModalStore,
|
||||
} from "@storyteller/ui-pricing-modal";
|
||||
import { HelpMenuButton } from "@storyteller/ui-help-menu";
|
||||
import { CreateEmptyState } from "~/components/generation-feed/CreateEmptyState";
|
||||
|
||||
type Mode = "image" | "text";
|
||||
type Variant = "object" | "world";
|
||||
|
|
@ -549,9 +550,9 @@ export const ImageTo3DExperience = ({
|
|||
role="button"
|
||||
tabIndex={0}
|
||||
className={twMerge(
|
||||
"flex flex-col items-center justify-center rounded-2xl border-[3px] border-dashed border-primary/40 bg-primary/5 text-center text-xs transition-all hover:border-primary hover:bg-primary/10 focus:outline-none focus:ring-2 focus:ring-primary/40",
|
||||
"flex flex-col items-center justify-center border border-dashed border-white/20 bg-white/[0.015] text-center text-xs transition-all hover:border-white/40 hover:bg-white/5 focus:border-white/60 focus:outline-none",
|
||||
hasResults ? "aspect-square w-24" : "aspect-square w-48",
|
||||
dragActive && "border-primary bg-primary/10",
|
||||
dragActive && "border-white/60 bg-white/5",
|
||||
)}
|
||||
onDragEnter={(event) => {
|
||||
event.preventDefault();
|
||||
|
|
@ -581,11 +582,11 @@ export const ImageTo3DExperience = ({
|
|||
<PlusIcon
|
||||
|
||||
className={twMerge(
|
||||
"text-base-fg opacity-90 drop-shadow",
|
||||
"text-base-fg opacity-80",
|
||||
hasResults ? "text-2xl" : "text-4xl",
|
||||
)} />
|
||||
{!hasResults && (
|
||||
<span className="mt-3 text-[15px] font-medium text-base-fg opacity-60">
|
||||
<span className="mt-3 font-mono text-[11px] font-semibold uppercase tracking-[0.12em] text-base-fg/60">
|
||||
Add Image
|
||||
</span>
|
||||
)}
|
||||
|
|
@ -598,7 +599,7 @@ export const ImageTo3DExperience = ({
|
|||
{uploadedPreview ? (
|
||||
<div
|
||||
className={twMerge(
|
||||
"group relative cursor-pointer overflow-hidden rounded-2xl border-[3px] border-primary/40 bg-black/30 transition-all",
|
||||
"group relative cursor-pointer overflow-hidden border border-white/30 bg-black/30 transition-all hover:border-white/60",
|
||||
hasResults ? "aspect-square w-24" : "aspect-square w-48",
|
||||
)}
|
||||
onClick={() => !isUploading && fileInputRef.current?.click()}
|
||||
|
|
@ -613,13 +614,13 @@ export const ImageTo3DExperience = ({
|
|||
/>
|
||||
{isUploading && (
|
||||
<div className="absolute inset-0 flex items-center justify-center">
|
||||
<div className="h-8 w-8 animate-spin rounded-full border-[3px] border-white/30 border-t-primary" />
|
||||
<div className="h-8 w-8 animate-spin rounded-full border-[3px] border-white/30 border-t-white" />
|
||||
</div>
|
||||
)}
|
||||
{!isUploading && (
|
||||
<button
|
||||
type="button"
|
||||
className="absolute right-2 top-2 flex h-6 w-6 items-center justify-center rounded-full bg-black/60 text-white opacity-0 transition-opacity group-hover:opacity-100"
|
||||
className="absolute right-2 top-2 flex h-6 w-6 items-center justify-center rounded-[3px] bg-black/60 text-white opacity-0 transition-all hover:bg-black group-hover:opacity-100"
|
||||
onClick={(event) => {
|
||||
event.stopPropagation();
|
||||
setUploadedPreview(null);
|
||||
|
|
@ -645,7 +646,7 @@ export const ImageTo3DExperience = ({
|
|||
ref={textareaRef}
|
||||
id={promptInputId}
|
||||
rows={1}
|
||||
className="text-md max-h-[5.5em] w-full resize-none overflow-y-auto rounded bg-transparent pr-2 pt-1 text-base-fg placeholder-base-fg/60 focus:outline-none"
|
||||
className="text-md max-h-[5.5em] w-full resize-none overflow-y-auto rounded-[3px] bg-transparent pr-2 pt-1 text-base-fg placeholder-base-fg/60 focus:outline-none"
|
||||
value={prompt}
|
||||
placeholder="Describe any object you want to generate from scratch..."
|
||||
onChange={(event) => setPrompt(event.target.value)}
|
||||
|
|
@ -691,8 +692,8 @@ export const ImageTo3DExperience = ({
|
|||
role="button"
|
||||
tabIndex={0}
|
||||
className={twMerge(
|
||||
"flex h-32 w-32 flex-col items-center justify-center rounded-xl border-2 border-dashed border-primary/40 bg-primary/5 transition-all hover:border-primary hover:bg-primary/10 focus:outline-none focus:ring-2 focus:ring-primary/40",
|
||||
dragActive && "border-primary bg-primary/10",
|
||||
"flex h-32 w-32 flex-col items-center justify-center border border-dashed border-white/20 bg-white/[0.015] transition-all hover:border-white/40 hover:bg-white/5 focus:border-white/60 focus:outline-none",
|
||||
dragActive && "border-white/60 bg-white/5",
|
||||
)}
|
||||
onDragEnter={(e) => {
|
||||
e.preventDefault();
|
||||
|
|
@ -730,8 +731,8 @@ export const ImageTo3DExperience = ({
|
|||
>
|
||||
<PlusIcon
|
||||
|
||||
className="text-3xl text-base-fg opacity-90 drop-shadow" />
|
||||
<span className="mt-2 text-sm font-medium text-base-fg/50">
|
||||
className="text-3xl text-base-fg opacity-80" />
|
||||
<span className="mt-2 font-mono text-[11px] font-semibold uppercase tracking-[0.12em] text-base-fg/60">
|
||||
Add Images
|
||||
</span>
|
||||
</div>
|
||||
|
|
@ -740,7 +741,7 @@ export const ImageTo3DExperience = ({
|
|||
<textarea
|
||||
ref={worldTextareaRef}
|
||||
rows={2}
|
||||
className="w-full resize-none overflow-y-auto rounded-lg bg-white/5 px-3 py-2.5 text-base text-base-fg placeholder-base-fg/60 outline-none ring-2 ring-transparent transition-all focus:ring-primary/80"
|
||||
className="w-full resize-none overflow-y-auto rounded-[3px] border border-white/15 bg-ui-controls px-3 py-2.5 text-base text-base-fg placeholder-base-fg/60 transition-colors focus:border-white/60 focus:outline-none"
|
||||
style={{ maxHeight: "5em" }}
|
||||
value={worldPrompt}
|
||||
placeholder="Describe your 3D world (optional)..."
|
||||
|
|
@ -754,7 +755,7 @@ export const ImageTo3DExperience = ({
|
|||
<div
|
||||
key={img.id}
|
||||
className={twMerge(
|
||||
"glass group relative aspect-square overflow-hidden rounded-lg border-2 border-white/30 transition-all",
|
||||
"group relative aspect-square overflow-hidden rounded-[3px] border border-white/30 bg-black/30 transition-all",
|
||||
hasResults ? "w-10" : "w-auto",
|
||||
img.isUploading
|
||||
? "cursor-default"
|
||||
|
|
@ -772,13 +773,13 @@ export const ImageTo3DExperience = ({
|
|||
/>
|
||||
{img.isUploading && (
|
||||
<div className="absolute inset-0 flex items-center justify-center bg-black/20">
|
||||
<div className="h-4 w-4 animate-spin rounded-full border-2 border-white/30 border-t-primary" />
|
||||
<div className="h-4 w-4 animate-spin rounded-full border-2 border-white/30 border-t-white" />
|
||||
</div>
|
||||
)}
|
||||
{!img.isUploading && (
|
||||
<button
|
||||
type="button"
|
||||
className="absolute right-[2px] top-[2px] flex h-4 w-4 cursor-pointer items-center justify-center rounded-full bg-black/50 text-white opacity-0 backdrop-blur-md transition-colors hover:bg-red/70 group-hover:opacity-100"
|
||||
className="absolute right-[2px] top-[2px] flex h-4 w-4 cursor-pointer items-center justify-center rounded-[3px] bg-black/60 text-white opacity-0 transition-colors hover:bg-red/70 group-hover:opacity-100"
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
removeWorldImage(img.id);
|
||||
|
|
@ -820,7 +821,7 @@ export const ImageTo3DExperience = ({
|
|||
<Button
|
||||
variant="action"
|
||||
className={twMerge(
|
||||
"aspect-square overflow-hidden rounded-lg border-2 border-dashed border-black/5 bg-ui-controls/40 transition-all hover:bg-ui-controls/60 dark:border-white/25",
|
||||
"aspect-square overflow-hidden rounded-[3px] border border-dashed border-black/15 bg-white/5 transition-all hover:border-white/40 hover:bg-white/10 dark:border-white/25",
|
||||
hasResults ? "w-10" : "w-14",
|
||||
)}
|
||||
onClick={() => worldFileInputRef.current?.click()}
|
||||
|
|
@ -836,7 +837,7 @@ export const ImageTo3DExperience = ({
|
|||
);
|
||||
|
||||
const imageCountRow = (
|
||||
<div className="flex items-center justify-center gap-2 text-xs text-base-fg/40">
|
||||
<div className="flex items-center justify-center gap-2 font-mono text-[11px] tabular-nums text-base-fg/40">
|
||||
<span>
|
||||
{worldImages.length}/{MAX_WORLD_IMAGES} images
|
||||
</span>
|
||||
|
|
@ -865,7 +866,7 @@ export const ImageTo3DExperience = ({
|
|||
<textarea
|
||||
ref={worldTextareaRef}
|
||||
rows={1}
|
||||
className="flex-1 resize-none overflow-y-auto rounded-lg bg-white/5 px-3 py-2 text-sm text-base-fg placeholder-base-fg/60 focus:outline-none focus:ring-2 focus:ring-primary/60"
|
||||
className="flex-1 resize-none overflow-y-auto rounded-[3px] border border-white/15 bg-ui-controls px-3 py-2 text-sm text-base-fg placeholder-base-fg/60 transition-colors focus:border-white/60 focus:outline-none"
|
||||
value={worldPrompt}
|
||||
placeholder="Describe world (optional)..."
|
||||
onChange={(e) => setWorldPrompt(e.target.value)}
|
||||
|
|
@ -887,7 +888,7 @@ export const ImageTo3DExperience = ({
|
|||
<textarea
|
||||
ref={worldTextareaRef}
|
||||
rows={2}
|
||||
className="w-full resize-none overflow-y-auto rounded-lg bg-white/5 px-3 py-2.5 text-base text-base-fg placeholder-base-fg/60 focus:outline-none focus:ring-2 focus:ring-primary/60"
|
||||
className="w-full resize-none overflow-y-auto rounded-[3px] border border-white/15 bg-ui-controls px-3 py-2.5 text-base text-base-fg placeholder-base-fg/60 transition-colors focus:border-white/60 focus:outline-none"
|
||||
style={{ maxHeight: "5em" }}
|
||||
value={worldPrompt}
|
||||
placeholder="Describe your 3D world (optional)..."
|
||||
|
|
@ -907,7 +908,7 @@ export const ImageTo3DExperience = ({
|
|||
results.find((r) => r.id === selectedResultId) || results[0];
|
||||
|
||||
return (
|
||||
<div className="bg-ui-panel-gradient flex h-[calc(100vh-56px)] w-full bg-ui-panel text-base-fg">
|
||||
<div className="flex h-[calc(100vh-56px)] w-full bg-ui-background text-base-fg">
|
||||
{backgroundImage && !hasResults && (
|
||||
<>
|
||||
<div className="pointer-events-none fixed inset-0 z-[1] overflow-hidden bg-[radial-gradient(50%_50%_at_50%_50%,_transparent_49%,_rgb(var(--st-controls-rgb)_/_var(--st-gallery-vignette-alpha))_100%)]" />
|
||||
|
|
@ -928,9 +929,12 @@ export const ImageTo3DExperience = ({
|
|||
|
||||
<div className="relative z-10 h-full w-full p-8">
|
||||
{!hasResults && (
|
||||
<div className="pointer-events-none absolute left-0 top-[calc(50%-280px)] w-full text-center">
|
||||
<h1 className="mb-3 text-7xl font-bold tracking-tight">{title}</h1>
|
||||
<p className="text-xl text-base-fg/70">{subtitle}</p>
|
||||
<div className="pointer-events-none absolute left-0 top-[calc(50%-280px)] flex w-full justify-center">
|
||||
<CreateEmptyState
|
||||
title={title}
|
||||
subtitle={subtitle}
|
||||
className="py-8 sm:py-8"
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
|
||||
|
|
@ -944,7 +948,7 @@ export const ImageTo3DExperience = ({
|
|||
style={{ height: `calc(100vh - ${bottomOffsetPx + 80}px)` }}
|
||||
>
|
||||
{/* Left: Viewer */}
|
||||
<div className="glass relative h-full overflow-hidden rounded-xl border border-ui-panel-border">
|
||||
<div className="relative h-full overflow-hidden border border-ui-panel-border bg-ui-panel">
|
||||
<Viewer3D
|
||||
key={activeResult?.id}
|
||||
modelUrl={activeResult?.modelUrl}
|
||||
|
|
@ -1002,13 +1006,15 @@ export const ImageTo3DExperience = ({
|
|||
</div>
|
||||
|
||||
{/* Right: History List */}
|
||||
<div className="glass flex h-full flex-col overflow-hidden rounded-xl border border-ui-panel-border">
|
||||
<div className="flex items-center justify-between p-4">
|
||||
<h3 className="font-semibold text-base-fg/80">History</h3>
|
||||
<div className="flex h-full flex-col overflow-hidden border border-ui-panel-border bg-ui-panel">
|
||||
<div className="flex items-center justify-between border-b border-ui-panel-border px-4 py-3">
|
||||
<h3 className="font-mono text-[11px] font-semibold uppercase tracking-[0.12em] text-base-fg/60">
|
||||
History
|
||||
</h3>
|
||||
{results.length > 0 && (
|
||||
<button
|
||||
onClick={resetResults}
|
||||
className="rounded-md bg-red/20 px-3 py-1 text-xs text-white/70 transition-colors hover:bg-red/30"
|
||||
className="rounded-[3px] border border-red/40 bg-red/10 px-3 py-1 text-xs text-white/70 transition-colors hover:bg-red/20"
|
||||
>
|
||||
Clear Session
|
||||
</button>
|
||||
|
|
@ -1024,13 +1030,13 @@ export const ImageTo3DExperience = ({
|
|||
key={result.id}
|
||||
onClick={() => setSelectedResultId(result.id)}
|
||||
className={twMerge(
|
||||
"group flex w-full items-center gap-3 rounded-xl border p-2 text-left transition-all hover:bg-ui-controls/40",
|
||||
"group flex w-full items-center gap-3 rounded-none border p-2 text-left transition-all hover:bg-white/5",
|
||||
isSelected
|
||||
? "border-primary/50 bg-primary/10"
|
||||
: "border-transparent bg-ui-controls/20",
|
||||
? "border-white bg-white/10"
|
||||
: "border-white/10 bg-transparent hover:border-white/30",
|
||||
)}
|
||||
>
|
||||
<div className="relative aspect-square h-14 w-14 shrink-0 overflow-hidden rounded-lg bg-black/30 ring-1 ring-white/5">
|
||||
<div className="relative aspect-square h-14 w-14 shrink-0 overflow-hidden border border-white/10 bg-black/30">
|
||||
{result.previewUrl ? (
|
||||
<img
|
||||
src={result.previewUrl}
|
||||
|
|
@ -1049,7 +1055,7 @@ export const ImageTo3DExperience = ({
|
|||
)}
|
||||
{isPending && (
|
||||
<div className="absolute inset-0 flex items-center justify-center bg-black/30">
|
||||
<div className="h-5 w-5 animate-spin rounded-full border-2 border-white/20 border-t-primary" />
|
||||
<div className="h-5 w-5 animate-spin rounded-full border-2 border-white/20 border-t-white" />
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
|
@ -1063,7 +1069,7 @@ export const ImageTo3DExperience = ({
|
|||
Generating…
|
||||
</span>
|
||||
) : (
|
||||
<span className="text-base-fg/45">
|
||||
<span className="font-mono tabular-nums text-base-fg/45">
|
||||
{formatTime(result.timestamp)}
|
||||
</span>
|
||||
)}
|
||||
|
|
@ -1093,7 +1099,7 @@ export const ImageTo3DExperience = ({
|
|||
<div ref={promptContentRef}>
|
||||
<div
|
||||
className={twMerge(
|
||||
"glass w-full rounded-xl shadow-2xl ring-1 ring-white/10",
|
||||
"glass w-full",
|
||||
hasResults ? "p-3" : "p-5",
|
||||
)}
|
||||
>
|
||||
|
|
@ -1138,7 +1144,7 @@ export const ImageTo3DExperience = ({
|
|||
activeTab={activeMode}
|
||||
onTabChange={(tabId) => setActiveMode(tabId as Mode)}
|
||||
className="w-fit"
|
||||
indicatorClassName="bg-primary/25"
|
||||
indicatorClassName="bg-white/15"
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
|
|
@ -1195,12 +1201,12 @@ export const ImageTo3DExperience = ({
|
|||
|
||||
{previewImage && (
|
||||
<div
|
||||
className="fixed inset-0 z-50 flex items-center justify-center bg-black/80 backdrop-blur-sm"
|
||||
className="fixed inset-0 z-50 flex items-center justify-center bg-black/90"
|
||||
onClick={() => setPreviewImage(null)}
|
||||
>
|
||||
<button
|
||||
type="button"
|
||||
className="absolute right-4 top-4 flex h-8 w-8 items-center justify-center rounded-full bg-black/60 text-white transition-colors hover:bg-black/80"
|
||||
className="absolute right-4 top-4 flex h-8 w-8 items-center justify-center rounded-[3px] border border-white/15 bg-black/60 text-white transition-colors hover:border-white/30 hover:bg-black"
|
||||
onClick={() => setPreviewImage(null)}
|
||||
>
|
||||
<XIcon className="h-4 w-4" />
|
||||
|
|
@ -1208,7 +1214,7 @@ export const ImageTo3DExperience = ({
|
|||
<img
|
||||
src={previewImage}
|
||||
alt="Preview"
|
||||
className="max-h-[80vh] max-w-[80vw] rounded-xl object-contain shadow-2xl"
|
||||
className="max-h-[80vh] max-w-[80vw] border border-white/15 object-contain"
|
||||
onClick={(e) => e.stopPropagation()}
|
||||
/>
|
||||
</div>
|
||||
|
|
|
|||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue