Feature: 数据库表展示

app/src/main/java/me/kbai/zhenxunui/api/ApiService.kt

@@ -2,6 +2,7 @@ package me.kbai.zhenxunui.api
 
 import me.kbai.zhenxunui.model.BotBaseInfo
 import me.kbai.zhenxunui.model.BotMessageCount
+import me.kbai.zhenxunui.model.ExecuteSql
 import me.kbai.zhenxunui.model.FriendListItem
 import me.kbai.zhenxunui.model.GroupInfo
 import me.kbai.zhenxunui.model.GroupListItem
@@ -12,6 +13,9 @@ import me.kbai.zhenxunui.model.PluginInfo
 import me.kbai.zhenxunui.model.PluginSwitch
 import me.kbai.zhenxunui.model.RequestListResult
 import me.kbai.zhenxunui.model.SendMessage
+import me.kbai.zhenxunui.model.SqlLog
+import me.kbai.zhenxunui.model.TableColumn
+import me.kbai.zhenxunui.model.TableListItem
 import me.kbai.zhenxunui.model.UpdateGroup
 import me.kbai.zhenxunui.model.UpdatePlugin
 import me.kbai.zhenxunui.model.UserDetail
@@ -103,4 +107,16 @@ interface ApiService {
 
     @POST("manage/send_message")
     suspend fun sendMessage(@Body message: SendMessage): ApiResponse<*>
+
+    @GET("database/get_table_list")
+    suspend fun getTableList(): ApiResponse<List<TableListItem>>
+
+    @GET("database/get_table_column")
+    suspend fun getTableColumn(@Query("table_name") table: String): ApiResponse<List<TableColumn>>
+
+    @GET("database/get_sql_log")
+    suspend fun getSqlLog(): ApiResponse<List<SqlLog>>
+
+    @POST("database/exec_sql")
+    suspend fun executeSql(@Body sql: ExecuteSql): RawApiResponse
 }

app/src/main/java/me/kbai/zhenxunui/model/ExecuteSql.kt

@@ -0,0 +1,5 @@
+package me.kbai.zhenxunui.model
+
+data class ExecuteSql(
+    val sql: String
+)

app/src/main/java/me/kbai/zhenxunui/model/SqlLog.kt

@@ -0,0 +1,17 @@
+package me.kbai.zhenxunui.model
+
+import com.google.gson.annotations.SerializedName
+
+data class SqlLog(
+    val id: Int,
+    /**
+     * The time for executed sql. sample: `2024-03-06T06:54:48.829937+00:00`
+     */
+    @SerializedName("create_time")
+    val createTime: String,
+    @SerializedName("is_suc")
+    val success: Boolean,
+    val ip: String,
+    val result: String,
+    val sql: String
+)

app/src/main/java/me/kbai/zhenxunui/model/TableColumn.kt

@@ -0,0 +1,14 @@
+package me.kbai.zhenxunui.model
+
+import com.google.gson.annotations.SerializedName
+
+data class TableColumn(
+    @SerializedName("column_name")
+    val name: String,
+    @SerializedName("data_type")
+    val type: String,
+    @SerializedName("max_length")
+    val maxLength: Int?,
+    @SerializedName("is_nullable")
+    val nullable: String
+)

app/src/main/java/me/kbai/zhenxunui/model/TableListItem.kt

@@ -0,0 +1,6 @@
+package me.kbai.zhenxunui.model
+
+data class TableListItem(
+    val name: String,
+    val desc: String?
+)

app/src/main/java/me/kbai/zhenxunui/repository/ApiRepository.kt

@@ -114,6 +114,10 @@ object ApiRepository {
         }
     }
 
