com.baulsupp.oksocial.authenticator.Authorisation.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oksocial Show documentation
Command Line Output Library
The newest version!
package com.baulsupp.oksocial.authenticator

import com.baulsupp.oksocial.commands.CommandLineClient
import com.baulsupp.oksocial.kotlin.client
import com.baulsupp.oksocial.output.UsageException
import com.baulsupp.oksocial.secrets.Secrets

class Authorisation(val main: CommandLineClient) {

  suspend fun authorize(
    auth: AuthInterceptor<*>?,
    token: String?,
    authArguments: List,
    tokenSet: String
  ) {
    if (auth == null) {
      throw UsageException(
        "unable to find authenticator. Specify name from " + main.authenticatingInterceptor.names().joinToString(", "))
    }

    if (token != null) {
      storeCredentials(auth, token, tokenSet)
    } else {
      authRequest(auth, authArguments, tokenSet)
    }
  }

  private fun  storeCredentials(auth: AuthInterceptor, token: String, tokenSet: String) {
    val credentials = auth.serviceDefinition.parseCredentialsString(token)
    main.credentialsStore.set(auth.serviceDefinition, tokenSet, credentials)
  }

  suspend fun  authRequest(auth: AuthInterceptor, authArguments: List, tokenSet: String) {
    auth.serviceDefinition.accountsLink()?.let { main.outputHandler.info("Accounts: $it") }

    val credentials = auth.authorize(client, main.outputHandler, authArguments)

    main.credentialsStore.set(auth.serviceDefinition, tokenSet, credentials)

    Secrets.instance.saveIfNeeded()

    // TODO validate credentials
  }

  suspend fun  renew(auth: AuthInterceptor?, tokenSet: String) {
    if (auth == null) {
      throw UsageException(
        "unable to find authenticator. Specify name from " + main.authenticatingInterceptor.names().joinToString(", "))
    }

    val serviceDefinition = auth.serviceDefinition

    val credentials = main.credentialsStore.get(serviceDefinition, tokenSet)
      ?: throw UsageException("no existing credentials")

    if (!auth.canRenew(credentials)) {
      throw UsageException("credentials not renewable")
    }

    val newCredentials = auth.renew(client, credentials) ?: throw UsageException("failed to renew")

    main.credentialsStore.set(serviceDefinition, tokenSet, newCredentials)
  }

  fun remove(auth: AuthInterceptor<*>?, tokenSet: String) {
    if (auth == null) {
      throw UsageException(
        "unable to find authenticator. Specify name from " + main.authenticatingInterceptor.names().joinToString(", "))
    }

    main.credentialsStore.remove(auth.serviceDefinition, tokenSet)
  }
}