updateBranchesState should use Pick<K> to be consistent with other me…

…thods
auntnacub · Oct 15, 2017 · 62121ba · 62121ba
1 parent b5e73f8
commit 62121ba
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/app/src/lib/stores/app-store.ts b/app/src/lib/stores/app-store.ts
@@ -445,13 +445,14 @@ export class AppStore {
     })
   }
 
-  private updateBranchesState(
+  private updateBranchesState<K extends keyof IBranchesState>(
     repository: Repository,
-    fn: (branchesState: IBranchesState) => IBranchesState
+    fn: (branchesState: IBranchesState) => Pick<IBranchesState, K>
   ) {
     this.updateRepositoryState(repository, state => {
-      const branchesState = fn(state.branchesState)
-      return { branchesState }
+      const changesState = state.branchesState
+      const newState = merge(changesState, fn(changesState))
+      return { branchesState: newState }
     })
   }