Rewrite package on typescript

.eslintignore

@@ -1 +1,2 @@
 /lib/browser/client-scripts/calibrate.min.js
+/build

.gitignore

@@ -1,3 +1,4 @@
 node_modules
+build
 *.log
 /lib/browser/client-scripts/calibrate.min.js

index.ts

@@ -0,0 +1,26 @@
+export { default as BaseStats } from "./lib/base-stats";
+export { default as SetsBuilder } from "./lib/sets-builder";
+
+import options from "./lib/config/options";
+export const config = { options };
+
+export { default as Calibrator } from "./lib/calibrator";
+export * as BrowserPool from "./lib/browser-pool";
+export { default as BrowserAgent } from "./lib/browser-agent";
+export { default as Image } from "./lib/image";
+export * as temp from "./lib/temp";
+export * as errors from "./lib/errors";
+export * as events from "./lib/events";
+export * as promiseUtils from "./lib/promise-utils";
+export * as clientBridge from "./lib/client-bridge";
+
+import * as coverageLevel from "./lib/coverage/coverage-level";
+export const coverage = { coverageLevel };
+
+import Camera from "./lib/browser/camera";
+export const browser = { Camera };
+export { Camera };
+
+export { default as ScreenShooter } from "./lib/screen-shooter";
+export { default as Viewport } from "./lib/screen-shooter/viewport";
+export { default as CoordValidator } from "./lib/screen-shooter/viewport/coord-validator";

lib/base-stats/index.ts

@@ -0,0 +1,72 @@
+import _ from "lodash";
+
+type StatNames<T extends string = 'passed' | 'failed' | 'skipped'> = Record<Uppercase<T>, Lowercase<T>>;
+
+type Stats = { [StatName in StatNames[keyof StatNames]]: number };
+type StatsResult = Stats & {
+    total: number;
+    retries: number;
+};
+
+export default class BaseStats {
+    private _tests: Set<string>;
+    private _retries: number;
+    private _statNames: StatNames;
+    private _stats: Stats;
+
+    public static create(statNames: StatNames): BaseStats {
+        return new this(statNames);
+    }
+
+    constructor(statNames: StatNames) {
+        this._tests = new Set();
+        this._retries = 0;
+        this._statNames = statNames;
+
+        this._stats = this._fillEmptyStats();
+    }
+
+    public addPassed(test: unknown) {
+        this._addStat(this._statNames.PASSED, test);
+    }
+
+    public addFailed(test: unknown) {
+        this._addStat(this._statNames.FAILED, test);
+    }
+
+    public addSkipped(test: unknown) {
+        this._addStat(this._statNames.SKIPPED, test);
+    }
+
+    public addRetries() {
+        this._retries++;
+    }
+
+    private _addStat(stat: StatNames[keyof StatNames], test: unknown, statsStorage = this._stats, testsStorage = this._tests) {
+        const key = `${this._buildSuiteKey(test)} ${this._buildStateKey(test)}`;
+
+        statsStorage[stat]++;
+        testsStorage.add(key);
+    }
+
+    private _fillEmptyStats(): Stats {
+        const statValues = _.values(this._statNames);
+
+        return _.zipObject(statValues, Array(statValues.length).fill(0)) as Stats;
+    }
+
+    private _buildStateKey(_test: unknown): string {
+        throw new Error('Method must be implemented in child classes');
+    }
+
+    private _buildSuiteKey(_test: unknown): string {
+        throw new Error('Method must be implemented in child classes');
+    }
+
+    public getResult(): StatsResult {
+        return _.extend(this._stats, {
+            total: this._tests.size,
+            retries: this._retries
+        });
+    }
+};

lib/browser-agent/index.ts

@@ -0,0 +1,34 @@
+import _ from 'lodash';
+
+import type { Pool } from '../types/pool';
+import type { NewBrowser } from '../types/new-browser';
+
+export default class BrowserAgent {
+    private _sessions: Array<string>;
+
+    public static create(browserId: string, pool: Pool): BrowserAgent {
+        return new BrowserAgent(browserId, pool);
+    }
+
+    constructor(public browserId: string, private _pool: Pool) {
+        this._sessions = [];
+    }
+
+    public async getBrowser(opts: any): Promise<NewBrowser> {
+        const browser = await this._pool.getBrowser(this.browserId, opts);
+
+        if (_.includes(this._sessions, browser.sessionId)) {
+            await this.freeBrowser(browser, {force: true});
+
+            return this.getBrowser(opts);
+        }
+
+        this._sessions.push(browser.sessionId);
+
+        return browser;
+    }
+
+    public async freeBrowser(browser: NewBrowser, opts: any): Promise<void> {
+        return this._pool.freeBrowser(browser, opts);
+    }
+}