import { expect, tap } from '@git.zone/tstest/tapbundle';

import * as plugins from '../ts/plugins.js';
import * as smartbucket from '../ts/index.js';

class FakeS3Client {
  private callIndex = 0;

  constructor(private readonly pages: Array<Partial<plugins.s3.ListObjectsV2Output>>) {}

  public async send(_command: any) {
    const page = this.pages[this.callIndex] || { Contents: [], CommonPrefixes: [], IsTruncated: false };
    this.callIndex += 1;
    return page;
  }
}

tap.test('MetaData.hasMetaData should return false when metadata file does not exist', async () => {
  const fakeFile = {
    name: 'file.txt',
    parentDirectoryRef: {
      async getFile() {
        throw new Error(`File not found at path 'file.txt.metadata'`);
      },
    },
  } as unknown as smartbucket.File;

  const hasMetaData = await smartbucket.MetaData.hasMetaData({ file: fakeFile });
  expect(hasMetaData).toBeFalse();
});

tap.test('getSubDirectoryByName should create correct parent chain for new nested directories', async () => {
  const fakeSmartbucket = { s3Client: new FakeS3Client([{ Contents: [], CommonPrefixes: [] }]) } as unknown as smartbucket.SmartBucket;
  const bucket = new smartbucket.Bucket(fakeSmartbucket, 'test-bucket');
  const baseDirectory = new smartbucket.Directory(bucket, null as any, '');

  const nestedDirectory = await baseDirectory.getSubDirectoryByName('level1/level2', { getEmptyDirectory: true });

  expect(nestedDirectory.name).toEqual('level2');
  expect(nestedDirectory.parentDirectoryRef.name).toEqual('level1');
  expect(nestedDirectory.getBasePath()).toEqual('level1/level2/');
});

tap.test('listFiles should aggregate results across paginated ListObjectsV2 responses', async () => {
  const firstPage = {
    Contents: Array.from({ length: 1000 }, (_, index) => ({ Key: `file-${index}` })),
    IsTruncated: true,
    NextContinuationToken: 'token-1',
  };
  const secondPage = {
    Contents: Array.from({ length: 200 }, (_, index) => ({ Key: `file-${1000 + index}` })),
    IsTruncated: false,
  };
  const fakeSmartbucket = { s3Client: new FakeS3Client([firstPage, secondPage]) } as unknown as smartbucket.SmartBucket;
  const bucket = new smartbucket.Bucket(fakeSmartbucket, 'test-bucket');
  const baseDirectory = new smartbucket.Directory(bucket, null as any, '');

  const files = await baseDirectory.listFiles();
  expect(files.length).toEqual(1200);
});

tap.test('listDirectories should aggregate CommonPrefixes across pagination', async () => {
  const fakeSmartbucket = {
    s3Client: new FakeS3Client([
      { CommonPrefixes: [{ Prefix: 'dirA/' }], IsTruncated: true, NextContinuationToken: 'token-1' },
      { CommonPrefixes: [{ Prefix: 'dirB/' }], IsTruncated: false },
    ]),
  } as unknown as smartbucket.SmartBucket;
  const bucket = new smartbucket.Bucket(fakeSmartbucket, 'test-bucket');
  const baseDirectory = new smartbucket.Directory(bucket, null as any, '');

  const directories = await baseDirectory.listDirectories();
  expect(directories.map((d) => d.name)).toEqual(['dirA', 'dirB']);
});

export default tap.start();