#21 Add LocalStore and can autologin padplus

Author: jcai

wechaty-puppet-padplus/pom.xml

@@ -46,7 +46,19 @@
       <artifactId>javase</artifactId>
       <version>3.4.0</version>
     </dependency>
-
+    <dependency>
+      <groupId>org.fusesource.leveldbjni</groupId>
+      <artifactId>leveldbjni-all</artifactId>
+      <version>1.8</version>
+    </dependency>
+    <dependency>
+      <groupId>org.junit.jupiter</groupId>
+      <artifactId>junit-jupiter-engine</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>org.junit.jupiter</groupId>
+      <artifactId>junit-jupiter-params</artifactId>
+    </dependency>
   </dependencies>
   <build>
     <plugins>

wechaty-puppet-padplus/src/main/scala/wechaty/padplus/PuppetPadplus.scala

@@ -2,7 +2,7 @@ package wechaty.padplus
 
 import com.typesafe.scalalogging.LazyLogging
 import wechaty.padplus.grpc.PadPlusServerOuterClass.ApiType
-import wechaty.padplus.support.{ContactRawSupport, GrpcEventSupport, GrpcSupport}
+import wechaty.padplus.support.{ContactRawSupport, GrpcEventSupport, GrpcSupport, LocalStoreSupport}
 import wechaty.puppet.schemas.Image.ImageType.Type
 import wechaty.puppet.schemas.Puppet.PuppetOptions
 import wechaty.puppet.schemas._
@@ -14,22 +14,36 @@ import wechaty.puppet.{Puppet, ResourceBox}
   * @author <a href="mailto:[email protected]">Jun Tsai</a>
   * @since 2020-06-21
   */
-class PuppetPadplus(val option:PuppetOptions)
+class PuppetPadplus(val option:PuppetOptions,val storePath:String="/tmp/padplus")
   extends Puppet
     with ContactRawSupport
     with ContactSupport
     with GrpcSupport
     with GrpcEventSupport
