OXYGENHUB
# 1870e4fa9808 oxygenhub / crates / hub / src / web / mod.rs
34.8 KiB · Rust · last changed Rerun one job of a run, with what depends on it 23 days ago source blame history raw
1
//! Pages (hub spec §6): the router, the shell every template extends, the
2
//! error page, viewer-aware lookups and askama filters. Handlers are thin:
3
//! they resolve names, call the view layer under the repository lock, and
4
//! render.
5
6
mod admin;
7
pub(crate) mod assistants;
8
mod avatar;
9
mod backups;
10
pub mod chain;
11
pub mod coverage;
12
mod device;
13
mod diffusion;
14
pub mod edit;
15
pub mod filters;
16
pub mod flow;
17
mod forks;
18
pub mod help;
19
pub mod history;
20
pub mod home;
21
pub mod icons;
22
23
/// A state as people read it (oxygenhub#170): `02a6c9d4746a+… (14 heads)`
24
/// for heads joined by `+`, which can run to hundreds of characters; a
25
/// single head, or text that is not heads, unchanged.
26
pub fn state_label(text: &str) -> String {
27
    let parts: Vec<&str> = text.split('+').collect();
28
    let heads = parts.len() > 1 && parts.iter().all(|p| p.len() >= 7 && p.chars().all(|c| c.is_ascii_hexdigit()));
29
    if !heads {
30
        return text.to_string();
31
    }
32
    format!("{}+… ({} heads)", &parts[0][..12.min(parts[0].len())], parts.len())
33
}
34
mod inbox;
35
pub mod issues;
36
mod orgs;
37
pub mod projects;
38
pub mod proposals;
39
mod recovery;
40
mod refs;
41
pub mod release_trust;
42
pub mod releases;
43
mod repo;
44
mod repo_settings;
45
pub mod reports;
46
mod resolve;
47
pub mod runners;
48
mod secrets;
49
mod session;
50
pub(crate) mod settings;
51
mod signin;
52
mod statics;
53
pub mod test_reports;
54
mod variables;
55
mod webhooks;
56
mod wiki;
57
58
use crate::auth::{self, Access, Role, Viewer};
59
use crate::repos::RepoError;
60
use crate::{db, Shared};
61
use askama::Template;
62
use axum::extract::DefaultBodyLimit;
63
use axum::http::{header, StatusCode};
64
use axum::response::{Html, IntoResponse, Redirect, Response};
65
use axum::routing::{any, get, post};
66
use axum::Router;
67
use tower_http::compression::CompressionLayer;
68
use tower_http::trace::TraceLayer;
69
70
pub fn router(app: Shared) -> Router {
71
    let push_limit = app.config.limits.push_body_limit.0 as usize;
72
    Router::new()
73
        .route("/", get(home::index))
74
        .route("/explore", get(home::explore))
75
        .route("/avatar/agent/{name}", get(avatar::agent))
76
        .route("/avatar/{name}", get(avatar::avatar))
77
        .route("/health", get(home::health))
78
        .route("/about", get(home::about))
79
        .route("/.well-known/oxyhub-key", get(home::hub_key))
80
        .route("/.well-known/oxyhub-ca.pem", get(home::hub_ca))
81
        .route("/.well-known/openid-configuration", get(crate::oidc::discovery))
82
        .route("/.well-known/jwks.json", get(crate::oidc::jwks))
83
        .route("/style.css", get(statics::style))
84
        .route("/favicon.svg", get(statics::favicon))
85
        .route("/favicon.ico", get(statics::favicon))
86
        .route("/static/{*path}", get(statics::asset))
87
        .route("/login", get(session::login_form).post(session::login))
88
        .route("/device", get(device::form).post(device::answer))
89
        .route("/logout", post(session::logout))
90
        .route("/register", get(session::register_form).post(session::register))
91
        .route("/settings", get(settings::profile).post(settings::update_profile))
92
        .route("/settings/password", post(settings::change_password))
93
        .route("/settings/appearance", post(settings::appearance))
94
        .route("/settings/security", get(settings::security))
95
        .route("/settings/security/totp/begin", post(settings::totp_begin))
96
        .route("/settings/security/totp/enable", post(settings::totp_enable))
97
        .route("/settings/security/totp/disable", post(settings::totp_disable))
98
        .route("/settings/security/sessions", post(settings::sessions_reset))
99
        .route("/settings/keys", get(settings::keys))
100
        .route("/settings/keys/upload", post(settings::keys_upload))
101
        .route("/settings/keys/verify", post(settings::keys_verify))
102
        .route("/settings/keys/unlink", post(settings::keys_unlink))
103
        .route("/settings/tokens", get(settings::tokens).post(settings::tokens_create))
104
        .route("/settings/backups", get(backups::user).post(backups::user_save))
105
        .route("/settings/backups/run", post(backups::user_run))
106
        .route("/settings/backups/test", post(backups::user_test))
107
        .route("/orgs/{name}/settings/backups", get(backups::org).post(backups::org_save))
108
        .route("/orgs/{name}/settings/backups/run", post(backups::org_run))
109
        .route("/orgs/{name}/settings/backups/test", post(backups::org_test))
110
        .route("/settings/tokens/{id}/delete", post(settings::tokens_delete))
111
        .route("/new", get(repo_settings::new_form).post(repo_settings::create))
112
        .route("/orgs/new", get(orgs::new_form).post(orgs::create))
113
        .route("/orgs/{name}/settings", get(orgs::settings).post(orgs::update))
114
        .route("/orgs/{name}/settings/members", post(orgs::member))
115
        .route("/orgs/{name}/settings/teams", post(orgs::team))
116
        .route("/orgs/{name}/settings/hooks", post(orgs::hooks))
117
        .route("/orgs/{name}/settings/assistants", post(orgs::assistants))
118
        .route("/orgs/{name}/settings/assistants/models", post(orgs::assistant_models))
119
        .route("/orgs/{name}/settings/variables", post(orgs::variables))
120
        .route("/orgs/{name}/settings/secrets", post(orgs::secrets))
121
        .route("/orgs/{name}/settings/storage", post(orgs::storage))
122
        .route("/orgs/{name}/settings/runners/token", post(orgs::runner_token))
123
        .route("/orgs/{name}/settings/runners/{id}/remove", post(orgs::runner_remove))
124
        .route("/orgs/{name}/settings/hooks/{id}/deliveries", get(orgs::deliveries))
125
        .route("/orgs/{name}/settings/hooks/{id}/redeliver/{delivery}", post(orgs::redeliver))
126
        .route("/orgs/{name}/settings/avatar", post(orgs::avatar).layer(DefaultBodyLimit::max(avatar::MAX_BYTES + (64 << 10))))
127
        .route("/settings/avatar", post(settings::avatar).layer(DefaultBodyLimit::max(avatar::MAX_BYTES + (64 << 10))))
128
        .route("/notifications", get(inbox::inbox))
129
        .route("/notifications/read", post(inbox::mark_read))
130
        .route("/settings/notifications", get(inbox::unsubscribe).post(inbox::prefs))
131
        .route("/search", get(inbox::search))
132
        .route("/admin", get(admin::index))
133
        .route("/admin/registration", post(admin::set_mode))
134
        .route("/admin/invite", post(admin::invite))
135
        .route("/admin/account", post(admin::account))
136
        .route("/admin/storage", get(admin::storage).post(admin::maintain))
137
        .route("/admin/runners", get(runners::list))
138
        .route("/admin/runners/settings", get(runners::settings).post(runners::save_settings))
139
        .route("/admin/runners/images", get(runners::images))
140
        .route("/admin/diffusion", get(diffusion::admin))
141
        .route("/{owner}/{name}/diffusion", get(diffusion::page))
142
        .route("/admin/runners/token", post(runners::new_token))
143
        .route("/admin/runners/{id}", get(runners::show))
144
        .route("/admin/runners/{id}/state", post(runners::set_state))
145
        .route("/admin/runners/{id}/scope", post(runners::set_scope))
146
        .route("/admin/runners/{id}/limits", post(runners::set_limits))
147
        .route("/admin/mail", post(admin::set_mail))
148
        .route("/admin/mail/test", post(admin::test_mail))
149
        .route("/admin/assistants", post(admin::assistants))
150
        .route("/admin/assistants/models", post(admin::assistant_models))
151
        .route("/admin/signin", post(admin::signin))
152
        .route("/login/{provider}", get(signin::start))
153
        .route("/login/{provider}/callback", get(signin::callback))
154
        .route("/settings/security/connect/{provider}", get(signin::connect))
155
        .route("/settings/security/disconnect/{provider}", post(signin::disconnect))
156
        .route("/admin/variables", post(admin::variables))
157
        .route("/admin/secrets", post(admin::secrets))
158
        .route("/admin/runners/{id}/remove", post(runners::remove))
159
        .nest("/api/v1", crate::api::router(app.config.limits.asset_limit.0 as usize))
160
        .route("/{owner}", get(home::owner))
161
        .route("/{owner}/{name}", get(repo::overview))
162
        .route("/{owner}/{name}/tree/{r}", get(repo::tree_root))
163
        .route("/{owner}/{name}/tree/{r}/{*path}", get(repo::tree))
164
        .route("/{owner}/{name}/file/{r}/{*path}", get(repo::file))
165
        .route("/{owner}/{name}/function/{r}/{*path}", get(history::function))
166
        .route("/{owner}/{name}/lanes/{r}", get(history::lanes))
167
        .route("/{owner}/{name}/lanes/{r}/plan", post(history::plan))
168
        .route("/{owner}/{name}/raw/{r}/{*path}", get(repo::raw_file))
169
        .route("/{owner}/{name}/last/{r}", get(repo::last_changes))
170
        .route("/{owner}/{name}/changes/{r}", get(history::history))
171
        .route("/{owner}/{name}/change/{hash}", get(history::change))
172
        .route("/apt", get(crate::apt::home))
173
        .route("/apt/{file}", get(crate::apt::key))
174
        .route("/apt/dists/{suite}/{*rest}", get(crate::apt::dists))
175
        .route("/apt/pool/{*path}", get(crate::apt::pool))
176
        .route("/help", get(help::help))
177
        .route("/projects/{owner}", get(projects::list).post(projects::create))
178
        .route("/projects/{owner}/{number}", get(projects::show).post(projects::act))
179
        .route("/notifications/thread", post(inbox::thread))
180
        .route("/notifications/open/{id}", get(inbox::open))
181
        .route("/login/reset", get(recovery::request_form).post(recovery::request))
182
        .route("/login/reset/{secret}", get(recovery::new_password_form).post(recovery::set_new_password))
183
        .route("/verify/{secret}", get(recovery::verify))
184
        .route("/settings/email/verify", post(settings::resend_verification))
185
        .route("/{owner}/{name}/compare", get(history::compare_index))
186
        .route("/{owner}/{name}/compare/{spec}", get(history::compare))
187
        .route("/{owner}/{name}/bookmarks", get(refs::bookmarks))
188
        .route("/{owner}/{name}/bookmarks/delete", post(refs::delete_bookmark))
189
        .route("/{owner}/{name}/tags", get(refs::tags))
190
        .route("/{owner}/{name}/conflicts", get(refs::conflicts))
191
        .route("/{owner}/{name}/trust", get(refs::trust))
192
        .route("/{owner}/{name}/pushes", get(refs::pushes))
193
        .route("/{owner}/{name}/fork", get(forks::form).post(forks::create))
194
        .route("/{owner}/{name}/forks", get(forks::list))
195
        .route("/{owner}/{name}/settings", get(repo_settings::settings).post(repo_settings::update))
196
        .route("/{owner}/{name}/settings/collaborators", post(repo_settings::collaborator))
197
        .route("/{owner}/{name}/settings/delete", post(repo_settings::delete))
198
        .route("/{owner}/{name}/settings/transfer", post(repo_settings::transfer))
199
        .route("/{owner}/{name}/settings/archive", post(repo_settings::archive))
200
        .route("/{owner}/{name}/settings/wiki/delete", post(repo_settings::wiki_delete))
201
        .route("/{owner}/{name}/wiki", get(wiki::home))
202
        .route("/{owner}/{name}/wiki/_pages", get(wiki::pages))
203
        .route("/{owner}/{name}/wiki/_new", get(wiki::new_form))
204
        .route("/{owner}/{name}/wiki/_raw/{*path}", get(wiki::raw))
205
        .route("/{owner}/{name}/wiki/{slug}", get(wiki::show))
206
        .route("/{owner}/{name}/wiki/{slug}/edit", get(wiki::edit_form).post(wiki::save))
207
        .route("/{owner}/{name}/wiki/{slug}/history", get(wiki::history))
208
        .route("/{owner}/{name}/wiki/{slug}/delete", post(wiki::delete))
209
        .route("/{owner}/{name}/settings/requirements", post(repo_settings::requirements))
210
        .route("/{owner}/{name}/proposals", get(proposals::list))
211
        .route("/{owner}/{name}/proposals/new", get(proposals::new_form).post(proposals::create))
212
        .route("/{owner}/{name}/proposals/new/draft", post(proposals::new_draft))
213
        .route("/{owner}/{name}/proposal/{number}", get(proposals::show))
214
        .route("/{owner}/{name}/proposal/{number}/comment", post(proposals::comment))
215
        .route("/{owner}/{name}/proposal/{number}/comment/{id}/resolve", post(proposals::resolve_comment))
216
        .route("/{owner}/{name}/proposal/{number}/review", post(proposals::review))
217
        .route("/{owner}/{name}/proposal/{number}/reviewers", post(proposals::reviewers))
218
        .route("/{owner}/{name}/proposal/{number}/assistant", post(proposals::assistant))
219
        .route("/{owner}/{name}/proposal/{number}/draft", post(proposals::draft))
220
        .route("/{owner}/{name}/proposal/{number}/edit", post(proposals::edit))
221
        .route("/{owner}/{name}/proposal/{number}/state", post(proposals::state))
222
        .route("/{owner}/{name}/proposal/{number}/land", post(proposals::land))
223
        .route("/{owner}/{name}/proposal/{number}/labels", post(proposals::set_labels))
224
        .route("/{owner}/{name}/proposal/{number}/project", post(projects::place_proposal))
225
        .route("/{owner}/{name}/issue/{number}/project", post(projects::place_issue))
226
        .route("/{owner}/{name}/proposal/{number}/events", get(proposals::events))
227
        .route("/{owner}/{name}/settings/hooks", post(repo_settings::hooks))
228
        .route("/{owner}/{name}/settings/assistants", post(repo_settings::assistants))
229
        .route("/{owner}/{name}/settings/assistants/models", post(repo_settings::assistant_models))
230
        .route("/{owner}/{name}/settings/variables", post(repo_settings::variables))
231
        .route("/{owner}/{name}/settings/secrets", post(repo_settings::secrets))
232
        .route("/{owner}/{name}/settings/hooks/{id}/deliveries", get(repo_settings::deliveries))
233
        .route("/{owner}/{name}/settings/hooks/{id}/redeliver/{delivery}", post(repo_settings::redeliver))
234
        .route("/{owner}/{name}/settings/checks", post(repo_settings::required_checks))
235
        .route("/{owner}/{name}/watch", post(issues::watch))
236
        .route("/{owner}/{name}/star", post(issues::star))
237
        .route("/{owner}/{name}/edit/{r}", post(edit::save))
238
        .route("/{owner}/{name}/edit/{r}/{*path}", get(edit::edit_form))
239
        .route("/{owner}/{name}/resolve/{bookmark}/{*path}", get(resolve::form).post(resolve::save))
240
        .route("/{owner}/{name}/resolve-live/{bookmark}/{*path}", get(resolve::live_events).post(resolve::live_choose))
241
        .route("/{owner}/{name}/newfile/{r}", get(edit::new_form))
242
        .route("/{owner}/{name}/flow", get(flow::runs))
243
        .route("/{owner}/{name}/flow/badge.svg", get(flow::badge))
244
        .route("/{owner}/{name}/tests", get(test_reports::overview))
245
        .route("/{owner}/{name}/tests/case", get(test_reports::case))
246
        .route("/{owner}/{name}/reports", get(reports::overview))
247
        .route("/{owner}/{name}/reports/files/{id}/{mac}/{*path}", get(reports::file))
248
        .route("/{owner}/{name}/reports/{report}", get(reports::show))
249
        .route("/{owner}/{name}/coverage", get(coverage::overview))
250
        .route("/{owner}/{name}/coverage/badge.svg", get(coverage::badge))
251
        .route("/{owner}/{name}/coverage/{id}/file/{*path}", get(coverage::file))
252
        .route("/{owner}/{name}/flow/{id}", get(flow::run))
253
        .route("/{owner}/{name}/flow/{id}/live", get(flow::live))
254
        .route("/{owner}/{name}/flow/start", post(flow::start))
255
        .route("/{owner}/{name}/flow/{id}/rerun", post(flow::rerun))
256
        .route("/{owner}/{name}/flow/{id}/jobs/{job}/rerun", post(flow::rerun_job))
257
        .route("/{owner}/{name}/flow/{id}/cancel", post(flow::cancel))
258
        .route("/{owner}/{name}/flow/stamp/{r}", get(flow::stamp))
259
        .route("/{owner}/{name}/flow/at/{r}", get(flow::at))
260
        .route("/{owner}/{name}/receipts/{attest}", get(flow::receipt_file))
261
        .route("/{owner}/{name}/flow/{id}/artifacts/{job}/{artifact}", get(flow::artifact_file))
262
        .route("/{owner}/{name}/flow/{id}/artifacts/{job}/{artifact}/attach", post(flow::attach))
263
        .route("/{owner}/{name}/releases", get(releases::list))
264
        .route("/{owner}/{name}/releases/new", get(releases::new_form).post(releases::create))
265
        .route("/{owner}/{name}/releases/tag/{*tag}", get(releases::show))
266
        .route("/{owner}/{name}/releases/trust/{*tag}", get(release_trust::page))
267
        .route("/{owner}/{name}/releases/edit/{id}", post(releases::edit))
268
        .route(
269
            "/{owner}/{name}/releases/upload/{id}",
270
            post(releases::upload).layer(DefaultBodyLimit::max(app.config.limits.asset_limit.0 as usize + (1 << 20))),
271
        )
272
        .route("/{owner}/{name}/asset/{id}/{file}", get(releases::download))
273
        .route("/{owner}/{name}/archive", get(releases::source_archive))
274
        .route("/{owner}/{name}/search", get(repo::code_search))
275
        .route("/{owner}/{name}/info/{*rest}", any(git_info))
276
        .route("/{owner}/{name}/HEAD", get(git_head))
277
        .route("/{owner}/{name}/objects/{*rest}", get(git_objects))
278
        .route("/{owner}/{name}/git-upload-pack", post(git_upload_pack))
279
        .route("/{owner}/{name}/git-receive-pack", post(git_receive_pack))
280
        .route("/{owner}/{name}/settings/mirror", post(repo_settings::mirror))
281
        .route("/{owner}/{name}/settings/flow", post(repo_settings::flow))
282
        .route("/{owner}/{name}/settings/diffusion", post(repo_settings::diffusion))
283
        .route("/{owner}/{name}/settings/storage", post(repo_settings::storage))
284
        .route("/{owner}/{name}/settings/coverage", post(repo_settings::coverage))
285
        .route("/{owner}/{name}/settings/runners/token", post(repo_settings::runner_token))
286
        .route("/{owner}/{name}/settings/runners/{id}/remove", post(repo_settings::runner_remove))
287
        .route("/{owner}/{name}/issues", get(issues::list))
288
        .route("/{owner}/{name}/issues/new", get(issues::new_form).post(issues::create))
289
        .route("/{owner}/{name}/issue/{number}", get(issues::show))
290
        .route("/{owner}/{name}/issue/{number}/comment", post(issues::comment))
291
        .route("/{owner}/{name}/issue/{number}/comment/{id}/delete", post(issues::delete_comment))
292
        .route("/{owner}/{name}/issue/{number}/edit", post(issues::edit))
293
        .route("/{owner}/{name}/issue/{number}/state", post(issues::state))
294
        .route("/{owner}/{name}/issue/{number}/labels", post(issues::set_labels))
295
        .route("/{owner}/{name}/issue/{number}/assign", post(issues::assign))
296
        .route("/{owner}/{name}/issue/{number}/milestone", post(issues::set_milestone))
297
        .route("/{owner}/{name}/issue/{number}/subscribe", post(issues::subscribe))
298
        .route("/{owner}/{name}/labels", get(issues::labels).post(issues::save_label))
299
        .route("/{owner}/{name}/milestones", get(issues::milestones).post(issues::save_milestone))
300
        .route("/{owner}/{name}/oxy/v1/{endpoint}", any(crate::protocol::dispatch).layer(DefaultBodyLimit::max(push_limit)))
301
        .fallback(not_found)
302
        .layer(axum::middleware::from_fn_with_state(app.clone(), same_origin))
303
        .layer(CompressionLayer::new())
304
        .layer(TraceLayer::new_for_http())
305
        .with_state(app)
306
}
307
308
// Git's smart HTTP paths under `/{owner}/{name}.git/`, forwarded to the
309
// mirror (hub spec §14).
310
async fn git_info(
311
    st: axum::extract::State<Shared>,
312
    axum::extract::Path((o, n, rest)): axum::extract::Path<(String, String, String)>,
313
    m: axum::http::Method,
314
    q: axum::extract::RawQuery,
315
    h: axum::http::HeaderMap,
316
    b: axum::body::Bytes,
317
) -> Response {
318
    crate::mirror::serve(st, axum::extract::Path((o, n, format!("info/{rest}"))), m, q, h, b).await
319
}
320
321
async fn git_head(
322
    st: axum::extract::State<Shared>,
323
    axum::extract::Path((o, n)): axum::extract::Path<(String, String)>,
324
    m: axum::http::Method,
325
    q: axum::extract::RawQuery,
326
    h: axum::http::HeaderMap,
327
    b: axum::body::Bytes,
328
) -> Response {
329
    crate::mirror::serve(st, axum::extract::Path((o, n, "HEAD".into())), m, q, h, b).await
330
}
331
332
async fn git_objects(
333
    st: axum::extract::State<Shared>,
334
    axum::extract::Path((o, n, rest)): axum::extract::Path<(String, String, String)>,
335
    m: axum::http::Method,
336
    q: axum::extract::RawQuery,
337
    h: axum::http::HeaderMap,
338
    b: axum::body::Bytes,
339
) -> Response {
340
    crate::mirror::serve(st, axum::extract::Path((o, n, format!("objects/{rest}"))), m, q, h, b).await
341
}
342
343
async fn git_upload_pack(
344
    st: axum::extract::State<Shared>,
345
    axum::extract::Path((o, n)): axum::extract::Path<(String, String)>,
346
    m: axum::http::Method,
347
    q: axum::extract::RawQuery,
348
    h: axum::http::HeaderMap,
349
    b: axum::body::Bytes,
350
) -> Response {
351
    crate::mirror::serve(st, axum::extract::Path((o, n, "git-upload-pack".into())), m, q, h, b).await
352
}
353
354
async fn git_receive_pack(
355
    st: axum::extract::State<Shared>,
356
    axum::extract::Path((o, n)): axum::extract::Path<(String, String)>,
357
    m: axum::http::Method,
358
    q: axum::extract::RawQuery,
359
    h: axum::http::HeaderMap,
360
    b: axum::body::Bytes,
361
) -> Response {
362
    crate::mirror::serve(st, axum::extract::Path((o, n, "git-receive-pack".into())), m, q, h, b).await
363
}
364
365
/// Cross-site form posts are refused (hub spec §15.2): a browser that
366
/// sends `Origin` or `Sec-Fetch-Site` must name this site. Protocol and
367
/// API clients send neither.
368
async fn same_origin(
369
    axum::extract::State(app): axum::extract::State<Shared>,
370
    req: axum::extract::Request,
371
    next: axum::middleware::Next,
372
) -> Response {
373
    if req.method() != axum::http::Method::GET && req.method() != axum::http::Method::HEAD {
374
        let headers = req.headers();
375
        if headers.get("sec-fetch-site").and_then(|v| v.to_str().ok()) == Some("cross-site") {
376
            return (StatusCode::FORBIDDEN, "cross-site request refused").into_response();
377
        }
378
        if let Some(origin) = headers.get(header::ORIGIN).and_then(|v| v.to_str().ok()) {
379
            let host_of = |u: &str| u.split("://").nth(1).unwrap_or(u).split('/').next().unwrap_or("").to_ascii_lowercase();
380
            let request_host = headers.get(header::HOST).and_then(|v| v.to_str().ok()).map(|h| h.to_ascii_lowercase());
381
            let ok = host_of(origin) == host_of(app.config.base_url()) || request_host.as_deref() == Some(host_of(origin).as_str());
382
            if !ok && origin != "null" {
383
                return (StatusCode::FORBIDDEN, "cross-site request refused").into_response();
384
            }
385
        }
386
    }
387
    next.run(req).await
388
}
389
390
/// What every page's chrome needs.
391
#[derive(Clone, Debug)]
392
pub struct Shell {
393
    pub site: String,
394
    pub base_url: String,
395
    pub path: String,
396
    pub version: &'static str,
397
    /// What was built: `oxy describe` at deploy, a Flow build's heads, or "local build".
398
    pub build: &'static str,
399
    /// The signed-in username, if any.
400
    pub user: Option<String>,
401
    pub is_admin: bool,
402
    pub csrf: String,
403
    pub unread: i64,
404
}
405
406
impl Shell {
407
    /// The version in the stylesheet and script URLs (`statics::assets_version`).
408
    pub fn assets(&self) -> &'static str {
409
        statics::assets_version()
410
    }
