@naturalcycles/abba 2.9.0 → 2.11.0

package/dist/abba.d.ts

@@ -41,6 +41,7 @@ export declare class Abba {
      * Cold method.
      */
     deleteExperiment(experimentId: string): Promise<void>;
+    getExperimentWithBuckets(experimentId: string): Promise<ExperimentWithBuckets | undefined>;
     /**
      * Get an assignment for a given user. If existingOnly is false, it will attempt to generate a new assignment
      * Cold method.
@@ -64,6 +65,13 @@ export declare class Abba {
      * Hot method.
      */
     generateUserAssignments(userId: string, segmentationData: SegmentationData, existingOnly?: boolean): Promise<DecoratedUserAssignment[]>;
+    /**
+     * Returns all user IDs that are assigned to any of the given buckets in the given experiment.
+     *
+     * @param experimentKey - The `key` field of the experiment (not the DB id).
+     * @param bucketKeys - The `key` fields of the buckets to include.
+     */
+    getUserIdsInBuckets(experimentKey: string, bucketKeys: readonly string[]): Promise<Set<string>>;
     /**
      * Get assignment statistics for an experiment.
      * Cold method.

package/dist/abba.js

@@ -156,6 +156,16 @@ export class Abba {
         await this.experimentDao.deleteById(experimentId);
         await this.updateExclusions(experimentId, []);
     }
+    async getExperimentWithBuckets(experimentId) {
+        const experiment = await this.experimentDao.getById(experimentId);
+        if (!experiment)
+            return;
+        const buckets = await this.bucketDao.getByExperimentId(experiment.id);
+        return {
+            ...experiment,
+            buckets,
+        };
+    }
     /**
      * Get an assignment for a given user. If existingOnly is false, it will attempt to generate a new assignment
      * Cold method.
@@ -281,6 +291,26 @@ export class Abba {
         await this.userAssignmentDao.saveBatch(newAssignments);
         return assignments;
     }
+    /**
+     * Returns all user IDs that are assigned to any of the given buckets in the given experiment.
+     *
+     * @param experimentKey - The `key` field of the experiment (not the DB id).
+     * @param bucketKeys - The `key` fields of the buckets to include.
+     */
+    async getUserIdsInBuckets(experimentKey, bucketKeys) {
+        if (!bucketKeys.length)
+            return new Set();
+        const experiment = await this.experimentDao.getByKey(experimentKey);
+        if (!experiment)
+            return new Set();
+        const bucketKeySet = new Set(bucketKeys);
+        const experimentBuckets = await this.bucketDao.getByExperimentId(experiment.id);
+        const matchedBucketIds = experimentBuckets.filter(b => bucketKeySet.has(b.key)).map(b => b.id);
+        if (!matchedBucketIds.length)
+            return new Set();
+        const userIds = await this.userAssignmentDao.getUserIdsByBucketIds(matchedBucketIds);
+        return new Set(userIds);
+    }
     /**
      * Get assignment statistics for an experiment.
      * Cold method.

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

@@ -7,5 +7,6 @@ export declare class UserAssignmentDao extends CommonDao<UserAssignment> {
     deleteByExperimentId(experimentId: string): Promise<void>;
     getCountByExperimentId(experimentId: string): Promise<number>;
     getCountByBucketId(bucketId: string): Promise<number>;
+    getUserIdsByBucketIds(bucketIds: string[]): Promise<string[]>;
 }
 export declare function userAssignmentDao(db: CommonDB): UserAssignmentDao;

package/dist/dao/userAssignment.dao.js

@@ -1,4 +1,5 @@
 import { CommonDao } from '@naturalcycles/db-lib/dao';
+import { _uniq } from '@naturalcycles/js-lib/array';
 export class UserAssignmentDao extends CommonDao {
     async getUserAssignmentByExperimentId(userId, experimentId) {
         const query = this.query().filterEq('userId', userId).filterEq('experimentId', experimentId);
@@ -18,6 +19,16 @@ export class UserAssignmentDao extends CommonDao {
     async getCountByBucketId(bucketId) {
         return await this.query().filterEq('bucketId', bucketId).runQueryCount();
     }
+    async getUserIdsByBucketIds(bucketIds) {
+        if (!bucketIds.length)
+            return [];
+        const userIds = await this.query()
+            .filterIn('bucketId', bucketIds)
+            .select(['userId'])
+            .distinct()
+            .runQuerySingleColumn();
+        return _uniq(userIds);
+    }
 }
 export function userAssignmentDao(db) {
     return new UserAssignmentDao({

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@naturalcycles/abba",
   "type": "module",
-  "version": "2.9.0",
+  "version": "2.11.0",
   "dependencies": {
     "@naturalcycles/db-lib": "^10",
     "@naturalcycles/js-lib": "^15",
@@ -11,7 +11,7 @@
   },
   "devDependencies": {
     "@types/semver": "^7",
-    "@typescript/native-preview": "7.0.0-dev.20260401.1",
+    "@typescript/native-preview": "beta",
     "@naturalcycles/dev-lib": "18.4.2"
   },
   "exports": {

package/src/abba.ts

@@ -242,6 +242,17 @@ export class Abba {
     await this.updateExclusions(experimentId, [])
   }
 
+  async getExperimentWithBuckets(experimentId: string): Promise<ExperimentWithBuckets | undefined> {
+    const experiment = await this.experimentDao.getById(experimentId)
+    if (!experiment) return
+
+    const buckets = await this.bucketDao.getByExperimentId(experiment.id)
+    return {
+      ...experiment,
+      buckets,
+    }
+  }
+
   /**
    * Get an assignment for a given user. If existingOnly is false, it will attempt to generate a new assignment
    * Cold method.
@@ -400,6 +411,31 @@ export class Abba {
     return assignments
   }
 
+  /**
+   * Returns all user IDs that are assigned to any of the given buckets in the given experiment.
+   *
+   * @param experimentKey - The `key` field of the experiment (not the DB id).
+   * @param bucketKeys - The `key` fields of the buckets to include.
+   */
+  async getUserIdsInBuckets(
+    experimentKey: string,
+    bucketKeys: readonly string[],
+  ): Promise<Set<string>> {
+    if (!bucketKeys.length) return new Set()
+
+    const experiment = await this.experimentDao.getByKey(experimentKey)
+    if (!experiment) return new Set()
+
+    const bucketKeySet = new Set(bucketKeys)
+    const experimentBuckets = await this.bucketDao.getByExperimentId(experiment.id)
+    const matchedBucketIds = experimentBuckets.filter(b => bucketKeySet.has(b.key)).map(b => b.id)
+
+    if (!matchedBucketIds.length) return new Set()
+
+    const userIds = await this.userAssignmentDao.getUserIdsByBucketIds(matchedBucketIds)
+    return new Set(userIds)
+  }
+
   /**
    * Get assignment statistics for an experiment.
    * Cold method.

package/src/dao/userAssignment.dao.ts

@@ -1,5 +1,6 @@
 import type { CommonDB } from '@naturalcycles/db-lib'
 import { CommonDao } from '@naturalcycles/db-lib/dao'
+import { _uniq } from '@naturalcycles/js-lib/array'
 import type { UserAssignment } from '../types.js'
 
 export class UserAssignmentDao extends CommonDao<UserAssignment> {
@@ -31,6 +32,18 @@ export class UserAssignmentDao extends CommonDao<UserAssignment> {
   async getCountByBucketId(bucketId: string): Promise<number> {
     return await this.query().filterEq('bucketId', bucketId).runQueryCount()
   }
+
+  async getUserIdsByBucketIds(bucketIds: string[]): Promise<string[]> {
+    if (!bucketIds.length) return []
+
+    const userIds = await this.query()
+      .filterIn('bucketId', bucketIds)
+      .select(['userId'])
+      .distinct()
+      .runQuerySingleColumn<string>()
+
+    return _uniq(userIds)
+  }
 }
 
 export function userAssignmentDao(db: CommonDB): UserAssignmentDao {