joystream/storage-node/packages/runtime-api/discovery.js

'use strict'

const debug = require('debug')('joystream:runtime:discovery')

/*
 * Add discovery related functionality to the substrate API.
 */
class DiscoveryApi {
  static async create (base) {
    const ret = new DiscoveryApi()
    ret.base = base
    await ret.init()
    return ret
  }

  async init () {
    debug('Init')
  }

  /*
   * Get Bootstrap endpoints
   */
  async getBootstrapEndpoints () {
    return this.base.api.query.discovery.bootstrapEndpoints()
  }

  /*
   * Set Bootstrap endpoints
   */
  async setBootstrapEndpoints (sudoAccount, endpoints) {
    const tx = this.base.api.tx.discovery.setBootstrapEndpoints(endpoints)
    // make sudo call
    return this.base.signAndSend(
      sudoAccount,
      this.base.api.tx.sudo.sudo(tx)
    )
  }

  /*
   * Get AccountInfo of a storage provider
   */
  async getAccountInfo (storageProviderId) {
    const info = await this.base.api.query.discovery.accountInfoByStorageProviderId(storageProviderId)
    // Not an Option so we use default value check to know if info was found
    return info.expires_at.eq(0) ? null : info
  }

  /*
   * Set AccountInfo of our storage provider
   */
  async setAccountInfo (ipnsId) {
    const roleAccountId = this.base.identities.key.address
    const storageProviderId = this.base.storageProviderId
    const isProvider = await this.base.workers.isStorageProvider(storageProviderId)
    if (isProvider) {
      const tx = this.base.api.tx.discovery.setIpnsId(storageProviderId, ipnsId)
      return this.base.signAndSend(roleAccountId, tx)
    } else {
      throw new Error('Cannot set AccountInfo, id is not a storage provider')
    }
  }

  /*
   * Clear AccountInfo of our storage provider
   */
  async unsetAccountInfo () {
    const roleAccountId = this.base.identities.key.address
    const storageProviderId = this.base.storageProviderId
    var tx = this.base.api.tx.discovery.unsetIpnsId(storageProviderId)
    return this.base.signAndSend(roleAccountId, tx)
  }
}

module.exports = {
  DiscoveryApi
}