add MemoryCard.kt

Author: diaozxin007

wechaty-puppet/src/main/kotlin/Puppet.kt

@@ -12,6 +12,7 @@ import io.github.wechaty.io.github.wechaty.watchdag.WatchDog
 import io.github.wechaty.io.github.wechaty.watchdag.WatchdogFood
 import io.github.wechaty.io.github.wechaty.watchdag.WatchdogListener
 import io.github.wechaty.listener.*
+import io.github.wechaty.memorycard.MemoryCard
 import io.github.wechaty.schemas.*
 import io.github.wechaty.utils.FutureUtils
 import io.github.wechaty.utils.JsonUtils
@@ -41,6 +42,7 @@ abstract class Puppet : EventEmitter {
     private val HEARTBEAT_COUNTER = AtomicLong()
     private val HOSTIE_KEEPALIVE_TIMEOUT = 15 * 1000L
     private val DEFAULT_WATCHDOG_TIMEOUT = 60L
+    private var memory: MemoryCard
 
     private val executorService = Executors.newSingleThreadScheduledExecutor()
 
@@ -65,6 +67,10 @@ abstract class Puppet : EventEmitter {
         count.addAndGet(1)
         this.puppetOptions = puppetOptions
 
+        this.memory = MemoryCard()
+        this.memory.load()
+
+
         val timeOut = puppetOptions.timeout ?: DEFAULT_WATCHDOG_TIMEOUT
         watchDog = WatchDog(1000 * timeOut, "puppet")
 
@@ -900,6 +906,10 @@ abstract class Puppet : EventEmitter {
         }
     }
 
+    fun setMemory(memoryCard: MemoryCard){
+        this.memory = memoryCard
+    }
+
 //    fun getEventBus(): EventBus {
 //        return eb
 //    }

wechaty-puppet/src/main/kotlin/io/github/wechaty/memorycard/MemoryCard.kt

@@ -5,44 +5,63 @@ import java.util.concurrent.Future
 
 class MemoryCard{
 
-    public var name:String? = null
+    private var name:String? = null
     protected var parent:MemoryCard? = null
     protected var payload:MemoryCardPayload ? = null
     protected var storage:StorageBackend? = null
     protected val multiplexNameList = mutableListOf<String>()
 
     private var options:MemoryCardOptions
 
-    constructor(name: String?,options:MemoryCardOptions){
+    constructor(name: String?=null,options:MemoryCardOptions? = null){
+
+        val _optiones:MemoryCardOptions = options ?: MemoryCardOptions()
+
         if(name != null){
-            options.name = name
+            if(options != null) {
+                _optiones.name = name
+            }
         }
-        this.options = options
-        this.name = options.name
+        this.options = _optiones
+        this.name = _optiones.name
 
-        (options.multiplex != null).let {
-            this.parent = options.multiplex!!.parent
+        (_optiones.multiplex != null).let {
+            this.parent = _optiones.multiplex!!.parent
             this.payload = this.parent!!.payload
             this.multiplexNameList.addAll(parent!!.multiplexNameList)
-            this.multiplexNameList.add(options.multiplex!!.name)
+            this.multiplexNameList.add(_optiones.multiplex!!.name)
             this.storage = null
         }
 
-        (options.multiplex == null).let {
+        (_optiones.multiplex == null).let {
             this.payload = null
             this.multiplexNameList.clear()
-
-
         }
     }
 
     private fun getStore():StorageBackend?{
         log.debug("getStorage() for storage type: %s'",this.options)
-        TODO()
+
+        return StorageBackend.getStorage(
+            this.options.name!!,
+            this.options.storageOptions
+        )
+    }
+
+    fun load(){
+        this.payload = this.storage!!.load()
+    }
+
+    fun save(){
+        this.storage!!.save(this.payload!!)
     }
 
     companion object{
         private val log = LoggerFactory.getLogger(MemoryCard::class.java)
+
+        fun multiplex(memory:MemoryCard,name:String):MemoryCard{
+            return MemoryCard()
+        }
     }
 
 

wechaty-puppet/src/main/kotlin/io/github/wechaty/memorycard/StorageOptions.kt

@@ -24,3 +24,7 @@ class StorageNopOptions: StorageBackendOptions() {
 }
 
 typealias StorageFileOptions = StorageNopOptions
+
+val BACKEND_DICT = mapOf(
+    "file" to StorageFile::class
+)

wechaty-puppet/src/main/kotlin/io/github/wechaty/memorycard/StrorageBackend.kt

@@ -1,6 +1,8 @@
 package io.github.wechaty.memorycard
 
 import org.slf4j.LoggerFactory
+import kotlin.reflect.full.createInstance
+import kotlin.reflect.full.primaryConstructor
 
 abstract class StorageBackend(name: String, option: StorageBackendOptions) {
 
@@ -14,6 +16,27 @@ abstract class StorageBackend(name: String, option: StorageBackendOptions) {
 
     companion object{
         private val log = LoggerFactory.getLogger(StorageBackend::class.java)
+
+        fun getStorage(name: String,options: StorageBackendOptions?):StorageBackend{
+
+            var _options = options
+
+            if(options == null) {
+                _options = StorageFileOptions()
+                _options.type = "file"
+            }
+
+            if(_options?.type == null || _options.type !in BACKEND_DICT.keys){
+                throw Exception("backed unknown : ${_options?.type}")
+            }
+
+            val kClass = BACKEND_DICT[_options.type]
+            val createInstance = kClass?.primaryConstructor
+            return createInstance?.call(name,_options)!!
+
+
+        }
+
     }
 
 }

wechaty/src/main/kotlin/io/github/wechaty/Wechaty.kt

@@ -6,6 +6,7 @@ import io.github.wechaty.eventEmitter.EventEmitter
 import io.github.wechaty.eventEmitter.Listener
 import io.github.wechaty.grpc.GrpcPuppet
 import io.github.wechaty.listener.*
+import io.github.wechaty.memorycard.MemoryCard
 import io.github.wechaty.schemas.*
 import io.github.wechaty.user.*
 import io.github.wechaty.user.manager.*
@@ -29,12 +30,18 @@ class Wechaty private constructor(private var wechatyOptions: WechatyOptions) :
     @Volatile
     private var status = StateEnum.OFF
 
+    private var memory:MemoryCard? = null
+
     val tagManager: TagManager = TagManager(this)
     val contactManager = ContactManager(this)
     val messageManager = MessageManager(this)
     val roomManager = RoomManager(this)
     val roomInvitationMessage = RoomInvitationManager(this)
 
+    init {
+        this.memory = wechatyOptions.memory
+    }
+
 
     fun start(await: Boolean = false) {
 

wechaty/src/main/kotlin/io/github/wechaty/WechatyOptions.kt

@@ -1,9 +1,12 @@
 package io.github.wechaty
 
+import io.github.wechaty.memorycard.MemoryCard
 import io.github.wechaty.schemas.PuppetOptions
 
 class WechatyOptions {
 
+    var memory:MemoryCard? = null
+
     var name:String = "Wechaty"
 
     var puppet:String = "wechaty-puppet-hostie"