Rewarded SDAI (#31)

* Rewarded feature

* Update localization

* Update

* Added rewarded to settings

* Update

Author: ShiftHackZ

build.gradle

@@ -1,6 +1,6 @@
 buildscript {
     ext {
-        appVersion = "0.2.1"
+        appVersion = "0.3.0"
         minSdk = 26
         targetSdk = 33
     }

data/src/main/java/com/shifthackz/aisdv1/data/local/CoinLocalDataSource.kt

@@ -2,19 +2,29 @@ package com.shifthackz.aisdv1.data.local
 
 import com.shifthackz.aisdv1.domain.datasource.CoinDataSource
 import com.shifthackz.aisdv1.storage.db.coins.dao.CoinDao
+import com.shifthackz.aisdv1.storage.db.coins.dao.EarnedCoinDao
 import com.shifthackz.aisdv1.storage.db.coins.entity.CoinEntity
+import com.shifthackz.aisdv1.storage.db.coins.entity.EarnedCoinEntity
 import io.reactivex.rxjava3.core.Completable
+import io.reactivex.rxjava3.core.Single
 import java.util.*
 
 internal class CoinLocalDataSource(
     private val coinDao: CoinDao,
+    private val earnedCoinDao: EarnedCoinDao,
 ) : CoinDataSource.Local {
 
-    override fun querySpentCoinsForPeriod(start: Long, end: Long) = coinDao
+    override fun querySpentDailyCoinsForPeriod(start: Long, end: Long) = coinDao
         .queryAvailableCoinsForPeriod(start, end)
         .map { it.size }
 
-    override fun onCoinSpent(): Completable = coinDao.insert(
-        CoinEntity(0L, Date())
-    )
+    override fun queryEarnedCoins(): Single<Int> = earnedCoinDao.queryCount()
+
+    override fun onDailyCoinSpent(): Completable = coinDao.insert(CoinEntity(0L, Date()))
+
+    override fun onEarnedCoinSpent(): Completable = earnedCoinDao.deleteLast()
+
+    override fun onEarnedCoinsRewarded(amount: Int): Completable = (1..amount)
+        .map { EarnedCoinEntity(0L, Date()) }
+        .let(earnedCoinDao::insert)
 }

data/src/main/java/com/shifthackz/aisdv1/data/repository/CoinRepositoryImpl.kt

@@ -20,47 +20,68 @@ internal class CoinRepositoryImpl(
     private val sessionPreference: SessionPreference,
 ) : CoinRepository {
 
-    private val eventCoinSpent: PublishSubject<Unit> = PublishSubject.create()
-
-    override fun observeAvailableCoinsForToday(): Flowable<Int> {
-        val coinsPerDayProducer: () -> Single<Int> = {
-            val availableCoinsPerDay = sessionPreference.coinsPerDay
-            if (availableCoinsPerDay != -1) Single.just(availableCoinsPerDay)
-            else coinRemoteDateSource
-                .fetchCoinsConfig()
-                .doOnSuccess { sessionPreference.coinsPerDay = it }
-        }
+    private val eventCoinsUpdated: PublishSubject<Unit> = PublishSubject.create()
 
+    override fun observeAvailableCoins(): Flowable<Int> {
         val coinsCalculationProducer: () -> Flowable<Int> = {
-            val (start, end) = Date().getDayRange()
-            coinsPerDayProducer()
-                .zipWith(
-                    coinLocalDataSource.querySpentCoinsForPeriod(start.time, end.time),
-                    ::Pair,
-                )
-                .map { (availableToday, spentToday) -> availableToday - spentToday }
-                .map { coins -> if (coins < 0) 0 else coins }
+            Single.zip(
+                calculateAvailableDailyCoins(),
+                calculateAvailableEarnedCoins(),
+                ::Pair,
+            )
+                .map { (daily, earned) -> daily + earned }
                 .toFlowable()
         }
 
-        val coinSpentEventProducer: () -> Flowable<*> = {
-            eventCoinSpent.toFlowable(BackpressureStrategy.LATEST)
+        val coinsUpdatedEventProducer: () -> Flowable<*> = {
+            eventCoinsUpdated.toFlowable(BackpressureStrategy.LATEST)
         }
 
         val coreTickProducer: () -> Flowable<*> = {
-            Flowable.interval(10L, TimeUnit.SECONDS)
+            Flowable.interval(5L, TimeUnit.SECONDS)
         }
 
         return Flowable
-            .merge(coinsCalculationProducer(), coreTickProducer(), coinSpentEventProducer())
+            .merge(coinsCalculationProducer(), coreTickProducer(), coinsUpdatedEventProducer())
             .flatMap { coinsCalculationProducer() }
             .replay(1)
             .refCount(1, TimeUnit.SECONDS)
     }
 
     override fun spendCoin(): Completable {
-        eventCoinSpent.onNext(Unit)
-        return if (preferenceManager.useSdAiCloud) coinLocalDataSource.onCoinSpent()
+        eventCoinsUpdated.onNext(Unit)
+        return if (preferenceManager.useSdAiCloud) {
+            calculateAvailableDailyCoins()
+                .flatMapCompletable {
+                    if (it > 0) coinLocalDataSource.onDailyCoinSpent()
+                    else coinLocalDataSource.onEarnedCoinSpent()
+                }
+                .doOnComplete { eventCoinsUpdated.onNext(Unit) }
+        }
         else Completable.complete()
     }
+
+    override fun earnCoins(amount: Int) = coinLocalDataSource.onEarnedCoinsRewarded(amount)
+
+    private fun getAvailableCoinsPerDay(): Single<Int> {
+        val availableCoinsPerDay = sessionPreference.coinsPerDay
+        return if (availableCoinsPerDay != -1) Single.just(availableCoinsPerDay)
+        else coinRemoteDateSource
+            .fetchCoinsConfig()
+            .doOnSuccess { sessionPreference.coinsPerDay = it }
+    }
+
+    private fun calculateAvailableDailyCoins(): Single<Int> {
+        val (start, end) = Date().getDayRange()
+        return getAvailableCoinsPerDay()
+            .zipWith(
+                coinLocalDataSource.querySpentDailyCoinsForPeriod(start.time, end.time),
+                ::Pair,
+            )
+            .map { (availableToday, spentToday) -> availableToday - spentToday }
+            .map { coins -> if (coins < 0) 0 else coins }
+    }
+
+    private fun calculateAvailableEarnedCoins(): Single<Int> =
+        coinLocalDataSource.queryEarnedCoins()
 }

demo/src/main/java/com/shifthackz/aisdv1/demo/ImageToImageDemoImpl.kt

@@ -6,7 +6,7 @@ import com.shifthackz.aisdv1.domain.entity.AiGenerationResult
 import com.shifthackz.aisdv1.domain.entity.ImageToImagePayload
 import java.util.*
 
-class ImageToImageDemoImpl(
+internal class ImageToImageDemoImpl(
     override val demoDataSerializer: DemoDataSerializer,
 ) : ImageToImageDemo, DemoFeature<ImageToImagePayload> {
 

demo/src/main/java/com/shifthackz/aisdv1/demo/serialize/DemoDataSerializer.kt

@@ -5,7 +5,7 @@ import com.google.gson.Gson
 import com.google.gson.reflect.TypeToken
 import java.io.BufferedReader
 
-class DemoDataSerializer(private val contextProvider: () -> Context) {
+internal class DemoDataSerializer(private val contextProvider: () -> Context) {
 
     fun readDemoAssets(): List<String> {
         val rawString = contextProvider().resources.assets

domain/src/main/java/com/shifthackz/aisdv1/domain/datasource/CoinDataSource.kt

@@ -6,8 +6,11 @@ import io.reactivex.rxjava3.core.Single
 sealed interface CoinDataSource {
 
     interface Local : CoinDataSource {
-        fun querySpentCoinsForPeriod(start: Long, end: Long): Single<Int>
-        fun onCoinSpent(): Completable
+        fun querySpentDailyCoinsForPeriod(start: Long, end: Long): Single<Int>
+        fun queryEarnedCoins(): Single<Int>
+        fun onDailyCoinSpent(): Completable
+        fun onEarnedCoinSpent(): Completable
+        fun onEarnedCoinsRewarded(amount: Int): Completable
     }
 
     interface Remote : CoinDataSource {

domain/src/main/java/com/shifthackz/aisdv1/domain/di/DomainModule.kt

@@ -4,6 +4,8 @@ import com.shifthackz.aisdv1.domain.usecase.caching.ClearAppCacheUseCase
 import com.shifthackz.aisdv1.domain.usecase.caching.ClearAppCacheUseCaseImpl
 import com.shifthackz.aisdv1.domain.usecase.caching.DataPreLoaderUseCase
 import com.shifthackz.aisdv1.domain.usecase.caching.DataPreLoaderUseCaseImpl
+import com.shifthackz.aisdv1.domain.usecase.coin.EarnRewardedCoinsUseCase
+import com.shifthackz.aisdv1.domain.usecase.coin.EarnRewardedCoinsUseCaseImpl
 import com.shifthackz.aisdv1.domain.usecase.coin.ObserveCoinsUseCase
 import com.shifthackz.aisdv1.domain.usecase.coin.ObserveCoinsUseCaseImpl
 import com.shifthackz.aisdv1.domain.usecase.connectivity.*
@@ -48,4 +50,5 @@ val domainModule = module {
     factoryOf(::ObserveSeverConnectivityUseCaseImpl) bind ObserveSeverConnectivityUseCase::class
     factoryOf(::CheckAppVersionUpdateUseCaseImpl) bind CheckAppVersionUpdateUseCase::class
     factoryOf(::ObserveCoinsUseCaseImpl) bind ObserveCoinsUseCase::class
+    factoryOf(::EarnRewardedCoinsUseCaseImpl) bind EarnRewardedCoinsUseCase::class
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/feature/ad/AdFeature.kt

@@ -9,6 +9,7 @@ interface AdFeature {
     fun getHomeScreenBannerAd(context: Context): Ad
     fun getGalleryDetailBannerAd(context: Context): Ad
     fun loadAd(ad: Ad)
+    fun showRewardedCoinsAd(activity: Activity, rewardCallback: (Int) -> Unit)
 
     data class Ad(
         val id: String = "",

domain/src/main/java/com/shifthackz/aisdv1/domain/repository/CoinRepository.kt

@@ -4,6 +4,7 @@ import io.reactivex.rxjava3.core.Completable
 import io.reactivex.rxjava3.core.Flowable
 
 interface CoinRepository {
-    fun observeAvailableCoinsForToday(): Flowable<Int>
+    fun observeAvailableCoins(): Flowable<Int>
     fun spendCoin(): Completable
+    fun earnCoins(amount: Int): Completable
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/coin/EarnRewardedCoinsUseCase.kt

@@ -0,0 +1,7 @@
+package com.shifthackz.aisdv1.domain.usecase.coin
+
+import io.reactivex.rxjava3.core.Completable
+
+interface EarnRewardedCoinsUseCase {
+    operator fun invoke(amount: Int): Completable
+}