Select current revision in selector if it is a branch

scm-ui/src/repos/containers/BranchSelector.js

@@ -17,6 +17,7 @@ const styles = {
 type Props = {
   branches: Branch[], // TODO: Use generics?
   selected: (branch?: Branch) => void,
+  selectedBranch: string,
 
   // context props
   classes: Object,
@@ -31,6 +32,12 @@ class BranchSelector extends React.Component<Props, State> {
     this.state = {};
   }
 
+  componentDidMount() {
+    this.props.branches
+      .filter(branch => branch.name === this.props.selectedBranch)
+      .forEach(branch => this.setState({ selectedBranch: branch }));
+  }
+
   render() {
     const { branches, classes, t } = this.props;