openreplay/frontend/app/player/mobile/managers/SnapshotManager.ts

import { TarFile } from "js-untar";
import ListWalker  from 'Player/common/ListWalker';

interface Snapshots {
  [timestamp: number]: TarFile
}

type Timestamp = { time: number }


export default class SnapshotManager extends ListWalker<Timestamp> {
  private snapshots: Snapshots = {}

  public mapToSnapshots(files: TarFile[]) {
    const filenameRegexp = /(\d+)_1_(\d+)\.jpeg$/;
    const firstPair = files[0].name.match(filenameRegexp)
    const sessionStart = firstPair ? parseInt(firstPair[1], 10) : 0
    files.forEach(file => {
      const [_, _2, imageTimestamp] = file
        .name
        .match(filenameRegexp)
        ?.map(n => parseInt(n, 10)) ?? [0, 0, 0]
      const messageTime = imageTimestamp - sessionStart
      this.snapshots[messageTime] = file
      this.append({ time: messageTime })
    })
  }

  public moveReady(t: number) {
    const msg = this.moveGetLast(t)
    if (msg) {
      return this.snapshots[msg.time]
    }
  }

  public clean() {
    this.snapshots = {}
    this.reset()
  }
}