diff --git a/crates/api_clients/seedance2pro/src/lib.rs b/crates/api_clients/seedance2pro/src/lib.rs index 872124d599..438641538f 100644 --- a/crates/api_clients/seedance2pro/src/lib.rs +++ b/crates/api_clients/seedance2pro/src/lib.rs @@ -5,3 +5,4 @@ pub (crate) mod test_utils; pub mod creds; pub mod error; pub mod requests; +pub mod utils; diff --git a/crates/api_clients/seedance2pro/src/requests/generate_video/generate_video.rs b/crates/api_clients/seedance2pro/src/requests/generate_video/generate_video.rs index 42817fb258..fadf37d8e8 100644 --- a/crates/api_clients/seedance2pro/src/requests/generate_video/generate_video.rs +++ b/crates/api_clients/seedance2pro/src/requests/generate_video/generate_video.rs @@ -4,12 +4,12 @@ use crate::error::seedance2pro_error::Seedance2ProError; use crate::error::seedance2pro_generic_api_error::Seedance2ProGenericApiError; use crate::error::seedance2pro_specific_api_error::Seedance2ProSpecificApiError; use crate::requests::generate_video::request_types::*; +use crate::utils::common_headers::FIREFOX_USER_AGENT; use log::info; use wreq::Client; use wreq_util::Emulation; const RUN_TASK_URL: &str = "https://seedance2-pro.com/api/trpc/workflow.runTask?batch=1"; -const FIREFOX_USER_AGENT: &str = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:147.0) Gecko/20100101 Firefox/147.0"; // --- Request args --- diff --git a/crates/api_clients/seedance2pro/src/requests/poll_orders/poll_orders.rs b/crates/api_clients/seedance2pro/src/requests/poll_orders/poll_orders.rs index efe7ebf1b7..de601f0799 100644 --- a/crates/api_clients/seedance2pro/src/requests/poll_orders/poll_orders.rs +++ b/crates/api_clients/seedance2pro/src/requests/poll_orders/poll_orders.rs @@ -3,15 +3,13 @@ use crate::error::seedance2pro_client_error::Seedance2ProClientError; use crate::error::seedance2pro_error::Seedance2ProError; use crate::error::seedance2pro_generic_api_error::Seedance2ProGenericApiError; use crate::requests::poll_orders::request_types::*; +use crate::utils::common_headers::FIREFOX_USER_AGENT; use log::info; use wreq::Client; use wreq_util::Emulation; const GET_ORDERS_BASE_URL: &str = "https://seedance2-pro.com/api/trpc/userOrder.getOrders"; -const FIREFOX_USER_AGENT: &str = - "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:147.0) Gecko/20100101 Firefox/147.0"; - /// Builds the tRPC `input` JSON for the getOrders endpoint. /// When `cursor` is `Some`, it is included in the JSON payload. fn build_input_json(cursor: Option) -> String { diff --git a/crates/api_clients/seedance2pro/src/requests/prepare_image_upload/prepare_image_upload.rs b/crates/api_clients/seedance2pro/src/requests/prepare_image_upload/prepare_image_upload.rs index 08ee8dd375..0b5bd642a5 100644 --- a/crates/api_clients/seedance2pro/src/requests/prepare_image_upload/prepare_image_upload.rs +++ b/crates/api_clients/seedance2pro/src/requests/prepare_image_upload/prepare_image_upload.rs @@ -3,6 +3,7 @@ use crate::error::seedance2pro_client_error::Seedance2ProClientError; use crate::error::seedance2pro_error::Seedance2ProError; use crate::error::seedance2pro_generic_api_error::Seedance2ProGenericApiError; use crate::requests::prepare_image_upload::request_types::*; +use crate::utils::common_headers::FIREFOX_USER_AGENT; use chrono::Utc; use log::info; use rand::Rng; @@ -10,7 +11,6 @@ use wreq::Client; use wreq_util::Emulation; const SIGNED_UPLOAD_URL: &str = "https://seedance2-pro.com/api/trpc/uploads.signedUploadUrl?batch=1"; -const FIREFOX_USER_AGENT: &str = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:147.0) Gecko/20100101 Firefox/147.0"; /// Generates a material path based on the current time. /// Format: `materials/YYYYMMDD/-.png` diff --git a/crates/api_clients/seedance2pro/src/requests/upload_image/upload_image.rs b/crates/api_clients/seedance2pro/src/requests/upload_image/upload_image.rs index 33d7456068..0cb8984fc9 100644 --- a/crates/api_clients/seedance2pro/src/requests/upload_image/upload_image.rs +++ b/crates/api_clients/seedance2pro/src/requests/upload_image/upload_image.rs @@ -1,12 +1,12 @@ use crate::error::seedance2pro_client_error::Seedance2ProClientError; use crate::error::seedance2pro_error::Seedance2ProError; use crate::error::seedance2pro_generic_api_error::Seedance2ProGenericApiError; +use crate::utils::common_headers::FIREFOX_USER_AGENT; use log::info; use url::Url; use wreq::Client; use wreq_util::Emulation; -const FIREFOX_USER_AGENT: &str = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:147.0) Gecko/20100101 Firefox/147.0"; const STATIC_BASE_URL: &str = "https://static.seedance2-pro.com"; pub struct UploadImageArgs { diff --git a/crates/api_clients/seedance2pro/src/utils/common_headers.rs b/crates/api_clients/seedance2pro/src/utils/common_headers.rs new file mode 100644 index 0000000000..d10f62ab06 --- /dev/null +++ b/crates/api_clients/seedance2pro/src/utils/common_headers.rs @@ -0,0 +1,2 @@ +pub const FIREFOX_USER_AGENT: &str = + "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:147.0) Gecko/20100101 Firefox/147.0"; diff --git a/crates/api_clients/seedance2pro/src/utils/mod.rs b/crates/api_clients/seedance2pro/src/utils/mod.rs new file mode 100644 index 0000000000..aae60bbf75 --- /dev/null +++ b/crates/api_clients/seedance2pro/src/utils/mod.rs @@ -0,0 +1 @@ +pub mod common_headers;