DragonOS-Community
/
aya
réplica de https://github.com/aya-rs/aya


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
							use anyhow::{anyhow, Context as _, Result};
use cargo_metadata::{Metadata, MetadataCommand};
use indoc::{indoc, writedoc};
use std::{ffi::OsString, fs, io::Write as _, process::Command};

pub fn exec(cmd: &mut Command) -> Result<()> {
    let status = cmd
        .status()
        .with_context(|| format!("failed to run {cmd:?}"))?;
    match status.code() {
        Some(code) => match code {
            0 => Ok(()),
            code => Err(anyhow!("{cmd:?} exited with code {code}")),
        },
        None => Err(anyhow!("{cmd:?} terminated by signal")),
    }
}

pub fn docs() -> Result<()> {
    const PACKAGE_TO_DESCRIPTION: &[(&str, &str)] = &[
        ("aya", "User-space BPF program loading and manipulation"),
        ("aya-bpf", "Kernel-space BPF program implementation toolkit"),
        ("aya-log-ebpf", "Kernel-space logging from BPF programs"),
        (
            "aya-log",
            "User-space consumption of logs from BPF programs",
        ),
    ];

    let Metadata {
        workspace_root,
        target_directory,
        ..
    } = MetadataCommand::new().exec().context("cargo metadata")?;

    exec(
        Command::new("cargo")
            .current_dir(&workspace_root)
            .args(["clean", "--doc"]),
    )?;

    let tmp = tempfile::tempdir().context("create tempdir")?;
    let header = tmp.path().join("header.html");
    fs::write(&header, r#"<meta name="robots" content="noindex">"#).context("write header.html")?;

    let mut rustdocflags = OsString::new();
    rustdocflags.push("--cfg docsrs --html-in-header ");
    rustdocflags.push(header);
    rustdocflags.push(" -D warnings");

    exec(
        Command::new("cargo")
            .current_dir(&workspace_root)
            .env("RUSTDOCFLAGS", rustdocflags)
            .args(["+nightly", "doc", "--no-deps", "--all-features"])
            .args(
                PACKAGE_TO_DESCRIPTION
                    .iter()
                    .flat_map(|(package, _)| ["--package", package]),
            ),
    )?;

    let site = workspace_root.join("site");
    match fs::remove_dir_all(&site) {
        Ok(()) => {}
        Err(err) => {
            if err.kind() != std::io::ErrorKind::NotFound {
                return Err(err).context(format!("remove {site:?}"));
            }
        }
    }
    let doc = target_directory.join("doc");
    fs::rename(&doc, &site).with_context(|| format!("rename {doc:?} to {site:?}"))?;

    exec(Command::new("sh").current_dir(&site).args([
        "-c",
        "grep -FRl crabby.svg | xargs sed -i s/crabby.svg/crabby_dev.svg/g",
    ]))?;

    fs::write(
        site.join("robots.txt"),
        indoc! {r#"
    User-Agent:*
    Disallow: /
    "#},
    )
    .context("can't write robots.txt")?;

    let mut index = fs::File::create(site.join("index.html"))
        .with_context(|| format!("create {site:?}/index.html"))?;
    writedoc! {&mut index, r#"
        <html>
            <meta name="robots" content="noindex">
            <body>
              <ul>
    "#}
    .context("write to index.html")?;

    for (package, description) in PACKAGE_TO_DESCRIPTION {
        let package = package.replace('-', "_");
        writedoc! {&mut index, r#"
            <li><a href="{package}/index.html">Aya {description}</a></li>
        "#}
        .context("write to string")?;
    }

    writedoc! {&mut index, r#"
              </ul>
            </body>
        </html>
    "#}
    .context("write to index.html")?;

    Ok(())
}