@naturalcycles/abba 1.27.0 → 2.0.0

package/dist/abba.d.ts

@@ -1,6 +1,5 @@
-import { Unsaved } from '@naturalcycles/js-lib';
-import { SegmentationData } from '.';
-import { AbbaConfig, Bucket, Experiment, ExperimentAssignmentStatistics, ExperimentWithBuckets, GeneratedUserAssignment } from './types';
+import type { Unsaved } from '@naturalcycles/js-lib';
+import type { AbbaConfig, Bucket, Experiment, ExperimentAssignmentStatistics, ExperimentWithBuckets, GeneratedUserAssignment, SegmentationData } from './types.js';
 export declare class Abba {
     cfg: AbbaConfig;
     private experimentDao;

package/dist/abba.js

@@ -1,24 +1,21 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Abba = void 0;
-const tslib_1 = require("tslib");
-const js_lib_1 = require("@naturalcycles/js-lib");
-const nodejs_lib_1 = require("@naturalcycles/nodejs-lib");
-const bucket_dao_1 = require("./dao/bucket.dao");
-const experiment_dao_1 = require("./dao/experiment.dao");
-const userAssignment_dao_1 = require("./dao/userAssignment.dao");
-const types_1 = require("./types");
-const util_1 = require("./util");
+import { __decorate } from "tslib";
+import { _assert, _Memo, _shuffle, localTime, pMap } from '@naturalcycles/js-lib';
+import { LRUMemoCache } from '@naturalcycles/nodejs-lib';
+import { bucketDao } from './dao/bucket.dao.js';
+import { experimentDao } from './dao/experiment.dao.js';
+import { userAssignmentDao } from './dao/userAssignment.dao.js';
+import { AssignmentStatus } from './types.js';
+import { canGenerateNewAssignments, generateUserAssignmentData, getUserExclusionSet, validateTotalBucketRatio, } from './util.js';
 /**
  * 10 minutes
  */
 const CACHE_TTL = 600_000;
-class Abba {
+export class Abba {
     constructor(cfg) {
         this.cfg = cfg;
-        this.experimentDao = (0, experiment_dao_1.experimentDao)(this.cfg.db);
-        this.bucketDao = (0, bucket_dao_1.bucketDao)(this.cfg.db);
-        this.userAssignmentDao = (0, userAssignment_dao_1.userAssignmentDao)(this.cfg.db);
+        this.experimentDao = experimentDao(this.cfg.db);
+        this.bucketDao = bucketDao(this.cfg.db);
+        this.userAssignmentDao = userAssignmentDao(this.cfg.db);
     }
     /**
      * Returns all experiments.
@@ -50,8 +47,8 @@ class Abba {
      * Cold method.
      */
     async createExperiment(experiment, buckets) {
-        if (experiment.status === types_1.AssignmentStatus.Active) {
-            (0, util_1.validateTotalBucketRatio)(buckets);
+        if (experiment.status === AssignmentStatus.Active) {
+            validateTotalBucketRatio(buckets);
         }
         const created = await this.experimentDao.save(experiment);
         const createdbuckets = await this.bucketDao.saveBatch(buckets.map(b => ({ ...b, experimentId: created.id })));
@@ -66,11 +63,11 @@ class Abba {
      * Cold method.
      */
     async saveExperiment(experiment, buckets) {
-        if (experiment.status === types_1.AssignmentStatus.Active) {
-            (0, util_1.validateTotalBucketRatio)(buckets);
+        if (experiment.status === AssignmentStatus.Active) {
+            validateTotalBucketRatio(buckets);
         }
         const updatedExperiment = await this.experimentDao.save(experiment, { saveMethod: 'update' });
-        const updatedBuckets = await (0, js_lib_1.pMap)(buckets, async (bucket) => {
+        const updatedBuckets = await pMap(buckets, async (bucket) => {
             return await this.bucketDao.save({ ...bucket, experimentId: updatedExperiment.id }, { saveMethod: bucket.id ? 'update' : undefined });
         });
         await this.updateExclusions(updatedExperiment.id, updatedExperiment.exclusions);
@@ -109,9 +106,9 @@ class Abba {
      */
     async deleteExperiment(experimentId) {
         const experiment = await this.experimentDao.requireById(experimentId);
-        const hasBeenInactiveFor15Mins = experiment.status === types_1.AssignmentStatus.Inactive &&
-            (0, js_lib_1.localTime)(experiment.updated).isOlderThan(15, 'minute');
-        (0, js_lib_1._assert)(hasBeenInactiveFor15Mins, 'Experiment must be inactive for at least 15 minutes before deletion');
+        const hasBeenInactiveFor15Mins = experiment.status === AssignmentStatus.Inactive &&
+            localTime(experiment.updated).isOlderThan(15, 'minute');
+        _assert(hasBeenInactiveFor15Mins, 'Experiment must be inactive for at least 15 minutes before deletion');
         const userAssignmentDeleteQuery = this.userAssignmentDao
             .query()
             .filterEq('experimentId', experimentId);
@@ -132,9 +129,9 @@ class Abba {
      */
     async getUserAssignment(experimentKey, userId, existingOnly, segmentationData) {
         const experiment = await this.experimentDao.getOneBy('key', experimentKey);
-        (0, js_lib_1._assert)(experiment, `Experiment does not exist: ${experimentKey}`);
+        _assert(experiment, `Experiment does not exist: ${experimentKey}`);
         // Inactive experiments should never return an assignment
-        if (experiment.status === types_1.AssignmentStatus.Inactive) {
+        if (experiment.status === AssignmentStatus.Inactive) {
             return {
                 experiment,
                 assignment: null,
@@ -156,23 +153,23 @@ class Abba {
             };
         }
         // No existing assignment, but we don't want to generate a new one
-        if (existingOnly || experiment.status === types_1.AssignmentStatus.Paused) {
+        if (existingOnly || experiment.status === AssignmentStatus.Paused) {
             return {
                 experiment,
                 assignment: null,
             };
         }
         const experiments = await this.getAllExperiments();
-        const exclusionSet = (0, util_1.getUserExclusionSet)(experiments, existingAssignments);
-        if (!(0, util_1.canGenerateNewAssignments)(experiment, exclusionSet)) {
+        const exclusionSet = getUserExclusionSet(experiments, existingAssignments);
+        if (!canGenerateNewAssignments(experiment, exclusionSet)) {
             return {
                 experiment,
                 assignment: null,
             };
         }
-        (0, js_lib_1._assert)(segmentationData, 'Segmentation data required when creating a new assignment');
+        _assert(segmentationData, 'Segmentation data required when creating a new assignment');
         const experimentWithBuckets = { ...experiment, buckets };
-        const assignment = (0, util_1.generateUserAssignmentData)(experimentWithBuckets, userId, segmentationData);
+        const assignment = generateUserAssignmentData(experimentWithBuckets, userId, segmentationData);
         if (!assignment) {
             return {
                 experiment,
@@ -199,7 +196,7 @@ class Abba {
      */
     async getAllExistingGeneratedUserAssignments(userId) {
         const assignments = await this.userAssignmentDao.getBy('userId', userId);
-        return await (0, js_lib_1.pMap)(assignments, async (assignment) => {
+        return await pMap(assignments, async (assignment) => {
             const experiment = await this.experimentDao.requireById(assignment.experimentId);
             const bucket = await this.bucketDao.getById(assignment.bucketId);
             return {
@@ -221,13 +218,13 @@ class Abba {
     async generateUserAssignments(userId, segmentationData, existingOnly = false) {
         const experiments = await this.getAllExperiments();
         const existingAssignments = await this.userAssignmentDao.getBy('userId', userId);
-        const exclusionSet = (0, util_1.getUserExclusionSet)(experiments, existingAssignments);
+        const exclusionSet = getUserExclusionSet(experiments, existingAssignments);
         const assignments = [];
         const newAssignments = [];
         // Shuffling means that randomisation occurs in the mutual exclusion
         // as experiments are looped through sequentially, this removes the risk of the same experiment always being assigned first in the list of mutually exclusive experiments
         // This is simmpler than trying to resolve after assignments have already been determined
-        const availableExperiments = (0, js_lib_1._shuffle)(experiments.filter(e => e.status === types_1.AssignmentStatus.Active || e.status === types_1.AssignmentStatus.Paused));
+        const availableExperiments = _shuffle(experiments.filter(e => e.status === AssignmentStatus.Active || e.status === AssignmentStatus.Paused));
         for (const experiment of availableExperiments) {
             const existing = existingAssignments.find(ua => ua.experimentId === experiment.id);
             if (existing) {
@@ -242,8 +239,8 @@ class Abba {
                     },
                 });
             }
-            else if (!existingOnly && (0, util_1.canGenerateNewAssignments)(experiment, exclusionSet)) {
-                const assignment = (0, util_1.generateUserAssignmentData)(experiment, userId, segmentationData);
+            else if (!existingOnly && canGenerateNewAssignments(experiment, exclusionSet)) {
+                const assignment = generateUserAssignmentData(experiment, userId, segmentationData);
                 if (assignment) {
                     const created = this.userAssignmentDao.create(assignment);
                     newAssignments.push(created);
@@ -275,7 +272,7 @@ class Abba {
             .filterEq('experimentId', experimentId)
             .runQueryCount();
         const buckets = await this.bucketDao.getBy('experimentId', experimentId);
-        const bucketAssignments = await (0, js_lib_1.pMap)(buckets, async (bucket) => {
+        const bucketAssignments = await pMap(buckets, async (bucket) => {
             const totalAssignments = await this.userAssignmentDao
                 .query()
                 .filterEq('bucketId', bucket.id)
@@ -291,7 +288,6 @@ class Abba {
         };
     }
 }
-exports.Abba = Abba;
-tslib_1.__decorate([
-    (0, js_lib_1._Memo)({ cacheFactory: () => new nodejs_lib_1.LRUMemoCache({ ttl: CACHE_TTL, max: 1 }) })
+__decorate([
+    _Memo({ cacheFactory: () => new LRUMemoCache({ ttl: CACHE_TTL, max: 1 }) })
 ], Abba.prototype, "getAllExperiments", null);

package/dist/dao/bucket.dao.d.ts

@@ -1,6 +1,7 @@
-import { CommonDao, CommonDB } from '@naturalcycles/db-lib';
-import { Saved } from '@naturalcycles/js-lib';
-import { BaseBucket, Bucket } from '../types';
+import type { CommonDB } from '@naturalcycles/db-lib';
+import { CommonDao } from '@naturalcycles/db-lib';
+import type { Saved } from '@naturalcycles/js-lib';
+import type { BaseBucket, Bucket } from '../types.js';
 type BucketDBM = Saved<BaseBucket> & {
     data: string | null;
 };

package/dist/dao/bucket.dao.js

@@ -1,11 +1,7 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.bucketDao = exports.BucketDao = void 0;
-const db_lib_1 = require("@naturalcycles/db-lib");
-class BucketDao extends db_lib_1.CommonDao {
+import { CommonDao } from '@naturalcycles/db-lib';
+export class BucketDao extends CommonDao {
 }
-exports.BucketDao = BucketDao;
-const bucketDao = (db) => new BucketDao({
+export const bucketDao = (db) => new BucketDao({
     db,
     table: 'Bucket',
     hooks: {
@@ -21,4 +17,3 @@ const bucketDao = (db) => new BucketDao({
         }),
     },
 });
-exports.bucketDao = bucketDao;

package/dist/dao/experiment.dao.d.ts

@@ -1,6 +1,7 @@
-import { CommonDao, CommonDB } from '@naturalcycles/db-lib';
-import { Saved } from '@naturalcycles/js-lib';
-import { BaseExperiment, Experiment } from '../types';
+import type { CommonDB } from '@naturalcycles/db-lib';
+import { CommonDao } from '@naturalcycles/db-lib';
+import type { Saved } from '@naturalcycles/js-lib';
+import type { BaseExperiment, Experiment } from '../types.js';
 type ExperimentDBM = Saved<BaseExperiment> & {
     rules: string | null;
     exclusions: string | null;

package/dist/dao/experiment.dao.js

@@ -1,12 +1,8 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.experimentDao = exports.ExperimentDao = void 0;
-const db_lib_1 = require("@naturalcycles/db-lib");
-const js_lib_1 = require("@naturalcycles/js-lib");
-class ExperimentDao extends db_lib_1.CommonDao {
+import { CommonDao } from '@naturalcycles/db-lib';
+import { localDate } from '@naturalcycles/js-lib';
+export class ExperimentDao extends CommonDao {
 }
-exports.ExperimentDao = ExperimentDao;
-const experimentDao = (db) => new ExperimentDao({
+export const experimentDao = (db) => new ExperimentDao({
     db,
     table: 'Experiment',
     hooks: {
@@ -34,7 +30,6 @@ const experimentDao = (db) => new ExperimentDao({
         }),
     },
 });
-exports.experimentDao = experimentDao;
 /**
  * https://nc1.slack.com/archives/CCNTHJT7V/p1682514277002739
  * MySQL Automatically parses Date fields as Date objects
@@ -43,6 +38,6 @@ exports.experimentDao = experimentDao;
 function parseMySQLDate(date) {
     // @ts-expect-error
     if (date instanceof Date)
-        return (0, js_lib_1.localDate)(date).toISODate();
+        return localDate(date).toISODate();
     return date;
 }

package/dist/dao/userAssignment.dao.d.ts

@@ -1,5 +1,6 @@
-import { CommonDao, CommonDB } from '@naturalcycles/db-lib';
-import { UserAssignment } from '../types';
+import type { CommonDB } from '@naturalcycles/db-lib';
+import { CommonDao } from '@naturalcycles/db-lib';
+import type { UserAssignment } from '../types.js';
 export declare class UserAssignmentDao extends CommonDao<UserAssignment> {
 }
 export declare const userAssignmentDao: (db: CommonDB) => UserAssignmentDao;

package/dist/dao/userAssignment.dao.js

@@ -1,12 +1,7 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.userAssignmentDao = exports.UserAssignmentDao = void 0;
-const db_lib_1 = require("@naturalcycles/db-lib");
-class UserAssignmentDao extends db_lib_1.CommonDao {
+import { CommonDao } from '@naturalcycles/db-lib';
+export class UserAssignmentDao extends CommonDao {
 }
-exports.UserAssignmentDao = UserAssignmentDao;
-const userAssignmentDao = (db) => new UserAssignmentDao({
+export const userAssignmentDao = (db) => new UserAssignmentDao({
     db,
     table: 'UserAssignment',
 });
-exports.userAssignmentDao = userAssignmentDao;

package/dist/index.d.ts

@@ -1,2 +1,2 @@
-export * from './abba';
-export * from './types';
+export * from './abba.js';
+export * from './types.js';

package/dist/index.js

@@ -1,5 +1,2 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const tslib_1 = require("tslib");
-tslib_1.__exportStar(require("./abba"), exports);
-tslib_1.__exportStar(require("./types"), exports);
+export * from './abba.js';
+export * from './types.js';

package/dist/types.d.ts

@@ -1,5 +1,5 @@
-import { CommonDB } from '@naturalcycles/db-lib';
-import { AnyObject, BaseDBEntity, IsoDate, Saved } from '@naturalcycles/js-lib';
+import type { CommonDB } from '@naturalcycles/db-lib';
+import type { AnyObject, BaseDBEntity, IsoDate, Saved } from '@naturalcycles/js-lib';
 export interface AbbaConfig {
     db: CommonDB;
 }

package/dist/types.js

@@ -1,7 +1,4 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SegmentationRuleOperator = exports.AssignmentStatus = void 0;
-var AssignmentStatus;
+export var AssignmentStatus;
 (function (AssignmentStatus) {
     /**
      * Will return existing assignments and generate new assignments
@@ -15,8 +12,8 @@ var AssignmentStatus;
      * Will not return any assignments
      */
     AssignmentStatus[AssignmentStatus["Inactive"] = 3] = "Inactive";
-})(AssignmentStatus || (exports.AssignmentStatus = AssignmentStatus = {}));
-var SegmentationRuleOperator;
+})(AssignmentStatus || (AssignmentStatus = {}));
+export var SegmentationRuleOperator;
 (function (SegmentationRuleOperator) {
     SegmentationRuleOperator["IsSet"] = "isSet";
     SegmentationRuleOperator["IsNotSet"] = "isNotSet";
@@ -26,4 +23,4 @@ var SegmentationRuleOperator;
     SegmentationRuleOperator["Regex"] = "regex";
     /* eslint-disable id-blacklist*/
     SegmentationRuleOperator["Boolean"] = "boolean";
-})(SegmentationRuleOperator || (exports.SegmentationRuleOperator = SegmentationRuleOperator = {}));
+})(SegmentationRuleOperator || (SegmentationRuleOperator = {}));

package/dist/util.d.ts

@@ -1,14 +1,18 @@
-import { Unsaved } from '@naturalcycles/js-lib';
-import { Bucket, ExclusionSet, Experiment, ExperimentWithBuckets, SegmentationData, SegmentationRule, SegmentationRuleFn, SegmentationRuleOperator, UserAssignment } from './types';
+import type { Unsaved } from '@naturalcycles/js-lib';
+import type { Bucket, ExclusionSet, Experiment, ExperimentWithBuckets, SegmentationData, SegmentationRule, SegmentationRuleFn, UserAssignment } from './types.js';
+import { SegmentationRuleOperator } from './types.js';
 /**
  * Generate a new assignment for a given user.
  * Doesn't save it.
  */
 export declare const generateUserAssignmentData: (experiment: ExperimentWithBuckets, userId: string, segmentationData: SegmentationData) => UserAssignment | null;
-/**
- * Generate a random number between 0 and 100
- */
-export declare const rollDie: () => number;
+declare class RandomService {
+    /**
+     * Generate a random number between 0 and 100
+     */
+    rollDie(): number;
+}
+export declare const randomService: RandomService;
 /**
  * Determines a users assignment for this experiment. Returns null if they are not considered to be in the sampling group
  */
@@ -42,3 +46,4 @@ export declare const canGenerateNewAssignments: (experiment: Experiment, exclusi
  * based on a combination of existing assignments and mutual exclusion configuration
  */
 export declare const getUserExclusionSet: (experiments: Experiment[], existingAssignments: UserAssignment[]) => ExclusionSet;
+export {};

package/dist/util.js

@@ -1,49 +1,46 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getUserExclusionSet = exports.canGenerateNewAssignments = exports.segmentationRuleMap = exports.validateSegmentationRules = exports.validateTotalBucketRatio = exports.determineBucket = exports.determineAssignment = exports.rollDie = exports.generateUserAssignmentData = void 0;
-const js_lib_1 = require("@naturalcycles/js-lib");
-const semver_1 = require("semver");
-const types_1 = require("./types");
+import { localDate } from '@naturalcycles/js-lib';
+import { satisfies } from 'semver';
+import { AssignmentStatus, SegmentationRuleOperator } from './types.js';
 /**
  * Generate a new assignment for a given user.
  * Doesn't save it.
  */
-const generateUserAssignmentData = (experiment, userId, segmentationData) => {
-    const segmentationMatch = (0, exports.validateSegmentationRules)(experiment.rules, segmentationData);
+export const generateUserAssignmentData = (experiment, userId, segmentationData) => {
+    const segmentationMatch = validateSegmentationRules(experiment.rules, segmentationData);
     if (!segmentationMatch)
         return null;
-    const bucket = (0, exports.determineAssignment)(experiment.sampling, experiment.buckets);
+    const bucket = determineAssignment(experiment.sampling, experiment.buckets);
     return {
         userId,
         experimentId: experiment.id,
         bucketId: bucket?.id || null,
     };
 };
-exports.generateUserAssignmentData = generateUserAssignmentData;
-/**
- * Generate a random number between 0 and 100
- */
-const rollDie = () => {
-    return Math.random() * 100;
-};
-exports.rollDie = rollDie;
+class RandomService {
+    /**
+     * Generate a random number between 0 and 100
+     */
+    rollDie() {
+        return Math.random() * 100;
+    }
+}
+export const randomService = new RandomService();
 /**
  * Determines a users assignment for this experiment. Returns null if they are not considered to be in the sampling group
  */
-const determineAssignment = (sampling, buckets) => {
+export const determineAssignment = (sampling, buckets) => {
     // Should this person be considered for the experiment?
-    if ((0, exports.rollDie)() > sampling) {
+    if (randomService.rollDie() > sampling) {
         return null;
     }
     // get their bucket
-    return (0, exports.determineBucket)(buckets);
+    return determineBucket(buckets);
 };
-exports.determineAssignment = determineAssignment;
 /**
  * Determines which bucket a user assignment will recieve
  */
-const determineBucket = (buckets) => {
-    const bucketRoll = (0, exports.rollDie)();
+export const determineBucket = (buckets) => {
+    const bucketRoll = randomService.rollDie();
     let range;
     const bucket = buckets.find(b => {
         if (!range) {
@@ -61,17 +58,15 @@ const determineBucket = (buckets) => {
     }
     return bucket;
 };
-exports.determineBucket = determineBucket;
 /**
  * Validate the total ratio of the buckets equals 100
  */
-const validateTotalBucketRatio = (buckets) => {
+export const validateTotalBucketRatio = (buckets) => {
     const bucketSum = buckets.reduce((sum, current) => sum + current.ratio, 0);
     if (bucketSum !== 100) {
         throw new Error('Total bucket ratio must be 100 before you can activate an experiment');
     }
 };
-exports.validateTotalBucketRatio = validateTotalBucketRatio;
 /**
  * Validate a users segmentation data against multiple rules. Returns false if any fail
  *
@@ -79,38 +74,37 @@ exports.validateTotalBucketRatio = validateTotalBucketRatio;
  * @param segmentationData
  * @returns
  */
-const validateSegmentationRules = (rules, segmentationData) => {
+export const validateSegmentationRules = (rules, segmentationData) => {
     for (const rule of rules) {
         const { key, value, operator } = rule;
-        if (!exports.segmentationRuleMap[operator](segmentationData[key], value))
+        if (!segmentationRuleMap[operator](segmentationData[key], value))
             return false;
     }
     return true;
 };
-exports.validateSegmentationRules = validateSegmentationRules;
 /**
  * Map of segmentation rule validators
  */
-exports.segmentationRuleMap = {
-    [types_1.SegmentationRuleOperator.IsSet](keyValue) {
+export const segmentationRuleMap = {
+    [SegmentationRuleOperator.IsSet](keyValue) {
         return !!keyValue;
     },
-    [types_1.SegmentationRuleOperator.IsNotSet](keyValue) {
+    [SegmentationRuleOperator.IsNotSet](keyValue) {
         return !keyValue;
     },
-    [types_1.SegmentationRuleOperator.EqualsText](keyValue, ruleValue) {
+    [SegmentationRuleOperator.EqualsText](keyValue, ruleValue) {
         return keyValue?.toString() === ruleValue.toString();
     },
-    [types_1.SegmentationRuleOperator.NotEqualsText](keyValue, ruleValue) {
+    [SegmentationRuleOperator.NotEqualsText](keyValue, ruleValue) {
         return keyValue?.toString() !== ruleValue.toString();
     },
-    [types_1.SegmentationRuleOperator.Semver](keyValue, ruleValue) {
-        return (0, semver_1.satisfies)(keyValue?.toString() || '', ruleValue.toString());
+    [SegmentationRuleOperator.Semver](keyValue, ruleValue) {
+        return satisfies(keyValue?.toString() || '', ruleValue.toString());
     },
-    [types_1.SegmentationRuleOperator.Regex](keyValue, ruleValue) {
+    [SegmentationRuleOperator.Regex](keyValue, ruleValue) {
         return new RegExp(ruleValue).test(keyValue?.toString() || '');
     },
-    [types_1.SegmentationRuleOperator.Boolean](keyValue, ruleValue) {
+    [SegmentationRuleOperator.Boolean](keyValue, ruleValue) {
         // If it's true, then must be true
         if (ruleValue === 'true')
             return keyValue?.toString() === 'true';
@@ -121,17 +115,16 @@ exports.segmentationRuleMap = {
 /**
  * Returns true if an experiment is able to generate new assignments based on status and start/end dates
  */
-const canGenerateNewAssignments = (experiment, exclusionSet) => {
+export const canGenerateNewAssignments = (experiment, exclusionSet) => {
     return (!exclusionSet.has(experiment.id) &&
-        experiment.status === types_1.AssignmentStatus.Active &&
-        js_lib_1.localDate.today().isBetween(experiment.startDateIncl, experiment.endDateExcl, '[)'));
+        experiment.status === AssignmentStatus.Active &&
+        localDate.today().isBetween(experiment.startDateIncl, experiment.endDateExcl, '[)'));
 };
-exports.canGenerateNewAssignments = canGenerateNewAssignments;
 /**
  * Returns an object that includes keys of all experimentIds a user should not be assigned to
  * based on a combination of existing assignments and mutual exclusion configuration
  */
-const getUserExclusionSet = (experiments, existingAssignments) => {
+export const getUserExclusionSet = (experiments, existingAssignments) => {
     const exclusionSet = new Set();
     existingAssignments.forEach(assignment => {
         // Users who are excluded from an experiment due to sampling
@@ -143,4 +136,3 @@ const getUserExclusionSet = (experiments, existingAssignments) => {
     });
     return exclusionSet;
 };
-exports.getUserExclusionSet = getUserExclusionSet;

package/package.json

@@ -1,6 +1,7 @@
 {
   "name": "@naturalcycles/abba",
-  "version": "1.27.0",
+  "type": "module",
+  "version": "2.0.0",
   "scripts": {
     "prepare": "husky",
     "build": "dev-lib build",
@@ -10,16 +11,18 @@
     "lbt": "dev-lib lbt"
   },
   "dependencies": {
-    "@naturalcycles/db-lib": "^9.14.1",
-    "@naturalcycles/js-lib": "^14.237.0",
-    "@naturalcycles/nodejs-lib": "^13.1.3",
-    "semver": "^7.3.5"
+    "@naturalcycles/db-lib": "^10",
+    "@naturalcycles/js-lib": "^14",
+    "@naturalcycles/nodejs-lib": "^13",
+    "semver": "^7"
   },
   "devDependencies": {
-    "@naturalcycles/dev-lib": "^15.19.0",
-    "@types/node": "^22.7.4",
-    "@types/semver": "^7.3.9",
-    "jest": "^29.3.1"
+    "@naturalcycles/dev-lib": "^17",
+    "@types/node": "^22",
+    "@types/semver": "^7",
+    "@vitest/coverage-v8": "^3",
+    "tsx": "^4",
+    "vitest": "^3"
   },
   "files": [
     "dist",

package/src/abba.ts

@@ -1,26 +1,27 @@
-import { _assert, _Memo, _shuffle, localTime, pMap, Unsaved } from '@naturalcycles/js-lib'
+import type { Unsaved } from '@naturalcycles/js-lib'
+import { _assert, _Memo, _shuffle, localTime, pMap } from '@naturalcycles/js-lib'
 import { LRUMemoCache } from '@naturalcycles/nodejs-lib'
-import { SegmentationData } from '.'
-import { bucketDao } from './dao/bucket.dao'
-import { experimentDao } from './dao/experiment.dao'
-import { userAssignmentDao } from './dao/userAssignment.dao'
-import {
+import { bucketDao } from './dao/bucket.dao.js'
+import { experimentDao } from './dao/experiment.dao.js'
+import { userAssignmentDao } from './dao/userAssignment.dao.js'
+import type {
   AbbaConfig,
-  AssignmentStatus,
   Bucket,
   BucketAssignmentStatistics,
   Experiment,
   ExperimentAssignmentStatistics,
   ExperimentWithBuckets,
   GeneratedUserAssignment,
+  SegmentationData,
   UserAssignment,
-} from './types'
+} from './types.js'
+import { AssignmentStatus } from './types.js'
 import {
   canGenerateNewAssignments,
   generateUserAssignmentData,
   getUserExclusionSet,
   validateTotalBucketRatio,
-} from './util'
+} from './util.js'
 
 /**
  * 10 minutes

package/src/dao/bucket.dao.ts

@@ -1,6 +1,7 @@
-import { CommonDao, CommonDB } from '@naturalcycles/db-lib'
-import { Saved } from '@naturalcycles/js-lib'
-import { BaseBucket, Bucket } from '../types'
+import type { CommonDB } from '@naturalcycles/db-lib'
+import { CommonDao } from '@naturalcycles/db-lib'
+import type { Saved } from '@naturalcycles/js-lib'
+import type { BaseBucket, Bucket } from '../types.js'
 
 type BucketDBM = Saved<BaseBucket> & {
   data: string | null

package/src/dao/experiment.dao.ts

@@ -1,6 +1,8 @@
-import { CommonDao, CommonDB } from '@naturalcycles/db-lib'
-import { IsoDate, localDate, Saved } from '@naturalcycles/js-lib'
-import { BaseExperiment, Experiment } from '../types'
+import type { CommonDB } from '@naturalcycles/db-lib'
+import { CommonDao } from '@naturalcycles/db-lib'
+import type { IsoDate, Saved } from '@naturalcycles/js-lib'
+import { localDate } from '@naturalcycles/js-lib'
+import type { BaseExperiment, Experiment } from '../types.js'
 
 type ExperimentDBM = Saved<BaseExperiment> & {
   rules: string | null

package/src/dao/userAssignment.dao.ts

@@ -1,5 +1,6 @@
-import { CommonDao, CommonDB } from '@naturalcycles/db-lib'
-import { UserAssignment } from '../types'
+import type { CommonDB } from '@naturalcycles/db-lib'
+import { CommonDao } from '@naturalcycles/db-lib'
+import type { UserAssignment } from '../types.js'
 
 export class UserAssignmentDao extends CommonDao<UserAssignment> {}
 

package/src/index.ts

@@ -1,2 +1,2 @@
-export * from './abba'
-export * from './types'
+export * from './abba.js'
+export * from './types.js'

package/src/types.ts

@@ -1,5 +1,5 @@
-import { CommonDB } from '@naturalcycles/db-lib'
-import { AnyObject, BaseDBEntity, IsoDate, Saved } from '@naturalcycles/js-lib'
+import type { CommonDB } from '@naturalcycles/db-lib'
+import type { AnyObject, BaseDBEntity, IsoDate, Saved } from '@naturalcycles/js-lib'
 
 export interface AbbaConfig {
   db: CommonDB

package/src/util.ts

@@ -1,7 +1,7 @@
-import { localDate, Unsaved } from '@naturalcycles/js-lib'
+import type { Unsaved } from '@naturalcycles/js-lib'
+import { localDate } from '@naturalcycles/js-lib'
 import { satisfies } from 'semver'
-import {
-  AssignmentStatus,
+import type {
   Bucket,
   ExclusionSet,
   Experiment,
@@ -9,9 +9,9 @@ import {
   SegmentationData,
   SegmentationRule,
   SegmentationRuleFn,
-  SegmentationRuleOperator,
   UserAssignment,
-} from './types'
+} from './types.js'
+import { AssignmentStatus, SegmentationRuleOperator } from './types.js'
 
 /**
  * Generate a new assignment for a given user.
@@ -34,19 +34,23 @@ export const generateUserAssignmentData = (
   } as UserAssignment
 }
 
-/**
- * Generate a random number between 0 and 100
- */
-export const rollDie = (): number => {
-  return Math.random() * 100
+class RandomService {
+  /**
+   * Generate a random number between 0 and 100
+   */
+  rollDie(): number {
+    return Math.random() * 100
+  }
 }
 
+export const randomService = new RandomService()
+
 /**
  * Determines a users assignment for this experiment. Returns null if they are not considered to be in the sampling group
  */
 export const determineAssignment = (sampling: number, buckets: Bucket[]): Bucket | null => {
   // Should this person be considered for the experiment?
-  if (rollDie() > sampling) {
+  if (randomService.rollDie() > sampling) {
     return null
   }
 
@@ -58,7 +62,7 @@ export const determineAssignment = (sampling: number, buckets: Bucket[]): Bucket
  * Determines which bucket a user assignment will recieve
  */
 export const determineBucket = (buckets: Bucket[]): Bucket => {
-  const bucketRoll = rollDie()
+  const bucketRoll = randomService.rollDie()
   let range: [number, number] | undefined
   const bucket = buckets.find(b => {
     if (!range) {