@@ -3,19 +3,28 @@ package toolbox.gateway.sample
33import com.jetbrains.toolbox.gateway.EnvironmentVisibilityState
44import com.jetbrains.toolbox.gateway.RemoteProviderEnvironment
55import com.jetbrains.toolbox.gateway.environments.EnvironmentContentsView
6- import com.jetbrains.toolbox.gateway.states.StateConsumer
6+ import com.jetbrains.toolbox.gateway.states.EnvironmentStateConsumer
7+ import com.jetbrains.toolbox.gateway.ui.ActionListener
78import java.util.concurrent.CompletableFuture
89
910class SampleRemoteEnvironment (
1011private val environment : EnvironmentDTO
1112) : RemoteProviderEnvironment {
13+ private val stateListeners= mutableSetOf<EnvironmentStateConsumer >()
14+ private val actionListeners= mutableSetOf<ActionListener >()
1215override fun getId ():String = environment.id
1316override fun getName ():String = environment.name
14-
15- override fun addStateListener (consumer : StateConsumer ) {
17+ override fun addStateListener (p0 : EnvironmentStateConsumer ? ):Boolean {
18+ return if (p0!= null ) {
19+ stateListeners+ = p0
20+ true
21+ }else false
1622 }
1723
18- override fun removeStateListener (consumer : StateConsumer ) {
24+ override fun removeStateListener (p0 : EnvironmentStateConsumer ? ) {
25+ if (p0!= null ) {
26+ stateListeners- = p0
27+ }
1928 }
2029
2130override fun getContentsView ():CompletableFuture <EnvironmentContentsView > {
@@ -24,4 +33,12 @@ class SampleRemoteEnvironment(
2433
2534override fun setVisible (visibilityState : EnvironmentVisibilityState ) {
2635 }
36+
37+ override fun registerActionListener (p0 : ActionListener ) {
38+ actionListeners+ = p0
39+ }
40+
41+ override fun unregisterActionListener (p0 : ActionListener ) {
42+ actionListeners- = p0
43+ }
2744}