From 330ae2c1e24367bd951b2d82436cb4dd84263364 Mon Sep 17 00:00:00 2001
From: Boris Cherny <boris@performancejs.com>
Date: Sun, 22 Apr 2018 22:54:59 -0700
Subject: [PATCH] add getState API

---
 src/index.js.flow  |  3 +++
 src/index.ts       |  7 +++++++
 test/stateless.tsx | 28 ++++++++++++++++++++++++++++
 3 files changed, 38 insertions(+)

diff --git a/src/index.js.flow b/src/index.js.flow
index f873850..c429fca 100644
--- a/src/index.js.flow
+++ b/src/index.js.flow
@@ -16,6 +16,7 @@ export interface Store<Actions: {}> {
   onAll<K: $Keys<Actions>>(): Observable<$Values<Undux<Actions>>>;
   before<K: $Keys<Actions>>(key: K): Observable<$ElementType<Actions, K>>;
   beforeAll<K: $Keys<Actions>>(): Observable<$Values<Undux<Actions>>>;
+  getState(): $ReadOnly<Actions>;
 }
 
 declare export class StoreSnapshot<Actions: {}> implements Store<Actions> {
@@ -25,6 +26,7 @@ declare export class StoreSnapshot<Actions: {}> implements Store<Actions> {
   onAll<K: $Keys<Actions>>(): Observable<$Values<Undux<Actions>>>;
   before<K: $Keys<Actions>>(key: K): Observable<$ElementType<Actions, K>>;
   beforeAll<K: $Keys<Actions>>(): Observable<$Values<Undux<Actions>>>;
+  getState(): $ReadOnly<Actions>;
 }
 
 declare export class StoreDefinition<Actions: {}> implements Store<Actions> {
@@ -34,6 +36,7 @@ declare export class StoreDefinition<Actions: {}> implements Store<Actions> {
   onAll<K: $Keys<Actions>>(): Observable<$Values<Undux<Actions>>>;
   before<K: $Keys<Actions>>(key: K): Observable<$ElementType<Actions, K>>;
   beforeAll<K: $Keys<Actions>>(): Observable<$Values<Undux<Actions>>>;
+  getState(): $ReadOnly<Actions>;
 }
 
 declare export function createStore<Actions: Object>(initialState: Actions): StoreDefinition<Actions>
diff --git a/src/index.ts b/src/index.ts
index 48d8e28..04c3945 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -16,6 +16,7 @@ export interface Store<Actions extends object> {
   onAll<K extends keyof Actions>(): RxJS.Observable<Undux<Actions>[keyof Actions]>
   before<K extends keyof Actions>(key: K): RxJS.Observable<Undux<Actions>[K]>
   beforeAll<K extends keyof Actions>(): RxJS.Observable<Undux<Actions>[keyof Actions]>
+  getState(): Readonly<Actions>
 }
 
 export class StoreSnapshot<Actions extends object> implements Store<Actions> {
@@ -41,6 +42,9 @@ export class StoreSnapshot<Actions extends object> implements Store<Actions> {
   beforeAll<K extends keyof Actions>() {
     return this.store.beforeAll()
   }
+  getState() {
+    return Object.freeze(Object.assign({}, this.state))
+  }
 
   private assign<Actions extends object, K extends keyof Actions>(
     key: K, value: Actions[K]) {
@@ -80,6 +84,9 @@ export class StoreDefinition<Actions extends object> implements Store<Actions> {
       this.alls.emit(key, { key, previousValue, value })
     }
   }
+  getState() {
+    return this.store.getState()
+  }
 }
 
 export function createStore<Actions extends object>(
diff --git a/test/stateless.tsx b/test/stateless.tsx
index 8ea64aa..ae085eb 100644
--- a/test/stateless.tsx
+++ b/test/stateless.tsx
@@ -168,3 +168,31 @@ test('[stateless] it should typecheck with additional props', t => {
 
   t.pass()
 })
+
+test('#getState should return up to date state', t => {
+  let A = connect(store)(({ store }) =>
+    <div>
+      {store.get('isTrue') ? 'True' : 'False'}
+      <button onClick={() => store.set('isTrue')(!store.get('isTrue'))}>Update</button>
+    </div>
+  )
+
+  withElement(A, _ => {
+    t.deepEqual(store.getState(), { isTrue: true, users: [] })
+    Simulate.click(_.querySelector('button')!)
+    t.deepEqual(store.getState(), { isTrue: false, users: [] })
+    Simulate.click(_.querySelector('button')!)
+    t.deepEqual(store.getState(), { isTrue: true, users: [] })
+    Simulate.click(_.querySelector('button')!)
+    t.deepEqual(store.getState(), { isTrue: false, users: [] })
+  })
+})
+
+test('#getState should not be writeable', t => {
+  let A = connect(store)(({ store }) =>
+    <div />
+  )
+  withElement(A, _ =>
+    t.throws(() => (store.getState() as any).isTrue = false)
+  )
+})