byokey-proxy 1.1.0

Bring Your Own Keys — AI subscription-to-API proxy gateway
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Amp CLI compatibility layer — served on the main byokey port alongside
//! the REST AI proxy and `ConnectRPC` management service.
//!
//! Routes (all paths are served **without** an `/amp` prefix, matching
//! what the amp CLI sends on the wire):
//! - `GET  /v1/login`              -> 302 redirect to ampcode.com/login.
//! - `GET  /auth/cli-login`        -> 302 redirect to ampcode.com/auth/cli-login.
//! - `ANY  /v0/management/{*path}` -> proxy to ampcode.com/v0/management/*.
//! - `POST /api/provider/*`        -> provider-specific handlers.
//! - `ANY  /api/{*path}`           -> catch-all proxy to ampcode.com.
//!
//! Submodules:
//! - [`provider`] — `AmpCode` provider-namespaced AI endpoints (`/api/provider/*`).
//! - [`threads`]  — parser + in-memory index for local Amp CLI thread files
//!   (consumed by the `ConnectRPC` management service).

pub mod provider;
pub mod threads;

use axum::{
    extract::RawQuery,
    http::{HeaderValue, StatusCode},
    response::IntoResponse,
};

/// Redirects Amp CLI to the web login page.
pub async fn login_redirect() -> impl IntoResponse {
    (
        StatusCode::FOUND,
        [(
            axum::http::header::LOCATION,
            HeaderValue::from_static("https://ampcode.com/login"),
        )],
    )
}

/// Handles `GET /auth/cli-login?authToken=...&callbackPort=...`
///
/// `amp login` opens this URL in the browser. We forward it to `AmpCode`'s
/// own login endpoint so `AmpCode` can authenticate the user and then
/// callback to `http://localhost:{callbackPort}/...` directly.
pub async fn cli_login_redirect(RawQuery(query): RawQuery) -> impl IntoResponse {
    let url = match query {
        Some(q) => format!("https://ampcode.com/auth/cli-login?{q}"),
        None => "https://ampcode.com/auth/cli-login".to_string(),
    };
    let location = HeaderValue::from_str(&url)
        .unwrap_or_else(|_| HeaderValue::from_static("https://ampcode.com/amp/auth/cli-login"));
    (
        StatusCode::FOUND,
        [(axum::http::header::LOCATION, location)],
    )
}