+    fun getTableList() = networkFlow { BotApi.service.getTableList() }
+
+    fun getTableColumn(table: String) = networkFlow { BotApi.service.getTableColumn(table) }
+
     private fun <T> networkFlow(
         tempData: T? = null,
         f: suspend () -> ApiResponse<T>

app/src/main/java/me/kbai/zhenxunui/ui/MainActivity.kt

@@ -41,6 +41,7 @@ class MainActivity : BaseActivity<ActivityMainBinding>() {
                 R.id.nav_plugin,
                 R.id.nav_friend_list,
                 R.id.nav_request,
+                R.id.nav_db_manage,
 //                R.id.nav_info
             ), viewBinding.drawerLayout
         )

app/src/main/java/me/kbai/zhenxunui/ui/db/DbManageFragment.kt

@@ -0,0 +1,43 @@
+package me.kbai.zhenxunui.ui.db
+
+import android.view.LayoutInflater
+import android.view.ViewGroup
+import androidx.fragment.app.viewModels
+import me.kbai.zhenxunui.base.BaseFragment
+import me.kbai.zhenxunui.databinding.FragmentDbManageBinding
+import me.kbai.zhenxunui.extends.setOnDebounceClickListener
+import me.kbai.zhenxunui.extends.viewLifecycleScope
+import me.kbai.zhenxunui.viewmodel.DbManageViewModel
+
+/**
+ * @author Sean on 2023/5/30
+ */
+class DbManageFragment : BaseFragment<FragmentDbManageBinding>() {
+
+    private val mViewModel by viewModels<DbManageViewModel>()
+
+    override fun getViewBinding(
+        inflater: LayoutInflater,
+        container: ViewGroup?
+    ): FragmentDbManageBinding = FragmentDbManageBinding.inflate(inflater, container, false)
+
+    override fun initView() = viewBinding.run {
+
+        srlRefresh.setOnRefreshListener {
+            mViewModel.requestTableList()
+        }
+
+        btnSql.setOnDebounceClickListener {
+            //TODO
+        }
+    }
+
+    override fun initData() {
+        mViewModel.tables.observe(viewLifecycleOwner) {
+            viewBinding.srlRefresh.isRefreshing = false
+            viewBinding.rvTables.adapter = DbTableAdapter(it, mViewModel, viewLifecycleScope)
+        }
+
+        mViewModel.requestTableList()
+    }
+}

app/src/main/java/me/kbai/zhenxunui/ui/db/DbTableAdapter.kt

@@ -0,0 +1,69 @@
+package me.kbai.zhenxunui.ui.db
+
+import android.view.LayoutInflater
+import android.view.ViewGroup
+import androidx.core.view.isVisible
+import androidx.core.view.size
+import androidx.lifecycle.Observer
+import androidx.recyclerview.widget.RecyclerView
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Job
+import kotlinx.coroutines.launch
+import me.kbai.zhenxunui.databinding.ItemDbManageTableBinding
+import me.kbai.zhenxunui.extends.setOnDebounceClickListener
+import me.kbai.zhenxunui.model.TableListItem
+import me.kbai.zhenxunui.viewmodel.DbManageViewModel
+
+class DbTableAdapter(
+    private val data: List<TableListItem>,
+    private val viewModel: DbManageViewModel,
+    private val coroutineScope: CoroutineScope
+) : RecyclerView.Adapter<DbTableAdapter.ItemViewHolder>() {
+
+    override fun onCreateViewHolder(parent: ViewGroup, viewType: Int) = ItemViewHolder(
+        ItemDbManageTableBinding.inflate(LayoutInflater.from(parent.context), parent, false)
+    )
+
+    override fun getItemCount() = data.size
+
+    override fun onBindViewHolder(holder: ItemViewHolder, position: Int) {
+        val item = data[position]
+
+        holder.binding.run {
+            tvTableName.text = item.name
+            tvDescription.text = item.desc
+
+            root.setOnDebounceClickListener {
+                elFields.toggleExpand()
+
+                if (rvFields.adapter != null) {
+                    return@setOnDebounceClickListener
+                }
+                coroutineScope.launch {
+                    viewModel.getColumns(item.name)
+                        .collect {
+                            if (!holder.recycled) {
+                                pbWaiting.isVisible = false
+                                rvFields.adapter = TableFieldAdapter(it)
+                                rvFields.post { elFields.expand() }
+                            }
+                        }
+                }
+            }
+
+            rvFields.adapter = null
+            elFields.collapse(false)
+            pbWaiting.isVisible = true
+            holder.recycled = false
+        }
+    }
+
+    override fun onViewRecycled(holder: ItemViewHolder) {
+        holder.recycled = true
+    }
+
+    class ItemViewHolder(val binding: ItemDbManageTableBinding) :
+        RecyclerView.ViewHolder(binding.root) {
+        var recycled: Boolean = false
+    }
+}

