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;
 

scm-ui/src/repos/sources/containers/Sources.js

@@ -79,11 +79,12 @@ class Sources extends React.Component<Props> {
   }
 
   renderBranchSelector = () => {
-    const { repository, branches } = this.props;
+    const { repository, branches, revision } = this.props;
     if (repository._links.branches) {
       return (
         <BranchSelector
           branches={branches}
+          selectedBranch={revision}
           selected={(b: Branch) => {
             this.branchSelected(b);
           }}