process.env.DATABASE_PATH = ':memory:';
process.env.THEMES_DIR = './themes';
process.env.FRONTEND_DIST = './frontend/dist';

import { RegistrationRateLimitService } from '../../backend/src/common/services/registration-rate-limit.service';
import { Test } from '@nestjs/testing';
import { INestApplication } from '@nestjs/common';
import { HttpAdapterHost } from '@nestjs/core';
import cookieParser from 'cookie-parser';
import * as express from 'express';
import request from 'supertest';
import { AppModule } from '../../backend/src/app.module';
import { GlobalExceptionFilter } from '../../backend/src/common/filters/global-exception.filter';
import { CsrfMiddleware } from '../../backend/src/common/middleware/csrf.middleware';
import { ConfigService } from '@nestjs/config';
import { initDb } from './db-helper';

describe('RegistrationRateLimitService', () => {
  it('allows up to 10 per minute per IP', () => {
    const s = new RegistrationRateLimitService();
    const now = 1_000_000;
    for (let i = 0; i < 10; i++) {
      expect(s.isAllowed('1.1.1.1', now + i)).toBe(true);
      s.record('1.1.1.1', now + i);
    }
    expect(s.isAllowed('1.1.1.1', now + 10)).toBe(false);
  });

  it('isolates per IP', () => {
    const s = new RegistrationRateLimitService();
    const now = 1_000_000;
    for (let i = 0; i < 10; i++) s.record('1.1.1.1', now + i);
    expect(s.isAllowed('2.2.2.2', now + 10)).toBe(true);
  });
});

describe('register-first-admin registration rate limit (integration)', () => {
  let app: INestApplication;
  let limiter: RegistrationRateLimitService;

  beforeAll(async () => {
    const moduleRef = await Test.createTestingModule({ imports: [AppModule] }).compile();
    app = moduleRef.createNestApplication();
    app.use(cookieParser());
    app.use(express.json({ limit: '1mb' }));
    const csrfMw = new CsrfMiddleware(app.get(ConfigService));
    app.use((req: any, res: any, next: any) => csrfMw.use(req, res, next));
    const httpAdapterHost = app.get(HttpAdapterHost);
    app.useGlobalFilters(new GlobalExceptionFilter(httpAdapterHost));
    await app.init();
    await initDb(app);
    limiter = app.get(RegistrationRateLimitService);
  });

  afterAll(async () => {
    await app.close();
  });

  it('returns 429 once the per-IP cap is reached', async () => {
    // Prime the limiter for the IP that supertest will use. The default
    // Express socket address is IPv6-mapped IPv4 ('::ffff:127.0.0.1') on
    // most platforms; priming both forms keeps the test stable.
    const now = Date.now();
    for (let i = 0; i < 10; i++) {
      limiter.record('127.0.0.1', now + i);
      limiter.record('::ffff:127.0.0.1', now + i);
    }

    const server = app.getHttpServer();
    const agent = request.agent(server);
    await agent.get('/api/v1/auth/csrf');
    const cookies: any = agent.jar.getCookies(require('cookiejar').CookieAccessInfo.All);
    const csrf = cookies.find((c: any) => c.name === 'csrf');

    await agent.post('/api/v1/auth/register-first-admin')
      .set('X-CSRF-Token', csrf.value)
      .send({ username: 'noone', password: 'Sup3rSecret!Pass' })
      .expect(429);
  });
});