All files / src/storage hub.js

62.3% Statements 38/61
39.13% Branches 9/23
82.35% Functions 14/17
62.3% Lines 38/61

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180                          1x                       5x 5x                   5x 2x   3x     2x 2x         5x                                                             2x 2x   2x   2x       2x 2x             2x 2x           3x   3x 3x   2x 2x 2x   2x                                                                           4x 4x     4x 4x 4x             1x 1x         1x 1x 1x   1x 1x 1x 1x      
/* @flow */
import bitcoin from 'bitcoinjs-lib'
import crypto from 'crypto'
 
import { TokenSigner } from 'jsontokens'
import { ecPairToAddress } from '../utils'
import { getPublicKeyFromPrivate, hexStringToECPair } from '../index'
import { BLOCKSTACK_DEFAULT_GAIA_HUB_URL } from '../auth/authConstants'
 
import { UserSession } from '../auth/userSession'
import { Logger } from '../logger'
import { InvalidStateError } from '../errors'
 
export const BLOCKSTACK_GAIA_HUB_LABEL = 'blockstack-gaia-hub-config'
 
export type GaiaHubConfig = {
  address: string,
  url_prefix: string,
  token: string,
  server: string
}
 
export function uploadToGaiaHub(filename: string, contents: any,
                                hubConfig: GaiaHubConfig,
                                contentType: string = 'application/octet-stream'): Promise<*> {
  Logger.debug(`uploadToGaiaHub: uploading ${filename} to ${hubConfig.server}`)
  return fetch(`${hubConfig.server}/store/${hubConfig.address}/${filename}`,
               {
                 method: 'POST',
                 headers: {
                   'Content-Type': contentType,
                   Authorization: `bearer ${hubConfig.token}`
                 },
                 body: contents
               })
    .then((response) => {
      if (response.ok) {
        return response.text()
      } else {
        throw new Error('Error when uploading to Gaia hub')
      }
    })
    .then(responseText => JSON.parse(responseText))
    .then(responseJSON => responseJSON.publicURL)
}
 
export function getFullReadUrl(filename: string,
                               hubConfig: GaiaHubConfig): string {
  return `${hubConfig.url_prefix}${hubConfig.address}/${filename}`
}
 
function makeLegacyAuthToken(challengeText: string, signerKeyHex: string): string {
  // only sign specific legacy auth challenges.
  let parsedChallenge
  try {
    parsedChallenge = JSON.parse(challengeText)
  } catch (err) {
    throw new Error('Failed in parsing legacy challenge text from the gaia hub.')
  }
  if (parsedChallenge[0] === 'gaiahub'
      && parsedChallenge[3] === 'blockstack_storage_please_sign') {
    const signer = hexStringToECPair(signerKeyHex
                                     + (signerKeyHex.length === 64 ? '01' : ''))
    const digest = bitcoin.crypto.sha256(challengeText)
    const signature = signer.sign(digest).toDER().toString('hex')
    const publickey = getPublicKeyFromPrivate(signerKeyHex)
    const token = Buffer.from(JSON.stringify(
      { publickey, signature }
    )).toString('base64')
    return token
  } else {
    throw new Error('Failed to connect to legacy gaia hub. If you operate this hub, please update.')
  }
}
 
function makeV1GaiaAuthToken(hubInfo: Object,
                             signerKeyHex: string,
                             hubUrl: string,
                             associationToken?: string): string {
  const challengeText = hubInfo.challenge_text
  const handlesV1Auth = (hubInfo.latest_auth_version
                         && parseInt(hubInfo.latest_auth_version.slice(1), 10) >= 1)
  const iss = getPublicKeyFromPrivate(signerKeyHex)
 
  Iif (!handlesV1Auth) {
    return makeLegacyAuthToken(challengeText, signerKeyHex)
  }
 
  const salt = crypto.randomBytes(16).toString('hex')
  const payload = {
    gaiaChallenge: challengeText,
    hubUrl,
    iss,
    salt,
    associationToken
  }
  const token = new TokenSigner('ES256K', signerKeyHex).sign(payload)
  return `v1:${token}`
}
 
export function connectToGaiaHub(gaiaHubUrl: string,
                                 challengeSignerHex: string,
                                 associationToken?: string): Promise<GaiaHubConfig> {
  Logger.debug(`connectToGaiaHub: ${gaiaHubUrl}/hub_info`)
 
  return fetch(`${gaiaHubUrl}/hub_info`)
    .then(response => response.json())
    .then((hubInfo) => {
      const readURL = hubInfo.read_url_prefix
      const token = makeV1GaiaAuthToken(hubInfo, challengeSignerHex, gaiaHubUrl, associationToken)
      const address = ecPairToAddress(hexStringToECPair(challengeSignerHex
                                        + (challengeSignerHex.length === 64 ? '01' : '')))
      return {
        url_prefix: readURL,
        address,
        token,
        server: gaiaHubUrl
      }
    })
}
 
/**
 * These two functions are app-specific connections to gaia hub,
 *   they read the user data object for information on setting up
 *   a hub connection, and store the hub config to localstorage
 * @param {UserSession} caller - the instance calling this function
 * @private
 * @returns {Promise} that resolves to the new gaia hub connection
 */
export function setLocalGaiaHubConnection(caller: UserSession): Promise<GaiaHubConfig> {
  const userData = caller.loadUserData()
 
  if (!userData) {
    throw new InvalidStateError('Missing userData')
  }
 
  if (!userData.hubUrl) {
    userData.hubUrl = BLOCKSTACK_DEFAULT_GAIA_HUB_URL
  }
 
  return connectToGaiaHub(userData.hubUrl,
                          userData.appPrivateKey,
                          userData.associationToken)
    .then((gaiaConfig) => {
      userData.gaiaHubConfig = gaiaConfig
      return gaiaConfig
    })
}
 
export function getOrSetLocalGaiaHubConnection(caller: UserSession): Promise<GaiaHubConfig> {
  const userData = caller.store.getSessionData().userData
  Iif (!userData) {
    throw new InvalidStateError('Missing userData')
  }
  const hubConfig = userData.gaiaHubConfig
  Eif (hubConfig) {
    return Promise.resolve(hubConfig)
  }
  return setLocalGaiaHubConnection(caller)
}
 
export function getBucketUrl(gaiaHubUrl: string, appPrivateKey: string): Promise<string> {
  let challengeSigner
  try {
    challengeSigner = bitcoin.ECPair.fromPrivateKey(new Buffer(appPrivateKey, 'hex'))
  } catch (e) {
    return Promise.reject(e)
  }
 
  return fetch(`${gaiaHubUrl}/hub_info`)
    .then(response => response.text())
    .then(responseText => JSON.parse(responseText))
    .then((responseJSON) => {
      const readURL = responseJSON.read_url_prefix
      const address = ecPairToAddress(challengeSigner)
      const bucketUrl = `${readURL}${address}/`
      return bucketUrl
    })
}