import { describe, expect, it } from "vitest";
import { ensureNewSshKey, isSshPublicKey, sshFingerprint } from "#lib/api/settings.ts";
const ed25519 =
"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIHEisipfUi+zRyW1reo72Rt4+EUbM5mKZrI9/PkyHMpR qa";
const ecdsa =
"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBPa17IfyrTk7DrKJLEhhFFDho8BE0v7Wk07xsmqWfoi+W6OuaFyPrSNJ2oYbmp1ZHIOOFSq9VY3Tu/eEk8VGhQY= qa";
const withBlob = (key: string, edit: (blob: string) => string) => {
const [type, blob, ...rest] = key.split(" ");
return [type, edit(blob), ...rest].join(" ");
};
describe("isSshPublicKey", () => {
it("accepts well-formed keys", () => {
expect(isSshPublicKey(ed25519)).toBe(true);
expect(isSshPublicKey(ecdsa)).toBe(true);
expect(isSshPublicKey(ed25519.split(" ").slice(0, 2).join(" "))).toBe(true);
});
it("rejects keys with characters removed", () => {
expect(isSshPublicKey(withBlob(ed25519, (blob) => blob.slice(0, -4)))).toBe(false);
expect(
isSshPublicKey(withBlob(ed25519, (blob) => blob.slice(0, 20) + blob.slice(24)))
).toBe(false);
expect(isSshPublicKey(withBlob(ecdsa, (blob) => blob.slice(0, -8) + "="))).toBe(false);
});
it("rejects keys with trailing bytes", () => {
expect(isSshPublicKey(withBlob(ed25519, (blob) => blob + "AAAA"))).toBe(false);
});
it("rejects a type that does not match the blob", () => {
expect(isSshPublicKey(ed25519.replace("ssh-ed25519", "ssh-rsa"))).toBe(false);
expect(isSshPublicKey("ssh-ed25519 not-base64!")).toBe(false);
expect(isSshPublicKey("ssh-ed25519")).toBe(false);
});
});
describe("ensureNewSshKey", () => {
it("returns the fingerprint of a key that is not added yet", async () => {
await expect(ensureNewSshKey(ed25519, [])).resolves.toMatch(/^SHA256:/);
});
it("rejects a key whose content is already added under any name", async () => {
const fingerprint = await sshFingerprint(ed25519);
await expect(
ensureNewSshKey(ed25519.replace(" qa", " another name"), [fingerprint])
).rejects.toThrow(`This key is already added (${fingerprint}).`);
});
});