types: Add DistributionBucketState

types/augment/all/defs.json

@@ -601,7 +601,7 @@
         "assigned_bags": "u64"
     },
     "DistributionBucketFamily": {
-        "distribution_buckets": "BTreeSet<DistributionBucketId>"
+        "distribution_buckets": "BTreeMap<DistributionBucketId,{\"accepting_new_bags\":\"bool\"}>"
     },
     "DataObjectIdMap": "BTreeMap<DataObjectId,DataObject>",
     "DistributionBucketIdSet": "BTreeSet<DistributionBucketId>",

types/augment/all/types.ts

@@ -416,7 +416,9 @@ export interface DistributionBucket extends Struct {
 
 /** @name DistributionBucketFamily */
 export interface DistributionBucketFamily extends Struct {
-  readonly distribution_buckets: BTreeSet<DistributionBucketId>;
+  readonly distribution_buckets: BTreeMap<DistributionBucketId, {
+    readonly accepting_new_bags: bool;
+  } & Struct>;
 }
 
 /** @name DistributionBucketFamilyId */

types/src/storage.ts

@@ -220,13 +220,23 @@ export class DistributionBucket
   })
   implements IDistributionBucket {}
 
+export type IDistributionBucketState = {
+  accepting_new_bags: bool
+}
+
+export class DistributionBucketState
+  extends JoyStructDecorated({
+    accepting_new_bags: bool,
+  })
+  implements IDistributionBucketState {}
+
 export type IDistributionBucketFamily = {
-  distribution_buckets: BTreeSet<DistributionBucketId>
+  distribution_buckets: BTreeMap<DistributionBucketId, DistributionBucketState>
 }
 
 export class DistributionBucketFamily
   extends JoyStructDecorated({
-    distribution_buckets: BTreeSet.with(DistributionBucketId),
+    distribution_buckets: BTreeMap.with(DistributionBucketId, DistributionBucketState),
   })
   implements IDistributionBucketFamily {}