import {
  GenericContainer,
  type StartedTestContainer,
  Wait,
} from 'testcontainers';

const GREENMAIL_IMAGE =
  'greenmail/standalone:2.1.9@sha256:3ac5a83dd6727cf95e4d50e18907fb8ee7bbf5f67e8534714dee2fb1b5b2e1d4';
const GREENMAIL_IMAP_PORT = 3143;
const GREENMAIL_SMTP_PORT = 3025;

export type GreenmailServer = {
  host: string;
  imapPort: number;
  smtpPort: number;
  stop: () => Promise<void>;
};

// The mailbox has to be declared up front: the account save authenticates
// against IMAP immediately, before any mail exists to create a user implicitly.
export const startGreenmailContainer = async ({
  username,
  password,
}: {
  username: string;
  password: string;
}): Promise<GreenmailServer> => {
  const [localPart, domain] = username.split('@');

  const container: StartedTestContainer = await new GenericContainer(
    GREENMAIL_IMAGE,
  )
    .withEnvironment({
      GREENMAIL_OPTS: [
        '-Dgreenmail.setup.test.imap',
        '-Dgreenmail.setup.test.smtp',
        '-Dgreenmail.hostname=0.0.0.0',
        `-Dgreenmail.users=${localPart}:${password}@${domain}`,
      ].join(' '),
    })
    .withExposedPorts(GREENMAIL_IMAP_PORT, GREENMAIL_SMTP_PORT)
    .withWaitStrategy(Wait.forListeningPorts())
    .start();

  return {
    host: container.getHost(),
    imapPort: container.getMappedPort(GREENMAIL_IMAP_PORT),
    smtpPort: container.getMappedPort(GREENMAIL_SMTP_PORT),
    stop: async () => {
      await container.stop();
    },
  };
};