Initial

app/src/main/java/net/mezimmah/wkt9/WKT9.kt

@@ -0,0 +1,310 @@
+package net.mezimmah.wkt9
+
+import android.annotation.SuppressLint
+import android.inputmethodservice.InputMethodService
+import android.text.InputType
+import android.util.Log
+import android.view.KeyEvent
+import android.view.View
+import android.view.ViewConfiguration
+import android.view.inputmethod.EditorInfo
+import android.widget.LinearLayout
+import android.widget.TextView
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.Job
+import kotlinx.coroutines.SupervisorJob
+import kotlinx.coroutines.launch
+import net.mezimmah.wkt9.dao.SettingDao
+import net.mezimmah.wkt9.dao.WordDao
+import net.mezimmah.wkt9.db.AppDatabase
+import net.mezimmah.wkt9.inputmode.InputMode
+import net.mezimmah.wkt9.inputmode.AlphaInputMode
+import net.mezimmah.wkt9.inputmode.NumericInputMode
+import net.mezimmah.wkt9.inputmode.WordInputMode
+import net.mezimmah.wkt9.inputmode.WKT9InputMode
+import net.mezimmah.wkt9.keypad.KeyCodeMapping
+import net.mezimmah.wkt9.keypad.KeyEventResult
+import net.mezimmah.wkt9.keypad.KeyLayout
+import net.mezimmah.wkt9.keypad.Keypad
+import net.mezimmah.wkt9.t9.T9
+import java.lang.StringBuilder
+
+class WKT9: InputMethodService() {
+    private val tag = "WKT9"
+
+    // Dao - Database
+    private lateinit var db: AppDatabase
+    private lateinit var wordDao: WordDao
+    private lateinit var settingDao: SettingDao
+
+    // Coroutines
+    private val job = SupervisorJob()
+    private val scope = CoroutineScope(Dispatchers.Main + job)
+    private var queryJob: Job? = null
+
+    private var cursorPosition = 0
+    private var longPressTimeout = 700
+
+    // Keypad
+    private lateinit var keypad: Keypad
+
+    // T9
+    private lateinit var t9: T9
+
+    // Input
+    private var languageTag = "en_US"
+    private var lastInputMode: WKT9InputMode = WKT9InputMode.WORD
+    private var inputMode: InputMode? = null
+    private lateinit var alphaInputMode: AlphaInputMode
+    private lateinit var numericInputMode: NumericInputMode
+    private lateinit var wordInputMode: WordInputMode
+    private var composing = false
+    private val candidates: MutableList<String> = mutableListOf()
+    private var candidateIndex = 0
+
+    // UI
+    private lateinit var inputView: View
+
+    override fun onCreate() {
+        Log.d(tag, "WKT9 is loading")
+
+        db = AppDatabase.getInstance(this)
+        wordDao = db.getWordDao()
+        settingDao = db.getSettingDao()
+        keypad  = Keypad(KeyCodeMapping(KeyCodeMapping.basic), KeyLayout.en_US)
+        t9 = T9(this, keypad, settingDao, wordDao)
+        alphaInputMode = AlphaInputMode()
+        numericInputMode = NumericInputMode()
+        wordInputMode = WordInputMode()
+        longPressTimeout = ViewConfiguration.getLongPressTimeout()
+
+        t9.initializeWords(languageTag)
+
+        super.onCreate()
+    }
+
+    @SuppressLint("InflateParams")
+    override fun onCreateInputView(): View {
+        inputView = layoutInflater.inflate(R.layout.suggestions, null)
+
+        return inputView
+    }
+
+    override fun onFinishInput() {
+        super.onFinishInput()
+
+        inputMode = null
+        cursorPosition = 0
+    }
+
+    override fun onKeyDown(keyCode: Int, event: KeyEvent?): Boolean {
+        val key = keypad.getKey(keyCode) ?: return super.onKeyDown(keyCode, event)
+        val repeatCount = event?.repeatCount ?: 0
+
+        return inputMode?.let {
+            val keyEventResult =
+                if (repeatCount > 0) it.onKeyDownRepeatedly(key, repeatCount)
+                else {
+                    event?.startTracking()
+                    it.onKeyDown(key)
+                }
+
+            handleKeyEventResult(keyEventResult)
+        } ?: super.onKeyDown(keyCode, event)
+    }
+
+    override fun onKeyUp(keyCode: Int, event: KeyEvent?): Boolean {
+        val key = keypad.getKey(keyCode) ?: return super.onKeyUp(keyCode, event)
+        val eventTime = event?.eventTime ?: 0
+        val downTime = event?.downTime ?: 0
+        val keyDownMS = eventTime - downTime
+
+        return inputMode?.let {
+            val keyEventResult =
+                if (keyDownMS >= longPressTimeout) it.afterKeyLongDown(key, keyDownMS)
+                else it.afterKeyDown(key)
+
+            handleKeyEventResult(keyEventResult)
+        } ?: super.onKeyUp(keyCode, event)
+    }
+
+    override fun onKeyLongPress(keyCode: Int, event: KeyEvent?): Boolean {
+        val key = keypad.getKey(keyCode) ?: return super.onKeyLongPress(keyCode, event)
+
+        return inputMode?.let {
+            val keyEventResult = it.onKeyLongDown(key)
+
+            handleKeyEventResult(keyEventResult)
+        } ?: super.onKeyLongPress(keyCode, event)
+    }
+
+    override fun onStartInput(attribute: EditorInfo?, restarting: Boolean) {
+        val inputType = attribute?.inputType?.and(InputType.TYPE_MASK_CLASS) ?: 0
+
+        cursorPosition = attribute?.initialSelEnd ?: 0
+
+        when (inputType) {
+            InputType.TYPE_CLASS_DATETIME,
+            InputType.TYPE_CLASS_NUMBER,
+            InputType.TYPE_CLASS_PHONE -> enableInputMode(WKT9InputMode.NUMERIC, inputType)
+            InputType.TYPE_CLASS_TEXT ->
+                if (lastInputMode == WKT9InputMode.ALPHA) enableInputMode(WKT9InputMode.ALPHA, inputType)
+                else enableInputMode(WKT9InputMode.WORD, inputType)
+            else -> Log.d(tag, "Mode without input...")
+        }
+
+        super.onStartInput(attribute, restarting)
+    }
+
+    override fun onUpdateSelection(
+        oldSelStart: Int,
+        oldSelEnd: Int,
+        newSelStart: Int,
+        newSelEnd: Int,
+        candidatesStart: Int,
+        candidatesEnd: Int
+    ) {
+        cursorPosition = newSelEnd
+
+        super.onUpdateSelection(
+            oldSelStart,
+            oldSelEnd,
+            newSelStart,
+            newSelEnd,
+            candidatesStart,
+            candidatesEnd
+        )
+    }
+
+    private fun cancelComposing() {
+        composing = false
+
+        currentInputConnection.let {
+            it.setComposingText("", 1)
+            it.finishComposingText()
+        }
+    }
+
+    private fun clearCandidates() {
+        clearCandidateUI()
+
+        candidates.clear()
+        candidateIndex = 0
+    }
+
+    private fun clearCandidateUI() {
+        val candidatesView = inputView.findViewById<LinearLayout>(R.id.suggestions)
+
+        candidatesView.removeAllViews()
+    }
+
+    private fun commitText(text: CharSequence, start: Int, end: Int, cursorPosition: Int): Boolean {
+        return (markComposingRegion(start, end) && composeText(text, cursorPosition) && finishComposingText())
+    }
+
+    private fun composeText(text: CharSequence, cursorPosition: Int = 1): Boolean {
+        if (!composing) return false
+
+        return currentInputConnection?.setComposingText(text, cursorPosition) ?: false
+    }
+
+    private fun deleteText(beforeCursor: Int, afterCursor: Int): Boolean {
+        val cursorPosition = currentInputEditorInfo?.initialSelEnd ?: 0
+
+        return commitText("", cursorPosition - beforeCursor, cursorPosition + afterCursor, 1)
+    }
+
+    // Todo: inputType
+    private fun enableInputMode(mode: WKT9InputMode, inputType: Int) {
+        lastInputMode = mode
+
+        inputMode = when(mode) {
+            WKT9InputMode.ALPHA -> alphaInputMode
+            WKT9InputMode.NUMERIC -> numericInputMode
+            WKT9InputMode.WORD -> wordInputMode
+        }
+    }
+
+    private fun finishComposingText(): Boolean {
+        return if (composing) {
+            composing = false
+            currentInputConnection?.finishComposingText() ?: false
+        } else false
+    }
+
+    private fun handleKeyEventResult(res: KeyEventResult): Boolean {
+        if (res.finishComposing) finishComposingText()
+        if (res.startComposing) markComposingRegion()
+        if (!res.codeWord.isNullOrEmpty()) onCodeWordUpdate(res.codeWord)
+        if (!res.candidates.isNullOrEmpty()) onCandidates(res.candidates)
+        if (res.right) onRight()
+
+        return res.consumed
+    }
+
+    private fun loadCandidates(highLight: Int? = null) {
+        val candidatesView = inputView.findViewById<LinearLayout>(R.id.suggestions)
+
+        candidates.forEachIndexed { index, candidate ->
+            val layout = if (index == highLight) R.layout.current_suggestion else R.layout.suggestion
+            val candidateView = layoutInflater.inflate(layout, null) as TextView
+
+            candidateView.text  = candidate
+
+            candidatesView.addView(candidateView)
+        }
+    }
+
+    private fun markComposingRegion(start: Int? = null, end: Int? = null): Boolean {
+        if (composing) return false
+
+        val composeStart = start ?: cursorPosition
+        val composeEnd = end ?: cursorPosition
+
+        composing = currentInputConnection?.setComposingRegion(composeStart, composeEnd) ?: false
+
+        return composing
+    }
+
+    private fun onCandidates(candidates: List<String>) {
+        clearCandidates()
+
+        candidates.forEach {
+            this.candidates.add(it)
+        }
+
+        loadCandidates(candidateIndex)
+        composeText(candidates[candidateIndex])
+    }
+
+    private fun onCodeWordUpdate(codeWord: StringBuilder) {
+        clearCandidates()
+
+        queryJob?.cancel()
+        queryJob = scope.launch {
+            val hasCandidates = queryT9Candidates(codeWord, 10)
+
+            if (!hasCandidates) return@launch
+
+            loadCandidates(candidateIndex)
+            composeText(candidates[candidateIndex], 1)
+        }
+    }
+
+    private fun onRight() {
+        if (candidates.isEmpty()) return
+
+        Log.d(tag, "Let's have some fun!!")
+    }
+
+    private suspend fun queryT9Candidates(codeWord: StringBuilder, limit: Int = 10): Boolean {
+        val words = wordDao.findCandidates(codeWord.toString(), limit)
+
+        words.forEach {
+            candidates.add(it.word)
+        }
+
+        return words.isNotEmpty()
+    }
+}

