-
Notifications
You must be signed in to change notification settings - Fork 2.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor(cred): Move macos impl to inline mod
- Loading branch information
Showing
1 changed file
with
37 additions
and
35 deletions.
There are no files selected for viewing
72 changes: 37 additions & 35 deletions
72
crates/credential/cargo-credential-macos-keychain/src/main.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,50 +1,52 @@ | ||
//! Cargo registry macos keychain credential process. | ||
|
||
use cargo_credential::{Credential, Error}; | ||
use security_framework::os::macos::keychain::SecKeychain; | ||
mod macos { | ||
use cargo_credential::{Credential, Error}; | ||
use security_framework::os::macos::keychain::SecKeychain; | ||
|
||
struct MacKeychain; | ||
pub(crate) struct MacKeychain; | ||
|
||
/// The account name is not used. | ||
const ACCOUNT: &'static str = ""; | ||
/// The account name is not used. | ||
const ACCOUNT: &'static str = ""; | ||
|
||
fn registry(registry_name: &str) -> String { | ||
format!("cargo-registry:{}", registry_name) | ||
} | ||
|
||
impl Credential for MacKeychain { | ||
fn name(&self) -> &'static str { | ||
env!("CARGO_PKG_NAME") | ||
fn registry(registry_name: &str) -> String { | ||
format!("cargo-registry:{}", registry_name) | ||
} | ||
|
||
fn get(&self, index_url: &str) -> Result<String, Error> { | ||
let keychain = SecKeychain::default().unwrap(); | ||
let service_name = registry(index_url); | ||
let (pass, _item) = keychain.find_generic_password(&service_name, ACCOUNT)?; | ||
String::from_utf8(pass.as_ref().to_vec()) | ||
.map_err(|_| "failed to convert token to UTF8".into()) | ||
} | ||
impl Credential for MacKeychain { | ||
fn name(&self) -> &'static str { | ||
env!("CARGO_PKG_NAME") | ||
} | ||
|
||
fn store(&self, index_url: &str, token: &str, name: Option<&str>) -> Result<(), Error> { | ||
let keychain = SecKeychain::default().unwrap(); | ||
let service_name = registry(name.unwrap_or(index_url)); | ||
if let Ok((_pass, mut item)) = keychain.find_generic_password(&service_name, ACCOUNT) { | ||
item.set_password(token.as_bytes())?; | ||
} else { | ||
keychain.add_generic_password(&service_name, ACCOUNT, token.as_bytes())?; | ||
fn get(&self, index_url: &str) -> Result<String, Error> { | ||
let keychain = SecKeychain::default().unwrap(); | ||
let service_name = registry(index_url); | ||
let (pass, _item) = keychain.find_generic_password(&service_name, ACCOUNT)?; | ||
String::from_utf8(pass.as_ref().to_vec()) | ||
.map_err(|_| "failed to convert token to UTF8".into()) | ||
} | ||
|
||
fn store(&self, index_url: &str, token: &str, name: Option<&str>) -> Result<(), Error> { | ||
let keychain = SecKeychain::default().unwrap(); | ||
let service_name = registry(name.unwrap_or(index_url)); | ||
if let Ok((_pass, mut item)) = keychain.find_generic_password(&service_name, ACCOUNT) { | ||
item.set_password(token.as_bytes())?; | ||
} else { | ||
keychain.add_generic_password(&service_name, ACCOUNT, token.as_bytes())?; | ||
} | ||
Ok(()) | ||
} | ||
Ok(()) | ||
} | ||
|
||
fn erase(&self, index_url: &str) -> Result<(), Error> { | ||
let keychain = SecKeychain::default().unwrap(); | ||
let service_name = registry(index_url); | ||
let (_pass, item) = keychain.find_generic_password(&service_name, ACCOUNT)?; | ||
item.delete(); | ||
Ok(()) | ||
fn erase(&self, index_url: &str) -> Result<(), Error> { | ||
let keychain = SecKeychain::default().unwrap(); | ||
let service_name = registry(index_url); | ||
let (_pass, item) = keychain.find_generic_password(&service_name, ACCOUNT)?; | ||
item.delete(); | ||
Ok(()) | ||
} | ||
} | ||
} | ||
|
||
fn main() { | ||
cargo_credential::main(MacKeychain); | ||
cargo_credential::main(macos::MacKeychain); | ||
} |