Jack Amadeo
116 lines
3.7 KiB
Rust
116 lines
3.7 KiB
Rust
use crate::state::AppState;
|
|
use axum::{http::StatusCode, routing::post, Json, Router};
|
|
use goose::config::signup_openrouter::OpenRouterAuth;
|
|
use goose::config::signup_tetrate::{configure_tetrate, TetrateAuth};
|
|
use goose::config::{configure_openrouter, Config};
|
|
use serde::Serialize;
|
|
use std::sync::Arc;
|
|
use utoipa::ToSchema;
|
|
|
|
#[derive(Serialize, ToSchema)]
|
|
pub struct SetupResponse {
|
|
pub success: bool,
|
|
pub message: String,
|
|
}
|
|
|
|
pub fn routes(state: Arc<AppState>) -> Router {
|
|
Router::new()
|
|
.route("/handle_openrouter", post(start_openrouter_setup))
|
|
.route("/handle_tetrate", post(start_tetrate_setup))
|
|
.with_state(state)
|
|
}
|
|
|
|
#[utoipa::path(
|
|
post,
|
|
path = "/handle_openrouter",
|
|
responses(
|
|
(status = 200, body=SetupResponse)
|
|
),
|
|
)]
|
|
async fn start_openrouter_setup() -> Result<Json<SetupResponse>, StatusCode> {
|
|
tracing::info!("Starting OpenRouter setup flow");
|
|
|
|
let mut auth_flow = OpenRouterAuth::new().map_err(|e| {
|
|
tracing::error!("Failed to initialize auth flow: {}", e);
|
|
StatusCode::INTERNAL_SERVER_ERROR
|
|
})?;
|
|
|
|
tracing::info!("Auth flow initialized, starting complete_flow");
|
|
|
|
match auth_flow.complete_flow().await {
|
|
Ok(api_key) => {
|
|
tracing::info!("Got API key, configuring OpenRouter...");
|
|
|
|
let config = Config::global();
|
|
|
|
if let Err(e) = configure_openrouter(config, api_key) {
|
|
tracing::error!("Failed to configure OpenRouter: {}", e);
|
|
return Ok(Json(SetupResponse {
|
|
success: false,
|
|
message: format!("Failed to configure OpenRouter: {}", e),
|
|
}));
|
|
}
|
|
|
|
tracing::info!("OpenRouter setup completed successfully");
|
|
Ok(Json(SetupResponse {
|
|
success: true,
|
|
message: "OpenRouter setup completed successfully".to_string(),
|
|
}))
|
|
}
|
|
Err(e) => {
|
|
tracing::error!("OpenRouter setup failed: {}", e);
|
|
Ok(Json(SetupResponse {
|
|
success: false,
|
|
message: format!("Setup failed: {}", e),
|
|
}))
|
|
}
|
|
}
|
|
}
|
|
|
|
#[utoipa::path(
|
|
post,
|
|
path = "/handle_tetrate",
|
|
responses(
|
|
(status = 200, body=SetupResponse)
|
|
),
|
|
)]
|
|
async fn start_tetrate_setup() -> Result<Json<SetupResponse>, StatusCode> {
|
|
tracing::info!("Starting Tetrate Agent Router Service setup flow");
|
|
|
|
let mut auth_flow = TetrateAuth::new().map_err(|e| {
|
|
tracing::error!("Failed to initialize auth flow: {}", e);
|
|
StatusCode::INTERNAL_SERVER_ERROR
|
|
})?;
|
|
|
|
tracing::info!("Auth flow initialized, starting complete_flow");
|
|
|
|
match auth_flow.complete_flow().await {
|
|
Ok(api_key) => {
|
|
tracing::info!("Got API key, configuring Tetrate Agent Router Service...");
|
|
|
|
let config = Config::global();
|
|
|
|
if let Err(e) = configure_tetrate(config, api_key) {
|
|
tracing::error!("Failed to configure Tetrate Agent Router Service: {}", e);
|
|
return Ok(Json(SetupResponse {
|
|
success: false,
|
|
message: format!("Failed to configure Tetrate Agent Router Service: {}", e),
|
|
}));
|
|
}
|
|
|
|
tracing::info!("Tetrate Agent Router Service setup completed successfully");
|
|
Ok(Json(SetupResponse {
|
|
success: true,
|
|
message: "Tetrate Agent Router Service setup completed successfully".to_string(),
|
|
}))
|
|
}
|
|
Err(e) => {
|
|
tracing::error!("Tetrate Agent Router Service setup failed: {}", e);
|
|
Ok(Json(SetupResponse {
|
|
success: false,
|
|
message: format!("Setup failed: {}", e),
|
|
}))
|
|
}
|
|
}
|
|
}
|