PDFHelper:new:优化脚本导入和处理逻辑、优化缩略图显示

Author: BlackCCCat

PDFHelper/components/PDFHelperView.tsx

@@ -115,6 +115,17 @@ export function PDFHelperView() {
     () => selectedPages.filter((page) => page.kind === "image").length,
     [selectedPages]
   )
+  const selectedTotalPageCount = useMemo(() => {
+    let count = 0
+    for (const page of selectedPages) {
+      if (page.kind === "pdf-whole") {
+        count += page.pageCount
+      } else {
+        count += 1
+      }
+    }
+    return count
+  }, [selectedPages])
 
   const isBusy = processing || loadingMessage !== null
   const canConvert = selectedImageCount > 0 && !isBusy
@@ -332,7 +343,7 @@ export function PDFHelperView() {
             <VStack spacing={8} padding={24}>
               <Text>点右上角「+」添加文件或照片</Text>
               <Text foregroundStyle="secondaryLabel">
-                图片将显示整图，PDF 将按页展示缩略图，可逐页勾选后执行转换或合并
+                图片将显示整图；PDF 可选择按页展示缩略图(文件页数较大时不显示缩略图)或作为整体导入，支持自定义导入页码范围
               </Text>
             </VStack>
           ) : (
@@ -352,7 +363,7 @@ export function PDFHelperView() {
         <VStack spacing={8} padding={12}>
           <HStack>
             <Text foregroundStyle="secondaryLabel">
-              已选择 {selectedPages.length} 页（图片 {selectedImageCount}）
+              已选择 {selectedPages.length} 项 · 共 {selectedTotalPageCount} 页（图片 {selectedImageCount}）
             </Text>
             <Spacer />
             {loadingMessage ? (

PDFHelper/components/SourceBlockView.tsx

@@ -45,7 +45,7 @@ function rowSwipeDelete(onDelete: () => void) {
 }
 
 function PdfPreview(props: { page: PageItem }) {
-  if (props.page.kind !== "pdf") return null
+  if (props.page.kind === "image") return null
 
   if (props.page.previewImage) {
     return (
@@ -147,6 +147,37 @@ function PdfRow(props: {
   )
 }
 
+function PdfWholeRow(props: {
+  source: SourceItem
+  page: PageItem
+  onTogglePage: (sourceId: string, pageId: string) => void
+  onDeletePage: (sourceId: string, pageId: string) => void
+}) {
+  if (props.page.kind !== "pdf-whole") return null
+  const onToggle = () => props.onTogglePage(props.source.id, props.page.id)
+
+  return (
+    <HStack
+      key={props.page.id}
+      spacing={10}
+      padding={{ top: 8, bottom: 8 }}
+      frame={{ maxWidth: "infinity", alignment: "leading" }}
+      background={"rgba(0,0,0,0.001)"}
+      onTapGesture={onToggle}
+      trailingSwipeActions={rowSwipeDelete(() => props.onDeletePage(props.source.id, props.page.id))}
+    >
+      <PdfPreview page={props.page} />
+      <VStack spacing={6} frame={{ maxWidth: "infinity", alignment: "leading" }}>
+        <Text>{props.page.title}</Text>
+        <Text font="footnote" foregroundStyle="secondaryLabel">
+          {props.source.name}
+        </Text>
+      </VStack>
+      <SelectionMark selected={props.page.selected} selectedOrder={props.page.selectedOrder} />
+    </HStack>
+  )
+}
+
 export function SourceBlockView(props: {
   source: SourceItem
   onTogglePage: (sourceId: string, pageId: string) => void
@@ -164,14 +195,22 @@ export function SourceBlockView(props: {
       )}
     >
       {props.source.pages.map((page) =>
-        props.source.kind === "image" ? (
+        page.kind === "image" ? (
           <ImageRow
             key={page.id}
             source={props.source}
             page={page}
             onTogglePage={props.onTogglePage}
             onDeletePage={props.onDeletePage}
           />
+        ) : page.kind === "pdf-whole" ? (
+          <PdfWholeRow
+            key={page.id}
+            source={props.source}
+            page={page}
+            onTogglePage={props.onTogglePage}
+            onDeletePage={props.onDeletePage}
+          />
         ) : (
           <PdfRow
             key={page.id}

PDFHelper/index.tsx

@@ -1,8 +1,10 @@
 import { Navigation, Script } from "scripting"
 import { PDFHelperView } from "./components/PDFHelperView"
+import { cleanUpPreviewDirectory } from "./utils/preview"
 
 async function run() {
   await Navigation.present({ element: <PDFHelperView /> })
+  await cleanUpPreviewDirectory()
   Script.exit()
 }
 

PDFHelper/types.ts

@@ -1,4 +1,4 @@
-export type PageItem = ImagePageItem | PdfPageItem
+export type PageItem = ImagePageItem | PdfPageItem | PdfWholeItem
 
 export type SourceItem = {
   id: string
@@ -31,6 +31,19 @@ export type PdfPageItem = {
   previewFilePath: string | null
 }
 
+export type PdfWholeItem = {
+  id: string
+  kind: "pdf-whole"
+  title: string
+  sourceName: string
+  selected: boolean
+  selectedOrder?: number
+  pdfPath: string
+  pageCount: number
+  previewImage: UIImage | null
+  previewFilePath: string | null
+}
+
 export type ImportResult = {
   sources: SourceItem[]
   notices: string[]