app/src/main/java/net/mezimmah/wkt9/dao/SettingDao.kt

@@ -0,0 +1,20 @@
+package net.mezimmah.wkt9.dao
+
+import androidx.room.Dao
+import androidx.room.Delete
+import androidx.room.Insert
+import androidx.room.OnConflictStrategy
+import androidx.room.Query
+import net.mezimmah.wkt9.entity.Setting
+
+@Dao
+interface SettingDao {
+    @Insert(onConflict = OnConflictStrategy.REPLACE)
+    suspend fun insert(setting: Setting)
+
+    @Query("SELECT `setting`.* FROM `setting` WHERE `key` = :key")
+    suspend fun getByKey(key: String): Setting?
+
+    @Delete
+    suspend fun delete(setting: Setting)
+}

app/src/main/java/net/mezimmah/wkt9/dao/WordDao.kt

@@ -0,0 +1,26 @@
+package net.mezimmah.wkt9.dao
+
+import androidx.room.Dao
+import androidx.room.Delete
+import androidx.room.Insert
+import androidx.room.Query
+import net.mezimmah.wkt9.entity.Word
+
+@Dao
+interface WordDao {
+    @Insert
+    suspend fun insert(vararg words: Word)
+
+    @Delete
+    fun delete(word: Word)
+
+    @Query("SELECT * FROM word")
+    fun getAll(): List<Word>
+
+    @Query("SELECT * FROM word WHERE code LIKE :code || '%' " +
+            "ORDER BY length, weight DESC LIMIT :limit")
+    suspend fun findCandidates(code: String, limit: Int = 10): List<Word>
+
+    @Query("UPDATE word SET weight = weight + 1 WHERE word=:word")
+    suspend fun increaseWeight(word: String)
+}

