@@ -5,6 +5,7 @@ import io.github.wechaty.Puppet
55import io.github.wechaty.StateEnum
66import io.github.wechaty.filebox.FileBox
77import io.github.wechaty.grpc.puppet.*
8+ import io.github.wechaty.io.github.wechaty.schemas.EventEnum
89import io.github.wechaty.schemas.*
910import io.github.wechaty.utils.JsonUtils
1011import io.grpc.ManagedChannel
@@ -19,6 +20,7 @@ import java.util.concurrent.CompletableFuture
1920import java.util.concurrent.Executors.newFixedThreadPool
2021import java.util.concurrent.Future
2122import java.util.concurrent.TimeUnit
23+ import kotlin.system.exitProcess
2224
2325/* *
2426 * puppet
@@ -97,7 +99,7 @@ class GrpcPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
9799
98100 try {
99101 if (logonoff()) {
100- emit(" logout " ,EventLogoutPayload (getId()!! ," logout" ))
102+ emit(EventEnum . LOGOUT ,EventLogoutPayload (getId()!! ," logout" ))
101103
102104 this .setId(null )
103105 }
@@ -148,7 +150,7 @@ class GrpcPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
148150
149151 if (StringUtils .isEmpty(discoverHostieIp.first) || StringUtils .equals(discoverHostieIp.first, " 0.0.0.0" )) {
150152 log.error(" cannot get ip by token, check token" )
151- throw Exception ( " cannot get ip by token, check token " )
153+ exitProcess( 1 )
152154 }
153155 val newFixedThreadPool = newFixedThreadPool(16 )
154156 channel = ManagedChannelBuilder .forAddress(discoverHostieIp.first,NumberUtils .toInt(discoverHostieIp.second)).usePlaintext().executor(newFixedThreadPool).build()
@@ -167,7 +169,7 @@ class GrpcPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
167169 override fun onError (t : Throwable ? ) {
168170 log.error(" error of grpc" ,t)
169171 val payload = EventResetPayload (t?.message ? : " " )
170- emit(" reset " ,payload)
172+ emit(EventEnum . RESET ,payload)
171173 }
172174
173175 override fun onCompleted () {
@@ -200,7 +202,7 @@ class GrpcPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
200202 } catch (e: Exception ) {
201203 log.error(" logout() rejection: %s" , e)
202204 } finally {
203- emit(" logout " ,EventLogoutPayload (getId()!! ," logout" ))
205+ emit(EventEnum . LOGOUT ,EventLogoutPayload (getId()!! ," logout" ))
204206 }
205207 return CompletableFuture .completedFuture(null )
206208 }
@@ -942,67 +944,67 @@ class GrpcPuppet(puppetOptions: PuppetOptions) : Puppet(puppetOptions) {
942944 log.debug(" PuppetHostie $type payload $payload " )
943945
944946 if (type != Event .EventType .EVENT_TYPE_HEARTBEAT ) {
945- emit(" heartbeat " , EventHeartbeatPayload (" heartbeat" ))
947+ emit(EventEnum . HEART_BEAT , EventHeartbeatPayload (" heartbeat" ))
946948 }
947949
948950 when (type) {
949- Event .EventType .EVENT_TYPE_DONG -> emit(" dong " , JsonUtils .readValue<EventDongPayload >(payload))
951+ Event .EventType .EVENT_TYPE_DONG -> emit(EventEnum . DONG , JsonUtils .readValue<EventDongPayload >(payload))
950952
951953 Event .EventType .EVENT_TYPE_ERROR -> {
952- emit(" error " , JsonUtils .readValue<EventErrorPayload >(payload))
954+ emit(EventEnum . ERROR , JsonUtils .readValue<EventErrorPayload >(payload))
953955 }
954956
955957 Event .EventType .EVENT_TYPE_HEARTBEAT -> {
956958 val heartbeatPayload = JsonUtils .readValue<EventHeartbeatPayload >(payload)
957- emit(" heartbeat " , heartbeatPayload)
959+ emit(EventEnum . HEART_BEAT , heartbeatPayload)
958960 }
959961
960962 Event .EventType .EVENT_TYPE_FRIENDSHIP -> {
961963 val friendshipPayload = JsonUtils .readValue<EventFriendshipPayload >(payload)
962- emit(" friendship " , friendshipPayload)
964+ emit(EventEnum . FRIENDSHIP , friendshipPayload)
963965 }
964966
965967 Event .EventType .EVENT_TYPE_LOGIN -> {
966968 val loginPayload = JsonUtils .readValue<EventLoginPayload >(payload)
967969 setId(loginPayload.contactId)
968- emit(" login " , loginPayload)
970+ emit(EventEnum . LOGIN , loginPayload)
969971 }
970972
971973 Event .EventType .EVENT_TYPE_LOGOUT -> {
972974 this .setId(" " )
973- emit(" logout " , JsonUtils .readValue<EventLogoutPayload >(payload))
975+ emit(EventEnum . LOGOUT , JsonUtils .readValue<EventLogoutPayload >(payload))
974976 }
975977
976978
977979 Event .EventType .EVENT_TYPE_MESSAGE -> {
978980 val eventMessagePayload = JsonUtils .readValue<EventMessagePayload >(payload)
979- emit(" message " , eventMessagePayload)
981+ emit(EventEnum . MESSAGE , eventMessagePayload)
980982 }
981983
982984 Event .EventType .EVENT_TYPE_READY -> {
983- emit(" ready " , JsonUtils .readValue<EventReadyPayload >(payload))
985+ emit(EventEnum . READY , JsonUtils .readValue<EventReadyPayload >(payload))
984986 }
985987
986988 Event .EventType .EVENT_TYPE_ROOM_INVITE -> {
987- emit(" room-invite " , JsonUtils .readValue<EventRoomInvitePayload >(payload))
989+ emit(EventEnum . ROOM_INVITE , JsonUtils .readValue<EventRoomInvitePayload >(payload))
988990 }
989991
990992 Event .EventType .EVENT_TYPE_ROOM_JOIN -> {
991- emit(" room-join " , JsonUtils .readValue<EventRoomJoinPayload >(payload))
993+ emit(EventEnum . ROOM_JOIN , JsonUtils .readValue<EventRoomJoinPayload >(payload))
992994 }
993995
994996 Event .EventType .EVENT_TYPE_ROOM_LEAVE -> {
995- emit(" room-leave " , JsonUtils .readValue<EventRoomLeavePayload >(payload))
997+ emit(EventEnum . ROOM_LEAVE , JsonUtils .readValue<EventRoomLeavePayload >(payload))
996998 }
997999
9981000 Event .EventType .EVENT_TYPE_ROOM_TOPIC -> {
999- emit(" room-topic " , JsonUtils .readValue<EventRoomTopicPayload >(payload))
1001+ emit(EventEnum . ROOM_TOPIC , JsonUtils .readValue<EventRoomTopicPayload >(payload))
10001002 }
10011003
10021004 Event .EventType .EVENT_TYPE_SCAN -> {
10031005 val eventScanPayload = JsonUtils .readValue<EventScanPayload >(payload)
10041006 log.debug(" scan pay load is {}" , eventScanPayload)
1005- emit(" scan " , eventScanPayload)
1007+ emit(EventEnum . SCAN , eventScanPayload)
10061008 }
10071009
10081010 Event .EventType .EVENT_TYPE_RESET -> {
0 commit comments