Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

7 changes: 7 additions & 0 deletions apps/labrinth/migrations/20260922120000_project_redirects.sql
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
CREATE TABLE project_redirects (
slug TEXT PRIMARY KEY CHECK (slug = LOWER(slug)),
target_project_id BIGINT NOT NULL REFERENCES mods(id) ON DELETE CASCADE
);

CREATE INDEX project_redirects_target_project_id
ON project_redirects(target_project_id);
104 changes: 104 additions & 0 deletions apps/labrinth/src/routes/maven.rs
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,86 @@ pub fn config(cfg: &mut actix_web::web::ServiceConfig) {
cfg.service(version_file);
}

async fn redirect_maven_ref(
req: &HttpRequest,
pool: &PgPool,
redis: &RedisPool,
) -> Result<Option<HttpResponse>, ApiError> {
let Some(project_ref) = req.match_info().get("id") else {
return Ok(None);
};
let Some(target_project_id) =
crate::routes::resolve_ref(project_ref, pool, redis).await?
else {
return Ok(None);
};

let Some(route_pattern) = req.match_pattern() else {
return Ok(None);
};
let route_segments = route_pattern.split('/').collect::<Vec<_>>();
let Some(project_ref_index) =
route_segments.iter().position(|segment| *segment == "{id}")
else {
return Ok(None);
};
let mut path_segments = req
.uri()
.path()
.split('/')
.map(ToOwned::to_owned)
.collect::<Vec<_>>();
let Some(project_ref_segment) = path_segments.get_mut(project_ref_index)
else {
return Ok(None);
};
let target_project_ref = target_project_id.to_string();
*project_ref_segment = target_project_ref.clone();

if let (Some(version), Some(filename)) = (
req.match_info().get("versionnum"),
req.match_info().get("file"),
) {
let source_prefix = format!("{project_ref}-{version}");
if filename
.get(..source_prefix.len())
.is_some_and(|prefix| prefix.eq_ignore_ascii_case(&source_prefix))
{
let target_prefix = format!("{target_project_ref}-{version}");
let rewritten_filename =
format!("{target_prefix}{}", &filename[source_prefix.len()..]);
if let Some(filename_index) = route_segments
.iter()
.position(|segment| segment.starts_with("{file}"))
{
let route_suffix = route_segments[filename_index]
.strip_prefix("{file}")
.unwrap_or_default();
if let Some(filename_segment) =
path_segments.get_mut(filename_index)
{
*filename_segment = format!(
"{}{route_suffix}",
urlencoding::encode(&rewritten_filename)
);
}
}
}
}

let mut location = path_segments.join("/");
if let Some(query) = req.uri().query() {
location.push('?');
location.push_str(query);
}

Ok(Some(
HttpResponse::PermanentRedirect()
.append_header((actix_web::http::header::LOCATION, location))
.finish(),
))
}

#[derive(Default, Debug, Clone, YaSerialize)]
#[yaserde(rename = "metadata")]
pub struct Metadata {
Expand Down Expand Up @@ -85,6 +165,12 @@ pub async fn maven_metadata(
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
if let Some(response) =
redirect_maven_ref(&req, pool.as_ref(), redis.as_ref()).await?
{
return Ok(response);
}

let project_id = params.into_inner().0;
let Some(project) =
database::models::DBProject::get(&project_id, &**pool, &redis)
Expand Down Expand Up @@ -323,6 +409,12 @@ pub async fn version_file(
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
if let Some(response) =
redirect_maven_ref(&req, pool.as_ref(), redis.as_ref()).await?
{
return Ok(response);
}

let (project_id, vnum, file) = params.into_inner();
let Some(project) =
database::models::DBProject::get(&project_id, &**pool, &redis)
Expand Down Expand Up @@ -410,6 +502,12 @@ pub async fn version_file_sha1(
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
if let Some(response) =
redirect_maven_ref(&req, pool.as_ref(), redis.as_ref()).await?
{
return Ok(response);
}

let (project_id, vnum, file) = params.into_inner();
let Some(project) =
database::models::DBProject::get(&project_id, &**pool, &redis)
Expand Down Expand Up @@ -476,6 +574,12 @@ pub async fn version_file_sha512(
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
if let Some(response) =
redirect_maven_ref(&req, pool.as_ref(), redis.as_ref()).await?
{
return Ok(response);
}

let (project_id, vnum, file) = params.into_inner();
let Some(project) =
database::models::DBProject::get(&project_id, &**pool, &redis)
Expand Down
Loading
Loading