app/src/main/java/me/kbai/zhenxunui/ui/db/TableFieldAdapter.kt

@@ -0,0 +1,37 @@
+package me.kbai.zhenxunui.ui.db
+
+import android.view.LayoutInflater
+import android.view.ViewGroup
+import androidx.recyclerview.widget.RecyclerView
+import me.kbai.zhenxunui.R
+import me.kbai.zhenxunui.databinding.ItemDbManageFieldBinding
+import me.kbai.zhenxunui.model.TableColumn
+
+class TableFieldAdapter(
+    private val data: List<TableColumn>
+) : RecyclerView.Adapter<TableFieldAdapter.ItemViewHolder>() {
+
+    override fun onCreateViewHolder(parent: ViewGroup, viewType: Int) = ItemViewHolder(
+        ItemDbManageFieldBinding.inflate(
+            LayoutInflater.from(parent.context), parent, false
+        )
+    )
+
+    override fun getItemCount() = data.size
+
+    override fun onBindViewHolder(holder: ItemViewHolder, position: Int) = holder.binding.run {
+        val item = data[position]
+
+        tvName.text = item.name
+        tvType.text = item.type
+        tvMaxLength.text =
+            tvMaxLength.context.getString(
+                R.string.max_length_format,
+                item.maxLength?.toString().orEmpty()
+            )
+        tvNullable.text = tvNullable.context.getString(R.string.nullable_format, item.nullable)
+    }
+
+    class ItemViewHolder(val binding: ItemDbManageFieldBinding) :
+        RecyclerView.ViewHolder(binding.root)
+}

app/src/main/java/me/kbai/zhenxunui/viewmodel/DbManageViewModel.kt

@@ -0,0 +1,43 @@
+package me.kbai.zhenxunui.viewmodel
+
+import androidx.lifecycle.LiveData
+import androidx.lifecycle.MutableLiveData
+import androidx.lifecycle.ViewModel
+import androidx.lifecycle.viewModelScope
+import androidx.webkit.internal.ApiFeature.M
+import kotlinx.coroutines.flow.flow
+import kotlinx.coroutines.launch
+import me.kbai.zhenxunui.extends.apiCollect
+import me.kbai.zhenxunui.model.TableColumn
+import me.kbai.zhenxunui.model.TableListItem
+import me.kbai.zhenxunui.repository.ApiRepository
+
+class DbManageViewModel : ViewModel() {
+
+    private val _tables: MutableLiveData<List<TableListItem>> = MutableLiveData()
+    val tables: LiveData<List<TableListItem>> = _tables
+
+    private val _columnMap: MutableMap<String, List<TableColumn>> = HashMap()
+
+    fun requestTableList() = viewModelScope.launch {
+        ApiRepository.getTableList().apiCollect {
+            if (it.success()) {
+                _tables.value = it.data ?: return@apiCollect
+                _columnMap.clear()
+            }
+        }
+    }
+
+    fun getColumns(table: String) = flow {
+        val data = _columnMap[table]
+        if (data != null) {
+            emit(data)
+            return@flow
+        }
+        val res = ApiRepository.getTableColumn(table).apiCollect()
+        if (res.success() && res.data != null) {
+            _columnMap[table] = res.data
+            emit(res.data)
+        }
+    }
+}