MWClash/android/app/src/main/kotlin/com/follow/clash/Service.kt

package com.follow.clash

import com.follow.clash.common.ServiceDelegate
import com.follow.clash.common.intent
import com.follow.clash.service.ICallbackInterface
import com.follow.clash.service.IMessageInterface
import com.follow.clash.service.IRemoteInterface
import com.follow.clash.service.RemoteService
import com.follow.clash.service.models.NotificationParams
import com.follow.clash.service.models.VpnOptions
import java.util.concurrent.atomic.AtomicBoolean

object Service {
    private val delegate by lazy {
        ServiceDelegate<IRemoteInterface>(
            RemoteService::class.intent, ::handleOnServiceCrash
        ) {
            IRemoteInterface.Stub.asInterface(it)
        }
    }

    var onServiceCrash: (() -> Unit)? = null

    private fun handleOnServiceCrash() {
        bindingState.set(false)
        onServiceCrash?.let {
            it()
        }
    }

    private val bindingState = AtomicBoolean(false)

    fun bind() {
        if (bindingState.compareAndSet(false, true)) {
            delegate.bind()
        }
    }

    suspend fun invokeAction(
        data: String, cb: (result: ByteArray?, isSuccess: Boolean) -> Unit
    ) {
        delegate.useService {
            it.invokeAction(data, object : ICallbackInterface.Stub() {
                override fun onResult(result: ByteArray?, isSuccess: Boolean) {
                    cb(result, isSuccess)
                }
            })
        }
    }

    suspend fun updateNotificationParams(
        params: NotificationParams
    ) {
        delegate.useService {
            it.updateNotificationParams(params)
        }
    }

    suspend fun setMessageCallback(
        cb: (result: String?) -> Unit
    ) {
        delegate.useService {
            it.setMessageCallback(object : IMessageInterface.Stub() {
                override fun onResult(result: String?) {
                    cb(result)
                }
            })
        }
    }

    suspend fun startService(options: VpnOptions, inApp: Boolean) {
        delegate.useService { it.startService(options, inApp) }
    }

    suspend fun stopService() {
        delegate.useService { it.stopService() }
    }
}