feat(auth): add credential mode (credentials/token) support
Introduces a credential mode for stored credentials and wire it to per-remote URL resolution and operation flows. The UI, storage, and remote interactions now track and persist the mode, enabling token based auth alongside username/password credentials. - Remote URL resolution now considers direction (pull/push) and mode - Credential dialog, saving, and keychain handling updated to pass and respect the mode - Unique askpass scripts generated per invocation to avoid clashes
This commit is contained in:
Notes:
Christoph Brandau
2026-08-13 22:48:24 +02:00
git note test
+134
-9
@@ -2432,6 +2432,8 @@ const CRED_SERVICE: &str = "tauri_git_lite";
|
||||
pub struct StoredCredential {
|
||||
pub username: String,
|
||||
pub password: String,
|
||||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||||
pub mode: Option<String>,
|
||||
}
|
||||
|
||||
fn cred_entry(key: &str) -> Result<keyring::Entry, String> {
|
||||
@@ -2445,17 +2447,27 @@ fn cred_entry(key: &str) -> Result<keyring::Entry, String> {
|
||||
/// Returns the remote URL used for auth key derivation (upstream remote of the
|
||||
/// current branch, falling back to `origin`, then the first configured remote).
|
||||
#[tauri::command(async)]
|
||||
pub fn get_remote_url(path: String) -> Result<Option<String>, String> {
|
||||
pub fn get_remote_url(
|
||||
path: String,
|
||||
remote: Option<String>,
|
||||
push: Option<bool>,
|
||||
) -> Result<Option<String>, String> {
|
||||
let repo = resolve_repo(&path)?;
|
||||
let remote = upstream_remote_name(&repo).unwrap_or_else(|| "origin".to_string());
|
||||
let remote = match remote
|
||||
.map(|value| value.trim().to_string())
|
||||
.filter(|value| !value.is_empty())
|
||||
{
|
||||
Some(remote) => validate_remote_name(&repo, &remote, true)?,
|
||||
None => upstream_remote_name(&repo).unwrap_or_else(|| "origin".to_string()),
|
||||
};
|
||||
|
||||
if let Some(url) = remote_url_for(&repo, &remote) {
|
||||
if let Some(url) = remote_url_for_auth(&repo, &remote, push.unwrap_or(false)) {
|
||||
return Ok(Some(url));
|
||||
}
|
||||
// origin missing → try the first configured remote
|
||||
if let Some(first) = first_remote_name(&repo) {
|
||||
if first != remote {
|
||||
if let Some(url) = remote_url_for(&repo, &first) {
|
||||
if let Some(url) = remote_url_for_auth(&repo, &first, push.unwrap_or(false)) {
|
||||
return Ok(Some(url));
|
||||
}
|
||||
}
|
||||
@@ -2463,6 +2475,20 @@ pub fn get_remote_url(path: String) -> Result<Option<String>, String> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
fn remote_url_for_auth(repo: &Path, remote: &str, push: bool) -> Option<String> {
|
||||
let mut command = git_command();
|
||||
command.arg("-C").arg(repo).args(["remote", "get-url"]);
|
||||
if push {
|
||||
command.arg("--push");
|
||||
}
|
||||
let out = command.arg(remote).output().ok()?;
|
||||
if !out.status.success() {
|
||||
return None;
|
||||
}
|
||||
let url = String::from_utf8_lossy(&out.stdout).trim().to_string();
|
||||
if url.is_empty() { None } else { Some(url) }
|
||||
}
|
||||
|
||||
fn remote_url_for(repo: &Path, remote: &str) -> Option<String> {
|
||||
let out = git_command()
|
||||
.arg("-C")
|
||||
@@ -2614,9 +2640,22 @@ pub fn cred_load(key: String) -> Result<Option<StoredCredential>, String> {
|
||||
}
|
||||
|
||||
#[tauri::command(async)]
|
||||
pub fn cred_save(key: String, username: String, password: String) -> Result<(), String> {
|
||||
pub fn cred_save(
|
||||
key: String,
|
||||
username: String,
|
||||
password: String,
|
||||
mode: Option<String>,
|
||||
) -> Result<(), String> {
|
||||
let entry = cred_entry(&key)?;
|
||||
let cred = StoredCredential { username, password };
|
||||
let mode = match mode.as_deref() {
|
||||
Some("token") => Some("token".to_string()),
|
||||
_ => Some("credentials".to_string()),
|
||||
};
|
||||
let cred = StoredCredential {
|
||||
username,
|
||||
password,
|
||||
mode,
|
||||
};
|
||||
let json = serde_json::to_string(&cred)
|
||||
.map_err(|err| format!("Could not serialize credentials: {err}"))?;
|
||||
entry
|
||||
@@ -4688,6 +4727,13 @@ fn run_git_clone(
|
||||
password: Option<&str>,
|
||||
) -> Result<(), String> {
|
||||
let mut command = git_command();
|
||||
let has_explicit_credentials = matches!(
|
||||
(username, password),
|
||||
(Some(user), Some(pass)) if !user.is_empty() || !pass.is_empty()
|
||||
);
|
||||
if has_explicit_credentials {
|
||||
command.arg("-c").arg("credential.helper=");
|
||||
}
|
||||
command
|
||||
.arg("clone")
|
||||
.arg("--")
|
||||
@@ -5649,10 +5695,20 @@ fn run_apply_patch_command(
|
||||
run_git(repo, args).map(|_| ())
|
||||
}
|
||||
|
||||
static ASKPASS_COUNTER: AtomicU64 = AtomicU64::new(1);
|
||||
|
||||
fn next_askpass_path(extension: &str) -> std::path::PathBuf {
|
||||
let sequence = ASKPASS_COUNTER.fetch_add(1, Ordering::Relaxed);
|
||||
std::env::temp_dir().join(format!(
|
||||
"gitty-askpass-{}-{sequence}.{extension}",
|
||||
std::process::id()
|
||||
))
|
||||
}
|
||||
|
||||
#[cfg(unix)]
|
||||
fn write_askpass_script() -> Result<std::path::PathBuf, String> {
|
||||
use std::os::unix::fs::PermissionsExt;
|
||||
let path = std::env::temp_dir().join("gitlite_askpass.sh");
|
||||
let path = next_askpass_path("sh");
|
||||
let script = "#!/bin/sh\ncase \"$1\" in\n *[Uu]sername*) printf '%s\\n' \"$GIT_CRED_USER\" ;;\n *) printf '%s\\n' \"$GIT_CRED_PASS\" ;;\nesac\n";
|
||||
std::fs::write(&path, script)
|
||||
.map_err(|e| format!("Could not write authentication script: {e}"))?;
|
||||
@@ -5663,8 +5719,17 @@ fn write_askpass_script() -> Result<std::path::PathBuf, String> {
|
||||
|
||||
#[cfg(not(unix))]
|
||||
fn write_askpass_script() -> Result<std::path::PathBuf, String> {
|
||||
let path = std::env::temp_dir().join("gitlite_askpass.bat");
|
||||
let script = "@echo off\necho %1 | findstr /I \"sername\" >nul 2>&1\nif %errorlevel% == 0 (echo %GIT_CRED_USER%) else (echo %GIT_CRED_PASS%)\n";
|
||||
let path = next_askpass_path("bat");
|
||||
// Reading the value from PowerShell avoids cmd.exe interpreting special
|
||||
// characters such as &, |, ^ or % from passwords and access tokens.
|
||||
let script = r#"@echo off
|
||||
echo %1 | findstr /I "sername" >nul 2>&1
|
||||
if %errorlevel% == 0 (
|
||||
powershell.exe -NoProfile -NonInteractive -Command "[Console]::Out.WriteLine($env:GIT_CRED_USER)"
|
||||
) else (
|
||||
powershell.exe -NoProfile -NonInteractive -Command "[Console]::Out.WriteLine($env:GIT_CRED_PASS)"
|
||||
)
|
||||
"#;
|
||||
std::fs::write(&path, script)
|
||||
.map_err(|e| format!("Could not write authentication script: {e}"))?;
|
||||
Ok(path)
|
||||
@@ -5706,6 +5771,8 @@ where
|
||||
let askpass = write_askpass_script()?;
|
||||
|
||||
let result = git_command()
|
||||
.arg("-c")
|
||||
.arg("credential.helper=")
|
||||
.arg("-C")
|
||||
.arg(repo)
|
||||
.args(args)
|
||||
@@ -7146,6 +7213,64 @@ mod tests {
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn credential_payload_remains_backward_compatible() {
|
||||
let legacy: StoredCredential =
|
||||
serde_json::from_str(r#"{"username":"alice","password":"secret"}"#)
|
||||
.expect("legacy credential should deserialize");
|
||||
assert_eq!(legacy.username, "alice");
|
||||
assert_eq!(legacy.password, "secret");
|
||||
assert_eq!(legacy.mode, None);
|
||||
|
||||
let token = StoredCredential {
|
||||
username: "alice".to_string(),
|
||||
password: "token".to_string(),
|
||||
mode: Some("token".to_string()),
|
||||
};
|
||||
let encoded = serde_json::to_string(&token).expect("credential should serialize");
|
||||
assert!(encoded.contains(r#""mode":"token""#));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn auth_remote_url_uses_the_requested_direction() {
|
||||
let repo = init_temp_repo("auth_remote_url_direction");
|
||||
run_git_test(
|
||||
&repo.path,
|
||||
[
|
||||
"remote",
|
||||
"add",
|
||||
"origin",
|
||||
"https://gitea.example/fetch/repo.git",
|
||||
],
|
||||
);
|
||||
run_git_test(
|
||||
&repo.path,
|
||||
[
|
||||
"remote",
|
||||
"set-url",
|
||||
"--push",
|
||||
"origin",
|
||||
"https://gitea.example/push/repo.git",
|
||||
],
|
||||
);
|
||||
|
||||
assert_eq!(
|
||||
remote_url_for_auth(&repo.path, "origin", false).as_deref(),
|
||||
Some("https://gitea.example/fetch/repo.git")
|
||||
);
|
||||
assert_eq!(
|
||||
remote_url_for_auth(&repo.path, "origin", true).as_deref(),
|
||||
Some("https://gitea.example/push/repo.git")
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn askpass_scripts_use_unique_paths() {
|
||||
let first = next_askpass_path("test");
|
||||
let second = next_askpass_path("test");
|
||||
assert_ne!(first, second);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
#[cfg_attr(
|
||||
windows,
|
||||
|
||||
Reference in New Issue
Block a user