package com.sml.data.datasource.bmartellib

import android.accounts.NetworkErrorException
import com.sml.data.datasource.base.SpeedTestListener
import com.sml.data.factory.SpeedTestSocketFactory
import com.sml.data.mapper.FileTransferModelMapper
import com.sml.domain.enums.FileTransferMarker
import fr.bmartel.speedtest.SpeedTestReport
import fr.bmartel.speedtest.SpeedTestSocket
import fr.bmartel.speedtest.inter.ISpeedTestListener
import fr.bmartel.speedtest.model.SpeedTestError
import javax.inject.Inject

class BmartelSpeedTestApiUpload @Inject constructor(
        private val mapper: FileTransferModelMapper,
        private val speedTestSocketFactory: SpeedTestSocketFactory
) {

    private lateinit var speedTestSocket: SpeedTestSocket

    fun initUploadSpeedtestSettings(speedtestListener: SpeedTestListener) {
        speedTestSocket = speedTestSocketFactory.build()
        speedTestSocket.socketTimeout = BmartelWayConst.SOCKET_TIMEOUT

        speedTestSocket.addSpeedTestListener(object : ISpeedTestListener {

            override fun onCompletion(report: SpeedTestReport) {
                speedTestSocket.removeSpeedTestListener(this)
                speedtestListener.onNext(mapper.map(report, false, FileTransferMarker.NONE))
                speedtestListener.onComplete()
            }

            override fun onProgress(percent: Float, report: SpeedTestReport) {
                speedtestListener.onNext(mapper.map(report, false, FileTransferMarker.UPLOAD))
            }

            override fun onError(speedTestError: SpeedTestError, errorMessage: String) {
                speedtestListener.onError(NetworkErrorException("Speed test error"))
            }

        })
    }

    fun runUploadSpeedtest(url: String) {
        speedTestSocket.startFixedUpload(url, BmartelWayConst.FILE_SIZE, BmartelWayConst.FIXED_TIME, BmartelWayConst.REPORT_INTERVAL)
    }

    fun stopTask() {
        speedTestSocket.forceStopTask()
        speedTestSocket.clearListeners()
    }
}