Annotate createDispatcher and composeStores

Author: acdlite

src/createDispatcher.js

@@ -1,15 +1,25 @@
+/* @flow */
+
 import compose from './utils/composeMiddleware';
 
-export default function createDispatcher(store, middlewares = []) {
-  return function dispatcher(initialState, setState) {
-    let state = setState(store(initialState, {}));
+import { Middleware, Store, Action, State, Dispatcher } from './types';
+
+export default function createDispatcher(
+  store: Store,
+  middlewares: (Middleware[] | (getState: () => State) => Middleware[]) = []
+): Dispatcher {
+  return function dispatcher(
+    initialState: State,
+    setState: (state: State) => State
+  ) {
+    var state: State = setState(store(initialState, {}));
 
-    function dispatch(action) {
+    function dispatch(action: Action): Action {
       state = setState(store(state, action));
       return action;
     }
 
-    function getState() {
+    function getState(): State {
       return state;
     }
 

src/utils/composeStores.js

@@ -1,9 +1,13 @@
+/* @flow */
+
 import mapValues from 'lodash/object/mapValues';
 
-export default function composeStores(stores) {
-  return function Composition(atom = {}, action) {
+import { Store, Action, State } from '../types';
+
+export default function composeStores(stores: Store[]): Store {
+  return function Composition(state: State = {}, action: Action) {
     return mapValues(stores, (store, key) =>
-      store(atom[key], action)
+      store(state[key], action)
     );
   };
 }