/*
 * Copyright (c) 2018-2020 DJI
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 */

package dji.v5.ux.core.panel.listitem.obstacleavoidance

import dji.sdk.keyvalue.key.FlightControllerKey
import dji.sdk.keyvalue.value.flightcontroller.ObstacleActionType
import dji.v5.common.callback.CommonCallbacks
import dji.v5.common.error.IDJIError
import dji.v5.et.create
import dji.v5.manager.aircraft.perception.PerceptionManager
import dji.v5.manager.aircraft.perception.data.ObstacleAvoidanceType
import dji.v5.ux.core.base.DJISDKModel
import dji.v5.ux.core.base.WidgetModel
import dji.v5.ux.core.communication.ObservableInMemoryKeyedStore
import dji.v5.ux.core.util.DataProcessor
import io.reactivex.rxjava3.core.Completable

private const val TAG = "ObstacleAvoidanceListItemWidgetModel"

/**
 * Widget Model for the [ObstacleAvoidanceListItemWidgetModel] used to define
 * the underlying logic and communication
 */
class ObstacleAvoidanceListItemWidgetModel(
    djiSdkModel: DJISDKModel,
    keyedStore: ObservableInMemoryKeyedStore,
) : WidgetModel(djiSdkModel, keyedStore) {

    val obstacleActionTypeProcessor = DataProcessor.create(ObstacleActionType.UNKNOWN)

    private val flyControlConnectionProcessor = DataProcessor.create(false)


    fun setObstacleActionType(type: ObstacleActionType): Completable {
        return Completable.create {
            PerceptionManager.getInstance().setObstacleAvoidanceType(getObstacleAvoidanceType(type), object :
                CommonCallbacks.CompletionCallback {
                override fun onSuccess() {
                    it.onComplete()
                }

                override fun onFailure(error: IDJIError) {
                    it.onError(Throwable("setObstacleAvoidanceType failed!"))
                }
            })
        }
    }

    override fun inSetup() {
        bindDataProcessor(FlightControllerKey.KeyConnection.create(), flyControlConnectionProcessor)
    }

    private fun updateObstacleActionType() {
        if (!flyControlConnectionProcessor.value) {
            obstacleActionTypeProcessor.onNext(ObstacleActionType.UNKNOWN)
            return
        }
        PerceptionManager.getInstance().getObstacleAvoidanceType(object :
            CommonCallbacks.CompletionCallbackWithParam<ObstacleAvoidanceType> {
            override fun onSuccess(t: ObstacleAvoidanceType?) {
                if (t != null) {
                    obstacleActionTypeProcessor.onNext(getObstacleActionType(t))
                }
            }

            override fun onFailure(error: IDJIError) {
                //do nothing
            }
        })

    }

    override fun inCleanup() {
        // No clean up needed
    }

    override fun updateStates() {
        updateObstacleActionType()
    }

    fun getObstacleActionType(obstacleAvoidanceType: ObstacleAvoidanceType?): ObstacleActionType {
        val obstacleActionType: ObstacleActionType = when (obstacleAvoidanceType) {
            ObstacleAvoidanceType.BRAKE -> ObstacleActionType.STOP
            ObstacleAvoidanceType.CLOSE -> ObstacleActionType.CLOSE
            else -> ObstacleActionType.APAS
        }
        return obstacleActionType
    }

    private fun getObstacleAvoidanceType(obstacleActionType: ObstacleActionType?): ObstacleAvoidanceType? {
        if (obstacleActionType == null) {
            return null
        }
        val obstacleAvoidanceType: ObstacleAvoidanceType = when (obstacleActionType) {
                ObstacleActionType.STOP -> ObstacleAvoidanceType.BRAKE
                ObstacleActionType.APAS -> ObstacleAvoidanceType.BYPASS
                else -> ObstacleAvoidanceType.CLOSE
            }
        return obstacleAvoidanceType
    }
}