...

app/src/main/java/com/base/pdfviewerscannerwhite/ui/document/pdf/PdfActivity.kt

@@ -68,18 +68,32 @@ class PdfActivity : BaseActivity<ActivityPdfBinding>(), PdfView {
         changeNormalUI()
 
         muPDFCore = pdfPresenter.openFile(path)
-        muPDFCore?.needsPassword()
+
+
         // 搜索设为空
         SearchTaskResult.set(null)
 
-        if (muPDFCore == null) {
-            toast("can't open pdf")
+        if (muPDFCore?.needsPassword() == true) {
+            val pwd = intent.extras?.getString("pwd") ?: ""
+            val flag = muPDFCore?.authenticatePassword(pwd) ?: false
+            if (flag) {
+                pdfPresenter.password = pwd
+                pdfPageAdapter.setPassword(pwd)
+                createPdfUI()
+                binding.root.postDelayed({
+                    pdfPresenter.iniPdfPage(path)
+                }, 500)
+            } else {
+                toast("unknown error")
+                finish()
+            }
         } else {
             createPdfUI()
             binding.root.postDelayed({
                 pdfPresenter.iniPdfPage(path)
             }, 500)
         }
+
     }
 
     override fun initListener() {

app/src/main/java/com/base/pdfviewerscannerwhite/ui/document/pdf/PdfBoxUtils.kt

@@ -42,8 +42,8 @@ object PdfBoxUtils {
         return drawableList
     }
 
-    fun getNumberOfPages(filePath: String): Int {
-        val document = PDDocument.load(File(filePath))
+    fun getNumberOfPages(filePath: String, password: String? = null): Int {
+        val document = if (password == null) PDDocument.load(File(filePath)) else PDDocument.load(File(filePath), password)
         return document.numberOfPages
     }
 
@@ -51,9 +51,10 @@ object PdfBoxUtils {
         context: Context,
         filePath: String,
         pageIndex: Int,
-        scale: Float = 1f
+        scale: Float = 1f,
+        password: String? = null
     ): Drawable {
-        val document = PDDocument.load(File(filePath))
+        val document = if (password == null) PDDocument.load(File(filePath)) else PDDocument.load(File(filePath), password)
         val renderer = PDFRenderer(document)
 
         val bitmap: Bitmap = renderer.renderImage(pageIndex, scale, ImageType.RGB, RenderDestination.EXPORT)
@@ -115,6 +116,44 @@ object PdfBoxUtils {
         } catch (e: Exception) {
 
         }
+    }
+
+    fun clearPassword(filePath: String, password: String) {
+        try {
+            PDDocument.load(File(filePath), password).use { document ->
+                if (document.isEncrypted) {
+                    val ap: AccessPermission = document.getCurrentAccessPermission()
+                    if (ap.isOwnerPermission) {
+                        // 创建一个新的保护策略，不设置密码
+                        val spp = StandardProtectionPolicy(
+                            "", "",
+                            AccessPermission.getOwnerAccessPermission()
+                        )
+                        document.protect(spp)
+                    }
+                }
+                // 保存PDF文件到新的位置，没有密码保护
+                document.save(File(filePath))
+            }
+        } catch (e: IOException) {
+            e.printStackTrace()
+        }
+    }
 
+    fun checkPwd(filePath: String, password: String): Boolean {
+        try {
+            // 尝试使用提供的密码加载PDF文件
+            PDDocument.load(File(filePath), password).use { document ->
+                // 如果没有抛出异常，说明密码正确
+                return true
+            }
+        } catch (e: Exception) {
+            // 加载文件时，如果密码保护策略异常，密码可能错误
+            println("Incorrect password or restricted permissions.")
+        } catch (e: IOException) {
+            // 其他I/O异常处理
+            println("An I/O error occurred: " + e.message)
+        }
+        return false
     }
 }
\ No newline at end of file

app/src/main/java/com/base/pdfviewerscannerwhite/ui/document/pdf/PdfPagerAdapter.kt

@@ -17,11 +17,14 @@ import java.util.concurrent.ThreadPoolExecutor
 import java.util.concurrent.TimeUnit
 
 
-class PdfPagerAdapter(val pdfPath: String, val itemLayout: Int = R.layout.item_pdf_pager) :
-    BaseQuickAdapter<PdfPageBean, PdfPagerAdapter.PdfPagerViewHolder>() {
+class PdfPagerAdapter(
+    val pdfPath: String,
+    val itemLayout: Int = R.layout.item_pdf_pager,
+) : BaseQuickAdapter<PdfPageBean, PdfPagerAdapter.PdfPagerViewHolder>() {
 
+    var mPassword: String? = null
     var selectAction: ((enable: Boolean, allSelect: Boolean) -> Unit)? = null
-    var clickAction: ((pageIndex:Int) -> Unit)? = null
+    var clickAction: ((pageIndex: Int) -> Unit)? = null
 
     inner class PdfPagerViewHolder(view: View) : ViewHolder(view)
 
@@ -85,7 +88,7 @@ class PdfPagerAdapter(val pdfPath: String, val itemLayout: Int = R.layout.item_p
     ) {
         threadPoolExecutor.execute {
             runCatching {
-                val drawable = PdfBoxUtils.getPdfDrawablePage(context, pdfPath, item.pageIndex, scale)
+                val drawable = PdfBoxUtils.getPdfDrawablePage(context, pdfPath, item.pageIndex, scale, mPassword)
                 item.pageDrawable = drawable
                 itemView.post {
                     item.pageDrawable?.let {
@@ -115,4 +118,10 @@ class PdfPagerAdapter(val pdfPath: String, val itemLayout: Int = R.layout.item_p
         }
         notifyDataSetChanged()
     }
+
+    @SuppressLint("NotifyDataSetChanged")
+    fun setPassword(password: String) {
+        mPassword = password
+        notifyDataSetChanged()
+    }
 }
\ No newline at end of file

app/src/main/java/com/base/pdfviewerscannerwhite/ui/document/pdf/PdfPresenter.kt

@@ -18,6 +18,7 @@ class PdfPresenter(
     val pdfView: PdfView? = null
 ) {
     private val TAG = "PdfPresenter"
+    var password: String? = null
 
     fun openFile(path: String): MuPDFCore? {
         var muPDFCore: MuPDFCore? = null
@@ -36,7 +37,7 @@ class PdfPresenter(
 
     fun iniPdfPage(filePath: String) {
         val list = arrayListOf<PdfPageBean>()
-        val number = PdfBoxUtils.getNumberOfPages(filePath)
+        val number = PdfBoxUtils.getNumberOfPages(filePath, password)
         repeat(number) {
             list.add(PdfPageBean(it))
         }
@@ -50,7 +51,7 @@ class PdfPresenter(
 
         try {
             // 加载现有 PDF 文档
-            val sourceDocument = PDDocument.load(File(srcPath))
+            val sourceDocument = PDDocument.load(File(srcPath), password)
 
             // 创建新的 PDF 文档
             val newDocument = PDDocument()

app/src/main/java/com/base/pdfviewerscannerwhite/ui/main/DocumentFragment.kt

@@ -12,6 +12,7 @@ import com.base.pdfviewerscannerwhite.ui.document.pdf.PdfActivity
 import com.base.pdfviewerscannerwhite.ui.document.pdf.PdfSplitActivity
 import com.base.pdfviewerscannerwhite.ui.view.DialogView.showDocumentRenameDialog
 import com.base.pdfviewerscannerwhite.ui.view.DialogView.showPdfDetailDialog
+import com.base.pdfviewerscannerwhite.ui.view.DialogView.showPdfPwdDialog
 import com.base.pdfviewerscannerwhite.utils.PermissionUtils.checkStorePermission
 import java.io.File
 
@@ -47,9 +48,23 @@ class DocumentFragment() : BaseFragment<FragmentDocumentBinding>(), DocumentView
             documentPresenter.saveBookmarkChange(addRemove, path)
         }
         adapter.itemClick = { item ->
-            startActivity(Intent(requireContext(), PdfActivity::class.java).apply {
-                putExtra("path", item.path)
-            })
+            if (type == TYPE_PDF) {
+                if (item.state == 0) {
+                    startActivity(Intent(requireContext(), PdfActivity::class.java).apply {
+                        putExtra("path", item.path)
+                    })
+                }
+                if (item.state == 1) {
+                    requireContext().showPdfPwdDialog(null, item, true, verificationAction = { pwd ->
+                        startActivity(Intent(requireContext(), PdfActivity::class.java).apply {
+                            putExtra("path", item.path)
+                            putExtra("pwd", pwd)
+                        })
+                    })
+                }
+
+            }
+
         }
         adapter.moreAction = { item ->
             if (item.type == TYPE_PDF) {

app/src/main/java/com/base/pdfviewerscannerwhite/ui/splash/SplashPresenter.kt

@@ -14,7 +14,7 @@ class SplashPresenter(
 
     private var jumpJob: Job? = null
 
-    var loadingTime = 5L
+    var loadingTime = 2L
 
     /**
      * 超时跳转

app/src/main/java/com/base/pdfviewerscannerwhite/ui/view/DialogView.kt

@@ -34,6 +34,7 @@ import com.base.pdfviewerscannerwhite.utils.LogEx
 import com.base.pdfviewerscannerwhite.utils.SpStringUtils
 import com.base.pdfviewerscannerwhite.utils.SpStringUtils.LAST_VIEW_KEY
 import com.base.pdfviewerscannerwhite.utils.ToastUtils.toast
+import com.google.android.gms.dynamic.IFragmentWrapper
 import com.google.android.material.bottomsheet.BottomSheetBehavior
 import com.google.android.material.bottomsheet.BottomSheetDialog
 import java.io.File
@@ -121,6 +122,12 @@ object DialogView {
                 documentView.deleteDocument(item)
             }
         }
+        if (item.state == 1) {
+            binding.tvLock.text = "Unlock PDF"
+        }
+        if (item.state == 0) {
+            binding.tvLock.text = "Lock PDF"
+        }
         binding.llLock.setOnClickListener {
             showPdfPwdDialog(dialog, item)
         }
@@ -179,7 +186,13 @@ object DialogView {
     }
 
 
-    private fun Context.showPdfPwdDialog(firstDialog: Dialog?, item: DocumentBean) {
+    @SuppressLint("SetTextI18n")
+    fun Context.showPdfPwdDialog(
+        firstDialog: Dialog?,
+        item: DocumentBean,
+        isCheckPwd: Boolean = false,
+        verificationAction: ((pwd: String) -> Unit)? = null
+    ) {
         val dialog = BottomSheetDialog(this, R.style.BottomSheetDialog)
         val binding = DialogPdfPasswordBinding.inflate(LayoutInflater.from(this))
         dialog.setContentView(binding.root)
@@ -195,11 +208,29 @@ object DialogView {
         //展开
         behavior.state = BottomSheetBehavior.STATE_EXPANDED
 
-        binding.edit.requestFocus()
+        if (!isCheckPwd) {
+            if (item.state == 1) {
+                binding.tvTittle.text = "Delete Password"
+                binding.tvTip.text = "Delete password, the file is not password protected"
+            }
+            if (item.state == 0) {
+                binding.tvTittle.text = "Set Password"
+                binding.tvTip.text = "Set password protection pdf"
+
+            }
+        } else {
+            binding.tvTittle.text = "Input Password"
+            val file = File(item.path)
+            binding.tvTip.text = "${file.name} password protected"
+            binding.tvInputTip.visibility = View.VISIBLE
+        }
 
+        binding.edit.requestFocus()
         binding.edit.addTextChangedListener {
             binding.tvConfirm.isEnabled = it.toString().isNotEmpty()
+            binding.tvErrorTip.visibility = View.GONE
         }
+
         binding.tvCancel.setOnClickListener {
             dialog.dismiss()
         }
@@ -207,7 +238,25 @@ object DialogView {
             dialog.dismiss()
             firstDialog?.dismiss()
             val pwd = binding.edit.text.toString()
-            PdfBoxUtils.setPassword(item.path, pwd, pwd)
+
+            if (!isCheckPwd) {
+                if (item.state == 0) {
+                    PdfBoxUtils.setPassword(item.path, pwd, pwd)
+                    toast("Success Encryption")
+                }
+                if (item.state == 1) {
+                    PdfBoxUtils.clearPassword(item.path, pwd)
+                    toast("clear Encryption")
+                }
+            } else {
+                val result = PdfBoxUtils.checkPwd(item.path, pwd)
+                if (!result) {
+                    binding.tvErrorTip.visibility = View.VISIBLE
+                    return@setOnClickListener
+                }
+                verificationAction?.invoke(pwd)
+            }
+
         }
     }
 

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

@@ -210,6 +210,7 @@
                 tools:ignore="ContentDescription" />
 
             <TextView
+                android:id="@+id/tv_lock"
                 android:layout_width="0dp"
                 android:layout_height="wrap_content"
                 android:layout_gravity="center_vertical"

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

@@ -60,7 +60,7 @@
                 android:layout_height="wrap_content"
                 android:layout_gravity="center_vertical"
                 android:layout_marginStart="13dp"
-                android:src="@mipmap/merge_tool"
+                android:src="@mipmap/merge"
                 tools:ignore="ContentDescription" />
 
             <TextView

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

@@ -45,6 +45,7 @@
         app:layout_constraintTop_toBottomOf="@id/fl">
 
         <TextView
+            android:id="@+id/tv_tip"
             android:layout_width="wrap_content"
             android:layout_height="wrap_content"
             android:layout_marginStart="16dp"
@@ -54,6 +55,16 @@
             android:textSize="18sp"
             tools:ignore="HardcodedText" />
 
+        <TextView
+            android:id="@+id/tv_input_tip"
+            android:layout_width="wrap_content"
+            android:layout_height="wrap_content"
+            android:layout_marginStart="16dp"
+            android:layout_marginTop="8dp"
+            android:text="Enter the password to open the file"
+            android:visibility="gone"
+            tools:ignore="HardcodedText" />
+
         <FrameLayout
             android:layout_width="match_parent"
             android:layout_height="55dp"
@@ -89,6 +100,17 @@
 
         </FrameLayout>
 
+        <TextView
+            android:id="@+id/tv_error_tip"
+            android:layout_width="wrap_content"
+            android:layout_height="wrap_content"
+            android:layout_marginStart="16dp"
+            android:layout_marginTop="8dp"
+            android:text="Password Error"
+            android:textColor="#FA2232"
+            android:visibility="gone"
+            tools:ignore="HardcodedText" />
+
         <LinearLayout
             android:layout_width="wrap_content"
             android:layout_height="wrap_content"