From 24d3c18205fc21d08811e62cf35ca978e6bf8720 Mon Sep 17 00:00:00 2001
From: Jaco Greeff <jacogr@gmail.com>
Date: Tue, 26 Mar 2019 12:51:08 +0100
Subject: [PATCH] Nominate/Stake for controllers (#846)

---
 .../app-staking/src/Account/Nominating.tsx    | 18 ++++-
 packages/app-staking/src/Account/Staking.tsx  | 13 +++-
 packages/app-staking/src/Account/index.tsx    | 65 +++++++++++--------
 3 files changed, 65 insertions(+), 31 deletions(-)

diff --git a/packages/app-staking/src/Account/Nominating.tsx b/packages/app-staking/src/Account/Nominating.tsx
index 703e70db73..ef1fcbb4b4 100644
--- a/packages/app-staking/src/Account/Nominating.tsx
+++ b/packages/app-staking/src/Account/Nominating.tsx
@@ -5,7 +5,8 @@
 import { I18nProps } from '@polkadot/ui-app/types';
 
 import React from 'react';
-import { Button, Input, Modal, TxButton } from '@polkadot/ui-app';
+import { AccountId } from '@polkadot/types';
+import { Button, Input, InputAddress, Modal, TxButton } from '@polkadot/ui-app';
 import keyring from '@polkadot/ui-keyring';
 
 import translate from '../translate';
@@ -15,6 +16,7 @@ type Props = I18nProps & {
   isOpen: boolean,
   onClose: () => void,
   intentions: Array<string>,
+  stashId: AccountId,
   validators: Array<string>
 };
 
@@ -79,7 +81,7 @@ class Nominating extends React.PureComponent<Props, State> {
   }
 
   renderContent () {
-    const { t } = this.props;
+    const { accountId, stashId, t } = this.props;
     const { isNomineeValid, nominees } = this.state;
 
     return (
@@ -88,6 +90,18 @@ class Nominating extends React.PureComponent<Props, State> {
           {t('Nominate Validator')}
         </Modal.Header>
         <Modal.Content className='ui--signer-Signer-Content'>
+          <InputAddress
+            className='medium'
+            isDisabled
+            label={t('account')}
+            value={accountId}
+          />
+          <InputAddress
+            className='medium'
+            isDisabled
+            label={t('stash account')}
+            value={stashId}
+          />
           <Input
             autoFocus
             className='medium'
diff --git a/packages/app-staking/src/Account/Staking.tsx b/packages/app-staking/src/Account/Staking.tsx
index 6f74231cb8..41c75a4112 100644
--- a/packages/app-staking/src/Account/Staking.tsx
+++ b/packages/app-staking/src/Account/Staking.tsx
@@ -6,7 +6,7 @@ import { I18nProps } from '@polkadot/ui-app/types';
 
 import BN from 'bn.js';
 import React from 'react';
-import { ValidatorPrefs } from '@polkadot/types';
+import { AccountId, ValidatorPrefs } from '@polkadot/types';
 import { Button, InputAddress, InputBalance, InputNumber, Modal, TxButton } from '@polkadot/ui-app';
 
 import translate from '../translate';
@@ -15,7 +15,8 @@ type Props = I18nProps & {
   accountId: string,
   isOpen: boolean,
   onClose: () => void,
-  preferences: ValidatorPrefs
+  preferences: ValidatorPrefs,
+  stashId: AccountId
 };
 
 type State = {
@@ -91,7 +92,7 @@ class Staking extends React.PureComponent<Props, State> {
   }
 
   private renderContent () {
-    const { accountId, t } = this.props;
+    const { accountId, stashId, t } = this.props;
     const { unstakeThreshold, validatorPayment } = this.state;
 
     return (
@@ -106,6 +107,12 @@ class Staking extends React.PureComponent<Props, State> {
             label={t('account')}
             value={accountId}
           />
+          <InputAddress
+            className='medium'
+            isDisabled
+            label={t('stash account')}
+            value={stashId}
+          />
           <InputNumber
             autoFocus
             bitLength={32}
diff --git a/packages/app-staking/src/Account/index.tsx b/packages/app-staking/src/Account/index.tsx
index 68bedc85f3..6d30ba840b 100644
--- a/packages/app-staking/src/Account/index.tsx
+++ b/packages/app-staking/src/Account/index.tsx
@@ -7,7 +7,7 @@ import { I18nProps } from '@polkadot/ui-app/types';
 import { ApiProps } from '@polkadot/ui-api/types';
 
 import React from 'react';
-import { AccountId, Balance, Option, ValidatorPrefs } from '@polkadot/types';
+import { AccountId, Balance, Option, StakingLedger, ValidatorPrefs } from '@polkadot/types';
 import { AddressMini, AddressSummary, Button } from '@polkadot/ui-app';
 import { withCalls } from '@polkadot/ui-api';
 
@@ -24,6 +24,7 @@ type Props = ApiProps & I18nProps & {
   name: string,
   session_nextKeyFor?: Option<AccountId>,
   staking_bonded?: Option<AccountId>,
+  staking_ledger?: Option<StakingLedger>,
   staking_nominating?: Option<AccountId>,
   staking_nominatorsFor?: Array<string>,
   staking_validators?: ValidatorPrefs,
@@ -39,7 +40,8 @@ type State = {
   isStakingOpen: boolean,
   bondedId: AccountId | null,
   controllerId: AccountId | null,
-  nomineeId: AccountId | null
+  nomineeId: AccountId | null,
+  stashId: AccountId | null
 };
 
 class Account extends React.PureComponent<Props, State> {
@@ -50,10 +52,11 @@ class Account extends React.PureComponent<Props, State> {
     isStakingOpen: false,
     bondedId: null,
     controllerId: null,
-    nomineeId: null
+    nomineeId: null,
+    stashId: null
   };
 
-  static getDerivedStateFromProps ({ session_nextKeyFor, staking_bonded, staking_nominating }: Props): Partial<State> {
+  static getDerivedStateFromProps ({ session_nextKeyFor, staking_bonded, staking_ledger, staking_nominating }: Props): Partial<State> {
     return {
       bondedId: staking_bonded
         ? staking_bonded.unwrapOr(null)
@@ -63,6 +66,9 @@ class Account extends React.PureComponent<Props, State> {
         : null,
       nomineeId: staking_nominating
         ? staking_nominating.unwrapOr(null)
+        : null,
+      stashId: staking_ledger && staking_ledger.isSome
+        ? staking_ledger.unwrap().stash
         : null
     };
   }
@@ -126,9 +132,9 @@ class Account extends React.PureComponent<Props, State> {
 
   private renderStaking () {
     const { accountId, staking_validators } = this.props;
-    const { isStakingOpen } = this.state;
+    const { isStakingOpen, stashId } = this.state;
 
-    if (!staking_validators || !isStakingOpen) {
+    if (!staking_validators || !isStakingOpen || !stashId) {
       return null;
     }
 
@@ -138,6 +144,7 @@ class Account extends React.PureComponent<Props, State> {
         isOpen
         onClose={this.toggleStaking}
         preferences={staking_validators}
+        stashId={stashId}
       />
     );
   }
@@ -182,7 +189,11 @@ class Account extends React.PureComponent<Props, State> {
 
   private renderNominating () {
     const { accountId, intentions, validators } = this.props;
-    const { isNominateOpen } = this.state;
+    const { isNominateOpen, stashId } = this.state;
+
+    if (!stashId) {
+      return null;
+    }
 
     return (
       <Nominating
@@ -190,6 +201,7 @@ class Account extends React.PureComponent<Props, State> {
         isOpen={isNominateOpen}
         onClose={this.toggleNominate}
         intentions={intentions}
+        stashId={stashId}
         validators={validators}
       />
     );
@@ -197,32 +209,32 @@ class Account extends React.PureComponent<Props, State> {
 
   private renderButtons () {
     const { t } = this.props;
-    const { bondedId, controllerId } = this.state;
-    const buttons = [
-      <Button
-        isPrimary
-        key='controller'
-        onClick={this.toggleController}
-        label={t('Controller')}
-      />
-    ];
+    const { controllerId, stashId } = this.state;
+    const buttons = [];
 
-    // only display bonmding if we already have a controller
-    if (controllerId) {
-      buttons.push(<Button.Or key='bond.or' />);
+    if (!stashId) {
       buttons.push(
         <Button
           isPrimary
-          key='bond'
-          onClick={this.toggleBonding}
-          label={t('Bond')}
+          key='controller'
+          onClick={this.toggleController}
+          label={t('Controller')}
         />
       );
-    }
 
-    // only stake/nominate if we already have bonded
-    if (bondedId) {
-      buttons.push(<Button.Or key='stake.or' />);
+      // only display bonding if we already have a controller
+      if (controllerId) {
+        buttons.push(<Button.Or key='bond.or' />);
+        buttons.push(
+          <Button
+            isPrimary
+            key='bond'
+            onClick={this.toggleBonding}
+            label={t('Bond')}
+          />
+        );
+      }
+    } else {
       buttons.push(
         <Button
           isPrimary
@@ -278,6 +290,7 @@ export default translate(
   withCalls<Props>(
     ['query.session.nextKeyFor', { paramName: 'accountId' }],
     ['query.staking.bonded', { paramName: 'accountId' }],
+    ['query.staking.ledger', { paramName: 'accountId' }],
     ['query.staking.nominatorsFor', { paramName: 'accountId' }],
     ['query.staking.nominating', { paramName: 'accountId' }],
     ['query.staking.validators', { paramName: 'accountId' }]