Something went wrong. Try again.
A fork of https://github.com/crosspoint-reader/crosspoint-reader
Something went wrong. Try again.
5.0 kB · 174 lines
C++
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175#include "Txt.h"
#include <FsHelpers.h>#include <JpegToBmpConverter.h>#include <Logging.h>
Txt::Txt(std::string path, std::string cacheBasePath) : filepath(std::move(path)), cacheBasePath(std::move(cacheBasePath)) { // Generate cache path from file path hash const size_t hash = std::hash<std::string>{}(filepath); cachePath = this->cacheBasePath + "/txt_" + std::to_string(hash);}
bool Txt::load() { if (loaded) { return true; }
if (!Storage.exists(filepath.c_str())) { LOG_ERR("TXT", "File does not exist: %s", filepath.c_str()); return false; }
FsFile file; if (!Storage.openFileForRead("TXT", filepath, file)) { LOG_ERR("TXT", "Failed to open file: %s", filepath.c_str()); return false; }
fileSize = file.size(); file.close();
loaded = true; LOG_DBG("TXT", "Loaded TXT file: %s (%zu bytes)", filepath.c_str(), fileSize); return true;}
std::string Txt::getTitle() const { // Extract filename without path and extension size_t lastSlash = filepath.find_last_of('/'); std::string filename = (lastSlash != std::string::npos) ? filepath.substr(lastSlash + 1) : filepath;
// Remove .txt extension if (FsHelpers::hasTxtExtension(filename)) { filename = filename.substr(0, filename.length() - 4); }
return filename;}
void Txt::setupCacheDir() const { if (!Storage.exists(cacheBasePath.c_str())) { Storage.mkdir(cacheBasePath.c_str()); } if (!Storage.exists(cachePath.c_str())) { Storage.mkdir(cachePath.c_str()); }}
std::string Txt::findCoverImage() const { // Get the folder containing the txt file size_t lastSlash = filepath.find_last_of('/'); std::string folder = (lastSlash != std::string::npos) ? filepath.substr(0, lastSlash) : ""; if (folder.empty()) { folder = "/"; }
// Get the base filename without extension (e.g., "mybook" from "/books/mybook.txt") std::string baseName = getTitle();
// Image extensions to try const char* extensions[] = {".bmp", ".jpg", ".jpeg", ".png", ".BMP", ".JPG", ".JPEG", ".PNG"};
// First priority: look for image with same name as txt file (e.g., mybook.jpg) for (const auto& ext : extensions) { std::string coverPath = folder + "/" + baseName + ext; if (Storage.exists(coverPath.c_str())) { LOG_DBG("TXT", "Found matching cover image: %s", coverPath.c_str()); return coverPath; } }
// Fallback: look for cover image files const char* coverNames[] = {"cover", "Cover", "COVER"}; for (const auto& name : coverNames) { for (const auto& ext : extensions) { std::string coverPath = folder + "/" + std::string(name) + ext; if (Storage.exists(coverPath.c_str())) { LOG_DBG("TXT", "Found fallback cover image: %s", coverPath.c_str()); return coverPath; } } }
return "";}
std::string Txt::getCoverBmpPath() const { return cachePath + "/cover.bmp"; }
bool Txt::generateCoverBmp() const { // Already generated, return true if (Storage.exists(getCoverBmpPath().c_str())) { return true; }
std::string coverImagePath = findCoverImage(); if (coverImagePath.empty()) { LOG_DBG("TXT", "No cover image found for TXT file"); return false; }
// Setup cache directory setupCacheDir();
if (FsHelpers::hasBmpExtension(coverImagePath)) { // Copy BMP file to cache LOG_DBG("TXT", "Copying BMP cover image to cache"); FsFile src, dst; if (!Storage.openFileForRead("TXT", coverImagePath, src)) { return false; } if (!Storage.openFileForWrite("TXT", getCoverBmpPath(), dst)) { return false; } uint8_t buffer[1024]; while (src.available()) { size_t bytesRead = src.read(buffer, sizeof(buffer)); dst.write(buffer, bytesRead); } LOG_DBG("TXT", "Copied BMP cover to cache"); return true; } else if (FsHelpers::hasJpgExtension(coverImagePath)) { // Convert JPG/JPEG to BMP (same approach as Epub) LOG_DBG("TXT", "Generating BMP from JPG cover image"); FsFile coverJpg, coverBmp; if (!Storage.openFileForRead("TXT", coverImagePath, coverJpg)) { return false; } if (!Storage.openFileForWrite("TXT", getCoverBmpPath(), coverBmp)) { return false; } const bool success = JpegToBmpConverter::jpegFileToBmpStream(coverJpg, coverBmp);
if (!success) { LOG_ERR("TXT", "Failed to generate BMP from JPG cover image"); Storage.remove(getCoverBmpPath().c_str()); } else { LOG_DBG("TXT", "Generated BMP from JPG cover image"); } return success; }
// PNG files are not supported (would need a PNG decoder) LOG_ERR("TXT", "Cover image format not supported (only BMP/JPG/JPEG)"); return false;}
bool Txt::readContent(uint8_t* buffer, size_t offset, size_t length) const { if (!loaded) { return false; }
FsFile file; if (!Storage.openFileForRead("TXT", filepath, file)) { return false; }
if (!file.seek(offset)) { return false; }
size_t bytesRead = file.read(buffer, length); return bytesRead > 0;}