robosats/frontend/src/components/UserGenPage.js

import React, { Component } from "react";
import { withTranslation } from "react-i18next";
import { Button , Tooltip, Grid, Typography, TextField, ButtonGroup, CircularProgress, IconButton} from "@mui/material"
import { Link } from 'react-router-dom'
import Image from 'material-ui-image'
import { InfoDialog } from './Dialogs'

import SmartToyIcon from '@mui/icons-material/SmartToy';
import CasinoIcon from '@mui/icons-material/Casino';
import ContentCopy from "@mui/icons-material/ContentCopy";
import BoltIcon from '@mui/icons-material/Bolt';
import DownloadIcon from '@mui/icons-material/Download';
import { RoboSatsNoTextIcon } from "./Icons";

import { sha256 } from 'js-sha256';
import { genBase62Token, tokenStrength } from "../utils/token";
import { genKey } from "../utils/pgp";
import { getCookie, writeCookie, deleteCookie } from "../utils/cookies";
import { saveAsJson } from "../utils/saveFile";
import { copyToClipboard } from "../utils/clipboard";

class UserGenPage extends Component {
  constructor(props) {
    super(props);
    this.state = {
      openInfo: false,
      tokenHasChanged: false,
      token: ""
    };

    this.refCode = this.props.match.params.refCode;
  }

  componentDidMount() {
    // Checks in parent HomePage if there is already a nick and token
    // Displays the existing one
    if (this.props.nickname != null){
      this.setState({
        nickname: this.props.nickname,
        token: this.props.token? this.props.token : "",
        avatarUrl: '/static/assets/avatars/' + this.props.nickname + '.png',
        loadingRobot: false
      });
    }
    else{
      var newToken = genBase62Token(36)
      this.setState({
        token: newToken
      });
      this.getGeneratedUser(newToken);
    }
  }

  getGeneratedUser=(token)=>{

    const strength = tokenStrength(token);
    const refCode = this.refCode

    const requestOptions = genKey(token).then(function(key) {
      return {
        method: 'POST',
        headers: {'Content-Type':'application/json', 'X-CSRFToken': getCookie('csrftoken')},
        body: JSON.stringify({
            token_sha256: sha256(token),
            public_key: key.publicKeyArmored,
            encrypted_private_key: key.encryptedPrivateKeyArmored,
            unique_values: strength.uniqueValues,
            counts: strength.counts,
            length: token.length,
            ref_code: refCode,
        })
      }}
    );

    console.log(requestOptions)
    
    requestOptions.then((options) =>
      fetch("/api/user/",options)
        .then((response) => response.json())
        .then((data) => { console.log(data) &
          this.setState({
              nickname: data.nickname,
              bit_entropy: data.token_bits_entropy,
              avatarUrl: '/static/assets/avatars/' + data.nickname + '.png',
              shannon_entropy: data.token_shannon_entropy,
              bad_request: data.bad_request,
              found: data.found,
              loadingRobot:false,
          })
          &
          // Add nick and token to App state (token only if not a bad request)
          (data.bad_request ? this.props.setAppState({
            nickname: data.nickname,
            avatarLoaded: false, 
            activeOrderId: data.active_order_id ? data.active_order_id : null,
            referralCode: data.referral_code,
            earnedRewards: data.earned_rewards,
            lastOrderId: data.last_order_id ? data.last_order_id : null,
          })
          :
          (this.props.setAppState({
            nickname: data.nickname,
            token: token,
            avatarLoaded: false,
            activeOrderId: data.active_order_id ? data.active_order_id : null,
            lastOrderId: data.last_order_id ? data.last_order_id : null,
            referralCode: data.referral_code,
            earnedRewards: data.earned_rewards,
          })) & writeCookie("robot_token",token) 
              & writeCookie("pub_key",data.public_key.split('\n').join('\\')) 
              & writeCookie("enc_priv_key",data.encrypted_private_key.split('\n').join('\\')))
          &
          // If the robot has been found (recovered) we assume the token is backed up
          (data.found ? this.props.setAppState({copiedToken:true}) : null)
      })
    );
  }

