All files / src/store Store.ts

100% Statements 79/79
100% Branches 6/6
100% Functions 44/44
100% Lines 72/72
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321  1x 1x 1x 1x 1x   1x                                                                                                                                                                                                 1x 13x         10x 10x 34x 34x 37x   34x         10x 10x 10x 10x 1x   10x 10x                   13x 13x 13x 13x         26x 26x       13x 14x 14x 14x 13x           1x       2x       9x       23x       18x       1x       1x           13x         2x       9x       13x       2x       13x         4x       1x       1x       4x       2x       4x             13x 13x         13x         18x 18x       8x       7x       9x       4x         1x       1x             13x 13x         14x 14x 14x 14x 13x   14x             10x         11x                
import {Observable} from "rxjs/Observable";
import {Subject} from "rxjs/Subject";
import {ReplaySubject} from "rxjs/ReplaySubject";
import 'rxjs/add/operator/map';
import 'rxjs/add/operator/filter';
import 'rxjs/add/operator/distinct';
 
export namespace StoRx {
 
    /**
     *
     */
    export interface Store<S> {
 
        /**
         * create a view on the store pointing of a substate extracted by the @param mapFunction function
         * @param {(s: S) => T} map function to extract the substate the view will be based on
         * @param {(state: T, parentState: S) => S} mapReverse mapFunction function to reinject the substate in the parent state, this is used when the substate is updated
         * @returns {Store<T>} a view store on the substate
         */
        mapFunction<T>(map: (s: S) => T, mapReverse: (state: T, parentState: S) => S): Store<T>;
 
        /**
         * create a view on the substate designed by the path
         * @param {string} path to the substate
         * @returns {Store<T>} a view store on 'path' substate
         */
        map<T>(path: string): Store<T>
 
        /**
         * @returns {Observable<S>} an observable with the store state will be trigger when the state is updated
         */
        observable(): Observable<S>;
 
        /**
         * trigger an action on the store that may then trigger reducer, the action will propagate to parent store
         * @param {A} action the action to trigger
         */
        dispatch<A>(action: A);
 
        /**
         * @returns {ActionManager<S, A>} an action manager to be used to bind reducer to dispatched action
         */
        action<A>(): ActionManager<S, A>;
 
        /**
         *
         * @param {{new() => A}} type type of action manage
         * @returns {ActionManager<S, A>} an action manager working on 'type' action
         */
        actionByType<A>(type: new () => A): ActionManager<S, A>;
 
        /**
         * directly subscribe a reducer on an observable
         * @param {Observable<A>} actions the observable the reducer will subscribe
         * @param {(s: S, a: A) => S} reducer function
         */
        subscribe<A>(actions: Observable<A>, reducer: (s: S, a: A) => S): void;
 
        /**
         * directly subscribe a reducer on an observable
         * @param {Observable<A>} actions the observable the reducer will subscribe
         * @param {(s: S, a: A) => S} reducer object
         */
        subscribeReducer<A>(actions: Observable<A>, reducer: Reducer<S, A>): void;
    }
 
    /**
     * an action manager is a class to bind to reducer to store internal action dispatcher
     * it can also be use to monitor emitted action via the observable()
     */
    export interface ActionManager<S, A> {
        /**
         * subscribe a reducer object to the action this manager handle
         * @param {Reducer<S, A>} reducer
         */
        subscribeReducer(reducer: Reducer<S, A>): void;
 
        /**
         * subscribe a reducer function to the action this manager handle
         * @param {Reducer<S, A>} reducer
         */
        subscribe(reducer: (s: S, a: A) => S): void;
 
        /**
         *
         * @returns {Observable<A>} the observable of the actions managed
         */
        observable(): Observable<A>;
 
        /**
         * build a new actionManager that will handle the subset of action that pass the filter function
         * @param {(a: A) => boolean} filter function to filter actions
         * @returns {ActionManager<S, A>} the new action manage that handle the filtered actions
         */
        filter(filter: (a: A) => boolean): ActionManager<S, A>;
 
        /**
         *
         * @returns {Store<S>} the store associated with this action manager
         */
        getStore(): Store<S>;
    }
 
    export function  createStore<S>(state: S): Store<S> {
        return new StoreImpl<S>(state);
    }
 
    class StoreUtils {
        public static createMapFunction<S, T>(path: string): (s: S) => T {
            let pathSplit: string[] = path.split(".");
            return s => {
                let current: any = s;
                for (let el of pathSplit) {
                    current = current[el];
                }
                return current;
            };
        }
 
        public static createMapReverseFunction<S, T>(path: string): (state: T, parentState: S) => S {
            let pathSplit: string[] = path.split(".");
            return (s, p) => {
                let current: any = p;
                for (let i: number = 0; i < pathSplit.length - 1; i++) {
                    current = current[pathSplit[i]];
                }
                current[pathSplit[pathSplit.length - 1]] = s;
                return p;
            };
        }
    }
 
    class StoreImpl<S> implements Store<S> {
        private stateSubject: Subject<S>;
        private currentState: S;
        private actionSubject: Subject<any>;
 
        constructor(private state: S) {
            this.stateSubject = new ReplaySubject(1);
            this.actionSubject = new Subject();
            this.updateState(state);
        }
 
 
        protected updateState(state: S) {
            this.currentState = state;
            this.stateSubject.next(state);
        }
 