411
412
    pub fn new(app: &Shared, path: &str, viewer: &Viewer) -> Shell {
413
        Shell {
414
            site: app.config.site.name.clone(),
415
            base_url: app.config.base_url().to_string(),
416
            path: path.to_string(),
417
            version: env!("CARGO_PKG_VERSION"),
418
            build: env!("OXY_BUILD_ID"),
419
            user: viewer.account().map(|a| a.username.clone()),
420
            is_admin: viewer.is_admin(),
421
            csrf: viewer.csrf().to_string(),
422
            unread: viewer.0.as_ref().map(|v| v.unread).unwrap_or(0),
423
        }
424
    }
425
426
    pub fn anonymous(app: &Shared, path: &str) -> Shell {
427
        Shell::new(app, path, &Viewer(None))
428
    }
429
}
430
431
/// The repository chrome: tabs and counts.
432
#[derive(Clone, Debug)]
433
pub struct RepoShell {
434
    pub owner: String,
435
    pub name: String,
436
    pub url: String,
437
    pub description: String,
438
    pub default_bookmark: String,
439
    pub visibility: String,
440
    pub tab: &'static str,
441
    pub clone_url: String,
442
    /// The viewer's role, for showing settings and write actions.
443
    pub role: Option<Role>,
444
    /// Open issues and proposals, shown on their tabs.
445
    pub open_issues: i64,
446
    pub open_proposals: i64,
447
    /// Read-only (#108).
448
    pub archived: bool,
449
    /// Whether the repository has a wiki (§6.11).
450
    pub wiki: bool,
451
    /// The bucket its data lives in, when not this hub's disk (#96), as
452
    /// the header's provider icon shows it (#123).
453
    pub stored_in: Option<crate::storage::StoreBadge>,
454
    /// Whether it has coverage reports, for the Coverage tab (#113).
455
    pub coverage: bool,
456
    /// Whether it has test reports, for the Tests tab (#114).
457
    pub tests: bool,
458
    /// Whether it has reports of its own (#127); with coverage or tests,
459
    /// the Reports tab.
460
    pub reports: bool,
461
    /// Whether it pins, or is pinned by, another repository here: the
462
    /// Diffusion tab (#150).
463
    pub diffusion: bool,
464
}
465
466
impl RepoShell {
467
    pub fn new(app: &Shared, row: &db::repos::RepoRow, tab: &'static str) -> RepoShell {
468
        RepoShell {
469
            owner: row.owner.clone(),
470
            name: row.name.clone(),
471
            url: format!("/{}/{}", row.owner, row.name),
472
            description: row.description.clone(),
473
            default_bookmark: row.default_bookmark.clone(),
474
            visibility: row.visibility.clone(),
475
            tab,
476
            clone_url: format!("{}/{}/{}", app.config.base_url(), row.owner, row.name),
477
            role: None,
478
            open_issues: row.open_issues,
479
            open_proposals: row.open_proposals,
480
            archived: row.archived,
481
            wiki: crate::repos::RepoName::new(&row.owner, &row.name).is_some_and(|n| app.wikis.exists(&n)),
482
            stored_in: crate::storage::location(app, &row.owner, &row.name).and_then(|u| crate::storage::StoreBadge::of(&u)),
483
            coverage: crate::coverage::has_reports(app, row.id),
484
            tests: crate::junit::has_reports(app, row.id),
485
            reports: crate::reports::has_reports(app, row.id),
486
            diffusion: crate::diffusion::involved(app, row.id),
487
        }
488
    }
489
490
    pub fn with_role(mut self, role: Role) -> RepoShell {
491
        self.role = Some(role);
492
        self
493
    }
494
495
    pub fn can(&self, access: Access) -> bool {
496
        auth::roles::allows(self.role, access)
497
    }
498
499
    pub fn can_maintain(&self) -> bool {
500
        self.can(Access::Maintain)
501
    }
502
}
503
504
/// Render a template as an HTML response.
505
/// One page of a long list: which page, how many, and links that keep
506
/// the rest of the query.
507
#[derive(Clone, Debug)]
508
pub struct Pager {
509
    pub page: usize,
510
    pub pages: usize,
511
    pub total: usize,
512
    /// The query without `page`, each pair encoded, joined with `&`.
513
    query: String,
514
}
515
516
impl Pager {
517
    pub const PER_PAGE: usize = 25;
518
519
    /// `page` counts from 1; out-of-range pages clamp to the last.
520
    pub fn new(total: usize, page: Option<usize>, query: &[(&str, &str)]) -> Pager {
521
        let pages = total.div_ceil(Self::PER_PAGE).max(1);
522
        let page = page.unwrap_or(1).clamp(1, pages);
523
        let query = query
524
            .iter()
525
            .filter(|(_, v)| !v.is_empty())
526
            .map(|(k, v)| format!("{k}={}", settings::urlencode(v).replace('/', "%2F")))
527
            .collect::<Vec<_>>()
528
            .join("&");
529
        Pager { page, pages, total, query }
530
    }
531
532
    /// This page's items.
533
    pub fn slice<T>(&self, all: Vec<T>) -> Vec<T> {
534
        all.into_iter().skip((self.page - 1) * Self::PER_PAGE).take(Self::PER_PAGE).collect()
535
    }
536
537
    pub fn url(&self, page: usize) -> String {
538
        if self.query.is_empty() {
539
            format!("?page={page}")
540
        } else {
541
            format!("?{}&page={page}", self.query)
542
        }
543
    }
544
545
    /// The same query with another sort, back on page one.
546
    pub fn sorted(&self, sort: &str) -> String {
547
        let rest: Vec<&str> = self.query.split('&').filter(|p| !p.is_empty() && !p.starts_with("sort=")).collect();
548
        let mut q = rest.join("&");
549
        if !q.is_empty() {
550
            q.push('&');
551
        }
552
        format!("?{q}sort={sort}")
553
    }
554
}
555
556
pub struct Page<T: Template>(pub T);
557
558
impl<T: Template> IntoResponse for Page<T> {
559
    fn into_response(self) -> Response {
560
        match self.0.render() {
561
            Ok(s) => Html(s).into_response(),
562
            Err(e) => {
563
                tracing::error!("template: {e}");
564
                (StatusCode::INTERNAL_SERVER_ERROR, "template error").into_response()
565
            }
566
        }
567
    }
568
}
569
570
#[derive(Template)]
571
#[template(path = "error.html")]
572
struct ErrorPage {
573
    shell: Shell,
574
    status: u16,
575
    title: String,
576
    detail: String,
577
}
578
579
/// A page-level failure, rendered inside the shell, or a redirect.
580
#[derive(Debug)]
581
pub struct Error {
582
    pub status: StatusCode,
583
    pub detail: String,
584
    shell: Option<Box<Shell>>,
585
    redirect: Option<String>,
586
}
587
588
impl Error {
589
    pub fn redirect(to: String) -> Error {
590
        Error { status: StatusCode::SEE_OTHER, detail: String::new(), shell: None, redirect: Some(to) }
591
    }
592
593
    pub fn not_found(shell: Shell) -> Error {
594
        Error {
595
            status: StatusCode::NOT_FOUND,
596
            detail: "There is nothing at this address.".into(),
597
            shell: Some(Box::new(shell)),
598
            redirect: None,
599
        }
600
    }
601
602
    pub fn bad(shell: Shell, detail: impl Into<String>) -> Error {
603
        Error { status: StatusCode::BAD_REQUEST, detail: detail.into(), shell: Some(Box::new(shell)), redirect: None }
604
    }
605
606
    pub fn from_repo(shell: Shell, e: RepoError) -> Error {
607
        match e {
608
            RepoError::NotFound => Error::not_found(shell),
609
            RepoError::Oxy(oxy_core::repo::RepoError::Msg(m)) => {
610
                Error { status: StatusCode::NOT_FOUND, detail: m, shell: Some(Box::new(shell)), redirect: None }
611
            }
612
            other => {
613
                Error { status: StatusCode::INTERNAL_SERVER_ERROR, detail: other.to_string(), shell: Some(Box::new(shell)), redirect: None }
614
            }
615
        }
616
    }
617
618
    pub fn from_db(shell: Shell, e: sqlx::Error) -> Error {
619
        Error { status: StatusCode::INTERNAL_SERVER_ERROR, detail: e.to_string(), shell: Some(Box::new(shell)), redirect: None }
620
    }
621
}
622
623
impl IntoResponse for Error {
624
    fn into_response(self) -> Response {
625
        if let Some(to) = self.redirect {
626
            return Redirect::to(&to).into_response();
627
        }
628
        let title = match self.status {
629
            StatusCode::NOT_FOUND => "Not found",
630
            StatusCode::BAD_REQUEST => "Bad request",
631
            _ => "Something went wrong",
632
        };
633
        if self.status.is_server_error() {
634
            tracing::error!("{}", self.detail);
635
        }
636
        let shell = self.shell.map(|b| *b).unwrap_or(Shell {
637
            site: "OxygenHub".into(),
638
            base_url: String::new(),
639
            path: String::new(),
640
            version: "",
641
            build: "",
642
            user: None,
643
            is_admin: false,
644
            csrf: String::new(),
645
            unread: 0,
646
        });
647
        let page = ErrorPage { shell, status: self.status.as_u16(), title: title.into(), detail: self.detail };
648
        match page.render() {
649
            Ok(s) => (self.status, Html(s)).into_response(),
650
            Err(_) => (self.status, title).into_response(),
651
        }
652
    }
653
}
654
655
async fn not_found(axum::extract::State(app): axum::extract::State<Shared>, viewer: Viewer, uri: axum::http::Uri) -> Response {
656
    Error::not_found(Shell::new(&app, uri.path(), &viewer)).into_response()
657
}
658
659
/// A raw file response with a safe content type (hub spec §15.2).
660
pub fn raw(bytes: Vec<u8>, name: &str, inline_image: bool) -> Response {
661
    let mime = mime_of(name);
662
    let disposition = if inline_image && mime.starts_with("image/") { "inline" } else { "attachment" };
663
    (
664
        [
665
            (header::CONTENT_TYPE, mime.to_string()),
666
            (header::CONTENT_DISPOSITION, format!("{disposition}; filename=\"{}\"", name.replace('"', ""))),
667
            (header::X_CONTENT_TYPE_OPTIONS, "nosniff".to_string()),
668
        ],
669
        bytes,
670
    )
671
        .into_response()
672
}
673
674
pub fn mime_of(name: &str) -> &'static str {
675
    match name.rsplit('.').next().map(|e| e.to_ascii_lowercase()).as_deref() {
676
        Some("png") => "image/png",
677
        Some("jpg" | "jpeg") => "image/jpeg",
678
        Some("gif") => "image/gif",
679
        Some("webp") => "image/webp",
680
        Some("svg") => "image/svg+xml",
681
        Some("pdf") => "application/pdf",
682
        Some(
683
            "txt" | "md" | "rs" | "toml" | "json" | "yml" | "yaml" | "c" | "h" | "cpp" | "hpp" | "py" | "js" | "ts" | "html" | "css" | "sh",
684
        ) => "text/plain; charset=utf-8",
685
        _ => "application/octet-stream",
686
    }
