Copy last result to clipboard

at-date-console/src/main/kotlin/com/amerharb/atdate/Main.kt

@@ -10,14 +10,16 @@ fun main(args: Array<String>) {
 	if (args.isEmpty()) {
 		println("enter @...@ to encode, 0x... to decode, C to copy last result or Q to quit")
 		while (true) {
-			mainMenu()
+			if (!mainMenu()) {
+				println("Exiting 👋...")
+				break
+			}
 		}
 	} else {
 		val arg1 = args[0]
 		when {
 			arg1 == "-h" || arg1 == "--help" -> {
 				printHelp()
-				exitProcess(0)
 			}
 
 			arg1.startsWith("@") -> {
@@ -38,22 +40,23 @@ fun main(args: Array<String>) {
 	}
 }
 
-fun mainMenu() {
+fun mainMenu(): Boolean {
 	print(">")
 	val command = readlnOrNull()
 	if (command.isNullOrBlank()) {
 		println("Invalid input")
-		return
+		return true
 	}
 	when (command[0].lowercase()) {
 		"@" -> encodeCommand(command)
 		"0" -> decodeCommand(command)
 		"c" -> copyLastResultToClipboard()
-		"q" -> exitProcess()
+		"q" -> return false
 		else -> {
 			println("Invalid input")
 		}
 	}
+	return true
 }
 
 fun encodeCommand(input: String) {
@@ -88,11 +91,6 @@ fun copyLastResultToClipboard() {
 	println("$lastResult\nhas been copied to clipboard!")
 }
 
-fun exitProcess(status: Int = 0) {
-	println("Exiting with status $status")
-	kotlin.system.exitProcess(status)
-}
-
 fun printEncodingResult(atDate: AtDate) {
 	println("Hex: ${getHex(atDate)}")
 	println("Bin: ${getBin(atDate)}")

at-date-console/src/test/kotlin/com/amerharb/atdate/MainTest.kt

@@ -1,6 +1,9 @@
 package com.amerharb.atdate
 
+import java.awt.Toolkit
+import java.awt.datatransfer.DataFlavor
 import java.io.ByteArrayOutputStream
+import java.io.InputStream
 import java.io.PrintStream
 import kotlin.test.Test
 import kotlin.test.assertEquals
@@ -282,4 +285,37 @@ class MainTest {
 		assertEquals(expected, actual)
 		System.setOut(originalOut)
 	}
+
+	@Test
+	fun testCopyResult() {
+		val outContent = ByteArrayOutputStream()
+		val originalIn = System.`in`
+		System.setOut(PrintStream(outContent))
+		val systemInMock = InputStreamMock("0xa0\nc\nq")
+		System.setIn(systemInMock)
+		main(emptyArray())
+		val expected = "@P-tp@"
+		val actual = getCliboard()
+		assertEquals(expected, actual)
+		System.setIn(originalIn)
+	}
+}
+
+class InputStreamMock(input: String) : InputStream() {
+	private val data: MutableList<Char> = mutableListOf()
+
+	init {
+		this.data.addAll(input.toList())
+	}
+	override fun read(): Int {
+		if (data.isEmpty()) {
+			return -1
+		}
+		return data.removeAt(0).code
+	}
+}
+
+fun getCliboard(): String {
+	val clipboard = Toolkit.getDefaultToolkit().systemClipboard
+	return clipboard.getData(DataFlavor.stringFlavor) as String
 }