        subscribeReducer<A>(actions: Observable<A>, reducer: Reducer<S, A>): void {
            actions.subscribe(a => {
                let state: S = this.currentState;
                let newState: S = reducer.reduce(state, a);
                if (state != newState) {
                    this.updateState(newState);
                }
            })
        }
 
        subscribe<A>(actions: Observable<A>, reducer: (s: S, a: A) => S): void {
            this.subscribeReducer(actions, new ReducerFunction(reducer));
        }
 
        mapFunction<T>(map: (s: S) => T, mapReverse: (state: T, parentState: S) => S): Store<T> {
            return new ViewStore(map, mapReverse, this);
        }
 
        map<T>(path: string): Store<T> {
            return new ViewStore<T, S>(<(s: S) => T>StoreUtils.createMapFunction(path), <(state: T, parentState: S) => S>StoreUtils.createMapReverseFunction(path), this);
        }
 
        public observable(): Observable<S> {
            return this.stateSubject;
        }
 
        dispatch<A>(action: A) {
            this.actionSubject.next(action);
        }
 
        action<A>(): ActionManager<S, A> {
            return new ActionManagerImpl(this, this.actionSubject);
        }
 
        actionByType<A>(type: new () => A): ActionManager<S, A> {
            return new ActionManagerImpl(this, this.actionSubject.asObservable().filter(v => v instanceof type));
        }
    }
 
    class ActionManagerImpl<S, A>  implements ActionManager<S, A> {
 
        constructor(private store: Store<S>, private obs: Observable<A>) {
        }
 
 
        subscribeReducer(reducer: Reducer<S, A>): void {
            this.getStore().subscribeReducer(this.observable(), reducer);
        }
 
        subscribe(reducer: (s: S, a: A) => S): void {
            this.getStore().subscribe(this.observable(), reducer);
        }
 
        observable(): Observable<A> {
            return this.obs;
        }
 
        filter(filter: (a: A) => boolean): ActionManager<S, A> {
            return new FilterActionManager(this, filter);
        }
 
        getStore(): Store<S> {
            return this.store;
        }
    }
 
    class FilterActionManager<S, A> implements ActionManager<S, A> {
        constructor(private parent: ActionManager<S, A>, private filterFunction: (a: A) => boolean) {
        }
 
        subscribeReducer(reducer: Reducer<S, A>): void {
            this.getStore().subscribeReducer(this.observable(), reducer);
        }
 
        subscribe(reducer: (s: S, a: A) => S): void {
            this.getStore().subscribe(this.observable(), reducer);
        }
 
        observable(): Observable<A> {
            return this.parent.observable().filter(this.filterFunction);
        }
 
        filter(filter: (a: A) => boolean): ActionManager<S, A> {
            return new FilterActionManager(this, filter);
        }
 
        getStore(): Store<S> {
            return this.parent.getStore();
        }
    }
 
    class ViewStore<S, P> implements Store<S> {
        private actionSubject: Subject<any>;
 
        constructor(private mapFunc: (s: P) => S, private mapReverseFunction: (state: S, parentState: P) => P, private store: Store<P>) {
            this.actionSubject = new Subject();
        }
 
 
        public observable(): Observable<S> {
            return this.store.observable().map(this.mapFunc).distinct();
        }
 
 
        dispatch<A>(action: A) {
            this.store.dispatch(action);
            this.actionSubject.next(action);
        }
 
        action<A>(): ActionManager<S, A> {
            return new ActionManagerImpl(this, this.actionSubject);
        }
 
        actionByType<A>(type: new () => A): ActionManager<S, A> {
            return new ActionManagerImpl(this, this.actionSubject.asObservable().filter(v => v instanceof type));
        }
 
        subscribe<A>(actions: Observable<A>, reducer: (s: S, a: A) => S): void {
            this.store.subscribeReducer(actions, new ViewReducer(this.mapFunc, this.mapReverseFunction, new ReducerFunction(reducer)));
        }
 
        subscribeReducer<A>(actions: Observable<A>, reducer: Reducer<S, A>): void {
            this.store.subscribeReducer(actions, new ViewReducer(this.mapFunc, this.mapReverseFunction, reducer));
        }
 
 
        public mapFunction<T>(map: (s: S) => T, mapReverse: (state: T, parentState: S) => S): Store<T> {
            return new ViewStore(map, mapReverse, this);
        }
 
        public map<T>(path: string): Store<T> {
            return new ViewStore(StoreUtils.createMapFunction(path), StoreUtils.createMapReverseFunction(path), this);
        }
 
    }
 
    class ViewReducer<P, S, A> implements Reducer<P, A> {
 
        constructor(private map: (s: P) => S, private mapReverse: (state: S, parentState: P) => P,
                    private reducer: Reducer<S, A>) {
        }
 
 
        reduce(parentState: P, a: A): P {
            let state: S = this.map(parentState);
            let reduceState: S = this.reducer.reduce(state, a);
            let result: P = parentState;
            if (state != reduceState) {
                result = this.mapReverse(reduceState, Object.assign({}, parentState));
            }
            return result;
        }
    }
 
 
    class ReducerFunction<S, A> implements Reducer<S, A> {
 
        constructor(private reducer: (s: S, a: A) => S) {
 
        }
 
        reduce(state: S, a: A): S {
            return this.reducer(state, a);
        }
    }
 
    export interface Reducer<S, A> {
        reduce(state: S, a: A): S;
    }
 
}