app/src/main/java/net/mezimmah/wkt9/db/AppDatabase.kt

@@ -0,0 +1,37 @@
+package net.mezimmah.wkt9.db
+
+import android.content.Context
+import androidx.room.Database
+import androidx.room.Room
+import androidx.room.RoomDatabase
+import net.mezimmah.wkt9.entity.*
+import net.mezimmah.wkt9.dao.*
+
+@Database(entities = [Word::class, Setting::class], version = 1)
+abstract class AppDatabase : RoomDatabase() {
+    abstract fun getWordDao(): WordDao
+    abstract fun getSettingDao(): SettingDao
+
+    companion object {
+        @Volatile
+        private var INSTANCE: AppDatabase? = null
+
+        fun getInstance(context: Context): AppDatabase {
+            synchronized(this) {
+                var instance = INSTANCE
+
+                if (instance == null) {
+                    instance = Room.databaseBuilder(
+                        context.applicationContext,
+                        AppDatabase::class.java,
+                        "wkt9_db"
+                    )
+                        .fallbackToDestructiveMigration()
+                        .build()
+                    INSTANCE = instance
+                }
+                return instance
+            }
+        }
+    }
+}

app/src/main/java/net/mezimmah/wkt9/entity/Setting.kt

@@ -0,0 +1,18 @@
+package net.mezimmah.wkt9.entity
+
+import androidx.room.Entity
+import androidx.room.Index
+import androidx.room.PrimaryKey
+import net.mezimmah.wkt9.dao.SettingDao
+
+@Entity(indices = [Index(value = ["key"], unique = true)])
+data class Setting(
+    @PrimaryKey val key: String,
+    var value: String
+) {
+    companion object {
+        suspend fun set(key: String, value: String, dao: SettingDao) {
+            dao.insert(Setting(key, value))
+        }
+    }
+}

