From 64782096df1baafd4583b1a9cc9134a854de9b1a Mon Sep 17 00:00:00 2001
From: Yanick Champoux <yanick@babyl.ca>
Date: Thu, 25 Aug 2022 14:19:45 -0400
Subject: [PATCH] add actions

---
 src/actions.test.ts | 14 +++++++++++++
 src/actions.ts      | 51 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+)
 create mode 100644 src/actions.test.ts
 create mode 100644 src/actions.ts

diff --git a/src/actions.test.ts b/src/actions.test.ts
new file mode 100644
index 0000000..7651c5a
--- /dev/null
+++ b/src/actions.test.ts
@@ -0,0 +1,14 @@
+import { test, expect } from 'vitest';
+
+import { action } from './actions.js';
+
+test('basic action', () => {
+    const foo = action('foo', (thing: string) => ({ thing }));
+
+    expect(foo('bar')).toEqual({
+        type: 'foo',
+        payload: {
+            thing: 'bar',
+        },
+    });
+});
diff --git a/src/actions.ts b/src/actions.ts
new file mode 100644
index 0000000..c50eba8
--- /dev/null
+++ b/src/actions.ts
@@ -0,0 +1,51 @@
+export type Action<T extends string = string, TPayload = any> = {
+    type: T;
+    meta?: Record<string, unknown>;
+    payload?: TPayload;
+};
+
+export type ActionGenerator<
+    TType extends string = string,
+    TPayloadGen = (...args: any[]) => any,
+> = {
+    type: TType;
+} & (TPayloadGen extends (...args: any) => any
+    ? ReturnType<TPayloadGen> extends void
+        ? (...args: Parameters<TPayloadGen>) => {
+              type: TType;
+          }
+        : (...args: Parameters<TPayloadGen>) => {
+              type: TType;
+              payload: ReturnType<TPayloadGen>;
+          }
+    : (...args: any[]) => { type: TType; payload?: unknown });
+
+export function action<
+    TType extends string,
+    TPayload extends (...args: any) => any,
+>(
+    type: TType,
+    payloadFunction?: TPayload,
+    transformer?: Function,
+): ActionGenerator<TType, TPayload> {
+    let generator: any = function (...payloadArg:any[]) {
+        const result: Action = { type };
+
+        if (payloadFunction) {
+            result.payload = payloadFunction(...payloadArg);
+        } else {
+            if (payloadArg[0] !== undefined) result.payload = payloadArg[0];
+        }
+
+        return result;
+    };
+
+    if (transformer) {
+        const orig = generator;
+        generator = (...args: any) => transformer(orig(...args), args);
+    }
+
+    generator.type = type;
+
+    return generator;
+}