forked from omni/OmniCards
81 lines
2.5 KiB
Kotlin
81 lines
2.5 KiB
Kotlin
package ru.omni_devel.cards
|
|
|
|
import android.content.ContentValues
|
|
import android.content.Context
|
|
import android.database.sqlite.SQLiteDatabase
|
|
import android.database.sqlite.SQLiteOpenHelper
|
|
import com.google.zxing.BarcodeFormat
|
|
|
|
class DbHelper(val context: Context, val factory: SQLiteDatabase.CursorFactory?) : SQLiteOpenHelper(context, "omni_cards", factory, 1) {
|
|
override fun onCreate(db: SQLiteDatabase?) {
|
|
db!!.execSQL("CREATE TABLE IF NOT EXISTS cards (id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT, codeType TEXT, codeValue TEXT)")
|
|
}
|
|
|
|
override fun onUpgrade(
|
|
db: SQLiteDatabase?,
|
|
p1: Int,
|
|
p2: Int
|
|
) {
|
|
db!!.execSQL("DROP TABLE IF EXISTS cards")
|
|
}
|
|
|
|
fun addCard(name: String, codeValue: String, codeType: BarcodeFormat) {
|
|
val values = ContentValues()
|
|
|
|
values.put("name", name)
|
|
values.put("codeValue", codeValue)
|
|
values.put("codeType", codeType.name)
|
|
|
|
val db = this.writableDatabase
|
|
|
|
db.insert("cards", null, values)
|
|
|
|
db.close()
|
|
}
|
|
|
|
fun getCards(): List<CardInfo> {
|
|
val db = this.readableDatabase
|
|
|
|
val cursor = db.rawQuery("SELECT * FROM cards", null)
|
|
val cards = mutableListOf<CardInfo>()
|
|
|
|
while (cursor.moveToNext()) {
|
|
cards.add(CardInfo(
|
|
id = cursor.getInt(cursor.getColumnIndexOrThrow("id")),
|
|
name = cursor.getString(cursor.getColumnIndexOrThrow("name")),
|
|
codeType = BarcodeFormat.valueOf(cursor.getString(cursor.getColumnIndexOrThrow("codeType"))),
|
|
codeValue = cursor.getString(cursor.getColumnIndexOrThrow("codeValue"))
|
|
))
|
|
}
|
|
|
|
cursor.close()
|
|
db.close()
|
|
|
|
return cards
|
|
}
|
|
|
|
fun getCard(id: Int): CardInfo? {
|
|
val db = this.readableDatabase
|
|
|
|
val cursor = db.rawQuery("SELECT * FROM cards WHERE id = ?", arrayOf(id.toString()))
|
|
|
|
if (!cursor.moveToFirst()) {
|
|
cursor.close()
|
|
db.close()
|
|
|
|
return null
|
|
}
|
|
|
|
val cardInfo = CardInfo(
|
|
id = cursor.getInt(cursor.getColumnIndexOrThrow("id")),
|
|
name = cursor.getString(cursor.getColumnIndexOrThrow("name")),
|
|
codeType = BarcodeFormat.valueOf(cursor.getString(cursor.getColumnIndexOrThrow("codeType"))),
|
|
codeValue = cursor.getString(cursor.getColumnIndexOrThrow("codeValue"))
|
|
)
|
|
|
|
cursor.close()
|
|
db.close()
|
|
|
|
return cardInfo
|
|
}
|
|
} |