app/src/main/java/net/mezimmah/wkt9/entity/Word.kt

@@ -0,0 +1,17 @@
+package net.mezimmah.wkt9.entity
+
+import androidx.room.Entity
+import androidx.room.Index
+import androidx.room.PrimaryKey
+
+@Entity(indices = [Index(value = ["word"]), Index(value = ["code"]), Index(value = ["locale"])])
+data class Word(
+    var word: String,
+    var code: String,
+    var length: Int,
+    var weight: Int,
+    var locale: String
+) {
+    @PrimaryKey(autoGenerate = true)
+    var id: Int = 0
+}

app/src/main/java/net/mezimmah/wkt9/exception/MissingLetterCode.kt

@@ -0,0 +1,5 @@
+package net.mezimmah.wkt9.exception
+
+import java.lang.Exception
+
+class MissingLetterCode(message: String): Exception(message)

app/src/main/java/net/mezimmah/wkt9/inputmode/AlphaInputMode.kt

@@ -0,0 +1,26 @@
+package net.mezimmah.wkt9.inputmode
+
+import net.mezimmah.wkt9.keypad.Key
+import net.mezimmah.wkt9.keypad.KeyEventResult
+
+class AlphaInputMode: InputMode {
+    override fun onKeyDown(key: Key): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun onKeyLongDown(key: Key): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun onKeyDownRepeatedly(key: Key, repeat: Int): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun afterKeyDown(key: Key): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun afterKeyLongDown(key: Key, keyDownMS: Long): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+}

app/src/main/java/net/mezimmah/wkt9/inputmode/InputMode.kt

@@ -0,0 +1,16 @@
+package net.mezimmah.wkt9.inputmode
+
+import net.mezimmah.wkt9.keypad.Key
+import net.mezimmah.wkt9.keypad.KeyEventResult
+
+interface InputMode {
+    fun onKeyDown(key: Key): KeyEventResult
+
+    fun onKeyLongDown(key: Key): KeyEventResult
+
+    fun onKeyDownRepeatedly(key: Key, repeat: Int): KeyEventResult
+
+    fun afterKeyDown(key: Key): KeyEventResult
+
+    fun afterKeyLongDown(key: Key, keyDownMS: Long): KeyEventResult
+}

app/src/main/java/net/mezimmah/wkt9/inputmode/NumericInputMode.kt

@@ -0,0 +1,47 @@
+package net.mezimmah.wkt9.inputmode
+
+import android.util.Log
+import net.mezimmah.wkt9.keypad.Command
+import net.mezimmah.wkt9.keypad.KeyCommandResolver
+import net.mezimmah.wkt9.keypad.Key
+import net.mezimmah.wkt9.keypad.KeyEventResult
+import java.lang.StringBuilder
+
+class NumericInputMode: InputMode {
+    private val tag = "WKT9"
+    private val keyCommandResolver: KeyCommandResolver = KeyCommandResolver.getBasic()
+    private val codeWord = StringBuilder()
+
+    override fun onKeyDown(key: Key): KeyEventResult {
+        return when(keyCommandResolver.getCommand(key)) {
+            Command.CHARACTER -> buildCodeWord(key)
+//            Command.SELECT -> true
+//            Command.DELETE -> delete()
+//            Command.SPACE -> finalizeWordOrSentence()
+//            Command.CYCLE_CANDIDATES -> cycleCandidates()
+            else -> KeyEventResult(consumed = true)
+        }
+    }
+
+    override fun onKeyLongDown(key: Key): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun onKeyDownRepeatedly(key: Key, repeat: Int): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun afterKeyDown(key: Key): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    override fun afterKeyLongDown(key: Key, keyDownMS: Long): KeyEventResult {
+        return KeyEventResult(consumed = false)
+    }
+
+    private fun buildCodeWord(key: Key): KeyEventResult {
+        Log.d(tag, "GR8!")
+
+        return KeyEventResult(consumed = true)
+    }
+}

