import * as fs from 'fs';
import * as path from 'path';
import * as semver from 'semver';

...
export default class AssetService {
   ...
  public static async updateAll(throwOnError = false): Promise<void> {
    ...
    await this.cleanUpOldBinaries();
    ...
  }

  public static async updateOne(assetId: AssetIdentifier): Promise<void> {
    ...
    await this.cleanUpOldBinaries();
    return result;
  }

  static async cleanUpOldBinaries() {
    await this.cleanUpOldBinariesFromDir(join(AssetService.getAppmapDir(), 'lib', 'appmap'));
    await this.cleanUpOldBinariesFromDir(join(AssetService.getAppmapDir(), 'lib', 'java'));
    await this.cleanUpOldBinariesFromDir(join(AssetService.getAppmapDir(), 'lib', 'scanner'));
  }

  private static async cleanUpOldBinariesFromDir(directory: string) {
    try {
      const files = await fs.promises.readdir(directory);
      const fileGroups: { [key: string]: { name: string; version: string }[] } = {};

      files.forEach((file) => {
        const version = this.extractVersion(file);
        if (version && semver.valid(version)) {
          const groupName = file.replace(version, '');
          if (!fileGroups[groupName]) {
            fileGroups[groupName] = [];
          }
          fileGroups[groupName].push({ name: file, version });
        }
      });

      for (const groupName in fileGroups) {
        // Sort in descending order of versions
        const binaries = fileGroups[groupName].sort((a, b) => semver.compare(b.version, a.version));

        // Keep the latest version, delete the rest
        for (let i = 1; i < binaries.length; i++) {
          const filePath = path.join(directory, binaries[i].name);
          await fs.promises.unlink(filePath);
          log.info(`Deleted old binary: ${binaries[i].name}`);
        }
      }
    } catch (error) {
      log.error(`Failed to clean up old binaries: ${error}`);
    }
  }

  private static extractVersion(fileName: string): string | null {
    // Match patterns like '-v1.2.3', '-v1.2.3-alpha', '-v1.2.3+build', '-1.2.3-alpha+build'
    // with or without file extension
    const versionMatch = fileName.match(
      /-v?(\d+\.\d+\.\d+(-[a-zA-Z0-9-]+)?(\+[a-zA-Z0-9-]+)?)\b(\.[^.]+)?$/
    );
    return versionMatch ? versionMatch[1] : null;
  }
}

it('cleans up old binaries', async () => {
  ...
  await AssetService.updateAll(true);

  // Verify the results
  const verifyFiles = async (directory: string, expectedFiles: string[]) => {
    const files = await readdir(join(homeDir, '.appmap', directory));
    const expectedSet = new Set(expectedFiles);
    const remainingSet = new Set(files);

    expect(expectedSet).to.deep.equal(remainingSet);
  };

  // Verify each directory's expected files
  await verifyFiles(join('lib', 'appmap'), ['appmap-v0.0.0-TEST']);
  await verifyFiles(join('lib', 'java'), [
    'appmap-1.26.2.jar',
    'appmap-agent-1.26.2.jar',
    'appmap.jar',
  ]);
  await verifyFiles(join('lib', 'scanner'), ['scanner-v1.88.0']);
});