com.baulsupp.oksocial.output.ResponseExtractor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of oksocial-output Show documentation

Command Line Output Library

There is a newer version: 4.19.0

package com.baulsupp.oksocial.output

import okio.BufferedSource
import okio.Okio
import java.io.ByteArrayInputStream
import java.io.File
import java.nio.file.Files

object ToStringResponseExtractor : ResponseExtractor {
  override fun mimeType(response: Any): String = "text/plain"

  override fun source(response: Any): BufferedSource =
    Okio.buffer(Okio.source(ByteArrayInputStream(response.toString().toByteArray())))

  override fun filename(response: Any): String? = null
}

object FileResponseExtractor : ResponseExtractor {
  override fun mimeType(response: File): String? {
    return Files.probeContentType(response.toPath())
  }

  override fun source(response: File): BufferedSource {
    return Okio.buffer(Okio.source(response))
  }

  override fun filename(response: File): String {
    return response.name
  }
}

interface ResponseExtractor {
  fun mimeType(response: R): String?

  fun source(response: R): BufferedSource

  fun filename(response: R): String?
}