support token %k #158

trzsz · Nov 30, 2024 · 95210d5 · 95210d5
1 parent 6145612
commit 95210d5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tssh/tokens.go b/tssh/tokens.go
@@ -130,6 +130,12 @@ func expandTokens(str string, args *sshArgs, param *sshParam, tokens string) (st
 				hashStr += param.proxy[len(param.proxy)-1]
 			}
 			buf.WriteString(fmt.Sprintf("%x", sha1.Sum([]byte(hashStr))))
+		case 'k':
+			if hostKeyAlias := getOptionConfig(args, "HostKeyAlias"); hostKeyAlias != "" {
+				buf.WriteString(hostKeyAlias)
+			} else {
+				buf.WriteString(args.Destination)
+			}
 		default:
 			return str, fmt.Errorf("token [%%%c] in [%s] is not supported yet", c, str)
 		}