From 7fae8a6f1c8e2aed33da4338dc6fa0da7d0a596a Mon Sep 17 00:00:00 2001
From: Jigar-f <jigar@getlantern.org>
Date: Tue, 12 Sep 2023 22:05:20 +0530
Subject: [PATCH] Fix ads fix and Use subscription pattern.

---
 .run/main.dart.run.xml                        |  1 +
 android/app/libs/liblantern-all.aar           |  4 +-
 android/app/proguard-rules.pro                | 12 +++-
 .../kotlin/io/lantern/model/SessionModel.kt   | 59 ++++++++++++++-----
 .../org/getlantern/lantern/MainActivity.kt    |  2 +
 lib/ad_helper.dart                            | 47 ++++++++-------
 lib/common/session_model.dart                 | 24 ++++----
 lib/main.dart                                 |  7 ++-
 lib/vpn/vpn_switch.dart                       | 34 +++++++----
 9 files changed, 127 insertions(+), 63 deletions(-)

diff --git a/.run/main.dart.run.xml b/.run/main.dart.run.xml
index 0d0028752..fe5d595ad 100644
--- a/.run/main.dart.run.xml
+++ b/.run/main.dart.run.xml
@@ -1,6 +1,7 @@
 <component name="ProjectRunConfigurationManager">
   <configuration default="false" name="main.dart" type="FlutterRunConfigurationType" factoryName="Flutter">
     <option name="additionalArgs" value="--dart-define=INTERSTITIAL_AD_UNIT_ID=ca-app-pub-2685698271254859/9922829329 --dart-define=DD_APPLICATION_ID=f8eabf3c-5db3-4f7e-8e6a-5a72433b46d2 --dart-define=DD_CLIENT_TOKEN=puba617ab01333a95a25a9d3709f04e1654" />
+    <option name="attachArgs" value="--profile" />
     <option name="buildFlavor" value="prod" />
     <option name="filePath" value="$PROJECT_DIR$/lib/main.dart" />
     <method v="2" />
diff --git a/android/app/libs/liblantern-all.aar b/android/app/libs/liblantern-all.aar
index 675f1e563..f9d72a574 100644
--- a/android/app/libs/liblantern-all.aar
+++ b/android/app/libs/liblantern-all.aar
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:589fd604df85087a334cd778b34fc7a0d10a1bd24641091ac7c1c6ffe18da0d1
-size 78809650
+oid sha256:07b53368d7a8fe85f6e10eb8eab68bb824301cc44ad79ead89593d749e7d63d0
+size 78256609
diff --git a/android/app/proguard-rules.pro b/android/app/proguard-rules.pro
index 7a4e2fd76..26cdad8d0 100644
--- a/android/app/proguard-rules.pro
+++ b/android/app/proguard-rules.pro
@@ -190,4 +190,14 @@
 
 ##---------------Begin: proguard configuration for Signal  ----------
 -keep class org.whispersystems.** { *; }
-##---------------End: proguard configuration for Signal  ----------
\ No newline at end of file
+##---------------End: proguard configuration for Signal  ----------
+
+-keep class com.google.ads.** # Don't proguard AdMob classes
+-dontwarn com.google.ads.
+-keep public class com.google.android.gms.ads.** {
+    public *;
+}
+
+-keep public class com.google.ads.** {
+    public *;
+}
\ No newline at end of file
diff --git a/android/app/src/main/kotlin/io/lantern/model/SessionModel.kt b/android/app/src/main/kotlin/io/lantern/model/SessionModel.kt
index ed0a22e69..db1bc44c6 100644
--- a/android/app/src/main/kotlin/io/lantern/model/SessionModel.kt
+++ b/android/app/src/main/kotlin/io/lantern/model/SessionModel.kt
@@ -1,9 +1,9 @@
 package io.lantern.model
 
 import android.app.Activity
+import android.content.Intent
 import android.os.AsyncTask
 import android.os.Build
-import android.content.Intent
 import androidx.core.content.ContextCompat
 import com.google.gson.JsonObject
 import com.google.protobuf.ByteString
@@ -23,7 +23,6 @@ import org.getlantern.lantern.MainActivity
 import org.getlantern.lantern.R
 import org.getlantern.lantern.activity.FreeKassaActivity_
 import org.getlantern.lantern.activity.WebViewActivity_
-import org.getlantern.mobilesdk.model.IssueReporter
 import org.getlantern.lantern.datadog.Datadog
 import org.getlantern.lantern.model.LanternHttpClient
 import org.getlantern.lantern.model.LanternHttpClient.ProCallback
