ai.mantik.elements.registry.api.MantikRegistryApi.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elements_2.13 Show documentation
elements
The newest version!
/*
 * This file is part of the Mantik Project.
 * Copyright (c) 2020-2021 Mantik UG (Haftungsbeschränkt)
 * Authors: See AUTHORS file
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License version 3.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.
 *
 * Additionally, the following linking exception is granted:
 *
 * If you modify this Program, or any covered work, by linking or
 * combining it with other code, such other code is not for that reason
 * alone subject to any of the requirements of the GNU Affero GPL
 * version 3.
 *
 * You can be released from the requirements of the license by purchasing
 * a commercial license.
 */
package ai.mantik.elements.registry.api

import ai.mantik.elements.errors.{ErrorCode, MantikException, MantikRemoteException, RemoteRegistryException}
import ai.mantik.elements.{MantikId, NamedMantikId}
import akka.http.scaladsl.model.{HttpRequest, HttpResponse, Uri}
import akka.stream.Materializer
import akka.stream.scaladsl.Source
import akka.util.ByteString
import net.reactivecore.fhttp.akka.ApiClient

import scala.concurrent.{ExecutionContext, Future}

/** Implements raw API Calls to Mantik Registry. */
class MantikRegistryApi(rootUri: Uri, executor: ApiClient.RequestExecutor)(
    implicit ec: ExecutionContext,
    mat: Materializer
) {
  val apiPath = Uri("api").resolvedAgainst(rootUri)
  val apiClient = new ApiClient(executor, apiPath)

  private def orFail[A, T](in: A => Future[Either[(Int, ApiErrorResponse), T]]): A => Future[T] = { arg =>
    in(arg).flatMap {
      case Left((httpCode, failure)) =>
        Future.failed(new RemoteRegistryException(httpCode, failure.code, failure.message.getOrElse("")))
      case Right(in) => Future.successful(in)
    }
  }

  /** Execute a login call. */
  val login: ApiLoginRequest => Future[ApiLoginResponse] = orFail(apiClient.prepare(MantikRegistryApiCalls.login))

  /** Execute a Login Status call. */
  val loginStatus: String => Future[ApiLoginStatusResponse] = orFail(
    apiClient.prepare(MantikRegistryApiCalls.loginStatus)
  )

  /** Retrieve an artifact. */
  val artifact: ((String, MantikId)) => Future[ApiGetArtifactResponse] = orFail(
    apiClient.prepare(MantikRegistryApiCalls.artifact)
  )

  /** Tag an Artifact. */
  val tag: ((String, ApiTagRequest)) => Future[ApiTagResponse] = orFail(apiClient.prepare(MantikRegistryApiCalls.tag))

  /**
    * Retrieve an artifact file.
    * (With token and fileId)
    * @return content type and byte source.
    */
  val file: ((String, String)) => Future[(String, Source[ByteString, _])] = orFail(
    apiClient.prepare(MantikRegistryApiCalls.file)
  )

  /** Prepares the upload of a file. */
  val prepareUpload: ((String, ApiPrepareUploadRequest)) => Future[ApiPrepareUploadResponse] = orFail(
    apiClient.prepare(MantikRegistryApiCalls.prepareUpload)
  )

  /** Uploads the payload of an artifact. */
  val uploadFile: ((String, String, String, Source[ByteString, _])) => Future[ApiFileUploadResponse] = orFail(
    apiClient.prepare(MantikRegistryApiCalls.uploadFile)
  )
}