reader-shell 1.0.2 → 1.0.4

package/.claude/settings.local.json

@@ -0,0 +1,14 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(git add:*)",
+      "Bash(git commit:*)",
+      "Bash(git push:*)",
+      "Bash(node:*)",
+      "Bash(timeout:*)",
+      "Read(//Users/xuejichang/Documents/book/**)"
+    ],
+    "deny": [],
+    "ask": []
+  }
+}

package/README.md

@@ -1,6 +1,8 @@
 # 终端电子书阅读器
 
-一个基于 Node.js 的终端电子书阅读器，支持导入 epub 文件、记住阅读进度、快捷键翻页和字体颜色切换。
+一个基于 Node.js 的终端电子书阅读器，支持导入 epub 文件、记住阅读进度、快捷键翻页。
+
+![reader-shell-demo.gif?raw=true](https://cdn.jsdelivr.net/gh/jichangee/gallery@master/imgur/reader-shell-demo.gif?raw=true)
 
 ## 功能特性
 - 支持导入 epub 电子书
@@ -48,4 +50,4 @@ node src/ui.js ./books/三体.epub
 - [chalk](https://www.npmjs.com/package/chalk) 终端字体颜色
 
 ## 许可协议
-MIT 
+MIT 

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.2",
+  "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/epubReader.js

@@ -19,7 +19,7 @@ function parseEpub(epubPath) {
                         chapters[idx] = '';
                     } else {
                         let text = data.toString('utf-8');
-                        chapters[idx] = text.replace(/<[^>]+>/g, '').replace(/\r\n/g, '') + '\n\n 本章完';
+                        chapters[idx] = text.replace(/<[^>]+>/g, '').replace(/\r\n/g, '');
                     }
                     loaded++;
                     if (loaded === chapterIds.length) {

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
   });
@@ -72,9 +83,18 @@ async function startReader(epubPath) {
   screen.append(box);
   screen.append(status);
 
-  function render() {
+  // 用于跟踪当前显示的章节，避免重复设置相同内容
+  let lastDisplayedChapter = -1;
+
+  function render(forceContentRefresh = false) {
     const colorFn = colorFns[fontColor] || chalk.white;
-    box.setContent(colorFn(chapters[currentChapter] || ""));
+
+    // 只有当章节改变或强制刷新时才重新设置内容
+    if (forceContentRefresh || lastDisplayedChapter !== currentChapter) {
+      box.setContent(colorFn(chapters[currentChapter] || ""));
+      lastDisplayedChapter = currentChapter;
+    }
+
     status.setContent(
       `progress: ${currentScroll}/${box.getScrollHeight()}  |  chapter: ${currentChapter + 1}/${
         chapters.length
@@ -86,14 +106,14 @@ async function startReader(epubPath) {
   }
 
   function saveProgress() {
-    setBookProgress(epubPath, { chapter: currentChapter, scroll: currentScroll });
+    setBookProgress(filePath, { chapter: currentChapter, scroll: currentScroll });
   }
 
   function changeChapter(chapter) {
     currentChapter += chapter;
     currentScroll = 0;
     saveProgress();
-    render();
+    render(true); // 切换章节时强制刷新内容
   }
 
   // 快捷键
@@ -113,7 +133,7 @@ async function startReader(epubPath) {
       changeChapter(-1)
       return
     }
-    render();
+    render(true);
   });
   screen.key(["down", "d"], () => {
     currentScroll++;
@@ -121,7 +141,7 @@ async function startReader(epubPath) {
       changeChapter(1)
       return
     }
-    render();
+    render(true);
   });
   screen.key(["a"], () => {
     // 老板键
@@ -134,7 +154,7 @@ async function startReader(epubPath) {
       box.height = 0
       status.height = 0
     }
-    render()
+    render(false) // 老板键切换时不强制刷新内容
   });
   screen.key(["q", "C-c"], () => {
     saveProgress();
@@ -145,15 +165,15 @@ async function startReader(epubPath) {
     saveProgress();
   });
 
-  render();
+  render(true); // 初始渲染时强制设置内容
 }
 
 // 允许命令行直接运行
-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 };