feat: log file upload

Author: cssxsh

src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/NetDiskFileSyncPlugin.kt

@@ -3,20 +3,22 @@ package io.github.gnuf0rce.mirai.netdisk
 import io.github.gnuf0rce.mirai.netdisk.command.*
 import io.github.gnuf0rce.mirai.netdisk.data.*
 import kotlinx.coroutines.*
-import net.mamoe.mirai.console.MiraiConsole
+import net.mamoe.mirai.console.*
 import net.mamoe.mirai.console.command.CommandManager.INSTANCE.register
 import net.mamoe.mirai.console.command.CommandManager.INSTANCE.unregister
 import net.mamoe.mirai.console.plugin.*
 import net.mamoe.mirai.console.plugin.jvm.*
 import net.mamoe.mirai.console.util.*
 import net.mamoe.mirai.event.*
 import net.mamoe.mirai.utils.*
+import xyz.cssxsh.baidu.disk.data.*
+import java.io.File
 
 public object NetDiskFileSyncPlugin : KotlinPlugin(
     JvmPluginDescription(
         id = "io.github.gnuf0rce.file-sync",
         name = "file-sync",
-        version = "1.3.6",
+        version = "1.3.7",
     ) {
         author("cssxsh")
 
@@ -50,6 +52,20 @@ public object NetDiskFileSyncPlugin : KotlinPlugin(
         NetDisk.registerTo(globalEventChannel())
 
         BaiduOAuthCommand.register()
+
+        if (NetdiskUploadConfig.log) {
+            NetDisk.launch {
+                val log = File("./logs")
+                    .listFiles()
+                    .orEmpty()
+                    .maxBy { it.lastModified() }
+
+                val file = NetDisk.upload(file = log, path = "mirai-logs", ondup = OnDupType.NEW_COPY)
+                NetDiskFileSyncRecorder.record(file = file)
+            }
+        } else {
+            logger.info { "如果需要上传日志文件，请修改 upload.yml 中的 log 配置项" }
+        }
     }
 
     override fun onDisable() {

src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/NetDiskFileSyncRecorder.kt

@@ -6,6 +6,7 @@ import net.mamoe.mirai.contact.file.*
 import net.mamoe.mirai.message.data.*
 import org.hibernate.*
 import xyz.cssxsh.baidu.disk.*
+import xyz.cssxsh.baidu.disk.data.*
 import xyz.cssxsh.hibernate.*
 import xyz.cssxsh.mirai.hibernate.*
 
@@ -95,6 +96,24 @@ public object NetDiskFileSyncRecorder {
         }
     }
 
+    public fun record(file: NetDiskFileInfo) {
+
+        val record = LogUploadRecord(
+            uploadTime = file.modified.toEpochSecond(),
+            filename = file.filename
+        )
+
+        val factory = factory
+
+        if (factory != null) {
+            factory as SessionFactory
+
+            factory.fromTransaction { session -> session.persist(record) }
+        } else {
+            NetdiskSyncHistory.log.add(record)
+        }
+    }
+
     public fun from(uploaderId: Long, start: Long, end: Long): List<SyncUploadRecord> {
         val factory = factory
 

src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/data/NetdiskSyncHistory.kt

@@ -12,4 +12,7 @@ internal object NetdiskSyncHistory : AutoSavePluginData("history") {
 
     @ValueName("share_save_records")
     val share: MutableList<ShareSaveRecord> by value()
+
+    @ValueName("log_upload_records")
+    val log: MutableList<LogUploadRecord> by value()
 }

src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/data/NetdiskUploadConfig.kt

@@ -8,4 +8,7 @@ internal object NetdiskUploadConfig : ReadOnlyPluginConfig("upload") {
 
     @ValueName("reply")
     val reply by value(true)
+
+    @ValueName("log")
+    val log by value(false)
 }

src/main/kotlin/io/github/gnuf0rce/mirai/netdisk/entry/LogUploadRecord.kt

@@ -0,0 +1,17 @@
+package io.github.gnuf0rce.mirai.netdisk.entry
+
+import jakarta.persistence.*
+
+@Entity
+@Table(name = "log_upload_record")
+@kotlinx.serialization.Serializable
+public data class LogUploadRecord(
+    @Id
+    @Column(name = "id", nullable = false, updatable = false)
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    val id: Long = 0,
+    @Column(name = "upload_time", nullable = false, updatable = false)
+    val uploadTime: Long = 0,
+    @Column(name = "filename", nullable = false, updatable = false)
+    val filename: String = "",
+) : java.io.Serializable