@@ -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 (
1011 private val environment : EnvironmentDTO
1112) : RemoteProviderEnvironment {
13+ private val stateListeners = mutableSetOf<EnvironmentStateConsumer >()
14+ private val actionListeners = mutableSetOf<ActionListener >()
1215 override fun getId (): String = environment.id
1316 override 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
2130 override fun getContentsView (): CompletableFuture <EnvironmentContentsView > {
@@ -24,4 +33,12 @@ class SampleRemoteEnvironment(
2433
2534 override 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}
0 commit comments