import assert from "node:assert/strict";
import test from "node:test";
import { DeleteObjectCommand, DeleteObjectsCommand } from "@aws-sdk/client-s3";
import { getExpiredObjectKeys } from "../../src/server/services/s3-storage-retention.service";
import {
  deleteS3ObjectsWithClient,
  isMissingContentMd5Error,
} from "../../src/server/services/s3-storage.service";

const now = Date.parse("2026-07-15T00:00:00.000Z");

function destination(overrides: Record<string, unknown> = {}) {
  return {
    id: "destination-1",
    userId: "user-1",
    organizationId: "org-1",
    name: "Backups",
    bucket: "bucket",
    endpoint: null,
    region: "ap-southeast-1",
    accessKeyId: "access-key",
    secretAccessKeyEnc: "encrypted",
    additionalFlags: [],
    retentionDays: null,
    maxBackupCount: null,
    objectPrefix: "doktainer/backups",
    ...overrides,
  };
}

test("retention removes objects older than the configured number of days", () => {
  const keys = getExpiredObjectKeys(
    [
      {
        key: "old.tar.gz",
        lastModified: new Date("2026-07-01T00:00:00.000Z"),
        size: 10,
      },
      {
        key: "recent.tar.gz",
        lastModified: new Date("2026-07-14T00:00:00.000Z"),
        size: 10,
      },
    ],
    destination({ retentionDays: 7 }),
    now,
  );

  assert.deepEqual(keys, ["old.tar.gz"]);
});

test("maximum backup count keeps the newest objects", () => {
  const keys = getExpiredObjectKeys(
    [
      {
        key: "oldest.tar.gz",
        lastModified: new Date("2026-07-10T00:00:00.000Z"),
        size: 10,
      },
      {
        key: "newest.tar.gz",
        lastModified: new Date("2026-07-14T00:00:00.000Z"),
        size: 10,
      },
      {
        key: "middle.tar.gz",
        lastModified: new Date("2026-07-12T00:00:00.000Z"),
        size: 10,
      },
    ],
    destination({ maxBackupCount: 2 }),
    now,
  );

  assert.deepEqual(keys, ["oldest.tar.gz"]);
});

test("S3 deletion falls back to individual requests when batch MD5 is rejected", async () => {
  const commands: unknown[] = [];
  const client = {
    async send(command: unknown) {
      commands.push(command);
      if (command instanceof DeleteObjectsCommand) {
        throw Object.assign(new Error("Missing required header for this request: Content-MD5"), {
          Code: "InvalidRequest",
        });
      }
      return {};
    },
  };

  await deleteS3ObjectsWithClient(
    client as never,
    "bucket",
    ["backups/one.tar.gz", "backups/two.tar.gz"],
  );

  assert.equal(commands.length, 3);
  assert.ok(commands[0] instanceof DeleteObjectsCommand);
  assert.ok(commands[1] instanceof DeleteObjectCommand);
  assert.ok(commands[2] instanceof DeleteObjectCommand);
});

test("S3 deletion does not hide unrelated batch failures", async () => {
  const failure = Object.assign(new Error("Access denied"), {
    Code: "AccessDenied",
  });
  const client = {
    async send() {
      throw failure;
    },
  };

  assert.equal(isMissingContentMd5Error(failure), false);
  await assert.rejects(
    deleteS3ObjectsWithClient(client as never, "bucket", ["backup.tar.gz"]),
    failure,
  );
});