fix: resolve state as primitive types

Author: pedronauck

src/reworm.tsx

@@ -4,6 +4,8 @@ import makeSubject from 'callbag-subject'
 import equal from 'fast-deep-equal'
 import merge from 'deepmerge'
 
+const isPrimitive = (test: any) => test !== Object(test)
+
 type PrevState<T> = (prevState: T) => T
 type GetFn<T> = (state: T) => React.ReactNode
 
@@ -35,10 +37,11 @@ export function create<T = any>(initial: T = {} as T): State<T> {
       return this.props.children(STATE)
     }
     private update = (next: T): void => {
-      const newState = merge(STATE, next)
+      const newState = !isPrimitive(next) ? merge(STATE, next) : next
+      const isEqual = !isPrimitive ? equal(STATE, newState) : STATE === newState
 
-      if (!equal(STATE, newState)) STATE = newState
-      this.setState({})
+      if (!isEqual) STATE = newState
+      this.forceUpdate()
     }
   }
 

test/reworm.test.tsx

@@ -81,4 +81,28 @@ describe('State', () => {
       '<div><div>Michael</div><input value="Michael"></div>'
     )
   })
+
+  it('should modify state with primitive types', () => {
+    const initial = 'John'
+    const user = create(initial)
+
+    const Users = () => <div>{user.get(val => val)}</div>
+    const App = () => (
+      <div>
+        <Users />
+        {user.get(val => (
+          <input value={val} onChange={ev => user.set(ev.target.value)} />
+        ))}
+      </div>
+    )
+
+    const result = mount(<App />)
+    const input = result.find('input')
+
+    input.simulate('change', { target: { value: 'Michael' } })
+
+    expect(result.html()).toEqual(
+      '<div><div>Michael</div><input value="Michael"></div>'
+    )
+  })
 })