import * as plugins from './plugins.js';
import * as paths from './paths.js';

export interface IDockerImageStoreConstructorOptions {
  dirPath: string;
}

export class DockerImageStore {
  public options: IDockerImageStoreConstructorOptions;

  constructor(optionsArg: IDockerImageStoreConstructorOptions) {
    this.options = optionsArg;
  }

  // Method to store tar stream
  public async storeImage(imageName: string, tarStream: NodeJS.ReadableStream): Promise<void> {
    const imagePath = plugins.path.join(this.options.dirPath, `${imageName}.tar`);

    // Create a write stream to store the tar file
    const writeStream = plugins.smartfile.fsStream.createWriteStream(imagePath);

    return new Promise((resolve, reject) => {
      tarStream.pipe(writeStream);

      writeStream.on('finish', resolve);
      writeStream.on('error', reject);
    });
  }

  // Method to retrieve tar stream
  public async getImage(imageName: string): Promise<plugins.smartstream.stream.Readable> {
    const imagePath = plugins.path.join(this.options.dirPath, `${imageName}.tar`);

    if (!(await plugins.smartfile.fs.fileExists(imagePath))) {
      throw new Error(`Image ${imageName} does not exist.`);
    }

    return plugins.smartfile.fsStream.createReadStream(imagePath);
  }
}