app/src/main/java/net/mezimmah/wkt9/inputmode/WKT9InputMode.kt

@@ -0,0 +1,7 @@
+package net.mezimmah.wkt9.inputmode
+
+enum class WKT9InputMode(val idx: Int) {
+    WORD(0),
+    ALPHA(1),
+    NUMERIC(2)
+}

app/src/main/java/net/mezimmah/wkt9/inputmode/WordInputMode.kt

@@ -0,0 +1,95 @@
+package net.mezimmah.wkt9.inputmode
+
+import android.util.Log
+import net.mezimmah.wkt9.keypad.Command
+import net.mezimmah.wkt9.keypad.Key
+import net.mezimmah.wkt9.keypad.KeyCommandResolver
+import net.mezimmah.wkt9.keypad.KeyEventResult
+import java.lang.StringBuilder
+
+class WordInputMode: InputMode {
+    private val tag = "WKT9"
+    private val keyCommandResolver: KeyCommandResolver = KeyCommandResolver.getBasic()
+    private val codeWord = StringBuilder()
+    private var newKey = true
+    private var keyIndex = 0
+    private var lastKey: Key? = null
+
+    override fun onKeyDown(key: Key): KeyEventResult {
+        keyStats(key)
+
+        return when(keyCommandResolver.getCommand(key)) {
+            Command.CHARACTER -> buildCodeWord(key)
+//            Command.SELECT -> true
+//            Command.DELETE -> delete()
+            Command.SPACE -> finalizeWordOrSentence()
+//            Command.CYCLE_CANDIDATES -> cycleCandidates()
+            else -> KeyEventResult()
+        }
+    }
+
+    override fun onKeyLongDown(key: Key): KeyEventResult {
+//        Log.d(tag, "onKeyLongDown")
+
+        return KeyEventResult()
+    }
+
+    override fun onKeyDownRepeatedly(key: Key, repeat: Int): KeyEventResult {
+//        Log.d(tag, "onKeyDownRepeatedly")
+
+        return KeyEventResult()
+    }
+
+    override fun afterKeyDown(key: Key): KeyEventResult {
+//        Log.d(tag, "afterKeyDown")
+
+        return KeyEventResult()
+    }
+
+    override fun afterKeyLongDown(key: Key, keyDownMS: Long): KeyEventResult {
+//        Log.d(tag, "afterKeyLongDown")
+
+        return KeyEventResult()
+    }
+
+    private fun buildCodeWord(key: Key): KeyEventResult {
+        val startComposing = codeWord.isEmpty()
+
+        codeWord.append(key.code)
+
+        return KeyEventResult(
+            codeWord = codeWord,
+            finishComposing = startComposing,
+            startComposing = startComposing
+        )
+    }
+
+    private fun finalizeWordOrSentence(): KeyEventResult {
+        if (!newKey) return KeyEventResult(right = true)
+
+        val finishComposing = codeWord.isNotEmpty()
+
+        codeWord.clear()
+
+        return KeyEventResult(
+            finishComposing = finishComposing,
+            startComposing = true,
+            candidates = listOf(" ", ". ", "? ", "! ", ", ", ": ", "; ")
+        )
+    }
+
+    private fun keyStats(key: Key) {
+        when (key != lastKey) {
+            true -> {
+                newKey = true
+                keyIndex = 0
+            }
+            false -> {
+                newKey = false
+                keyIndex++
+            }
+        }
+
+        lastKey = key
+    }
+}

app/src/main/java/net/mezimmah/wkt9/keypad/Command.kt

@@ -0,0 +1,16 @@
+package net.mezimmah.wkt9.keypad
+
+enum class Command {
+    CHARACTER,
+    NUMBER,
+    SPACE,
+    NEWLINE,
+    DELETE,
+    CYCLE_CANDIDATES,
+    SELECT,
+    SHIFT_MODE,
+    SWITCH_MODE,
+    NAVIGATE,
+    RECORD,
+    TRANSCRIBE
+}

app/src/main/java/net/mezimmah/wkt9/keypad/Key.kt

