From 80416fce75ef35adf398d33388a23db81d742576 Mon Sep 17 00:00:00 2001
From: rokashkovvd <rokashkovvd@gmail.com>
Date: Fri, 11 Oct 2024 00:37:22 +0700
Subject: [PATCH] feat: add skip & take

---
 src/skip/index.ts | 55 +++++++++++++++++++++++++++++++++++++++++++++++
 src/take/index.ts | 52 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 107 insertions(+)
 create mode 100644 src/skip/index.ts
 create mode 100644 src/take/index.ts

diff --git a/src/skip/index.ts b/src/skip/index.ts
new file mode 100644
index 00000000..9471d9d2
--- /dev/null
+++ b/src/skip/index.ts
@@ -0,0 +1,55 @@
+import {
+  Event,
+  Store,
+  Unit,
+  combine,
+  createEvent,
+  createStore,
+  is,
+  sample,
+} from 'effector';
+
+export function skip<T>({
+  clock,
+  count,
+  reset,
+}: {
+  clock?: Unit<T>;
+  count: Store<number> | number;
+  reset?: Unit<T>;
+}): Event<T> {
+  const $count = is.store(count) ? count : createStore(count);
+
+  const $skipped = createStore(0);
+
+  const $canTrigger = combine(
+    [$skipped, $count],
+    ([skipped, count]) => skipped >= count,
+  );
+
+  clock = is.unit(clock) ? clock : createEvent();
+
+  reset = is.unit(reset) ? reset : createEvent();
+
+  const event = sample({
+    clock,
+    source: $skipped,
+    filter: $canTrigger,
+    fn: (_, params) => params,
+  });
+
+  sample({
+    clock,
+    source: $skipped,
+    filter: $canTrigger.map((canTrigger) => !canTrigger),
+    fn: (skipped) => skipped + 1,
+    target: $skipped,
+  });
+
+  sample({
+    clock: [$count, reset],
+    target: $skipped.reinit,
+  });
+
+  return event;
+}
diff --git a/src/take/index.ts b/src/take/index.ts
new file mode 100644
index 00000000..e3a6b0e6
--- /dev/null
+++ b/src/take/index.ts
@@ -0,0 +1,52 @@
+import {
+  Event,
+  Store,
+  Unit,
+  combine,
+  createEvent,
+  createStore,
+  is,
+  sample,
+} from 'effector';
+
+export function skip<T>({
+  clock,
+  count,
+  reset,
+}: {
+  clock: Unit<T>;
+  count: Store<number> | number;
+  reset?: Unit<T>;
+}): Event<T> {
+  const $count = is.store(count) ? count : createStore(count);
+
+  const $taken = createStore(0);
+
+  const $canTrigger = combine([$taken, $count], ([taken, count]) => taken >= count);
+
+  clock = is.unit(clock) ? clock : createEvent();
+
+  reset = is.unit(reset) ? reset : createEvent();
+
+  const event = sample({
+    clock,
+    source: $taken,
+    filter: $canTrigger,
+    fn: (_, params) => params,
+  });
+
+  sample({
+    clock,
+    source: $taken,
+    filter: $canTrigger.map((canTrigger) => !canTrigger),
+    fn: (taken) => taken + 1,
+    target: $taken,
+  });
+
+  sample({
+    clock: [$count, reset],
+    target: $taken.reinit,
+  });
+
+  return event;
+}