Move login/logout actions to GithubTabController

Author: smashwilson

lib/controllers/github-tab-controller.js

@@ -2,23 +2,30 @@ import React from 'react';
 import PropTypes from 'prop-types';
 
 import {
-  GithubLoginModelPropType, RefHolderPropType, RemoteSetPropType, BranchSetPropType, RefresherPropType,
+  GithubLoginModelPropType, TokenPropType, RefHolderPropType,
+  RemoteSetPropType, RemotePropType, BranchSetPropType, BranchPropType,
+  RefresherPropType,
 } from '../prop-types';
 import GitHubTabView from '../views/github-tab-view';
+import {incrementCounter} from '../reporter-proxy';
 
 export default class GitHubTabController extends React.Component {
   static propTypes = {
     workspace: PropTypes.object.isRequired,
     refresher: RefresherPropType.isRequired,
     loginModel: GithubLoginModelPropType.isRequired,
+    token: TokenPropType,
     rootHolder: RefHolderPropType.isRequired,
 
     workingDirectory: PropTypes.string,
     repository: PropTypes.object.isRequired,
     allRemotes: RemoteSetPropType.isRequired,
+    githubRemotes: RemoteSetPropType.isRequired,
+    currentRemote: RemotePropType.isRequired,
     branches: BranchSetPropType.isRequired,
-    selectedRemoteName: PropTypes.string,
-    aheadCount: PropTypes.number,
+    currentBranch: BranchPropType.isRequired,
+    aheadCount: PropTypes.number.isRequired,
+    manyRemotesAvailable: PropTypes.bool.isRequired,
     pushInProgress: PropTypes.bool.isRequired,
     isLoading: PropTypes.bool.isRequired,
     currentWorkDir: PropTypes.string,
@@ -35,21 +42,11 @@ export default class GitHubTabController extends React.Component {
   }
 
   render() {
-    const gitHubRemotes = this.props.allRemotes.filter(remote => remote.isGithubRepo());
-    const currentBranch = this.props.branches.getHeadBranch();
-
-    let currentRemote = gitHubRemotes.withName(this.props.selectedRemoteName);
-    let manyRemotesAvailable = false;
-    if (!currentRemote.isPresent() && gitHubRemotes.size() === 1) {
-      currentRemote = Array.from(gitHubRemotes)[0];
-    } else if (!currentRemote.isPresent() && gitHubRemotes.size() > 1) {
-      manyRemotesAvailable = true;
-    }
-
     return (
       <GitHubTabView
         // Connection
-        loginModel={this.props.loginModel}
+        endpoint={this.currentEndpoint()}
+        token={this.props.token}
 
         workspace={this.props.workspace}
         refresher={this.props.refresher}
@@ -59,14 +56,17 @@ export default class GitHubTabController extends React.Component {
         contextLocked={this.props.contextLocked}
         repository={this.props.repository}
         branches={this.props.branches}
-        currentBranch={currentBranch}
-        remotes={gitHubRemotes}
-        currentRemote={currentRemote}
-        manyRemotesAvailable={manyRemotesAvailable}
+        currentBranch={this.props.currentBranch}
+        remotes={this.props.githubRemotes}
+        currentRemote={this.props.currentRemote}
+        manyRemotesAvailable={this.props.manyRemotesAvailable}
         aheadCount={this.props.aheadCount}
         pushInProgress={this.props.pushInProgress}
         isLoading={this.props.isLoading}
 
+        handleLogin={this.handleLogin}
+        handleLogout={this.handleLogout}
+        handleTokenRetry={this.handleTokenRetry}
         handlePushBranch={this.handlePushBranch}
         handleRemoteSelect={this.handleRemoteSelect}
         changeWorkingDirectory={this.props.changeWorkingDirectory}
@@ -94,4 +94,20 @@ export default class GitHubTabController extends React.Component {
   }
 
   openBoundPublishDialog = () => this.props.openPublishDialog(this.props.repository);
+
+  handleLogin = token => {
+    incrementCounter('github-login');
+    this.props.loginModel.setToken(this.currentEndpoint().getLoginAccount(), token);
+  }
+
+  handleLogout = () => {
+    incrementCounter('github-logout');
+    this.props.loginModel.removeToken(this.currentEndpoint().getLoginAccount());
+  }
+
+  handleTokenRetry = () => this.props.loginModel.didUpdate();
+
+  currentEndpoint() {
+    return this.props.currentRemote.getEndpointOrDotcom();
+  }
 }

test/controllers/github-tab-controller.test.js

@@ -6,11 +6,13 @@ import Repository from '../../lib/models/repository';
 import BranchSet from '../../lib/models/branch-set';
 import Branch, {nullBranch} from '../../lib/models/branch';
 import RemoteSet from '../../lib/models/remote-set';
-import Remote from '../../lib/models/remote';
-import {InMemoryStrategy} from '../../lib/shared/keytar-strategy';
+import Remote, {nullRemote} from '../../lib/models/remote';
+import {DOTCOM} from '../../lib/models/endpoint';
+import {InMemoryStrategy, UNAUTHENTICATED} from '../../lib/shared/keytar-strategy';
 import GithubLoginModel from '../../lib/models/github-login-model';
 import RefHolder from '../../lib/models/ref-holder';
 import Refresher from '../../lib/models/refresher';
+import * as reporterProxy from '../../lib/reporter-proxy';
 
 import {buildRepository, cloneRepository} from '../helpers';
 
@@ -34,12 +36,18 @@ describe('GitHubTabController', function() {
         workspace={atomEnv.workspace}
         refresher={new Refresher()}
         loginModel={new GithubLoginModel(InMemoryStrategy)}
+        token="1234"
         rootHolder={new RefHolder()}
 
         workingDirectory={repo.getWorkingDirectoryPath()}
         repository={repo}
         allRemotes={new RemoteSet()}
+        githubRemotes={new RemoteSet()}
+        currentRemote={nullRemote}
         branches={new BranchSet()}
+        currentBranch={nullBranch}
+        aheadCount={0}
+        manyRemotesAvailable={false}
         pushInProgress={false}
         isLoading={false}
         currentWorkDir={repo.getWorkingDirectoryPath()}
@@ -60,48 +68,15 @@ describe('GitHubTabController', function() {
   }
 
   describe('derived view props', function() {
-    const dotcom0 = new Remote('yes0', '[email protected]:aaa/bbb.git');
-    const dotcom1 = new Remote('yes1', 'https://github.com/ccc/ddd.git');
-    const nonDotcom = new Remote('no0', '[email protected]:eee/fff.git');
-
-    it('passes the current branch', function() {
-      const currentBranch = new Branch('aaa', nullBranch, nullBranch, true);
-      const otherBranch = new Branch('bbb');
-      const branches = new BranchSet([currentBranch, otherBranch]);
-      const wrapper = shallow(buildApp({branches}));
-
-      assert.strictEqual(wrapper.find('GitHubTabView').prop('currentBranch'), currentBranch);
-    });
-
-    it('passes remotes hosted on GitHub', function() {
-      const allRemotes = new RemoteSet([dotcom0, dotcom1, nonDotcom]);
-      const wrapper = shallow(buildApp({allRemotes}));
-
-      const passed = wrapper.find('GitHubTabView').prop('remotes');
-      assert.isTrue(passed.withName('yes0').isPresent());
-      assert.isTrue(passed.withName('yes1').isPresent());
-      assert.isFalse(passed.withName('no0').isPresent());
-    });
-
-    it('detects an explicitly specified current remote', function() {
-      const allRemotes = new RemoteSet([dotcom0, dotcom1, nonDotcom]);
-      const wrapper = shallow(buildApp({allRemotes, selectedRemoteName: 'yes1'}));
-      assert.strictEqual(wrapper.find('GitHubTabView').prop('currentRemote'), dotcom1);
-      assert.isFalse(wrapper.find('GitHubTabView').prop('manyRemotesAvailable'));
+    it('passes the endpoint from the current GitHub remote when one exists', function() {
+      const remote = new Remote('hub', '[email protected]:some/repo.git');
+      const wrapper = shallow(buildApp({currentRemote: remote}));
+      assert.strictEqual(wrapper.find('GitHubTabView').prop('endpoint'), remote.getEndpoint());
     });
 
-    it('uses a single GitHub-hosted remote', function() {
-      const allRemotes = new RemoteSet([dotcom0, nonDotcom]);
-      const wrapper = shallow(buildApp({allRemotes}));
-      assert.strictEqual(wrapper.find('GitHubTabView').prop('currentRemote'), dotcom0);
-      assert.isFalse(wrapper.find('GitHubTabView').prop('manyRemotesAvailable'));
-    });
-
-    it('indicates when multiple remotes are available', function() {
-      const allRemotes = new RemoteSet([dotcom0, dotcom1]);
-      const wrapper = shallow(buildApp({allRemotes}));
-      assert.isFalse(wrapper.find('GitHubTabView').prop('currentRemote').isPresent());
-      assert.isTrue(wrapper.find('GitHubTabView').prop('manyRemotesAvailable'));
+    it('defaults the endpoint to dotcom', function() {
+      const wrapper = shallow(buildApp({currentRemote: nullRemote}));
+      assert.strictEqual(wrapper.find('GitHubTabView').prop('endpoint'), DOTCOM);
     });
   });
 
@@ -139,5 +114,26 @@ describe('GitHubTabController', function() {
       wrapper.find('GitHubTabView').prop('openBoundPublishDialog')();
       assert.isTrue(openPublishDialog.calledWith(someRepo));
     });
+
+    it('handles and instruments a login', async function() {
+      sinon.stub(reporterProxy, 'incrementCounter');
+      const loginModel = new GithubLoginModel(InMemoryStrategy);
+
+      const wrapper = shallow(buildApp({loginModel}));
+      await wrapper.find('GitHubTabView').prop('handleLogin')('good-token');
+      assert.strictEqual(await loginModel.getToken(DOTCOM.getLoginAccount()), 'good-token');
+      assert.isTrue(reporterProxy.incrementCounter.calledWith('github-login'));
+    });
+
+    it('handles and instruments a logout', async function() {
+      sinon.stub(reporterProxy, 'incrementCounter');
+      const loginModel = new GithubLoginModel(InMemoryStrategy);
+      await loginModel.setToken(DOTCOM.getLoginAccount(), 'good-token');
+
+      const wrapper = shallow(buildApp({loginModel}));
+      await wrapper.find('GitHubTabView').prop('handleLogout')();
+      assert.strictEqual(await loginModel.getToken(DOTCOM.getLoginAccount()), UNAUTHENTICATED);
+      assert.isTrue(reporterProxy.incrementCounter.calledWith('github-logout'));
+    });
   });
 });