@@ -0,0 +1,22 @@
+package net.mezimmah.wkt9.keypad
+
+enum class Key(val code: Char) {
+    N0('0'),
+    N1('1'),
+    N2('2'),
+    N3('3'),
+    N4('4'),
+    N5('5'),
+    N6('6'),
+    N7('7'),
+    N8('8'),
+    N9('9'),
+    STAR('*'),
+    POUND('#'),
+    UP('a'),
+    DOWN('b'),
+    LEFT('c'),
+    RIGHT('d'),
+    SELECT('e'),
+    DELETE('f')
+}

app/src/main/java/net/mezimmah/wkt9/keypad/KeyCodeMapping.kt

@@ -0,0 +1,50 @@
+package net.mezimmah.wkt9.keypad
+
+import java.util.Properties
+
+class KeyCodeMapping(
+    private val keyMap: Map<Int, Key>,
+) {
+
+    fun key(keyCode: Int): Key? {
+        return keyMap[keyCode]
+    }
+
+    companion object {
+        val basic = mapOf(
+            7 to Key.N0,
+            8 to Key.N1,
+            9 to Key.N2,
+            10 to Key.N3,
+            11 to Key.N4,
+            12 to Key.N5,
+            13 to Key.N6,
+            14 to Key.N7,
+            15 to Key.N8,
+            16 to Key.N9,
+            17 to Key.STAR,
+            18 to Key.POUND,
+            82 to Key.DELETE,
+            19 to Key.UP,
+            20 to Key.DOWN,
+            21 to Key.LEFT,
+            22 to Key.RIGHT,
+            23 to Key.SELECT
+        )
+
+        fun fromProperties(props: Properties): KeyCodeMapping {
+            val keyMap = HashMap<Int, Key>()
+
+            this.basic.forEach {
+                val keyCode = props.getProperty("key.${it.value.name}")?.toInt() ?: it.key
+                keyMap[keyCode] = it.value
+            }
+
+            return KeyCodeMapping(keyMap)
+        }
+
+        fun default(): KeyCodeMapping {
+            return KeyCodeMapping(basic)
+        }
+    }
+}

app/src/main/java/net/mezimmah/wkt9/keypad/KeyCommandResolver.kt

@@ -0,0 +1,83 @@
+package net.mezimmah.wkt9.keypad
+
+class KeyCommandResolver (
+    private val onShort: HashMap<Key, Command> = HashMap(mapOf()),
+    private val onLong: HashMap<Key, Command> = HashMap(mapOf()),
+    private val afterShort: HashMap<Key, Command> = HashMap(mapOf()),
+    private val afterLong: HashMap<Key, Command> = HashMap(mapOf()),
+    private val onRepeat: HashMap<Key, Command> = HashMap(mapOf()),
+    private val parent: KeyCommandResolver? = null
+) {
+    fun getCommand(key: Key, longPress: Boolean = false, after: Boolean = false, repeat: Int = 0): Command? {
+        val command = when {
+            repeat > 0 -> onRepeat[key]
+            (longPress && after) -> afterLong[key]
+            (longPress) -> onLong[key]
+            (after) -> afterShort[key]
+            else -> onShort[key]
+        }
+
+        return when (command) {
+            null -> parent?.getCommand(key, longPress, after)
+            else -> command
+        }
+    }
+
+    companion object {
+        fun getBasic(): KeyCommandResolver {
+            return KeyCommandResolver(
+                onShort = HashMap(mapOf(
+                    Key.N0 to Command.SPACE,
+
+                    Key.N1 to Command.CHARACTER,
+                    Key.N2 to Command.CHARACTER,
+                    Key.N3 to Command.CHARACTER,
+                    Key.N4 to Command.CHARACTER,
+                    Key.N5 to Command.CHARACTER,
+                    Key.N6 to Command.CHARACTER,
+                    Key.N7 to Command.CHARACTER,
+                    Key.N8 to Command.CHARACTER,
+                    Key.N9 to Command.CHARACTER,
+
+                    Key.LEFT to Command.NAVIGATE,
+                    Key.RIGHT to Command.NAVIGATE,
+                    Key.UP to Command.NAVIGATE,
+                    Key.DOWN to Command.NAVIGATE,
+
+                    Key.STAR to Command.CYCLE_CANDIDATES,
+                    Key.DELETE to Command.DELETE,
+                    Key.SELECT to Command.SELECT
+                )),
+
+                onLong = HashMap(mapOf(
+                    Key.N0 to Command.NUMBER,
+                    Key.N1 to Command.NUMBER,
+                    Key.N2 to Command.NUMBER,
+                    Key.N3 to Command.NUMBER,
+                    Key.N4 to Command.NUMBER,
+                    Key.N5 to Command.NUMBER,
+                    Key.N6 to Command.NUMBER,
+                    Key.N7 to Command.NUMBER,
+                    Key.N8 to Command.NUMBER,
+                    Key.N9 to Command.NUMBER,
+
+                    Key.POUND to Command.SWITCH_MODE,
+                    Key.SELECT to Command.RECORD
+                )),
+
+                afterShort = HashMap(mapOf(
+                    Key.POUND to Command.SHIFT_MODE,
+                )),
+
+                afterLong = HashMap(mapOf(
+                    Key.DELETE to Command.DELETE,
+                    Key.SELECT to Command.TRANSCRIBE,
+                )),
+
+                onRepeat = HashMap(mapOf(
+                    Key.DELETE to Command.DELETE,
+                ))
+            )
+        }
+    }
+}

