...

app/src/main/java/com/base/datarecovery/activity/MainActivity.kt

@@ -2,6 +2,7 @@ package com.base.datarecovery.activity
 
 import android.graphics.Color
 import android.os.Bundle
+import android.os.Environment
 import androidx.activity.addCallback
 import androidx.activity.enableEdgeToEdge
 import androidx.appcompat.app.AppCompatActivity
@@ -11,6 +12,7 @@ import androidx.core.view.updatePadding
 import androidx.lifecycle.lifecycleScope
 import com.base.datarecovery.databinding.ActivityMainBinding
 import com.base.datarecovery.help.BaseActivity
+import com.base.datarecovery.help.FileHelp.getDirFiles
 import com.base.datarecovery.help.PermissionHelp.checkStorePermission
 import com.base.datarecovery.utils.BarUtils
 import com.base.datarecovery.view.DialogViews.showExitDialog
@@ -19,6 +21,8 @@ import com.base.datarecovery.view.PermissionDialog.showPermissionBottomSheet
 import com.base.datarecovery.view.RateStarPop
 import com.base.datarecovery.view.RateStarPop.showRateStarPopDialog
 import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.Job
+import kotlinx.coroutines.async
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.launch
 
@@ -50,6 +54,8 @@ class MainActivity : BaseActivity<ActivityMainBinding>() {
             if (dialog == null) {
                 dialog = showPermissionBottomSheet(launcher)
             }
+        } else {
+
         }
 
         lifecycleScope.launch(Dispatchers.Main) {
@@ -57,4 +63,23 @@ class MainActivity : BaseActivity<ActivityMainBinding>() {
             showRateStarPopDialog()
         }
     }
+
+    fun setScreenSize(action: (size: Long) -> Unit) {
+        if (!checkStorePermission()) {
+            action.invoke(0)
+        } else {
+            lifecycleScope.launch(Dispatchers.Main) {
+                action.invoke(getScreenSize())
+            }
+        }
+    }
+
+    private suspend fun getScreenSize(): Long {
+        val deferred = lifecycleScope.async(Dispatchers.IO) {
+            val dcim = Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_DCIM)
+            getDirFiles(dcim).sumOf { it.length() }
+        }
+        return deferred.await()
+    }
+
 }
\ No newline at end of file

app/src/main/java/com/base/datarecovery/fragment/HomeFragment.kt

@@ -7,6 +7,7 @@ import android.content.Intent
 import android.view.View
 import android.view.animation.AccelerateDecelerateInterpolator
 import com.base.datarecovery.R
+import com.base.datarecovery.activity.MainActivity
 import com.base.datarecovery.activity.SettingActivity
 import com.base.datarecovery.activity.junkclean.ScanJunkActivity
 import com.base.datarecovery.activity.privacyspace.PrivacyPinOneActivity
@@ -22,6 +23,7 @@ import com.base.datarecovery.bean.ConstObject.SCAN_VIDEOS
 import com.base.datarecovery.bean.ConstObject.privacyPinPassword
 import com.base.datarecovery.databinding.FragmentHomeBinding
 import com.base.datarecovery.help.BaseFragment
+import com.base.datarecovery.help.KotlinExt.toFormatSize
 
 class HomeFragment : BaseFragment<FragmentHomeBinding>() {
 
@@ -73,6 +75,7 @@ class HomeFragment : BaseFragment<FragmentHomeBinding>() {
         binding.flSetting.setOnClickListener {
             startActivity(Intent(requireContext(), SettingActivity::class.java))
         }
+
     }
 
     @SuppressLint("Recycle")
@@ -114,6 +117,11 @@ class HomeFragment : BaseFragment<FragmentHomeBinding>() {
             AdmobMaxHelper.admobMaxShowNativeAd(requireActivity(), binding.flAd, 0)
             resumeTime++
         }
+        (requireActivity() as MainActivity).setScreenSize {
+            if (it > 0) {
+                binding.tvScreenSize.text = it.toFormatSize(0)
+            }
+        }
     }
 
 }
\ No newline at end of file

app/src/main/java/com/base/datarecovery/help/FileHelp.kt

@@ -114,4 +114,5 @@ object FileHelp {
         }
         return result
     }
+
 }
\ No newline at end of file

app/src/main/java/com/base/datarecovery/view/DialogViews.kt

@@ -306,9 +306,11 @@ object DialogViews {
         val parentView = binding.root.parent as View
         BottomSheetBehavior.from(parentView).state = BottomSheetBehavior.STATE_EXPANDED
         binding.tvY.setOnClickListener {
+            dialog.dismiss()
             startActivity(Intent(this, ScanJunkActivity::class.java))
         }
         binding.tvExit.setOnClickListener {
+            dialog.dismiss()
             this.finish()
         }
     }

app/src/main/res/layout/dialog_exit_app.xml

@@ -40,31 +40,33 @@
             android:textSize="16sp"
             tools:ignore="HardcodedText" />
 
+
         <TextView
-            android:id="@+id/tv_y"
+            android:id="@+id/tv_exit"
             android:layout_width="300dp"
             android:layout_height="45dp"
             android:layout_gravity="center_horizontal"
-            android:layout_marginTop="50dp"
-            android:background="@drawable/bg_4773ff_22"
+            android:layout_marginVertical="15dp"
             android:gravity="center"
-            android:text="Remove now"
-            android:textColor="@color/white"
+            android:text="Exit"
+            android:textColor="#999999"
             android:textSize="17sp"
-            android:textStyle="bold"
             tools:ignore="HardcodedText" />
 
         <TextView
-            android:id="@+id/tv_exit"
+            android:layout_marginBottom="15dp"
+            android:id="@+id/tv_y"
             android:layout_width="300dp"
             android:layout_height="45dp"
             android:layout_gravity="center_horizontal"
-            android:layout_marginVertical="15dp"
+            android:background="@drawable/bg_4773ff_22"
             android:gravity="center"
-            android:text="Exit"
-            android:textColor="#999999"
+            android:text="Remove now"
+            android:textColor="@color/white"
             android:textSize="17sp"
+            android:textStyle="bold"
             tools:ignore="HardcodedText" />
+
     </LinearLayout>
 
     <ImageView

app/src/main/res/layout/fragment_home.xml

@@ -228,6 +228,12 @@
 
                     </LinearLayout>
 
+                    <TextView
+                        android:id="@+id/tv_screen_size"
+                        android:layout_width="wrap_content"
+                        android:layout_height="wrap_content"
+                        android:layout_gravity="end"
+                        android:layout_margin="8dp" />
 
                 </androidx.cardview.widget.CardView>