Fix usesetstate

src/__stories__/useSetState.story.tsx

@@ -11,14 +11,14 @@ const Demo = () => {
       <pre>{JSON.stringify(state, null, 2)}</pre>
       <button onClick={() => setState({hello: 'world'})}>hello</button>
       <button onClick={() => setState({foo: 'bar'})}>foo</button>
-      <button 
+      <button
         onClick={() => {
           setState((prevState) => ({
             count: prevState.count === undefined ? 0 : prevState.count + 1,
           }))
         }}
       >
-        count
+        increment
       </button>
     </div>
   );

src/useOrientation.ts

@@ -26,7 +26,7 @@ const useOrientation = (initialState: OrientationState = defaultState) => {
           setState({angle, type});
         } else if (window.orientation) {
           setState({
-            angle: window.orientation,
+            angle: typeof window.orientation === 'number' ? window.orientation : 0,
             type: ''
           });
         } else {

src/useSetState.ts

@@ -2,15 +2,9 @@ import {useState} from 'react';
 
 const useSetState = <T extends object>(initialState: T = {} as T): [T, (patch: Partial<T> | Function) => void]=> {
   const [state, set] = useState<T>(initialState);
-  const setState = (patch) => {
-    if (patch instanceof Function) {
-      set((prevState) => {
-          return Object.assign(state, patch(prevState))
-      })
-    } else {
-      Object.assign(state, patch);
-      set(state);
-    }
+  const setState = patch => {
+    if (patch instanceof Function) set(prevState => Object.assign({}, prevState, patch(prevState)));
+    else set(Object.assign({}, state, patch));
   };
 
   return [state, setState];