1 files changed, 50 insertions, 0 deletions
diff --git a/src/main/kotlin/CryptoHandler.kt b/src/main/kotlin/CryptoHandler.kt
new file mode 100644
index 0000000..03396f2
--- /dev/null
+++ b/src/main/kotlin/CryptoHandler.kt
@@ -0,0 +1,50 @@
+package space.anity
+
+import java.io.*
+import java.security.*
+import javax.crypto.*
+import javax.crypto.spec.*
+
+class CryptoHandler @Throws(NoSuchPaddingException::class, NoSuchAlgorithmException::class)
+internal constructor(private val secretKey: SecretKey, cipher: String) {
+    private val cipher: Cipher = Cipher.getInstance(cipher)
+
+    @Throws(InvalidKeyException::class, IOException::class)
+    internal fun encrypt(content: String, fileName: String): ByteArray {
+        cipher.init(Cipher.ENCRYPT_MODE, secretKey)
+        val iv: ByteArray = cipher.iv
+
+        FileOutputStream(fileName).use { fileOut ->
+            CipherOutputStream(fileOut, cipher).use { cipherOut ->
+                cipherOut.write(content.toByteArray())
+            }
+        }
+
+        return iv
+    }
+
+    @Throws(InvalidAlgorithmParameterException::class, InvalidKeyException::class, IOException::class)
+    internal fun decrypt(fileName: String, iv: ByteArray): String {
+        var content = ""
+
+        FileInputStream(fileName).use { fileIn ->
+            cipher.init(Cipher.DECRYPT_MODE, secretKey, IvParameterSpec(iv))
+
+            CipherInputStream(fileIn, cipher).use { cipherIn ->
+                InputStreamReader(cipherIn).use { inputReader ->
+                    BufferedReader(inputReader).use { reader ->
+                        val sb = StringBuilder()
+                        var line: String? = reader.readLine()
+                        while (line != null) {
+                            sb.append(line)
+                            line = reader.readLine()
+                        }
+                        content = sb.toString()
+                    }
+                }
+            }
+        }
+
+        return content
+    }
+}