feat(time-slots): add generator utilities

src/__tests__/generator.test.ts

@@ -0,0 +1,30 @@
+import moment from "moment";
+
+import { Interval, slots } from "../slots";
+import { take, flatten } from "../generator";
+
+describe("generator", () => {
+  const interval: Interval = { amount: 30, unit: "minutes" };
+  const d1 = moment();
+  const start = d1;
+
+  let gen: Generator<string[], never, unknown>;
+
+  beforeEach(() => {
+    gen = slots(interval)([])(start);
+  });
+
+  describe("take", () => {
+    it("get n element from generator", () => {
+      expect(take(0)(gen)).toHaveLength(0);
+      expect(take(5)(gen)).toHaveLength(5);
+    });
+  });
+
+  describe("flatten", () => {
+    it("flatten elements from generator", () => {
+      expect(flatten(take(0)(gen))).toHaveLength(0);
+      expect(flatten(take(5)(gen))).toHaveLength(5);
+    });
+  });
+});

src/generator.ts

@@ -0,0 +1,10 @@
+import { Slot } from "./slots";
+
+import * as R from "ramda";
+
+const repeat = (n: number) => R.repeat(0, n);
+
+export const take = (n: number) => (iterable: IterableIterator<Slot[]>): string[][] =>
+  R.map(() => iterable.next().value as string[], repeat(n));
+
+export const flatten = R.flatten;