  delGeneratedUser() {
    const requestOptions = {
      method: 'DELETE',
      headers: {'Content-Type':'application/json', 'X-CSRFToken': getCookie('csrftoken')},
    };
    fetch("/api/user", requestOptions)
      .then((response) => response.json());

    deleteCookie("sessionid");
    deleteCookie("robot_token");
    deleteCookie("pub_key");
    deleteCookie("enc_priv_key");
  }

  handleClickNewRandomToken=()=>{
    var token = genBase62Token(36);
    this.setState({
      token: token,
      tokenHasChanged: true,
    });
    this.props.setAppState({copiedToken: true})
  }

  handleChangeToken=(e)=>{
    this.setState({
      token: e.target.value.split(' ').join(''),
      tokenHasChanged: true,
    })
  }

  handleClickSubmitToken=()=>{
    this.delGeneratedUser();
    this.getGeneratedUser(this.state.token);
    this.setState({loadingRobot: true, tokenHasChanged: false});
    this.props.setAppState({avatarLoaded: false, 
      nickname: null, 
      token: null, 
      copiedToken: false, 
      lastOrderId: null,
      activeOrderId: null});
  }

  handleClickOpenInfo = () => {
    this.setState({openInfo: true});
  };

  handleCloseInfo = () => {
    this.setState({openInfo: false});
  };

  createJsonFile = () => {
    return ({
      "token":getCookie('robot_token'),
      "token_shannon_entropy": this.state.shannon_entropy,
      "token_bit_entropy": this.state.bit_entropy,
      "public_key": getCookie('pub_key').split('\\').join('\n'), 
      "encrypted_private_key": getCookie('enc_priv_key').split('\\').join('\n'),
    })
  }

