EtchDroid/app/src/main/java/eu/depau/etchdroid/activities/UsbDrivePickerActivity.kt

package eu.depau.etchdroid.activities

import android.app.PendingIntent
import android.content.BroadcastReceiver
import android.content.Context
import android.content.Intent
import android.content.IntentFilter
import android.content.pm.PackageManager
import android.hardware.usb.UsbDevice
import android.hardware.usb.UsbManager
import android.net.Uri
import android.os.Bundle
import android.view.Menu
import android.view.MenuItem
import android.view.View
import androidx.recyclerview.widget.LinearLayoutManager
import androidx.recyclerview.widget.RecyclerView
import androidx.swiperefreshlayout.widget.SwipeRefreshLayout
import com.github.mjdev.libaums.UsbMassStorageDevice
import eu.depau.etchdroid.R
import eu.depau.etchdroid.StateKeeper
import eu.depau.etchdroid.adapters.UsbDrivesRecyclerViewAdapter
import eu.depau.etchdroid.enums.FlashMethod
import eu.depau.etchdroid.kotlin_exts.*
import eu.depau.etchdroid.utils.ClickListener
import eu.depau.etchdroid.utils.EmptyRecyclerView
import eu.depau.etchdroid.utils.RecyclerViewTouchListener
import kotlinx.android.synthetic.main.activity_usb_drive_picker.*
import java.io.File

class UsbDrivePickerActivity : ActivityBase(), SwipeRefreshLayout.OnRefreshListener {
    val USB_PERMISSION = "eu.depau.etchdroid.USB_PERMISSION"
    lateinit var mUsbPermissionIntent: PendingIntent

    private lateinit var recyclerView: EmptyRecyclerView
    private lateinit var viewAdapter: UsbDrivesRecyclerViewAdapter
    private lateinit var viewManager: RecyclerView.LayoutManager
    private lateinit var refreshLayout: SwipeRefreshLayout


    fun handleIntent(intent: Intent) {
        if (intent.action == Intent.ACTION_VIEW) {
            val uri = intent.data
            val ext = uri?.getExtension(contentResolver)

            when (ext) {
                in listOf("iso", "img") -> {
                    StateKeeper.flashMethod = FlashMethod.FLASH_API
                    StateKeeper.imageFile = uri
                }
                "dmg" -> {
                    val path: String?
                    try {
                        path = uri.getFilePath(this)
                    } catch (e: Exception) {
                        toast(getString(R.string.cannot_find_file_in_storage))
                        // Rethrow exception so it's logged in Google Developer Console
                        throw e
                    }
                    if (path == null) {
                        toast(getString(R.string.cannot_find_file_in_storage))
                        finish()
                    }

                    StateKeeper.flashMethod = FlashMethod.FLASH_DMG_API
                    StateKeeper.imageFile = Uri.fromFile(File(path))

                    checkAndRequestStorageReadPerm()
                }
                null -> {
                    return
                }
                else -> {
                    toast(getString(R.string.file_type_not_supported))
                    finish()
                }
            }

            if (shouldShowAndroidPieAlertDialog)
                showAndroidPieAlertDialog {}
        }
    }

    override fun onRequestPermissionsResult(requestCode: Int, permissions: Array<out String>, grantResults: IntArray) {
        when (requestCode) {
            READ_EXTERNAL_STORAGE_PERMISSION -> {
                if (grantResults[0] == PackageManager.PERMISSION_DENIED) {
                    toast(getString(R.string.storage_permission_required))
                    finish()
                }
                return
            }
            else -> {
            }
        }
    }

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)

        handleIntent(intent)

        setContentView(R.layout.activity_usb_drive_picker)
        actionBar?.setDisplayHomeAsUpEnabled(true)

        mUsbPermissionIntent = PendingIntent.getBroadcast(this, 0, Intent(USB_PERMISSION), 0)
        val usbPermissionFilter = IntentFilter(USB_PERMISSION)
        val usbAttachedFilter = IntentFilter(UsbManager.ACTION_USB_DEVICE_ATTACHED)
        val usbDetachedFilter = IntentFilter(UsbManager.ACTION_USB_DEVICE_DETACHED)

        registerReceiver(mUsbReceiver, usbPermissionFilter)
        registerReceiver(mUsbReceiver, usbAttachedFilter)
        registerReceiver(mUsbReceiver, usbDetachedFilter)

        refreshLayout = usbdevs_swiperefreshlayout
        refreshLayout.setOnRefreshListener(this)
        refreshLayout.post {
            refreshLayout.isRefreshing = true
            loadUsbDevices()
        }

        viewManager = LinearLayoutManager(this)
        recyclerView = usbdevs_recycler_view
        recyclerView.emptyView = usbdevs_recycler_empty_view

        recyclerView.addOnItemTouchListener(RecyclerViewTouchListener(this, recyclerView, object : ClickListener {
            override fun onClick(view: View, position: Int) {
                val device = viewAdapter.get(position)
                val manager = getSystemService(Context.USB_SERVICE) as UsbManager
                manager.requestPermission(device.usbDevice, mUsbPermissionIntent)
            }

            override fun onLongClick(view: View, position: Int) {}
        }))
    }

    override fun onDestroy() {
        super.onDestroy()
        unregisterReceiver(mUsbReceiver)
    }

    override fun onCreateOptionsMenu(menu: Menu): Boolean {
        menuInflater.inflate(R.menu.usb_devices_menu, menu)
        return super.onCreateOptionsMenu(menu)
    }

    override fun onContextItemSelected(item: MenuItem): Boolean {
        return when (item.itemId) {
            R.id.action_refresh -> {
                loadUsbDevices()
                true
            }
            else -> super.onOptionsItemSelected(item)
        }
    }

    override fun onRefresh() {
        loadUsbDevices()
    }


    private val mUsbReceiver = object : BroadcastReceiver() {

        override fun onReceive(context: Context, intent: Intent) {
            when (intent.action) {
                USB_PERMISSION -> synchronized(this) {
                    val device: UsbDevice? = intent.getParcelableExtra(UsbManager.EXTRA_DEVICE)

                    val result = intent.getBooleanExtra(UsbManager.EXTRA_PERMISSION_GRANTED, false)
                    if (result)
                        device?.apply {
                            StateKeeper.usbDevice = this
                        }

                    if (!result) {
                        if (device != null) {
                            recyclerView.snackbar(getString(R.string.usb_perm_denied) + device.name)
                        } else {
                            recyclerView.snackbar(getString(R.string.usb_perm_denied_noname))
                        }
                        return
                    }

                    StateKeeper.usbDevice = device
                    StateKeeper.usbMassStorageDevice = UsbMassStorageDevice.getMassStorageDevices(context).find { it.usbDevice == device }

                    nextStep()
                }
                UsbManager.ACTION_USB_DEVICE_ATTACHED, UsbManager.ACTION_USB_DEVICE_DETACHED -> loadUsbDevices()
            }
        }
    }

    fun loadUsbDevices() {
        try {
            viewAdapter = UsbDrivesRecyclerViewAdapter(UsbMassStorageDevice.getMassStorageDevices(this))

            recyclerView.apply {
                setHasFixedSize(true)
                layoutManager = viewManager
                adapter = viewAdapter
            }
        } finally {
            refreshLayout.isRefreshing = false
        }
    }


    fun nextStep() {
        val intent = Intent(this, ConfirmationActivity::class.java)
        startActivity(intent)
    }
}