app/src/main/java/net/mezimmah/wkt9/keypad/KeyEventResult.kt

@@ -0,0 +1,12 @@
+package net.mezimmah.wkt9.keypad
+
+import java.lang.StringBuilder
+
+data class KeyEventResult(
+    val consumed: Boolean = true,
+    val finishComposing: Boolean = false,
+    val startComposing: Boolean = false,
+    val codeWord: StringBuilder? = null,
+    val candidates: List<String>? = null,
+    val right: Boolean = false
+)

app/src/main/java/net/mezimmah/wkt9/keypad/KeyLayout.kt

@@ -0,0 +1,33 @@
+package net.mezimmah.wkt9.keypad
+
+object KeyLayout {
+    // Map for number input mode
+    val numeric = mapOf(
+        Key.N0 to '0',
+        Key.N1 to '1',
+        Key.N2 to '2',
+        Key.N3 to '3',
+        Key.N4 to '4',
+        Key.N5 to '5',
+        Key.N6 to '6',
+        Key.N7 to '7',
+        Key.N8 to '8',
+        Key.N9 to '9',
+    )
+
+    val en_US = mapOf(
+        Key.N1 to listOf('.','?','!',',','-','\'','"','@','$','/','%',':','(',')'),
+        Key.N2 to listOf('a','b','c','ä','æ','å','à','á','â','ã','ç'),
+        Key.N3 to listOf('d','e','f','è','é','ê','ë','đ'),
+        Key.N4 to listOf('g','h','i','ì','í','î','ï'),
+        Key.N5 to listOf('j','k','l','£'),
+        Key.N6 to listOf('m','n','o','ö','ø','ò','ó','ô','õ','õ'),
+        Key.N7 to listOf('p','q','r','s','ß','$'),
+        Key.N8 to listOf('t','u','v','ù','ú','û','ü'),
+        Key.N9 to listOf('w','x','y','z','ý','þ'),
+        Key.STAR to listOf('*'),
+        Key.POUND to listOf('#'),
+    )
+
+    val nonAlphaNumeric = setOf('*','#','\'','"','.','?','!',',','-','@','$','/','%',':','(',')')
+}

app/src/main/java/net/mezimmah/wkt9/keypad/Keypad.kt

@@ -0,0 +1,75 @@
+package net.mezimmah.wkt9.keypad
+
+import net.mezimmah.wkt9.exception.MissingLetterCode
+import java.lang.StringBuilder
+
+class Keypad(
+    private val keyCodeMapping: KeyCodeMapping,
+    private val letterLayout: Map<Key, List<Char>>
+) {
+    private lateinit var letterKeyMap: Map<Char, Key>
+    private lateinit var keyIsLetterMap: Map<Key, Boolean>
+    private var emojiCodes = mapOf(
+        "❤️" to "143278",
+        "\uD83D\uDE18" to "15477"
+    )
+
+    init {
+        initKeyMaps(letterLayout)
+    }
+
+    fun getKey(keyCode: Int): Key? {
+        return keyCodeMapping.key(keyCode)
+    }
+
+    fun getCharacter(key: Key, idx: Int): Char {
+        val chars = letterLayout[key]!!
+        val length = chars.size
+        val wrappedIdx = idx % (length - 1) + 1
+        return chars[wrappedIdx]
+    }
+
+    fun getDigit(key: Key): Char? {
+        val letters = letterLayout[key] ?: return null
+        return if (letters.isNotEmpty() && letters[0].isDigit()) letters[0] else null
+    }
+
+    fun getCodeForWord(word: String): String {
+        if (emojiCodes.keys.contains(word)) return emojiCodes[word]!!
+
+        val builder = StringBuilder()
+
+        for (letter in word) {
+            val code = codeForLetter(letter) ?: throw MissingLetterCode("No code found for '$letter'")
+
+            builder.append(code)
+        }
+
+        return builder.toString()
+    }
+
+    private fun codeForLetter(letter: Char): Char? {
+        return letterKeyMap[letter]?.code
+    }
+
+    private fun initKeyMaps(layout: Map<Key, List<Char>>): Map<Char, Key> {
+        val letterKeyMap = HashMap<Char, Key>()
+        val keyIsLetterMap = HashMap<Key, Boolean>()
+
+        this.letterKeyMap = letterKeyMap
+        this.keyIsLetterMap = keyIsLetterMap
+
+        for ((key, characters) in layout) {
+            for (char in characters) {
+                letterKeyMap[char] = key
+
+                if (!KeyLayout.nonAlphaNumeric.contains(char)) {
+                    letterKeyMap[char.uppercaseChar()] = key
+                    keyIsLetterMap[key] = true
+                } else if (keyIsLetterMap[key] == null) keyIsLetterMap[key] = false
+            }
+        }
+
+        return letterKeyMap
+    }
+}

