package ani.dantotsu.notifications

import android.content.Context
import androidx.work.Constraints
import androidx.work.PeriodicWorkRequest
import ani.dantotsu.notifications.anilist.AnilistNotificationWorker
import ani.dantotsu.notifications.comment.CommentNotificationWorker
import ani.dantotsu.notifications.TaskScheduler.TaskType

class WorkManagerScheduler(private val context: Context) : TaskScheduler {
    override fun scheduleRepeatingTask(taskType: TaskType, interval: Long) {
        val constraints = Constraints.Builder()
            .setRequiredNetworkType(androidx.work.NetworkType.CONNECTED)
            .build()

        when (taskType) {
            TaskType.COMMENT_NOTIFICATION -> {
                val recurringWork = PeriodicWorkRequest.Builder(
                    CommentNotificationWorker::class.java,
                    interval, java.util.concurrent.TimeUnit.MINUTES,
                    PeriodicWorkRequest.MIN_PERIODIC_FLEX_MILLIS, java.util.concurrent.TimeUnit.MINUTES
                )
                    .setConstraints(constraints)
                    .build()
                androidx.work.WorkManager.getInstance(context).enqueueUniquePeriodicWork(
                    CommentNotificationWorker.WORK_NAME,
                    androidx.work.ExistingPeriodicWorkPolicy.UPDATE,
                    recurringWork
                )
            }

            TaskType.ANILIST_NOTIFICATION -> {
                val recurringWork = PeriodicWorkRequest.Builder(
                    AnilistNotificationWorker::class.java,
                    interval, java.util.concurrent.TimeUnit.MINUTES,
                    PeriodicWorkRequest.MIN_PERIODIC_FLEX_MILLIS, java.util.concurrent.TimeUnit.MINUTES
                )
                    .setConstraints(constraints)
                    .build()
                androidx.work.WorkManager.getInstance(context).enqueueUniquePeriodicWork(
                    AnilistNotificationWorker.WORK_NAME,
                    androidx.work.ExistingPeriodicWorkPolicy.UPDATE,
                    recurringWork
                )
            }
        }
    }

    override fun cancelTask(taskType: TaskType) {
        when (taskType) {
            TaskType.COMMENT_NOTIFICATION -> {
                androidx.work.WorkManager.getInstance(context)
                    .cancelUniqueWork(CommentNotificationWorker.WORK_NAME)
            }

            TaskType.ANILIST_NOTIFICATION -> {
                androidx.work.WorkManager.getInstance(context)
                    .cancelUniqueWork(AnilistNotificationWorker.WORK_NAME)
            }
        }
    }
}