@@ -40,6 +39,7 @@ import org.getlantern.lantern.util.restartApp
 import org.getlantern.lantern.util.showAlertDialog
 import org.getlantern.lantern.util.showErrorDialog
 import org.getlantern.mobilesdk.Logger
+import org.getlantern.mobilesdk.model.IssueReporter
 import org.getlantern.mobilesdk.model.SessionManager
 
 /**
@@ -66,7 +66,11 @@ class SessionModel(
         const val PATH_USER_LEVEL = "userLevel"
 
         const val PATH_SPLIT_TUNNELING = "/splitTunneling"
+        const val SHOULD_SHOW_CAS_ADS = "shouldShowCASAds"
+        const val SHOULD_SHOW_GOOGLE_ADS = "shouldShowGoogleAds"
         const val PATH_APPS_DATA = "/appsData/"
+
+
     }
 
     init {
@@ -87,30 +91,43 @@ class SessionModel(
             // hard disable chat
             tx.put(SessionManager.CHAT_ENABLED, false)
             tx.put(PATH_SDK_VERSION, Internalsdk.sdkVersion())
+            //By Default set it to false
+            tx.put(SHOULD_SHOW_GOOGLE_ADS, false)
+            tx.put(SHOULD_SHOW_CAS_ADS, false)
         }
         updateAppsData()
+        checkAdsAvailability()
     }
 
+    fun checkAdsAvailability() {
+        Logger.debug(TAG, "checkAdsAvailability called")
+        val googleAds = shouldShowAdsBasedRegion { LanternApp.getSession().shouldShowAdsEnabled() }
+        Logger.debug(TAG, "checkAdsAvailability with googleAds values $googleAds enable ${LanternApp.getSession().shouldShowAdsEnabled()}")
+        val casAds = shouldShowAdsBasedRegion { LanternApp.getSession().shouldCASShowAdsEnabled() }
+        Logger.debug(TAG, "checkAdsAvailability with cas values $googleAds enable ${LanternApp.getSession().shouldCASShowAdsEnabled()}")
+        db.mutate { tx ->
+            tx.put(SHOULD_SHOW_GOOGLE_ADS, googleAds)
+            tx.put(SHOULD_SHOW_CAS_ADS, casAds)
+        }
+    }
+
+
     override fun doOnMethodCall(call: MethodCall, result: MethodChannel.Result) {
         when (call.method) {
             "authorizeViaEmail" -> authorizeViaEmail(call.argument("emailAddress")!!, result)
-            "shouldShowAds" -> {
-                result.success(shouldShowAdsBasedRegion {
-                    LanternApp.getSession().shouldShowAdsEnabled()
-                })
-            }
-            "shouldCASShowAds" -> {
-                result.success(shouldShowAdsBasedRegion {
-                    LanternApp.getSession().shouldCASShowAdsEnabled()
-                })
-            }
             "checkEmailExists" -> checkEmailExists(call.argument("emailAddress")!!, result)
             "requestLinkCode" -> requestLinkCode(result)
             "resendRecoveryCode" -> sendRecoveryCode(result)
             "validateRecoveryCode" -> validateRecoveryCode(call.argument("code")!!, result)
             "approveDevice" -> approveDevice(call.argument("code")!!, result)
             "removeDevice" -> removeDevice(call.argument("deviceId")!!, result)
-            "reportIssue" -> reportIssue(call.argument("email")!!, call.argument("issue")!!, call.argument("description")!!, result)
+            "reportIssue" -> reportIssue(
+                call.argument("email")!!,
+                call.argument("issue")!!,
+                call.argument("description")!!,
+                result
+            )
+
             "applyRefCode" -> paymentsUtil.applyRefCode(call.argument("refCode")!!, result)
             "redeemResellerCode" -> paymentsUtil.redeemResellerCode(
                 call.argument("email")!!,
@@ -158,12 +175,15 @@ class SessionModel(
                     activity.startActivity(intent)
                 }
             }
+
             "trackUserAction" -> {
                 Datadog.trackUserClick(call.argument("message")!!)
             }
+
             "acceptTerms" -> {
                 LanternApp.getSession().acceptTerms()
             }
+
             "setLanguage" -> {
                 LanternApp.getSession().setLanguage(call.argument("lang"))
             }
@@ -572,9 +592,18 @@ class SessionModel(
         )
     }
 
-    private fun reportIssue(email: String, issue: String, description: String, methodCallResult: MethodChannel.Result) {
+    private fun reportIssue(
+        email: String,
+        issue: String,
+        description: String,
+        methodCallResult: MethodChannel.Result
+    ) {
         if (!Utils.isNetworkAvailable(activity)) {
-            methodCallResult.error("errorReportingIssue", activity.getString(R.string.no_internet_connection), null)
+            methodCallResult.error(
+                "errorReportingIssue",
+                activity.getString(R.string.no_internet_connection),
+                null
+            )
             return
         }
         Logger.debug(TAG, "Reporting $issue issue on behalf of $email")
diff --git a/android/app/src/main/kotlin/org/getlantern/lantern/MainActivity.kt b/android/app/src/main/kotlin/org/getlantern/lantern/MainActivity.kt
index 7de549beb..b241614ac 100644
--- a/android/app/src/main/kotlin/org/getlantern/lantern/MainActivity.kt
+++ b/android/app/src/main/kotlin/org/getlantern/lantern/MainActivity.kt
@@ -543,6 +543,7 @@ class MainActivity :
                 //If user come here it mean user has all permissions needed
                 // Also user given permission for VPN service dialog as well
                 LanternApp.getSession().setHasFirstSessionCompleted(true)
+                sessionModel.checkAdsAvailability()
                 updateStatus(true)
                 startVpnService()
             }
@@ -606,6 +607,7 @@ class MainActivity :
                 // this mean user has already given
                 // system permissions
                 LanternApp.getSession().setHasFirstSessionCompleted(true)
+                sessionModel.checkAdsAvailability()
             }
 
         }
diff --git a/lib/ad_helper.dart b/lib/ad_helper.dart
index f3b990085..76eaac952 100644
--- a/lib/ad_helper.dart
+++ b/lib/ad_helper.dart
@@ -5,18 +5,14 @@ import 'package:clever_ads_solutions/public/AdCallback.dart';
 import 'package:clever_ads_solutions/public/AdImpression.dart';
 import 'package:clever_ads_solutions/public/AdTypes.dart';
 import 'package:clever_ads_solutions/public/Audience.dart';
-import 'package:clever_ads_solutions/public/ConsentFlow.dart';
 import 'package:clever_ads_solutions/public/InitConfig.dart';
 import 'package:clever_ads_solutions/public/InitializationListener.dart';
 import 'package:clever_ads_solutions/public/MediationManager.dart';
-import 'package:clever_ads_solutions/public/OnDismissListener.dart';
 import 'package:flutter/foundation.dart';
 import 'package:google_mobile_ads/google_mobile_ads.dart';
 import 'package:lantern/common/datadog.dart';
 import 'package:lantern/replica/common.dart';
 
-import 'common/session_model.dart';
-
 enum AdType { Google, CAS }
 
 const privacyPolicy = 'https://lantern.io/privacy';
@@ -59,10 +55,9 @@ class AdHelper {
   }
 
   // Private methods to decide whether to load or show Google Ads or CAS ads based on conditions
-  Future<void> _decideAndLoadAds() async {
-    final shouldShowGoogleAds = await sessionModel.shouldShowAds();
-    final shouldShowCASAds = await sessionModel.shouldCASShowAds();
-
+  Future<void> _decideAndLoadAds(
+      {required bool shouldShowGoogleAds,
+      required bool shouldShowCASAds}) async {
     logger.d(
         '[Ads Manager] Google Ads enable $shouldShowGoogleAds: CAS Ads $shouldShowCASAds');
     if (shouldShowGoogleAds) {
@@ -95,6 +90,7 @@ class AdHelper {
 
   Future<void> _loadInterstitialAd() async {
     //To avoid calling multiple ads request repeatedly
+    assert(interstitialAdUnitId!="","interstitialAdUnitId should not be null or empty");
     if (_interstitialAd == null && _failedLoadAttempts < _maxFailAttempts) {
       logger.i('[Ads Manager] Request: Making Google Ad request.');
       await InterstitialAd.load(
@@ -102,20 +98,24 @@ class AdHelper {
         request: const AdRequest(),
         adLoadCallback: InterstitialAdLoadCallback(
           onAdLoaded: (ad) {
+            _failedLoadAttempts = 0;
             ad.fullScreenContentCallback = FullScreenContentCallback(
               onAdClicked: (ad) {
                 logger.i('[Ads Manager] onAdClicked callback');
-                Datadog.trackUserTap('User tapped on interstitial ad', googleAttributes);
+                Datadog.trackUserTap(
+                    'User tapped on interstitial ad', googleAttributes);
               },
               onAdShowedFullScreenContent: (ad) {
                 logger.i('[Ads Manager] Showing Ads');
-                Datadog.trackUserCustom('User shown interstitial ad', googleAttributes);
+                Datadog.trackUserCustom(
+                    'User shown interstitial ad', googleAttributes);
               },
               onAdFailedToShowFullScreenContent: (ad, error) {
                 logger.i(
                     '[Ads Manager] onAdFailedToShowFullScreenContent callback');
-                Datadog.addError('Ad failed to show full screen content: $error',
-                  attributes: googleAttributes);
+                Datadog.addError(
+                    'Ad failed to show full screen content: $error',
+                    attributes: googleAttributes);
                 //if ads fail to load let user turn on VPN
                 _postShowingAds();
               },
@@ -131,7 +131,8 @@ class AdHelper {
           onAdFailedToLoad: (err) {
             _failedLoadAttempts++; // increment the count on failure
             logger.i('[Ads Manager] failed to load $err');
-            Datadog.addError('failed to load interstitial ad: $err', attributes: googleAttributes);
+            Datadog.addError('failed to load interstitial ad: $err',
+                attributes: googleAttributes);
             _postShowingAds();
           },
         ),
@@ -143,12 +144,10 @@ class AdHelper {
     if (_currentAdType == AdType.Google) {
       _interstitialAd?.dispose();
       _interstitialAd = null;
-      _failedLoadAttempts = 0; // Reset counter for Google Ads
       logger.i(
           '[Ads Manager] Post-show: Google Ad displayed. Resetting failed load attempts and requesting a new ad.');
       _loadInterstitialAd();
     } else if (_currentAdType == AdType.CAS) {
-      _failedCASLoadAttempts = 0; // Reset counter for CAS Ads
       logger.i(
           '[Ads Manager] Post-show: CAS Ad displayed. Resetting failed load attempts and requesting a new ad.');
       _loadCASInterstitial();
@@ -162,8 +161,13 @@ class AdHelper {
   }
 
   // Public methods
-  Future<void> loadAds() async {
-    await _decideAndLoadAds();
+  Future<void> loadAds(
+      {required bool shouldShowGoogleAds,
+      required bool shouldShowCASAds}) async {
+    await _decideAndLoadAds(
+      shouldShowCASAds: shouldShowCASAds,
+      shouldShowGoogleAds: shouldShowGoogleAds,
+    );
   }
 
   Future<void> showAds() async {
@@ -211,14 +215,16 @@ class AdHelper {
 
   void _onCASAdShowFailed() {
     logger.e('[Ads Manager] Error: CAS Interstitial failed to display.');
-    Datadog.addError('Failed to display interstitial ad', attributes: casAttributes);
+    Datadog.addError('Failed to display interstitial ad',
+        attributes: casAttributes);
     _failedCASLoadAttempts++;
     _postShowingAds(); // Reload or decide the next action
   }
 
   void _onCASAdClosedOrComplete() {
     logger.i('[Ads Manager] Completion: CAS Interstitial closed or completed.');
-    Datadog.trackUserCustom('Interstitial ad closed or completed', casAttributes);
+    Datadog.trackUserCustom(
+        'Interstitial ad closed or completed', casAttributes);
     // Reset the counter when the ad successfully shows and closes/completes
     _failedCASLoadAttempts = 0;
     _postShowingAds();
@@ -278,7 +284,8 @@ class InterstitialListenerWrapper extends AdCallback {
     onFailed.call();
     logger.i(
         '[CASIntegrationHelper] - InterstitialListenerWrapper onShowFailed-:$message');
-    Datadog.addError('Interstitial ad onShowFailed: $message', attributes: casAttributes);
+    Datadog.addError('Interstitial ad onShowFailed: $message',
+        attributes: casAttributes);
   }
 
   @override
diff --git a/lib/common/session_model.dart b/lib/common/session_model.dart
index 406bf81b5..8a7fd7ec6 100644
--- a/lib/common/session_model.dart
+++ b/lib/common/session_model.dart
@@ -161,16 +161,6 @@ class SessionModel extends Model {
         .invokeMethod('getCountryCode', <String, dynamic>{});
   }
 
-  Future<bool> shouldShowAds() async {
-    return await methodChannel
-        .invokeMethod('shouldShowAds', <String, dynamic>{});
-  }
-
-  Future<bool> shouldCASShowAds() async {
-    return await methodChannel
-        .invokeMethod('shouldCASShowAds', <String, dynamic>{});
-  }
-
   Future<void> setLanguage(String lang) {
     return methodChannel.invokeMethod('setLanguage', <String, dynamic>{
       'lang': lang,
@@ -212,6 +202,20 @@ class SessionModel extends Model {
     });
   }
 
+  Widget shouldShowGoogleAds(ValueWidgetBuilder<bool> builder) {
+    return subscribedSingleValueBuilder<bool>(
+      'shouldShowGoogleAds',
+      builder: builder,
+    );
+  }
+
+  Widget shouldShowCASAds(ValueWidgetBuilder<bool> builder) {
+    return subscribedSingleValueBuilder<bool>(
+      'shouldShowCASAds',
+      builder: builder,
+    );
+  }
+
   Widget selectedTab(ValueWidgetBuilder<String> builder) {
     return subscribedSingleValueBuilder<String>(
       '/selectedTab',
diff --git a/lib/main.dart b/lib/main.dart
index 1497af6f8..ebb60f19f 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -3,7 +3,6 @@ import 'package:google_mobile_ads/google_mobile_ads.dart';
 import 'package:lantern/app.dart';
 import 'package:lantern/common/common.dart';
 
-
 Future<void> main() async {
   // CI will be true only when running appium test
   var CI = const String.fromEnvironment('CI', defaultValue: 'false');
@@ -11,7 +10,6 @@ Future<void> main() async {
   if (CI == 'true') {
     enableFlutterDriverExtension();
   }
- 
   WidgetsFlutterBinding.ensureInitialized();
   await _initGoogleMobileAds();
   await SystemChrome.setPreferredOrientations([DeviceOrientation.portraitUp]);
@@ -21,5 +19,8 @@ Future<void> main() async {
 Future<void> _initGoogleMobileAds() async {
   await MobileAds.instance.initialize();
   await MobileAds.instance.setAppMuted(true);
-  // await MobileAds.instance.updateRequestConfiguration(RequestConfiguration(testDeviceIds: ['D79728264130CE0918737B5A2178D362']));
+  await MobileAds.instance.updateRequestConfiguration(RequestConfiguration(testDeviceIds: ['D79728264130CE0918737B5A2178D362']));
+  MobileAds.instance.openAdInspector((p0) {
+    print('ad error $p0');
+  });
 }
diff --git a/lib/vpn/vpn_switch.dart b/lib/vpn/vpn_switch.dart
index 59594956e..b0f813830 100644
--- a/lib/vpn/vpn_switch.dart
+++ b/lib/vpn/vpn_switch.dart
@@ -1,4 +1,5 @@
 import 'package:lantern/vpn/vpn.dart';
+
 import '../ad_helper.dart';
 
 class VPNSwitch extends StatefulWidget {
@@ -12,7 +13,6 @@ class _VPNSwitchState extends State<VPNSwitch> {
   @override
   void initState() {
     super.initState();
-    adHelper.loadAds();
   }
 
   bool isIdle(String vpnStatus) =>
@@ -36,16 +36,26 @@ class _VPNSwitchState extends State<VPNSwitch> {
 
   @override
   Widget build(BuildContext context) {
-    return Transform.scale(
-        scale: 2,
-        child: vpnModel
-            .vpnStatus((BuildContext context, String vpnStatus, Widget? child) {
-          return FlutterSwitch(
-            value: vpnStatus == 'connected' || vpnStatus == 'disconnecting',
-            activeColor: onSwitchColor,
-            inactiveColor: offSwitchColor,
-            onToggle: (bool newValue) => onSwitchTap(newValue, vpnStatus),
-          );
-        }));
+    return sessionModel
+        .shouldShowGoogleAds((context, isGoogleAdsEnable, child) {
+          print('Ads Manager Google Ad $isGoogleAdsEnable');
+      return sessionModel.shouldShowCASAds((context, isCasAdsEnable, child) {
+        print('Ads Manager CAS Ad $isCasAdsEnable');
+        adHelper.loadAds(
+            shouldShowGoogleAds: isGoogleAdsEnable,
+            shouldShowCASAds: isCasAdsEnable);
+        return Transform.scale(
+            scale: 2,
+            child: vpnModel.vpnStatus(
+                (BuildContext context, String vpnStatus, Widget? child) {
+              return FlutterSwitch(
+                value: vpnStatus == 'connected' || vpnStatus == 'disconnecting',
+                activeColor: onSwitchColor,
+                inactiveColor: offSwitchColor,
+                onToggle: (bool newValue) => onSwitchTap(newValue, vpnStatus),
+              );
+            }));
+      });
+    });
   }
 }