687
}
688
689
/// Resolve `{owner}/{name}` for the viewer with at least `access`, or 404:
690
/// a repository the viewer may not see does not exist for them (hub spec
691
/// §4.4). Returns the viewer's role too.
692
pub async fn lookup_role(
693
    app: &Shared,
694
    shell: &Shell,
695
    viewer: &Viewer,
696
    owner: &str,
697
    name: &str,
698
    access: Access,
699
) -> Result<(crate::repos::RepoName, db::repos::RepoRow, Role), Error> {
700
    let repo = crate::repos::RepoName::new(owner, name).ok_or_else(|| Error::not_found(shell.clone()))?;
701
    let row = match db::repos::get(&app.db, owner, name).await.map_err(|e| Error::from_db(shell.clone(), e))? {
702
        Some(row) => row,
703
        None => {
704
            // A transferred repository: the old URL goes to the new one (#108),
705
            // the rest of the path kept, when the viewer may see it.
706
            let moved = db::repos::moved_to(&app.db, owner, name).await.map_err(|e| Error::from_db(shell.clone(), e))?;
707
            let Some(moved) = moved else { return Err(Error::not_found(shell.clone())) };
708
            let role = auth::roles::role_for(&app.db, viewer.account(), &moved).await.map_err(|e| Error::from_db(shell.clone(), e))?;
709
            if !auth::roles::allows(role, Access::Read) {
710
                return Err(Error::not_found(shell.clone()));
711
            }
712
            let old = format!("/{owner}/{name}");
713
            let rest = shell.path.strip_prefix(&old).unwrap_or("");
714
            return Err(Error::redirect(format!("/{}/{}{rest}", moved.owner, moved.name)));
715
        }
716
    };
717
    let role = auth::roles::role_for(&app.db, viewer.account(), &row).await.map_err(|e| Error::from_db(shell.clone(), e))?;
718
    if !auth::roles::allows(role, access) {
719
        return Err(Error::not_found(shell.clone()));
720
    }
721
    if row.archived && matches!(access, Access::Write | Access::Maintain) {
722
        return Err(archived(shell));
723
    }
724
    Ok((repo, row, role.unwrap_or(Role::Read)))
725
}
726
727
/// The error for a write to an archived repository (#108).
728
pub fn archived(shell: &Shell) -> Error {
729
    Error::bad(shell.clone(), "This repository is archived and read-only. Its admins can bring it back under Settings › Advanced.")
730
}
731
732
/// Refuse a change to an archived repository, for handlers that open it
733
/// for reading and then write.
734
pub fn writable(row: &db::repos::RepoRow, shell: &Shell) -> Result<(), Error> {
735
    if row.archived {
736
        Err(archived(shell))
737
    } else {
738
        Ok(())
739
    }
740
}
741
742
/// `lookup_role` for reading.
743
pub async fn lookup(
744
    app: &Shared,
745
    shell: &Shell,
746
    viewer: &Viewer,
747
    owner: &str,
748
    name: &str,
749
) -> Result<(crate::repos::RepoName, db::repos::RepoRow, Role), Error> {
750
    lookup_role(app, shell, viewer, owner, name, Access::Read).await
751
}
752
753
/// Fill `account` on change authors whose key is linked to an account.
754
pub async fn attribute(app: &Shared, changes: &mut [&mut crate::repos::view::ChangeView]) {
755
    if changes.is_empty() {
756
        return;
757
    }
758
    let Ok(links) = db::accounts::identity_accounts(&app.db).await else { return };
759
    let map: std::collections::HashMap<String, String> = links.into_iter().collect();
760
    for c in changes.iter_mut() {
761
        if let Some(u) = map.get(&c.author.key) {
762
            c.author.account = Some(u.clone());
763
        }
764
    }
765
}