app/src/main/java/net/mezimmah/wkt9/t9/T9.kt

@@ -0,0 +1,98 @@
+package net.mezimmah.wkt9.t9
+
+import android.util.Log
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.SupervisorJob
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.runBlocking
+import net.mezimmah.wkt9.WKT9
+import net.mezimmah.wkt9.dao.SettingDao
+import net.mezimmah.wkt9.dao.WordDao
+import net.mezimmah.wkt9.entity.Setting
+import net.mezimmah.wkt9.entity.Word
+import net.mezimmah.wkt9.exception.MissingLetterCode
+import net.mezimmah.wkt9.keypad.Keypad
+
+class T9 (
+    private val context: WKT9,
+    private val keypad: Keypad,
+    private val settingDao: SettingDao,
+    private val wordDao: WordDao
+) {
+    // Debugging
+    private val tag = "WKT9"
+
+    // Coroutines
+    private val job = SupervisorJob()
+    private val scope = CoroutineScope(Dispatchers.IO + job)
+
+    fun initializeWords(locale: CharSequence) {
+        scope.launch {
+            val setting = settingDao.getByKey("initialized")
+
+            if (setting == null) {
+                Log.d(tag,"Initializing word database...")
+                readWords(locale.toString())
+                Setting.set("initialized", "t", settingDao)
+            } else Log.d(tag, "Word database already initialized")
+        }
+    }
+
+    private fun getWord(word: String, weight: Int = 1, locale: String) = Word(
+        word = word,
+        code = keypad.getCodeForWord(word),
+        length = word.length,
+        weight = weight,
+        locale = locale
+    )
+
+    private fun readWords(locale: String) {
+        val fileName = "$locale/words.txt"
+        val batchSize = 1000
+        val wordBatch = ArrayList<Word>(batchSize)
+
+        context.assets.open(fileName).bufferedReader().useLines { lines ->
+            for (chunk in lines.chunked(batchSize)) {
+                var arrayListIndex = 0
+
+                wordBatch.clear()
+
+                chunk.forEach { line ->
+                    val parts = line.split("\\s".toRegex())
+
+                    try {
+                        wordBatch.add(
+                            arrayListIndex,
+
+                            when (parts.size) {
+                                // Emoji support
+                                3 -> Word(
+                                    word = parts[0],
+                                    code = parts[2],
+                                    length = 1,
+                                    weight = parts[1].toInt(),
+                                    locale = locale
+                                )
+                                // Regular dictionary words
+                                else -> getWord(
+                                    word = parts[0],
+                                    weight = if (parts.size > 1) parts[1].toInt() else 0,
+                                    locale = locale
+                                )
+                            }
+                        )
+
+                        arrayListIndex++
+                    } catch (ex: MissingLetterCode) {
+                        Log.w(tag, "Problem adding ${parts[0]}: " + ex.message)
+                    }
+                }
+
+                runBlocking {
+                    wordDao.insert(*wordBatch.toTypedArray())
+                }
+            }
+        }
+    }
+}

app/src/main/java/net/mezimmah/wkt9/ui/Suggestions.kt

@@ -0,0 +1,9 @@
+package net.mezimmah.wkt9.ui
+
+import android.content.Context
+import android.util.AttributeSet
+import android.widget.RelativeLayout
+
+class Suggestions(context: Context, attrs: AttributeSet): RelativeLayout(context, attrs) {
+    private val tag = "WKT9"
+}