Merge pull request #1776 from blockstack/hotfix/gaia-config-profile

Hotfix to correct gaiaHubConfig info in profile.json

app/js/account/utils/blockstack-inc.js

@@ -1,6 +1,6 @@
 // @flow
 import log4js from 'log4js'
-import { connectToGaiaHub, GaiaHubConfig } from 'blockstack'
+import { connectToGaiaHub as bsConnectToGaiaHub, GaiaHubConfig, uploadToGaiaHub } from 'blockstack'
 
 const logger = log4js.getLogger(__filename)
 
@@ -11,4 +11,6 @@ export function redirectToConnectToGaiaHub() {
   window.top.location.href = `http://${host}:${port}/account/storage#gaiahub`
 }
 
-export { connectToGaiaHub, GaiaHubConfig }
+const connectToGaiaHub = (hubUrl: string, key: string) => bsConnectToGaiaHub(hubUrl, key)
+
+export { connectToGaiaHub, GaiaHubConfig, uploadToGaiaHub }

app/js/account/utils/index.js

@@ -1,8 +1,8 @@
 // @flow
 import { parseZoneFile } from 'zone-file'
 
-import type { GaiaHubConfig } from 'blockstack'
-import { connectToGaiaHub, uploadToGaiaHub } from 'blockstack'
+import type { GaiaHubConfig } from './blockstack-inc'
+import { connectToGaiaHub, uploadToGaiaHub } from './blockstack-inc'
 
 import { getTokenFileUrlFromZoneFile } from '@utils/zone-utils'
 