  render() {
    const { t, i18n} = this.props;
    return (
      <Grid container spacing={1}>
        <Grid item>
          <div className='clickTrough'/>
        </Grid>
        <Grid item xs={12} align="center" sx={{width:370, height:260}}>
          {!this.state.loadingRobot ?
            <div>
              <Grid item xs={12} align="center">
                <Typography component="h5" variant="h5">
                  <b>{this.state.nickname && getCookie("sessionid") ?
                    <div style={{display:'flex', alignItems:'center', justifyContent:'center', flexWrap:'wrap', height:'45px'}}>
                      <BoltIcon sx={{ color: "#fcba03", height: '33px',width: '33px'}}/><a>{this.state.nickname}</a><BoltIcon sx={{ color: "#fcba03", height: '33px',width: '33px'}}/>
                    </div>
                    : ""}</b>
                </Typography>
              </Grid>
              <Grid item xs={12} align="center">
              <Tooltip enterTouchDelay={0} title={t("This is your trading avatar")}>
                <div style={{ maxWidth: 200, maxHeight: 200 }}>
                  <Image 
                    className='newAvatar'
                    disableError={true}
                    cover={true}
                    color='null'
                    src={getCookie("sessionid") ? this.state.avatarUrl || "" : ""}
                  />
                </div>
                </Tooltip><br/>
              </Grid>
            </div>
            : <CircularProgress sx={{position: 'relative', top: 100, }}/>}
          </Grid>
          {
            this.state.found ?
              <Grid item xs={12} align="center">
                <Typography variant="subtitle2" color='primary'>
                  {this.state.found ? t("A robot avatar was found, welcome back!"):null}<br/>
                </Typography>
              </Grid>
             :
             ""
          }
          <Grid container align="center">
            <Grid item xs={12} align="center">
              <TextField sx={{maxWidth: 280}}
                error={this.state.bad_request ? true : false}
                label={t("Store your token safely")}
                required={true}
                value={this.state.token}
                variant='standard'
                helperText={this.state.bad_request}
                size='small'
                onChange={this.handleChangeToken}
                onKeyPress={(e) => {
                  if (e.key === 'Enter') {
                    this.handleClickSubmitToken();
                  }
                }}
                InputProps={{
                  startAdornment:
                  <div style={{width:50, minWidth:50, position:'relative',left:-6}}>
                    <Grid container>
                    <Grid item xs={6}>
                        <Tooltip enterTouchDelay={250} title={t("Save token and PGP credentials to file")}>
                          <span>
                          <IconButton  
                            color="primary" 
                            disabled={!(getCookie('robot_token')==this.state.token) || !this.props.avatarLoaded} 
                            onClick={()=> saveAsJson(this.state.nickname+'.json', this.createJsonFile())}
                            >
                            <DownloadIcon sx={{width:22, height:22}}/>
                          </IconButton>
                          </span>
                        </Tooltip>
                      </Grid>
                      <Grid item xs={6}>
                        <Tooltip disableHoverListener enterTouchDelay={0} title={t("Copied!")}>
                          <IconButton 
                            color={this.props.copiedToken ? "inherit":"primary"}
                            disabled={!(getCookie('robot_token')==this.state.token) || !this.props.avatarLoaded} 
                            onClick={()=> (copyToClipboard(getCookie('robot_token')) & this.props.setAppState({copiedToken:true}))}
                            >
                            <ContentCopy sx={{width:18, height:18}}/>
                          </IconButton>
                        </Tooltip>
                      </Grid>
                    </Grid>
                  </div>,
                  endAdornment:
                  <Tooltip enterTouchDelay={250} title={t("Generate a new token")}>
                    <IconButton onClick={this.handleClickNewRandomToken}><CasinoIcon/></IconButton>
                  </Tooltip>,
                  }}
              />
            </Grid>
          </Grid>
          <Grid item xs={12} align="center">
            {this.state.tokenHasChanged ?
            <Button type="submit" size='small'  onClick= {this.handleClickSubmitToken}>
              <SmartToyIcon sx={{width:18, height:18}} />
              <span> {t("Generate Robot")}</span>
            </Button>
            :
            <Tooltip enterTouchDelay={0} enterDelay={500} enterNextDelay={2000} title={t("You must enter a new token first")}>
              <div>
              <Button disabled={true} type="submit" size='small' >
                <SmartToyIcon sx={{width:18, height:18}} />
                <span>{t("Generate Robot")}</span>
              </Button>
              </div>
            </Tooltip>
            }
          </Grid>
          <Grid item xs={12} align="center">
            <ButtonGroup variant="contained" aria-label="outlined primary button group">
              <Button disabled={this.state.loadingRobot || !(this.props.token ? getCookie('robot_token')==this.props.token : true )} color='primary' to='/make/' component={Link}>{t("Make Order")}</Button>
              <Button color='inherit' style={{color: '#111111'}} onClick={this.handleClickOpenInfo}>{t("Info")}</Button>
              <InfoDialog open={Boolean(this.state.openInfo)} maxAmount='3,000,000' onClose = {this.handleCloseInfo}/>
              <Button disabled={this.state.loadingRobot || !(this.props.token ? getCookie('robot_token')==this.props.token : true )} color='secondary' to='/book/' component={Link}>{t("View Book")}</Button>
            </ButtonGroup>
          </Grid>

          <Grid item xs={12} align="center" sx={{width:370}}>
            <Grid item>
              <div style={{height:40}}/>
            </Grid>
            <div style={{width:370, left:30}}>
              <Grid container align="center">
                <Grid item xs={0.8}/>
                <Grid item xs={7.5} align="right">
                  <Typography component="h5" variant="h5">
                     {t("Simple and Private LN P2P Exchange")}
                  </Typography>
                </Grid>
                <Grid item xs={2.5} align="left">
                    <RoboSatsNoTextIcon color="primary" sx={{height:72, width:72}}/>
                </Grid>
              </Grid>
            </div>
          </Grid>
      </Grid>
    );
  }
}

export default withTranslation()(UserGenPage);