+    with LocalStoreSupport
     with LazyLogging {
+  private var selfUni:Option[String] = None
   def start(): Unit ={
     startGrpc(option.endPoint.get)
-    request(ApiType.GET_QRCODE)
+    //waiting stream start....
+    logger.info("waiting stream start....")
+    awaitStreamStart()
+    startLocalStore()
+    getUin match{
+      case Some(str) =>
+        selfUni = Some(str)
+        logger.debug("found uin in local store:{}",selfUni)
+        request(ApiType.INIT)
+      case _ =>
+        request(ApiType.GET_QRCODE)
+    }
   }
   def stop(): Unit = {
     stopGrpc()
+    stopLocalStore()
   }
 
-  override def selfIdOpt(): Option[String] = None
+  override def selfIdOpt(): Option[String] = selfUni
 
   override def roomAdd(roomId: String, contactId: String): Unit = ???
 

wechaty-puppet-padplus/src/main/scala/wechaty/padplus/internal/LocalStore.scala

@@ -0,0 +1,49 @@
+package wechaty.padplus.internal
+
+import java.io.File
+
+import com.google.protobuf.ByteString
+import org.fusesource.leveldbjni.JniDBFactory._
+import org.iq80.leveldb.{DB, Options}
+
+
+/**
+  *
+  * @author <a href="mailto:[email protected]">Jun Tsai</a>
+  * @since 2020-06-22
+  */
+class LocalStore(storePath:String) {
+  private var dbOpt :Option[DB] = None
+  def start(): Unit ={
+    val options = new Options()
+    options.createIfMissing(true)
+    val db= factory.open(new File(storePath), options)
+    dbOpt = Some(db)
+  }
+  def put(key: ByteString, value: ByteString): Unit = {
+    db.put(key.toByteArray,value.toByteArray)
+  }
+  def put(key: String, value: String): Unit = {
+    put(ByteString.copyFromUtf8(key),ByteString.copyFromUtf8(value))
+  }
+  def get(key:String): Option[ByteString]={
+    get(ByteString.copyFromUtf8(key))
+  }
+  def get(key:ByteString): Option[ByteString]={
+    val value = db.get(key.toByteArray)
+    if(value != null) Some(ByteString.copyFrom(value))
+    else None
+  }
+  def delete(key:ByteString): Unit ={
+    db.delete(key.toByteArray)
+  }
+  private def db={
+    dbOpt match{
+      case Some(d) => d
+      case _=> throw new IllegalStateException("db is null,database not init?")
+    }
+  }
+  def close(): Unit ={
+    db.close()
+  }
+}

wechaty-puppet-padplus/src/main/scala/wechaty/padplus/support/GrpcEventSupport.scala

@@ -2,6 +2,7 @@ package wechaty.padplus.support
 
 import java.io.{File, FileOutputStream}
 import java.util
+import java.util.concurrent.{CountDownLatch, TimeUnit}
 
 import com.google.zxing.client.j2se.BufferedImageLuminanceSource
 import com.google.zxing.common.HybridBinarizer
@@ -23,11 +24,15 @@ import wechaty.puppet.{Puppet, ResourceBox}
   * @since 2020-06-21
   */
 trait GrpcEventSupport extends StreamObserver[StreamResponse]{
-  self: GrpcSupport with Puppet with LazyLogging=>
+  self: GrpcSupport with LocalStoreSupport with Puppet with LazyLogging=>
 
+  private val countDownLatch = new CountDownLatch(1)
 
+
+  protected def awaitStreamStart()=countDownLatch.await(10,TimeUnit.SECONDS)
   override def onNext(response: StreamResponse): Unit = {
     logger.debug("stream response:{}",response)
+    countDownLatch.countDown()
 
     val traceId = response.getTraceId
     if(!isBlank(traceId)){
@@ -62,7 +67,7 @@ trait GrpcEventSupport extends StreamObserver[StreamResponse]{
           logger.debug("Scan QR Code to login: %s\nhttps://api.qrserver.com/v1/create-qr-code/?data=%s\n".format(payload.status, payload.qrcode))
           emit(PuppetEventName.SCAN,payload)
         case _ =>
-//          val uin = response.getUin()
+          saveUin(response.getUinBytes)
 //          val user = objectMapper.readTree(response.getData())
 //          val userName = user.get("userName").asText()
 

wechaty-puppet-padplus/src/main/scala/wechaty/padplus/support/GrpcSupport.scala

@@ -132,6 +132,8 @@ trait GrpcSupport {
     request.setRequestId(requestId)
     val traceId= UUID.randomUUID().toString
     request.setTraceId(traceId)
-    grpcClient.request(request.build())
+    val response = grpcClient.request(request.build())
+    logger.debug("request->response:{}",response)
+    response
   }
 }

wechaty-puppet-padplus/src/main/scala/wechaty/padplus/support/LocalStoreSupport.scala

@@ -0,0 +1,32 @@
+package wechaty.padplus.support
+
+import com.google.protobuf.ByteString
+import wechaty.padplus.PuppetPadplus
+import wechaty.padplus.internal.LocalStore
+
+/**
+  *
+  * @author <a href="mailto:[email protected]">Jun Tsai</a>
+  * @since 2020-06-22
+  */
+trait LocalStoreSupport {
+  self:PuppetPadplus=>
+  protected var store:LocalStore =  _
+  private val uinKey = ByteString.copyFromUtf8("uin")
+  protected def saveUin(uin:ByteString): Unit ={
+    if(!uin.isEmpty){
+      store.put(uinKey,uin)
+    }
+  }
+  protected def getUin: Option[String]={
+    store.get(uinKey).map(_.toStringUtf8)
+  }
+  protected def startLocalStore(): Unit ={
+    store = new LocalStore(storePath)
+    store.start()
+  }
+  protected def stopLocalStore(): Unit ={
+    if(store != null)
+      store.close()
+  }
+}

wechaty-puppet-padplus/src/test/scala/wechaty/padplus/internal/LocalStoreTest.scala

@@ -0,0 +1,24 @@
+// Copyright 2014,2015,2016 the original author or authors. All rights reserved.
+// site: http://www.ganshane.com
+package wechaty.padplus.internal
+
+import org.junit.jupiter.api.{Assertions, Test}
+
+
+/**
+ * Created by jcai on 14-8-17.
+ */
+class LocalStoreTest {
+  @Test
+  def test_version() {
+    val store = new LocalStore("/tmp/test.local")
+    store.start()
+    for (v <- 0 until 100) {
+      store.put("asdf", "fdsa")
+      store.put("sdf", "fdsa")
+      store.put("df", "fdsa")
+    }
+    Assertions.assertEquals("fdsa", store.get("asdf").get.toStringUtf8)
+    store.close()
+  }
+}