Only put platforms list is less than 6, otherwise load with AJAX

GuillaumeGomez · GuillaumeGomez · commit c717cec42708 · 2023-09-07T11:32:50.000+02:00
diff --git a/src/web/crate_details.rs b/src/web/crate_details.rs
@@ -1,5 +1,7 @@
 use super::{markdown, match_version, MatchSemver, MetaData};
 use crate::utils::{get_correct_docsrs_style_file, report_error, spawn_blocking};
+use crate::web::rustdoc::RustdocHtmlParams;
+use crate::web::{axum_cached_redirect, match_version_axum};
 use crate::{
     db::Pool,
     impl_axum_webpage,
@@ -15,6 +17,7 @@ use crate::{
 use anyhow::{Context, Result};
 use axum::{
     extract::{Extension, Path},
+    http::Uri,
     response::{IntoResponse, Response as AxumResponse},
 };
 use chrono::{DateTime, Utc};
@@ -24,6 +27,7 @@ use serde::Deserialize;
 use serde::{ser::Serializer, Serialize};
 use serde_json::Value;
 use std::sync::Arc;
+use tracing::{instrument, trace};
 
 // TODO: Add target name and versions
 
@@ -475,6 +479,167 @@ pub(crate) async fn get_all_releases(
     Ok(res.into_response())
 }
 
+#[derive(Debug, Clone, PartialEq, Serialize)]
+struct ShortMetadata {
+    name: String,
+    version_or_latest: String,
+    doc_targets: Vec<String>,
+}
+
+#[derive(Debug, Clone, PartialEq, Serialize)]
+struct PlatformList {
+    metadata: ShortMetadata,
+    inner_path: String,
+    use_direct_platform_links: bool,
+    current_target: String,
+}
+
+impl_axum_webpage! {
+    PlatformList = "rustdoc/platforms.html",
+    cpu_intensive_rendering = true,
+}
+
+#[tracing::instrument]
+pub(crate) async fn get_all_platforms(
+    Path(params): Path<RustdocHtmlParams>,
+    Extension(pool): Extension<Pool>,
+    uri: Uri,
+) -> AxumResult<AxumResponse> {
+    // since we directly use the Uri-path and not the extracted params from the router,
+    // we have to percent-decode the string here.
+    let original_path = percent_encoding::percent_decode(uri.path().as_bytes())
+        .decode_utf8()
+        .map_err(|_| AxumNope::BadRequest)?;
+    let mut req_path: Vec<&str> = original_path.split('/').collect();
+
+    let release_found = match_version_axum(&pool, &params.name, Some(&params.version)).await?;
+    trace!(?release_found, "found release");
+
+    // Remove the empty start, "releases", the name and the version from the path
+    req_path.drain(..4).for_each(drop);
+
+    // Convenience function to allow for easy redirection
+    #[instrument]
+    fn redirect(
+        name: &str,
+        vers: &str,
+        path: &[&str],
+        cache_policy: CachePolicy,
+    ) -> AxumResult<AxumResponse> {
+        trace!("redirect");
+        // Format and parse the redirect url
+        Ok(axum_cached_redirect(
+            encode_url_path(&format!("/platforms/{}/{}/{}", name, vers, path.join("/"))),
+            cache_policy,
+        )?
+        .into_response())
+    }
+
+    let (version, version_or_latest) = match release_found.version {
+        MatchSemver::Exact((version, _)) => {
+            // Redirect when the requested crate name isn't correct
+            if let Some(name) = release_found.corrected_name {
+                return redirect(&name, &version, &req_path, CachePolicy::NoCaching);
+            }
+
+            (version.clone(), version)
+        }
+
+        MatchSemver::Latest((version, _)) => {
+            // Redirect when the requested crate name isn't correct
+            if let Some(name) = release_found.corrected_name {
+                return redirect(&name, "latest", &req_path, CachePolicy::NoCaching);
+            }
+
+            (version, "latest".to_string())
+        }
+
+        // Redirect when the requested version isn't correct
+        MatchSemver::Semver((v, _)) => {
+            // to prevent cloudfront caching the wrong artifacts on URLs with loose semver
+            // versions, redirect the browser to the returned version instead of loading it
+            // immediately
+            return redirect(&params.name, &v, &req_path, CachePolicy::ForeverInCdn);
+        }
+    };
+
+    let (name, doc_targets, releases, default_target): (String, Vec<String>, Vec<Release>, String) =
+        spawn_blocking({
+            let pool = pool.clone();
+            move || {
+                let mut conn = pool.get()?;
+                let query = "
+            SELECT
+                crates.id,
+                crates.name,
+                releases.default_target,
+                releases.doc_targets
+            FROM releases
+            INNER JOIN crates ON releases.crate_id = crates.id
+            WHERE crates.name = $1 AND releases.version = $2;";
+
+                let rows = conn.query(query, &[&params.name, &version])?;
+
+                let krate = if rows.is_empty() {
+                    return Err(AxumNope::CrateNotFound.into());
+                } else {
+                    &rows[0]
+                };
+
+                // get releases, sorted by semver
+                let releases = releases_for_crate(&mut *conn, krate.get("id"))?;
+
+                Ok((
+                    krate.get("name"),
+                    MetaData::parse_doc_targets(krate.get("doc_targets")),
+                    releases,
+                    krate.get("default_target"),
+                ))
+            }
+        })
+        .await?;
+
+    let latest_release = releases
+        .iter()
+        .find(|release| release.version.pre.is_empty() && !release.yanked)
+        .unwrap_or(&releases[0]);
+
+    // The path within this crate version's rustdoc output
+    let (target, inner_path) = {
+        let mut inner_path = req_path.clone();
+
+        let target = if inner_path.len() > 1 && doc_targets.iter().any(|s| s == inner_path[0]) {
+            inner_path.remove(0)
+        } else {
+            ""
+        };
+
+        (target, inner_path.join("/"))
+    };
+
+    let current_target = if latest_release.build_status {
+        if target.is_empty() {
+            default_target
+        } else {
+            target.to_owned()
+        }
+    } else {
+        String::new()
+    };
+
+    let res = PlatformList {
+        metadata: ShortMetadata {
+            name,
+            version_or_latest: version_or_latest.to_string(),
+            doc_targets,
+        },
+        inner_path,
+        use_direct_platform_links: true,
+        current_target,
+    };
+    Ok(res.into_response())
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
diff --git a/src/web/routes.rs b/src/web/routes.rs
@@ -181,7 +181,15 @@ pub(super) fn build_axum_routes() -> AxumRouter {
             get_internal(super::crate_details::crate_details_handler),
         )
         .route(
-            "/:name/releases",
+            "/platforms/:name/:version/:target/",
+            get_internal(super::crate_details::get_all_platforms),
+        )
+        .route(
+            "/platforms/:name/:version/:target/*path",
+            get_internal(super::crate_details::get_all_platforms),
+        )
+        .route(
+            "/releases/list/:name",
             get_internal(super::crate_details::get_all_releases),
         )
         .route_with_tsr(
diff --git a/src/web/rustdoc.rs b/src/web/rustdoc.rs
@@ -349,17 +349,17 @@ impl RustdocPage {
     }
 }
 
-#[derive(Clone, Deserialize)]
+#[derive(Clone, Deserialize, Debug)]
 pub(crate) struct RustdocHtmlParams {
-    name: String,
-    version: String,
+    pub(crate) name: String,
+    pub(crate) version: String,
     // both target and path are only used for matching the route.
     // The actual path is read from the request `Uri` because
     // we have some static filenames directly in the routes.
     #[allow(dead_code)]
-    target: Option<String>,
+    pub(crate) target: Option<String>,
     #[allow(dead_code)]
-    path: Option<String>,
+    pub(crate) path: Option<String>,
 }
 
 /// Serves documentation generated by rustdoc.
diff --git a/static/menu.js b/static/menu.js
@@ -5,13 +5,17 @@ const updateMenuPositionForSubMenu = (currentMenuSupplier) => {
     subMenu?.style.setProperty('--menu-x', `${currentMenu.getBoundingClientRect().x}px`);
 }
 
-function generateReleaseList(data, crateName) {
-}
+const loadedMenus = new Set();
 
-let loadReleases = function() {
-    const releaseListElem = document.getElementById('releases-list');
-    // To prevent reloading the list unnecessarily.
-    loadReleases = function() {};
+function loadReleases(menu, id, msg, path, extra) {
+    if (loadedMenus.has(id)) {
+        return;
+    }
+    loadedMenus.add(id);
+    if (!menu.querySelector(".rotate")) {
+        return;
+    }
+    const releaseListElem = document.getElementById(id);
     if (!releaseListElem) {
         // We're not in a documentation page, so no need to do anything.
         return;
@@ -25,11 +29,11 @@ let loadReleases = function() {
         if (xhttp.status === 200) {
             releaseListElem.innerHTML = xhttp.responseText;
         } else {
-            console.error(`Failed to load release list: [${xhttp.status}] ${xhttp.responseText}`);
-            document.getElementById('releases-list').innerHTML = "Failed to load release list";
+            console.error(`Failed to load ${msg}: [${xhttp.status}] ${xhttp.responseText}`);
+            document.getElementById(id).innerHTML = `Failed to load ${msg}`;
         }
     };
-    xhttp.open("GET", `/${crateName}/releases`, true);
+    xhttp.open("GET", `/${path}/${crateName}${extra}`, true);
     xhttp.send();
 };
 
@@ -81,7 +85,18 @@ let loadReleases = function() {
         currentMenu = newMenu;
         newMenu.className += " pure-menu-active";
         backdrop.style.display = "block";
-        loadReleases();
+        if (newMenu.querySelector("#releases-list")) {
+            loadReleases(newMenu, "releases-list", "release list", "releases/list", "");
+        } else if (newMenu.querySelector("#platforms")) {
+            loadReleases(
+                newMenu,
+                "platforms",
+                "platforms list",
+                "platforms",
+                // We get everything except the first crate name.
+                "/" + window.location.pathname.split("/").slice(2).join("/")
+            );
+        }
     }
     function menuOnClick(e) {
         if (this.getAttribute("href") != "#") {
diff --git a/templates/rustdoc/platforms.html b/templates/rustdoc/platforms.html
@@ -0,0 +1,25 @@
+{%- for target in metadata.doc_targets -%}
+    {#
+        The crate-detail page is the only page where we want to allow google to follow
+        the target-links. On that page we also don't have to use `/target-redirect/`
+        because the documentation root page is guaranteed to exist for all targets.
+    #}
+    {%- if use_direct_platform_links -%}
+        {%- set target_url = "/" ~ metadata.name ~ "/" ~ metadata.version_or_latest ~ "/" ~ target ~ "/" ~ inner_path -%}
+        {%- set target_no_follow = "" -%}
+    {%- else -%}
+        {%- set target_url = "/crate/" ~ metadata.name ~ "/" ~ metadata.version_or_latest ~ "/target-redirect/" ~ target ~ "/" ~ inner_path -%}
+        {%- set target_no_follow = "nofollow" -%}
+    {%- endif -%}
+    {%- if current_target is defined and current_target == target -%}
+        {%- set current = " current" -%}
+    {%- else -%}
+        {%- set current = "" -%}
+    {%- endif -%}
+
+    <li class="pure-menu-item">
+        <a href="{{ target_url | safe }}" class="pure-menu-link{{ current | safe }}" data-fragment="retain" rel="{{ target_no_follow }}">
+            {{- target -}}
+        </a>
+    </li>
+{%- endfor -%}
diff --git a/templates/rustdoc/topbar.html b/templates/rustdoc/topbar.html
@@ -209,31 +209,11 @@
 
         {# Build the dropdown list showing available targets #}
         <ul class="pure-menu-children" id="platforms">
-            {%- for target in metadata.doc_targets -%}
-                {#
-                    The crate-detail page is the only page where we want to allow google to follow
-                    the target-links. On that page we also don't have to use `/target-redirect/`
-                    because the documentation root page is guaranteed to exist for all targets.
-                #}
-                {%- if use_direct_platform_links -%}
-                    {%- set target_url = "/" ~ metadata.name ~ "/" ~ metadata.version_or_latest ~ "/" ~ target ~ "/" ~ inner_path -%}
-                    {%- set target_no_follow = "" -%}
-                {%- else -%}
-                    {%- set target_url = "/crate/" ~ metadata.name ~ "/" ~ metadata.version_or_latest ~ "/target-redirect/" ~ target ~ "/" ~ inner_path -%}
-                    {%- set target_no_follow = "nofollow" -%}
-                {%- endif -%}
-                {%- if current_target is defined and current_target == target -%}
-                    {%- set current = " current" -%}
-                {%- else -%}
-                    {%- set current = "" -%}
-                {%- endif -%}
-
-                <li class="pure-menu-item">
-                    <a href="{{ target_url | safe }}" class="pure-menu-link{{ current | safe }}" data-fragment="retain" rel="{{ target_no_follow }}">
-                        {{- target -}}
-                    </a>
-                </li>
-            {%- endfor -%}
+            {%- if metadata.doc_targets|length < 6 -%}
+                {%- include "rustdoc/targets.tml" -%}
+            {%- else -%}
+                <span class="rotate">{{ "spinner" | fas }}</span>
+            {%- endif -%}
         </ul>
     </li>{#
     Display the features available in current build
