import { describe, expect, it } from "vitest";
import {
  isSafeScpRemoteHost,
  isSafeScpRemotePath,
  normalizeScpRemoteHost,
  normalizeScpRemotePath,
} from "./scp-host.js";

describe("scp remote host", () => {
  it.each([
    { value: "gateway-host", expected: "gateway-host" },
    { value: " bot@gateway-host ", expected: "bot@gateway-host" },
    { value: "bot@192.168.64.3", expected: "bot@192.168.64.3" },
    { value: "bot@[fe80::1]", expected: "bot@[fe80::1]" },
  ])("normalizes safe hosts for %j", ({ value, expected }) => {
    expect(normalizeScpRemoteHost(value)).toBe(expected);
  });

  it.each([
    null,
    undefined,
    "",
    "   ",
    "-oProxyCommand=whoami",
    "bot@gateway-host -oStrictHostKeyChecking=no",
    "bot@host:22",
    "bot@/tmp/host",
    "bot@@host",
    "@host",
    "bot@",
    "bot@host\\name",
    "bot@-gateway-host",
    "bot@fe80::1",
    "bot@[fe80::1%en0]",
    "bot name@gateway-host",
  ])("rejects unsafe host tokens: %j", (value) => {
    expect(normalizeScpRemoteHost(value)).toBeUndefined();
    expect(isSafeScpRemoteHost(value)).toBe(false);
  });
});

describe("scp remote path", () => {
  it.each([
    {
      value: "/Users/demo/Library/Messages/Attachments/ab/cd/photo.jpg",
      expected: "/Users/demo/Library/Messages/Attachments/ab/cd/photo.jpg",
    },
    {
      value: " /Users/demo/Library/Messages/Attachments/ab/cd/IMG 1234 (1).jpg ",
      expected: "/Users/demo/Library/Messages/Attachments/ab/cd/IMG 1234 (1).jpg",
    },
  ])("normalizes safe paths for %j", ({ value, expected }) => {
    expect(normalizeScpRemotePath(value)).toBe(expected);
    expect(isSafeScpRemotePath(value)).toBe(true);
  });

  it.each([
    null,
    undefined,
    "",
    "   ",
    "relative/path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad$path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad`path`.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad;path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad|path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad&path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad<path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad>path.jpg",
    '/Users/demo/Library/Messages/Attachments/ab/cd/bad"path.jpg',
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad'path.jpg",
    "/Users/demo/Library/Messages/Attachments/ab/cd/bad\\path.jpg",
  ])("rejects unsafe path tokens: %j", (value) => {
    expect(normalizeScpRemotePath(value)).toBeUndefined();
    expect(isSafeScpRemotePath(value)).toBe(false);
  });
});