@@ -88,8 +88,6 @@ export function uploadProfile(
   signedProfileTokenData: string
 ) {
   return connectToGaiaHub(api.gaiaHubUrl, identityKeyPair.key).then(identityHubConfig => {
-    const globalHubConfig = api.gaiaHubConfig
-
     const urlToWrite = getProfileUploadLocation(identity, identityHubConfig)
 
     let uploadAttempt = tryUpload(
@@ -102,7 +100,7 @@ export function uploadProfile(
       uploadAttempt = tryUpload(
         urlToWrite,
         signedProfileTokenData,
-        globalHubConfig,
+        identityHubConfig,
         'application/json'
       )
     }

app/js/store/reducers.js

@@ -39,7 +39,7 @@ export function initializeStateVersion() {
  * and other state is regenerated.
  * @type {number}
  */
-export const CURRENT_VERSION: number = 16
+export const CURRENT_VERSION: number = 17
 
 const AppReducer = combineReducers({
   account: AccountReducer,

app/js/update/index.js

@@ -10,7 +10,8 @@ import { AppHomeWrapper, ShellParent } from '@blockstack/ui'
 import {
   selectAccountCreated,
   selectEncryptedBackupPhrase,
-  selectIdentityAddresses
+  selectIdentityAddresses,
+  selectIdentityKeypairs
 } from '@common/store/selectors/account'
 import {
   selectDefaultIdentity,
@@ -28,7 +29,8 @@ import {
   hasLegacyCoreStateVersion,
   migrateLegacyCoreEndpoints
 } from '@utils/api-utils'
-import { decrypt } from '@utils'
+import { uploadProfile } from '../account/utils'
+import { decrypt, signProfileForUpload } from '@utils'
 const VIEWS = {
   INITIAL: 0,
   SUCCESS: 1,
@@ -43,7 +45,8 @@ const mapStateToProps = state => ({
   localIdentities: selectLocalIdentities(state),
   defaultIdentityIndex: selectDefaultIdentity(state),
   accountCreated: selectAccountCreated(state),
-  identityAddresses: selectIdentityAddresses(state)
+  identityAddresses: selectIdentityAddresses(state),
+  identityKeypairs: selectIdentityKeypairs(state)
 })
 
 const mapDispatchToProps = dispatch =>
@@ -137,51 +140,84 @@ class UpdatePage extends React.Component {
     const dataBuffer = new Buffer(encryptedBackupPhrase, 'hex')
     const { password } = this.state
 
-    return decrypt(dataBuffer, password)
-      .then(backupPhraseBuffer => {
-        this.setState(
-          {
-            upgradeInProgress: true
-          },
-          () =>
-            setTimeout(() => {
-              console.debug('decryptKeyAndResetState: correct password!')
-              const backupPhrase = backupPhraseBuffer.toString()
-              const numberOfIdentities =
-                localIdentities.length >= 1 ? localIdentities.length : 1
-              this.setState({
-                encryptedBackupPhrase,
-                backupPhrase,
-                defaultIdentityIndex,
-                numberOfIdentities
-              })
-              if (hasLegacyCoreStateVersion()) {
-                const migratedApi = migrateLegacyCoreEndpoints(api)
-                this.props.migrateAPIEndpoints(migratedApi)
-              }
-              // clear our state
-              this.props.updateState()
+    const updateProfileUrls = localIdentities.map((identity, index) =>
+      new Promise(async (resolve, reject) => {
+        try {
+          const signedProfileTokenData = signProfileForUpload(
+            identity.profile,
+            this.props.identityKeypairs[index],
+            this.props.api
+          )
+          uploadProfile(
+            this.props.api,
+            identity,
+            this.props.identityKeypairs[index],
+            signedProfileTokenData
+          ).then(resolve).catch(reject)
+        } catch (error) {
+          reject(error)
+        }
+
+      }))
 
-              // generate new account and IDs
-              this.createAccount().then(() => this.createNewIds())
-              .then(() => this.props.refreshIdentities(
-                  this.props.api, 
-                  this.props.identityAddresses
-                ))
-            }, 150)
-        )
-      })
-      .catch(error => {
-        console.error('decryptKeyAndResetState: invalid password', error)
-        this.setState({
-          loading: false,
-          password: null,
-          errors: {
-            password: 'Incorrect Password'
-          },
-          status: 'error'
+    return Promise.all(updateProfileUrls).then(() => {
+      console.log('updated profile URLs')
+      return decrypt(dataBuffer, password)
+        .then(backupPhraseBuffer => {
+          this.setState(
+            {
+              upgradeInProgress: true
+            },
+            () =>
+              setTimeout(() => {
+                console.debug('decryptKeyAndResetState: correct password!')
+                const backupPhrase = backupPhraseBuffer.toString()
+                const numberOfIdentities =
+                  localIdentities.length >= 1 ? localIdentities.length : 1
+                this.setState({
+                  encryptedBackupPhrase,
+                  backupPhrase,
+                  defaultIdentityIndex,
+                  numberOfIdentities
+                })
+                if (hasLegacyCoreStateVersion()) {
+                  const migratedApi = migrateLegacyCoreEndpoints(api)
+                  this.props.migrateAPIEndpoints(migratedApi)
+                }
+                // clear our state
+                this.props.updateState()
+
+                // generate new account and IDs
+                this.createAccount().then(() => this.createNewIds())
+                .then(() => this.props.refreshIdentities(
+                    this.props.api, 
+                    this.props.identityAddresses
+                  ))
+              }, 150)
+          )
         })
+        .catch(error => {
+          console.error('decryptKeyAndResetState: invalid password', error)
+          this.setState({
+            loading: false,
+            password: null,
+            errors: {
+              password: 'Incorrect Password'
+            },
+            status: 'error'
+          })
+        })
+    }).catch(error => {
+      console.error('upgradeBlockstackState: error updating profile', error)
+      this.setState({
+        loading: false,
+        password: null,
+        errors: {
+          password: 'Unable to update profile'
+        },
+        status: 'error'
       })
+    })
   }
 
   /**

app/js/utils/profile-utils.js

@@ -171,11 +171,19 @@ export function signProfileForUpload(profile, keypair, api) {
   const privateKey = keypair.key
   const publicKey = keypair.keyID
 
+  if (profile.api && profile.api.gaiaHubConfig) {
+    profile.api.gaiaHubConfig = {
+      url_prefix: profile.api.gaiaHubConfig.url_prefix
+    }
+  }
+
   if (api) {
     profile = {
       ...profile,
       api: {
-        gaiaHubConfig: api.gaiaHubConfig,
+        gaiaHubConfig: {
+          url_prefix: api.gaiaHubConfig.url_prefix
+        },
         gaiaHubUrl: api.gaiaHubUrl
       }
     }