market/src/@hooks/usePublish.ts

import { Logger } from '@oceanprotocol/lib'
import {
  Service,
  ServiceComputePrivacy,
  ServiceType
} from '@oceanprotocol/lib/dist/node/ddo/interfaces/Service'
import { useEffect, useState } from 'react'
import { publishFeedback } from '@utils/feedback'
import { useOcean } from '@context/Ocean'
import { useWeb3 } from '@context/Web3'
import { getOceanConfig } from '@utils/ocean'

export interface DataTokenOptions {
  cap?: string
  name?: string
  symbol?: string
}

interface UsePublish {
  publish: (
    asset: DDO,
    serviceType: ServiceType,
    dataTokenOptions?: DataTokenOptions,
    timeout?: number,
    providerUri?: string
  ) => Promise<Asset>
  publishStep?: number
  publishStepText?: string
  publishError?: string
  isLoading: boolean
}

function usePublish(): UsePublish {
  const { networkId, web3Loading } = useWeb3()
  const { connect, ocean, account } = useOcean()
  const [isLoading, setIsLoading] = useState(false)
  const [publishStep, setPublishStep] = useState<number | undefined>()
  const [publishStepText, setPublishStepText] = useState<string | undefined>()
  const [publishError, setPublishError] = useState<string | undefined>()

  function setStep(index?: number) {
    setPublishStep(index)
    index && setPublishStepText(publishFeedback[index])
  }

  //
  // Initiate OceanProvider based on user wallet
  //
  useEffect(() => {
    if (web3Loading || !connect) return

    async function initOcean() {
      const config = getOceanConfig(networkId)
      await connect(config)
    }
    initOcean()
  }, [web3Loading, networkId, connect])

  /**
   * Publish an asset. It also creates the datatoken, mints tokens and gives the market allowance
   * @param  {Metadata} asset The metadata of the asset.
   * @param  {PriceOptions}  priceOptions : number of tokens to mint, datatoken weight , liquidity fee, type : fixed, dynamic
   * @param  {ServiceType} serviceType Desired service type of the asset access or compute
   * @param  {DataTokenOptions} dataTokenOptions custom name, symbol and cap for datatoken
   * @return {Promise<Asset>} Returns the newly published ddo
   */
  async function publish(
    asset: Asset,
    serviceType: ServiceType,
    dataTokenOptions?: DataTokenOptions,
    timeout?: number,
    providerUri?: string
  ): Promise<Asset> {
    if (!ocean || !account) return null
    setIsLoading(true)
    setPublishError(undefined)
    setStep(0)

    try {
      const publishedDate =
        new Date(Date.now()).toISOString().split('.')[0] + 'Z'
      const services: Service[] = []
      const price = '1'
      asset.metadata.created = publishedDate

      switch (serviceType) {
        case 'access': {
          if (!timeout) timeout = 0
          const accessService =
            await ocean.assets.createAccessServiceAttributes(
              account,
              price,
              publishedDate,
              timeout,
              providerUri,
              null
            )
          Logger.log('access service created', accessService)
          services.push(accessService)
          break
        }
        case 'compute': {
          if (!timeout) timeout = 3600
          const provider = {}
          const origComputePrivacy: ServiceComputePrivacy = {
            allowRawAlgorithm: false,
            allowNetworkAccess: false,
            allowAllPublishedAlgorithms: false,
            publisherTrustedAlgorithms: []
          }
          const computeService = ocean.compute.createComputeService(
            account,
            price,
            publishedDate,
            provider,
            origComputePrivacy,
            timeout,
            providerUri
          )
          services.push(computeService)
          break
        }
      }

      Logger.log('services created', services)

      // const ddo = await ocean.assets
      //   .create(
      //     asset,
      //     account,
      //     services,
      //     undefined,
      //     dataTokenOptions?.cap,
      //     dataTokenOptions?.name,
      //     dataTokenOptions?.symbol,
      //     providerUri
      //   )
      //   .next(setStep)
      // Logger.log('ddo created', ddo)
      // await ocean.assets.publishDdo(ddo, account.getId())
      // Logger.log('ddo published')
      // await sleep(20000)
      // setStep(7)
      // return ddo
    } catch (error) {
      setPublishError(error.message)
      Logger.error(error)
      setStep()
    } finally {
      setIsLoading(false)
    }
  }

  return {
    publish,
    publishStep,
    publishStepText,
    isLoading,
    publishError
  }
}

export { usePublish }
export default usePublish