mirror of
https://github.com/dani-garcia/vaultwarden.git
synced 2026-05-08 04:23:59 +02:00
Aurora DSQL speaks a PostgreSQL-compatible wire protocol, but it uses IAM authentication and has DDL limits that differ from normal PostgreSQL. Serverless deployments need to select it with a dsql:// DATABASE_URL while the Diesel layer keeps using the PostgreSQL backend. Add a dsql feature, classify dsql:// URLs as PostgreSQL, and convert them to libpq URLs with cached IAM auth tokens. Route DSQL databases through a dedicated migration set instead of the normal PostgreSQL migrations. Keep the deployed DSQL migration versions and add DSQL migrations for upstream schema changes needed by this rebase. Split newer DSQL migrations into single-DDL, transactionless steps so they can run on Aurora DSQL.
100 lines
3.9 KiB
Rust
100 lines
3.9 KiB
Rust
use std::env;
|
|
use std::process::Command;
|
|
|
|
fn main() {
|
|
// These allow using e.g. #[cfg(mysql)] instead of #[cfg(feature = "mysql")], which helps when trying to add them through macros
|
|
#[cfg(feature = "sqlite_system")] // The `sqlite` feature implies this one.
|
|
println!("cargo:rustc-cfg=sqlite");
|
|
#[cfg(feature = "mysql")]
|
|
println!("cargo:rustc-cfg=mysql");
|
|
#[cfg(feature = "postgresql")]
|
|
println!("cargo:rustc-cfg=postgresql");
|
|
#[cfg(feature = "dsql")]
|
|
println!("cargo:rustc-cfg=dsql");
|
|
#[cfg(not(any(feature = "sqlite_system", feature = "mysql", feature = "postgresql", feature = "dsql")))]
|
|
compile_error!(
|
|
"You need to enable one DB backend. To build with previous defaults do: cargo build --features sqlite"
|
|
);
|
|
|
|
#[cfg(feature = "s3")]
|
|
println!("cargo:rustc-cfg=s3");
|
|
#[cfg(feature = "ses")]
|
|
println!("cargo:rustc-cfg=ses");
|
|
#[cfg(feature = "aws")]
|
|
println!("cargo:rustc-cfg=aws");
|
|
|
|
// Use check-cfg to let cargo know which cfg's we define,
|
|
// and avoid warnings when they are used in the code.
|
|
println!("cargo::rustc-check-cfg=cfg(sqlite)");
|
|
println!("cargo::rustc-check-cfg=cfg(mysql)");
|
|
println!("cargo::rustc-check-cfg=cfg(postgresql)");
|
|
println!("cargo::rustc-check-cfg=cfg(dsql)");
|
|
println!("cargo::rustc-check-cfg=cfg(s3)");
|
|
println!("cargo::rustc-check-cfg=cfg(ses)");
|
|
println!("cargo::rustc-check-cfg=cfg(aws)");
|
|
|
|
// Rerun when these paths are changed.
|
|
// Someone could have checked-out a tag or specific commit, but no other files changed.
|
|
println!("cargo:rerun-if-changed=.git");
|
|
println!("cargo:rerun-if-changed=.git/HEAD");
|
|
println!("cargo:rerun-if-changed=.git/index");
|
|
println!("cargo:rerun-if-changed=.git/refs/tags");
|
|
|
|
// Support $BWRS_VERSION for legacy compatibility, but default to $VW_VERSION.
|
|
// If neither exist, read from git.
|
|
let maybe_vaultwarden_version =
|
|
env::var("VW_VERSION").or_else(|_| env::var("BWRS_VERSION")).or_else(|_| version_from_git_info());
|
|
|
|
if let Ok(version) = maybe_vaultwarden_version {
|
|
println!("cargo:rustc-env=VW_VERSION={version}");
|
|
println!("cargo:rustc-env=CARGO_PKG_VERSION={version}");
|
|
}
|
|
}
|
|
|
|
fn run(args: &[&str]) -> Result<String, std::io::Error> {
|
|
let out = Command::new(args[0]).args(&args[1..]).output()?;
|
|
if !out.status.success() {
|
|
use std::io::Error;
|
|
return Err(Error::other("Command not successful"));
|
|
}
|
|
Ok(String::from_utf8(out.stdout).unwrap().trim().to_string())
|
|
}
|
|
|
|
/// This method reads info from Git, namely tags, branch, and revision
|
|
/// To access these values, use:
|
|
/// - `env!("GIT_EXACT_TAG")`
|
|
/// - `env!("GIT_LAST_TAG")`
|
|
/// - `env!("GIT_BRANCH")`
|
|
/// - `env!("GIT_REV")`
|
|
/// - `env!("VW_VERSION")`
|
|
fn version_from_git_info() -> Result<String, std::io::Error> {
|
|
// The exact tag for the current commit, can be empty when
|
|
// the current commit doesn't have an associated tag
|
|
let exact_tag = run(&["git", "describe", "--abbrev=0", "--tags", "--exact-match"]).ok();
|
|
if let Some(ref exact) = exact_tag {
|
|
println!("cargo:rustc-env=GIT_EXACT_TAG={exact}");
|
|
}
|
|
|
|
// The last available tag, equal to exact_tag when
|
|
// the current commit is tagged
|
|
let last_tag = run(&["git", "describe", "--abbrev=0", "--tags"])?;
|
|
println!("cargo:rustc-env=GIT_LAST_TAG={last_tag}");
|
|
|
|
// The current branch name
|
|
let branch = run(&["git", "rev-parse", "--abbrev-ref", "HEAD"])?;
|
|
println!("cargo:rustc-env=GIT_BRANCH={branch}");
|
|
|
|
// The current git commit hash
|
|
let rev = run(&["git", "rev-parse", "HEAD"])?;
|
|
let rev_short = rev.get(..8).unwrap_or_default();
|
|
println!("cargo:rustc-env=GIT_REV={rev_short}");
|
|
|
|
// Combined version
|
|
if let Some(exact) = exact_tag {
|
|
Ok(exact)
|
|
} else if &branch != "main" && &branch != "master" && &branch != "HEAD" {
|
|
Ok(format!("{last_tag}-{rev_short} ({branch})"))
|
|
} else {
|
|
Ok(format!("{last_tag}-{rev_short}"))
|
|
}
|
|
}
|