zhenyi
dbbfb747a4
- Replace InternalAuthService with TokenService using JWT tokens - Add support for token issuance, refresh, verification and revocation - Implement automatic signing key rotation with Redis storage - Add database migration checks for indexes and foreign key constraints - Update gRPC endpoints to use token-based authentication - Remove deprecated API key based authentication system - Add JSON Web Token support with HMAC-SHA256 signing - Implement refresh token handling with automatic rotation - Add token revocation by JTI and user ID - Update build configuration to include core proto files - Migrate database schema to handle token-based authentication - Add comprehensive token validation and verification logic
86 lines
2.8 KiB
Rust
86 lines
2.8 KiB
Rust
use std::fs;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let manifest_dir = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR")?);
|
|
let out_dir = PathBuf::from(std::env::var("OUT_DIR")?);
|
|
fs::create_dir_all(&out_dir)?;
|
|
let email_dir = manifest_dir.join("proto/email");
|
|
let email_protos = proto_files(&email_dir)?;
|
|
for proto in &email_protos {
|
|
println!("cargo:rerun-if-changed={}", proto.display());
|
|
}
|
|
tonic_prost_build::configure()
|
|
.build_client(true)
|
|
.build_server(false)
|
|
.out_dir(&out_dir)
|
|
.compile_protos(&email_protos, &[email_dir])?;
|
|
|
|
let git_dir = manifest_dir.join("proto/git");
|
|
let git_protos = proto_files(&git_dir)?;
|
|
for proto in &git_protos {
|
|
println!("cargo:rerun-if-changed={}", proto.display());
|
|
}
|
|
tonic_prost_build::configure()
|
|
.build_client(true)
|
|
.build_server(false)
|
|
.type_attribute(
|
|
".",
|
|
"#[derive(serde::Serialize, serde::Deserialize, utoipa::ToSchema)]",
|
|
)
|
|
.extern_path(".google.protobuf.Timestamp", "crate::pb::Timestamp")
|
|
.out_dir(&out_dir)
|
|
.compile_protos(&git_protos, &[git_dir])?;
|
|
|
|
// proto/core/ — JWT token service (server + client: appks serves, imks consumes)
|
|
let core_dir = manifest_dir.join("proto/core");
|
|
let core_protos = proto_files(&core_dir)?;
|
|
for proto in &core_protos {
|
|
println!("cargo:rerun-if-changed={}", proto.display());
|
|
}
|
|
tonic_prost_build::configure()
|
|
.build_client(true)
|
|
.build_server(true)
|
|
.out_dir(&out_dir)
|
|
.compile_protos(&core_protos, &[core_dir])?;
|
|
|
|
let this_dir = manifest_dir.join("proto/this");
|
|
let this_protos = proto_files(&this_dir)?;
|
|
for proto in &this_protos {
|
|
println!("cargo:rerun-if-changed={}", proto.display());
|
|
}
|
|
tonic_prost_build::configure()
|
|
.build_client(false)
|
|
.build_server(true)
|
|
.out_dir(&out_dir)
|
|
.compile_protos(&this_protos, &[this_dir])?;
|
|
|
|
let im_dir = manifest_dir.join("proto/this/im");
|
|
let im_protos = proto_files(&im_dir)?;
|
|
for proto in &im_protos {
|
|
println!("cargo:rerun-if-changed={}", proto.display());
|
|
}
|
|
tonic_prost_build::configure()
|
|
.build_client(false)
|
|
.build_server(true)
|
|
.out_dir(&out_dir)
|
|
.compile_protos(&im_protos, &[im_dir])?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn proto_files(proto_dir: &Path) -> Result<Vec<PathBuf>, Box<dyn std::error::Error>> {
|
|
let mut files = fs::read_dir(proto_dir)?
|
|
.map(|entry| entry.map(|entry| entry.path()))
|
|
.collect::<Result<Vec<_>, _>>()?;
|
|
|
|
files.retain(|path| path.extension().is_some_and(|ext| ext == "proto"));
|
|
files.sort();
|
|
|
|
if files.is_empty() {
|
|
return Err(format!("no .proto files found in {}", proto_dir.display()).into());
|
|
}
|
|
|
|
Ok(files)
|
|
}
|