Added lock

app/src/main/java/xyz/quaver/pupil/util/lock.kt

@@ -2,6 +2,11 @@ package xyz.quaver.pupil.util
 
 import android.content.Context
 import android.content.ContextWrapper
+import androidx.core.content.ContextCompat
+import kotlinx.serialization.*
+import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.JsonConfiguration
+import java.io.File
 import java.security.MessageDigest
 
 fun hash(password: String): String {
@@ -21,10 +26,20 @@ fun hashWithSalt(password: String): Pair<String, String> {
 
 val source = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
 
-data class Lock(private val type: Type, private val hash: String, private val salt: String) {
+@Serializable
+data class Lock(val type: Type, val hash: String, val salt: String) {
 
     enum class Type {
-        PATTERN
+        PATTERN,
+        PIN,
+        PASSWORD
+    }
+
+    companion object {
+        fun generate(type: Type, password: String): Lock {
+            val (hash, salt) = hashWithSalt(password)
+            return Lock(type, hash, salt)
+        }
     }
 
     fun match(password: String): Boolean {
@@ -34,6 +49,57 @@ data class Lock(private val type: Type, private val hash: String, private val sa
 
 class LockManager(base: Context): ContextWrapper(base) {
 
+    var locks: ArrayList<Lock>? = null
 
+    init {
+        load()
+    }
+
+    @UseExperimental(ImplicitReflectionSerializer::class)
+    private fun load() {
+        val lock = File(ContextCompat.getDataDir(this), "lock.json")
+
+        if (!lock.exists()) {
+            lock.createNewFile()
+            lock.writeText("[]")
+        }
+
+        locks = ArrayList(Json(JsonConfiguration.Stable).parseList(lock.readText()))
+    }
+
+    @UseExperimental(ImplicitReflectionSerializer::class)
+    private fun save() {
+        val lock = File(ContextCompat.getDataDir(this), "lock.json")
+
+        if (!lock.exists())
+            lock.createNewFile()
+
+        lock.writeText(Json(JsonConfiguration.Stable).stringify(locks?.toList() ?: listOf()))
+    }
+
+    fun add(lock: Lock) {
+        remove(lock.type)
+        locks?.add(lock)
+        save()
+    }
+
+    fun remove(type: Lock.Type) {
+        locks?.removeAll { it.type == type }
+        save()
+    }
+
+    fun check(password: String): Boolean? {
+        return locks?.any {
+            it.match(password)
+        }
+    }
+
+    fun empty(): Boolean {
+        return locks.isNullOrEmpty()
+    }
+
+    fun contains(type: Lock.Type): Boolean {
+        return locks?.any { it.type == type } ?: false
+    }
 
 }