com.baulsupp.oksocial.output.ConsoleHandler.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
The newest version!
package com.baulsupp.oksocial.output

import com.baulsupp.oksocial.output.iterm.ItermOutputHandler
import com.baulsupp.oksocial.output.iterm.itermIsAvailable
import com.baulsupp.oksocial.output.process.exec
import com.fasterxml.jackson.core.type.TypeReference
import com.fasterxml.jackson.databind.ObjectMapper
import com.fasterxml.jackson.dataformat.cbor.CBORFactory
import com.fasterxml.jackson.datatype.jdk8.Jdk8Module
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule
import com.fasterxml.jackson.module.paramnames.ParameterNamesModule
import kotlinx.coroutines.runBlocking
import okio.BufferedSource
import okio.buffer
import okio.sink
import okio.source
import java.awt.Desktop
import java.io.BufferedInputStream
import java.io.ByteArrayInputStream
import java.io.File
import java.io.IOException
import java.net.URI
import java.net.UnknownHostException
import java.util.Arrays.asList
import java.util.concurrent.TimeUnit
import java.util.logging.Level
import java.util.logging.Logger
import javax.sound.sampled.AudioSystem
import javax.sound.sampled.LineEvent
import kotlin.coroutines.resume

open class ConsoleHandler(protected var responseExtractor: ResponseExtractor) : OutputHandler {
  val jqInstalled by lazy {
    runBlocking {
      isInstalled("jq")
    }
  }

  override suspend fun showError(message: String?, e: Throwable?) {
    if (logger.isLoggable(Level.FINE)) {
      logger.log(Level.WARNING, message, e)
    }

    if (e is UsageException) {
      System.err.println(e.message)
    } else if (e is UnknownHostException && e.cause == null) {
      if (message != null) {
        System.err.print("$message: ")
      }
      System.err.println(e.toString())
    } else {
      if (message != null) {
        System.err.println(message)
      }
      e?.printStackTrace()
    }
  }

  override suspend fun showOutput(response: R) {
    var mimeType = responseExtractor.mimeType(response)

    var source = responseExtractor.source(response)

    if (mimeType != null) {
      if (isCbor(mimeType)) {
        source = convertCborToJson(source)
        mimeType = "application/json"
      }

      if (isMedia(mimeType)) {
        openPreview(response)
        return
      } else if (isAudio(mimeType)) {
        playAudio(response)
        return
      } else if (jqInstalled && isJson(mimeType)) {
        prettyPrintJson(source)
        return
      }
    }

    // TODO support a nice hex mode for binary files
    source.writeToSink(systemOut)
    println("")
  }

  private fun cborMapper(): ObjectMapper {
    return ObjectMapper(CBORFactory()).registerModule(ParameterNamesModule())
      .registerModule(Jdk8Module())
      .registerModule(JavaTimeModule())
  }

  private fun jsonMapper(): ObjectMapper {
    return ObjectMapper().registerModule(ParameterNamesModule())
      .registerModule(Jdk8Module())
      .registerModule(JavaTimeModule())
  }

  private fun convertCborToJson(source: BufferedSource): BufferedSource {
    // TODO consider adding streaming

    val cborMapper = cborMapper()

    val map = cborMapper.readValue(source.inputStream(),
      object : TypeReference>() {})

    val om = jsonMapper()

    val bytes = om.writeValueAsBytes(map)

    return ByteArrayInputStream(bytes).source().buffer()
  }

  suspend fun prettyPrintJson(response: BufferedSource) {
    val command = if (isTerminal) asList("jq", "-C", ".") else asList("jq", ".")

    val result = exec(command) {
      redirectInput(response.inputStream())
      redirectError(stdErrLogging)
      redirectOutput(System.out)
    }

    if (!result.success) {
      throw IOException("return code " + result.exitCode + " from " + command.joinToString(" "))
    }
  }

  open suspend fun playAudio(response: R) {
    val source = responseExtractor.source(response)

    // TODO proper IO handling continuations style also
    source.inputStream().use {
      return kotlinx.coroutines.suspendCancellableCoroutine { c ->
        val audioIn = AudioSystem.getAudioInputStream(BufferedInputStream(it, 512 * 1024))
        val clip = AudioSystem.getClip()
        clip.open(audioIn)
        clip.addLineListener { event ->
          if (event.type == LineEvent.Type.STOP) {
            c.resume(Unit)
          }
        }
        clip.start()
      }
    }
  }

  open suspend fun openPreview(response: R) {
    if (Desktop.isDesktopSupported()) {
      val tempFile = writeToFile(response)

      Desktop.getDesktop().open(tempFile)
    } else {
      System.err.println("Falling back to console output, use -r to avoid warning")

      responseExtractor.source(response).writeToSink(systemOut)
      println("")
    }
  }

  suspend fun writeToFile(response: R): File {
    val mimeType = responseExtractor.mimeType(response)

    val tempFile = File.createTempFile("output", getExtension(mimeType))

    tempFile.sink().use { out ->
      responseExtractor.source(response).writeToSink(out)
    }
    return tempFile
  }

  override suspend fun openLink(url: String) {
    if (Desktop.isDesktopSupported()) {
      Desktop.getDesktop().browse(URI.create(url))
    } else {
      System.err.println(url)
    }
  }

  suspend fun terminalWidth(): Int {
    val command = if (isOSX)
      listOf("/bin/stty", "-a", "-f", "/dev/tty")
    else
      listOf("/bin/stty", "-a", "-F", "/dev/tty")

    val result = exec(command) {
      timeout(5, TimeUnit.SECONDS)
      redirectError(stdErrLogging)
      readOutput(true)
    }

    val matchResult = "(\\d+) columns".toRegex(RegexOption.MULTILINE).find(result.outputString)
    return matchResult?.groupValues?.get(1)?.toIntOrNull() ?: 80
  }

  companion object {
    private val logger = Logger.getLogger(ConsoleHandler::class.java.name)

    fun  instance(re: ResponseExtractor): ConsoleHandler {
      return when {
        itermIsAvailable() -> ItermOutputHandler(re)
        isOSX -> OsxOutputHandler(re)
        else -> ConsoleHandler(re)
      }
    }

    fun instance(): ConsoleHandler = instance(ToStringResponseExtractor)
  }
}

fun main(args: Array) {
  val c = ConsoleHandler(FileResponseExtractor)

  runBlocking {
    c.playAudio(File("/Users/yuri/Downloads/07018050.wav"))
  }
}