reader-shell 1.0.3 → 1.0.4

package/.claude/settings.local.json

@@ -3,7 +3,10 @@
     "allow": [
       "Bash(git add:*)",
       "Bash(git commit:*)",
-      "Bash(git push:*)"
+      "Bash(git push:*)",
+      "Bash(node:*)",
+      "Bash(timeout:*)",
+      "Read(//Users/xuejichang/Documents/book/**)"
     ],
     "deny": [],
     "ask": []

package/cli.js

@@ -1,9 +1,9 @@
 #!/usr/bin/env node
 import { startReader } from './src/ui.js';
 import path from 'path';
-const epubPath = process.argv[2];
-if (!epubPath) {
-  console.error('用法: reader-shell <epub文件路径>');
+const filePath = process.argv[2];
+if (!filePath) {
+  console.error('用法: reader-shell <文件路径>');
   process.exit(1);
 }
-startReader(path.resolve(epubPath));
+startReader(path.resolve(filePath));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "reader-shell",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "main": "cli.js",
   "type": "module",
   "scripts": {
@@ -13,7 +13,8 @@
   "dependencies": {
     "blessed": "^0.1.81",
     "chalk": "^5.4.1",
-    "epub": "^1.3.0"
+    "epub": "^1.3.0",
+    "iconv-lite": "^0.7.0"
   },
   "bin": {
     "reader-shell": "./cli.js"

package/src/txtReader.js

@@ -0,0 +1,164 @@
+import fs from 'fs';
+import path from 'path';
+import iconv from 'iconv-lite';
+
+/**
+ * 检测文件编码
+ * @param {Buffer} buffer 文件内容
+ * @returns {string} 编码名称
+ */
+function detectEncoding(buffer) {
+    // 检查BOM标记
+    if (buffer.length >= 3 && buffer[0] === 0xEF && buffer[1] === 0xBB && buffer[2] === 0xBF) {
+        return 'utf8';
+    }
+
+    if (buffer.length >= 2 && buffer[0] === 0xFF && buffer[1] === 0xFE) {
+        return 'utf16le';
+    }
+
+    if (buffer.length >= 2 && buffer[0] === 0xFE && buffer[1] === 0xFF) {
+        return 'utf16be';
+    }
+
+    // 简单的编码检测：尝试UTF-8解码
+    try {
+        const utf8Content = buffer.toString('utf8');
+        // 检查是否包含UTF-8无效字符
+        if (!utf8Content.includes('�')) {
+            return 'utf8';
+        }
+    } catch (e) {
+        // UTF-8解码失败，尝试其他编码
+    }
+
+    // 尝试GBK/GB2312编码（中文Windows常用编码）
+    try {
+        const gbkContent = iconv.decode(buffer, 'gbk');
+        // 检查解码是否成功（没有太多乱码字符）
+        if (!gbkContent.includes('�') || gbkContent.indexOf('�') / gbkContent.length < 0.1) {
+            return 'gbk';
+        }
+    } catch (e) {
+        // GBK解码失败
+    }
+
+    // 默认使用UTF-8
+    return 'utf8';
+}
+
+/**
+ * 以正确的编码读取文件
+ * @param {string} filePath 文件路径
+ * @returns {Promise<string>} 文件内容
+ */
+function readFileWithCorrectEncoding(filePath) {
+    return new Promise((resolve, reject) => {
+        fs.readFile(filePath, (err, buffer) => {
+            if (err) {
+                reject(err);
+                return;
+            }
+
+            const encoding = detectEncoding(buffer);
+            let content;
+
+            try {
+                if (encoding === 'utf8') {
+                    content = buffer.toString('utf8');
+                } else {
+                    content = iconv.decode(buffer, encoding);
+                }
+                resolve(content);
+            } catch (e) {
+                // 如果指定编码解码失败，尝试UTF-8
+                try {
+                    content = buffer.toString('utf8');
+                    resolve(content);
+                } catch (utf8Err) {
+                    reject(new Error(`无法解码文件，尝试的编码: ${encoding}`));
+                }
+            }
+        });
+    });
+}
+
+/**
+ * 解析txt文件，返回章节文本数组
+ * @param {string} txtPath txt文件路径
+ * @returns {Promise<string[]>} 章节文本数组
+ */
+function parseTxt(txtPath) {
+    return new Promise((resolve, reject) => {
+        readFileWithCorrectEncoding(txtPath).then(data => {
+            let content = data;
+            let chapters = [];
+
+            // 先尝试按明确的章节标题分割
+            const lines = content.split('\n');
+            let currentChapter = '';
+            let chapterHeaders = [];
+
+            // 寻找所有章节标题的行号
+            lines.forEach((line, index) => {
+                if (/^第[一二三四五六七八九十百千万0-9]+[章节]/.test(line.trim())) {
+                    chapterHeaders.push(index);
+                }
+            });
+
+            if (chapterHeaders.length > 1) {
+                // 按找到的章节标题分割
+                for (let i = 0; i < chapterHeaders.length; i++) {
+                    const startLine = chapterHeaders[i];
+                    const endLine = i < chapterHeaders.length - 1 ? chapterHeaders[i + 1] : lines.length;
+
+                    const chapterLines = lines.slice(startLine, endLine);
+                    const chapterContent = chapterLines.join('\n').trim();
+
+                    if (chapterContent.length > 0) {
+                        chapters.push(chapterContent);
+                    }
+                }
+            }
+
+            // 如果没有找到章节分割，按段落分割
+            if (chapters.length === 0) {
+                // 按多个空行分割
+                const sections = content.split(/\n\s*\n\s*\n/);
+                chapters = sections.filter(section => section.trim().length > 0);
+
+                // 如果分割后章节太少，按单个空行分割，然后每10个段落合并
+                if (chapters.length <= 1) {
+                    const paragraphs = content.split(/\n\s*\n/);
+                    chapters = [];
+                    const chapterSize = 10;
+                    for (let i = 0; i < paragraphs.length; i += chapterSize) {
+                        const chapter = paragraphs.slice(i, i + chapterSize).join('\n\n');
+                        if (chapter.trim().length > 0) {
+                            chapters.push(chapter);
+                        }
+                    }
+                }
+            }
+
+            // 如果还是没有合适的章节，整个文件作为一章
+            if (chapters.length === 0) {
+                chapters = [content];
+            }
+
+            // 清理每章内容
+            chapters = chapters.map(chapter => {
+                return chapter
+                    .replace(/\r\n/g, '\n')  // 统一换行符
+                    .replace(/\n{3,}/g, '\n\n')  // 去除多余的空行
+                    .trim();
+            }).filter(chapter => chapter.length > 0);
+
+            resolve(chapters);
+        }).catch(err => {
+            reject(err);
+        });
+    });
+}
+
+export { parseTxt, detectEncoding, readFileWithCorrectEncoding };

package/src/ui.js

@@ -1,6 +1,7 @@
 import blessed from "blessed";
 import chalk from "chalk";
 import { parseEpub } from "./epubReader.js";
+import { parseTxt } from "./txtReader.js";
 import path from "path";
 import { getBookProgress, setBookProgress } from "./progressStore.js";
 import { config } from "./config.js";
@@ -16,9 +17,19 @@ const colorFns = {
   yellow: chalk.yellow,
 };
 
-async function startReader(epubPath) {
-  const chapters = await parseEpub(epubPath);
-  const progress = getBookProgress(epubPath);
+async function startReader(filePath) {
+  const ext = path.extname(filePath).toLowerCase();
+  let chapters;
+
+  if (ext === '.epub') {
+    chapters = await parseEpub(filePath);
+  } else if (ext === '.txt') {
+    chapters = await parseTxt(filePath);
+  } else {
+    throw new Error(`不支持的文件格式: ${ext}`);
+  }
+
+  const progress = getBookProgress(filePath);
   let currentChapter = (progress && progress.chapter) || 0;
   let currentScroll = (progress && progress.scroll) || 0;
 
@@ -34,7 +45,7 @@ async function startReader(epubPath) {
     top: 0,
     left: 0,
     width: "100%",
-    height: "80%",
+    height: "70%",
     border: "none",
     content: config.bossContent
   });
@@ -95,7 +106,7 @@ async function startReader(epubPath) {
   }
 
   function saveProgress() {
-    setBookProgress(epubPath, { chapter: currentChapter, scroll: currentScroll });
+    setBookProgress(filePath, { chapter: currentChapter, scroll: currentScroll });
   }
 
   function changeChapter(chapter) {
@@ -122,7 +133,7 @@ async function startReader(epubPath) {
       changeChapter(-1)
       return
     }
-    render();
+    render(true);
   });
   screen.key(["down", "d"], () => {
     currentScroll++;
@@ -130,7 +141,7 @@ async function startReader(epubPath) {
       changeChapter(1)
       return
     }
-    render();
+    render(true);
   });
   screen.key(["a"], () => {
     // 老板键
@@ -158,11 +169,11 @@ async function startReader(epubPath) {
 }
 
 // 允许命令行直接运行
-const epubPath = process.argv[2];
-if (!epubPath) {
-  console.error("用法: node src/ui.js <epub文件路径>");
+const filePath = process.argv[2];
+if (!filePath) {
+  console.error("用法: node src/ui.js <文件路径>");
   process.exit(1);
 }
-startReader(path.resolve(epubPath));
+startReader(path.resolve(filePath));
 
 export { startReader };