diff --git a/lib/KOReaderSync/KOReaderCredentialStore.cpp b/lib/KOReaderSync/KOReaderCredentialStore.cpp new file mode 100644 index 00000000..b9cbc564 --- /dev/null +++ b/lib/KOReaderSync/KOReaderCredentialStore.cpp @@ -0,0 +1,111 @@ +#include "KOReaderCredentialStore.h" + +#include +#include +#include +#include + +// Initialize the static instance +KOReaderCredentialStore KOReaderCredentialStore::instance; + +namespace { +// File format version +constexpr uint8_t KOREADER_FILE_VERSION = 1; + +// KOReader credentials file path +constexpr char KOREADER_FILE[] = "/.crosspoint/koreader.bin"; + +// Obfuscation key - "KOReader" in ASCII +// This is NOT cryptographic security, just prevents casual file reading +constexpr uint8_t OBFUSCATION_KEY[] = {0x4B, 0x4F, 0x52, 0x65, 0x61, 0x64, 0x65, 0x72}; +constexpr size_t KEY_LENGTH = sizeof(OBFUSCATION_KEY); +} // namespace + +void KOReaderCredentialStore::obfuscate(std::string& data) const { + for (size_t i = 0; i < data.size(); i++) { + data[i] ^= OBFUSCATION_KEY[i % KEY_LENGTH]; + } +} + +bool KOReaderCredentialStore::saveToFile() const { + // Make sure the directory exists + SdMan.mkdir("/.crosspoint"); + + FsFile file; + if (!SdMan.openFileForWrite("KRS", KOREADER_FILE, file)) { + return false; + } + + // Write header + serialization::writePod(file, KOREADER_FILE_VERSION); + + // Write username (plaintext - not particularly sensitive) + serialization::writeString(file, username); + Serial.printf("[%lu] [KRS] Saving username: %s\n", millis(), username.c_str()); + + // Write password (obfuscated) + std::string obfuscatedPwd = password; + obfuscate(obfuscatedPwd); + serialization::writeString(file, obfuscatedPwd); + + file.close(); + Serial.printf("[%lu] [KRS] Saved KOReader credentials to file\n", millis()); + return true; +} + +bool KOReaderCredentialStore::loadFromFile() { + FsFile file; + if (!SdMan.openFileForRead("KRS", KOREADER_FILE, file)) { + Serial.printf("[%lu] [KRS] No credentials file found\n", millis()); + return false; + } + + // Read and verify version + uint8_t version; + serialization::readPod(file, version); + if (version != KOREADER_FILE_VERSION) { + Serial.printf("[%lu] [KRS] Unknown file version: %u\n", millis(), version); + file.close(); + return false; + } + + // Read username + serialization::readString(file, username); + + // Read and deobfuscate password + serialization::readString(file, password); + obfuscate(password); // XOR is symmetric, so same function deobfuscates + + file.close(); + Serial.printf("[%lu] [KRS] Loaded KOReader credentials for user: %s\n", millis(), username.c_str()); + return true; +} + +void KOReaderCredentialStore::setCredentials(const std::string& user, const std::string& pass) { + username = user; + password = pass; + Serial.printf("[%lu] [KRS] Set credentials for user: %s\n", millis(), user.c_str()); +} + +std::string KOReaderCredentialStore::getMd5Password() const { + if (password.empty()) { + return ""; + } + + // Calculate MD5 hash of password using ESP32's MD5Builder + MD5Builder md5; + md5.begin(); + md5.add(password.c_str()); + md5.calculate(); + + return md5.toString().c_str(); +} + +bool KOReaderCredentialStore::hasCredentials() const { return !username.empty() && !password.empty(); } + +void KOReaderCredentialStore::clearCredentials() { + username.clear(); + password.clear(); + saveToFile(); + Serial.printf("[%lu] [KRS] Cleared KOReader credentials\n", millis()); +} diff --git a/lib/KOReaderSync/KOReaderCredentialStore.h b/lib/KOReaderSync/KOReaderCredentialStore.h new file mode 100644 index 00000000..0d3332fa --- /dev/null +++ b/lib/KOReaderSync/KOReaderCredentialStore.h @@ -0,0 +1,49 @@ +#pragma once +#include + +/** + * Singleton class for storing KOReader sync credentials on the SD card. + * Credentials are stored in /sd/.crosspoint/koreader.bin with basic + * XOR obfuscation to prevent casual reading (not cryptographically secure). + */ +class KOReaderCredentialStore { + private: + static KOReaderCredentialStore instance; + std::string username; + std::string password; + + // Private constructor for singleton + KOReaderCredentialStore() = default; + + // XOR obfuscation (symmetric - same for encode/decode) + void obfuscate(std::string& data) const; + + public: + // Delete copy constructor and assignment + KOReaderCredentialStore(const KOReaderCredentialStore&) = delete; + KOReaderCredentialStore& operator=(const KOReaderCredentialStore&) = delete; + + // Get singleton instance + static KOReaderCredentialStore& getInstance() { return instance; } + + // Save/load from SD card + bool saveToFile() const; + bool loadFromFile(); + + // Credential management + void setCredentials(const std::string& user, const std::string& pass); + const std::string& getUsername() const { return username; } + const std::string& getPassword() const { return password; } + + // Get MD5 hash of password for API authentication + std::string getMd5Password() const; + + // Check if credentials are set + bool hasCredentials() const; + + // Clear credentials + void clearCredentials(); +}; + +// Helper macro to access credential store +#define KOREADER_STORE KOReaderCredentialStore::getInstance() diff --git a/lib/KOReaderSync/KOReaderDocumentId.cpp b/lib/KOReaderSync/KOReaderDocumentId.cpp new file mode 100644 index 00000000..f75d7c8f --- /dev/null +++ b/lib/KOReaderSync/KOReaderDocumentId.cpp @@ -0,0 +1,69 @@ +#include "KOReaderDocumentId.h" + +#include +#include +#include + +size_t KOReaderDocumentId::getOffset(int i) { + // Offset = 1024 << (2*i) + // For i = -1: 1024 >> 2 = 256 + // For i >= 0: 1024 << (2*i) + if (i < 0) { + return CHUNK_SIZE >> (-2 * i); + } + return CHUNK_SIZE << (2 * i); +} + +std::string KOReaderDocumentId::calculate(const std::string& filePath) { + FsFile file; + if (!SdMan.openFileForRead("KODoc", filePath, file)) { + Serial.printf("[%lu] [KODoc] Failed to open file: %s\n", millis(), filePath.c_str()); + return ""; + } + + const size_t fileSize = file.fileSize(); + Serial.printf("[%lu] [KODoc] Calculating hash for file: %s (size: %zu)\n", millis(), filePath.c_str(), fileSize); + + // Initialize MD5 builder + MD5Builder md5; + md5.begin(); + + // Buffer for reading chunks + uint8_t buffer[CHUNK_SIZE]; + size_t totalBytesRead = 0; + + // Read from each offset (i = -1 to 10) + for (int i = -1; i < OFFSET_COUNT - 1; i++) { + const size_t offset = getOffset(i); + + // Skip if offset is beyond file size + if (offset >= fileSize) { + continue; + } + + // Seek to offset + if (!file.seekSet(offset)) { + Serial.printf("[%lu] [KODoc] Failed to seek to offset %zu\n", millis(), offset); + continue; + } + + // Read up to CHUNK_SIZE bytes + const size_t bytesToRead = std::min(CHUNK_SIZE, fileSize - offset); + const size_t bytesRead = file.read(buffer, bytesToRead); + + if (bytesRead > 0) { + md5.add(buffer, bytesRead); + totalBytesRead += bytesRead; + } + } + + file.close(); + + // Calculate final hash + md5.calculate(); + std::string result = md5.toString().c_str(); + + Serial.printf("[%lu] [KODoc] Hash calculated: %s (from %zu bytes)\n", millis(), result.c_str(), totalBytesRead); + + return result; +} diff --git a/lib/KOReaderSync/KOReaderDocumentId.h b/lib/KOReaderSync/KOReaderDocumentId.h new file mode 100644 index 00000000..434b9b63 --- /dev/null +++ b/lib/KOReaderSync/KOReaderDocumentId.h @@ -0,0 +1,36 @@ +#pragma once +#include + +/** + * Calculate KOReader document ID (partial MD5 hash). + * + * KOReader identifies documents using a partial MD5 hash of the file content. + * The algorithm reads 1024 bytes at specific offsets and computes the MD5 hash + * of the concatenated data. + * + * Offsets are calculated as: 1024 << (2*i) for i = -1 to 10 + * Producing: 256, 1024, 4096, 16384, 65536, 262144, 1048576, 4194304, + * 16777216, 67108864, 268435456, 1073741824 bytes + * + * If an offset is beyond the file size, it is skipped. + */ +class KOReaderDocumentId { + public: + /** + * Calculate the KOReader document hash for a file. + * + * @param filePath Path to the file (typically an EPUB) + * @return 32-character lowercase hex string, or empty string on failure + */ + static std::string calculate(const std::string& filePath); + + private: + // Size of each chunk to read at each offset + static constexpr size_t CHUNK_SIZE = 1024; + + // Number of offsets to try (i = -1 to 10, so 12 offsets) + static constexpr int OFFSET_COUNT = 12; + + // Calculate offset for index i: 1024 << (2*i) + static size_t getOffset(int i); +}; diff --git a/lib/KOReaderSync/KOReaderSyncClient.cpp b/lib/KOReaderSync/KOReaderSyncClient.cpp new file mode 100644 index 00000000..028957f8 --- /dev/null +++ b/lib/KOReaderSync/KOReaderSyncClient.cpp @@ -0,0 +1,177 @@ +#include "KOReaderSyncClient.h" + +#include +#include +#include +#include +#include + +#include "KOReaderCredentialStore.h" + +namespace { +// Base URL for KOReader sync server +constexpr char BASE_URL[] = "https://sync.koreader.rocks:443"; + +// Device identifier for CrossPoint reader +constexpr char DEVICE_NAME[] = "CrossPoint"; +constexpr char DEVICE_ID[] = "crosspoint-reader"; + +void addAuthHeaders(HTTPClient& http) { + http.addHeader("Accept", "application/vnd.koreader.v1+json"); + http.addHeader("x-auth-user", KOREADER_STORE.getUsername().c_str()); + http.addHeader("x-auth-key", KOREADER_STORE.getMd5Password().c_str()); +} +} // namespace + +KOReaderSyncClient::Error KOReaderSyncClient::authenticate() { + if (!KOREADER_STORE.hasCredentials()) { + Serial.printf("[%lu] [KOSync] No credentials configured\n", millis()); + return NO_CREDENTIALS; + } + + const std::unique_ptr client(new WiFiClientSecure); + client->setInsecure(); + HTTPClient http; + + std::string url = std::string(BASE_URL) + "/users/auth"; + Serial.printf("[%lu] [KOSync] Authenticating: %s\n", millis(), url.c_str()); + + http.begin(*client, url.c_str()); + addAuthHeaders(http); + + const int httpCode = http.GET(); + http.end(); + + Serial.printf("[%lu] [KOSync] Auth response: %d\n", millis(), httpCode); + + if (httpCode == 200) { + return OK; + } else if (httpCode == 401) { + return AUTH_FAILED; + } else if (httpCode < 0) { + return NETWORK_ERROR; + } + return SERVER_ERROR; +} + +KOReaderSyncClient::Error KOReaderSyncClient::getProgress(const std::string& documentHash, + KOReaderProgress& outProgress) { + if (!KOREADER_STORE.hasCredentials()) { + Serial.printf("[%lu] [KOSync] No credentials configured\n", millis()); + return NO_CREDENTIALS; + } + + const std::unique_ptr client(new WiFiClientSecure); + client->setInsecure(); + HTTPClient http; + + std::string url = std::string(BASE_URL) + "/syncs/progress/" + documentHash; + Serial.printf("[%lu] [KOSync] Getting progress: %s\n", millis(), url.c_str()); + + http.begin(*client, url.c_str()); + addAuthHeaders(http); + + const int httpCode = http.GET(); + + if (httpCode == 200) { + // Parse JSON response + JsonDocument doc; + const DeserializationError error = deserializeJson(doc, *client); + http.end(); + + if (error) { + Serial.printf("[%lu] [KOSync] JSON parse failed: %s\n", millis(), error.c_str()); + return JSON_ERROR; + } + + outProgress.document = documentHash; + outProgress.progress = doc["progress"].as(); + outProgress.percentage = doc["percentage"].as(); + outProgress.device = doc["device"].as(); + outProgress.deviceId = doc["device_id"].as(); + outProgress.timestamp = doc["timestamp"].as(); + + Serial.printf("[%lu] [KOSync] Got progress: %.2f%% at %s\n", millis(), outProgress.percentage * 100, + outProgress.progress.c_str()); + return OK; + } + + http.end(); + + Serial.printf("[%lu] [KOSync] Get progress response: %d\n", millis(), httpCode); + + if (httpCode == 401) { + return AUTH_FAILED; + } else if (httpCode == 404) { + return NOT_FOUND; + } else if (httpCode < 0) { + return NETWORK_ERROR; + } + return SERVER_ERROR; +} + +KOReaderSyncClient::Error KOReaderSyncClient::updateProgress(const KOReaderProgress& progress) { + if (!KOREADER_STORE.hasCredentials()) { + Serial.printf("[%lu] [KOSync] No credentials configured\n", millis()); + return NO_CREDENTIALS; + } + + const std::unique_ptr client(new WiFiClientSecure); + client->setInsecure(); + HTTPClient http; + + std::string url = std::string(BASE_URL) + "/syncs/progress"; + Serial.printf("[%lu] [KOSync] Updating progress: %s\n", millis(), url.c_str()); + + http.begin(*client, url.c_str()); + addAuthHeaders(http); + http.addHeader("Content-Type", "application/json"); + + // Build JSON body (timestamp not required per API spec) + JsonDocument doc; + doc["document"] = progress.document; + doc["progress"] = progress.progress; + doc["percentage"] = progress.percentage; + doc["device"] = DEVICE_NAME; + doc["device_id"] = DEVICE_ID; + + std::string body; + serializeJson(doc, body); + + Serial.printf("[%lu] [KOSync] Request body: %s\n", millis(), body.c_str()); + + const int httpCode = http.PUT(body.c_str()); + http.end(); + + Serial.printf("[%lu] [KOSync] Update progress response: %d\n", millis(), httpCode); + + if (httpCode == 200 || httpCode == 202) { + return OK; + } else if (httpCode == 401) { + return AUTH_FAILED; + } else if (httpCode < 0) { + return NETWORK_ERROR; + } + return SERVER_ERROR; +} + +const char* KOReaderSyncClient::errorString(Error error) { + switch (error) { + case OK: + return "Success"; + case NO_CREDENTIALS: + return "No credentials configured"; + case NETWORK_ERROR: + return "Network error"; + case AUTH_FAILED: + return "Authentication failed"; + case SERVER_ERROR: + return "Server error (try again later)"; + case JSON_ERROR: + return "JSON parse error"; + case NOT_FOUND: + return "No progress found"; + default: + return "Unknown error"; + } +} diff --git a/lib/KOReaderSync/KOReaderSyncClient.h b/lib/KOReaderSync/KOReaderSyncClient.h new file mode 100644 index 00000000..6a75843c --- /dev/null +++ b/lib/KOReaderSync/KOReaderSyncClient.h @@ -0,0 +1,67 @@ +#pragma once +#include + +/** + * Progress data from KOReader sync server. + */ +struct KOReaderProgress { + std::string document; // Document hash + std::string progress; // XPath-like progress string + float percentage; // Progress percentage (0.0 to 1.0) + std::string device; // Device name + std::string deviceId; // Device ID + int64_t timestamp; // Unix timestamp of last update +}; + +/** + * HTTP client for KOReader sync API. + * + * Base URL: https://sync.koreader.rocks:443/ + * + * API Endpoints: + * GET /users/auth - Authenticate (validate credentials) + * GET /syncs/progress/:document - Get progress for a document + * PUT /syncs/progress - Update progress for a document + * + * Authentication: + * x-auth-user: username + * x-auth-key: MD5 hash of password + */ +class KOReaderSyncClient { + public: + enum Error { + OK = 0, + NO_CREDENTIALS, + NETWORK_ERROR, + AUTH_FAILED, + SERVER_ERROR, + JSON_ERROR, + NOT_FOUND + }; + + /** + * Authenticate with the sync server (validate credentials). + * @return OK on success, error code on failure + */ + static Error authenticate(); + + /** + * Get reading progress for a document. + * @param documentHash The document hash (from KOReaderDocumentId) + * @param outProgress Output: the progress data + * @return OK on success, NOT_FOUND if no progress exists, error code on failure + */ + static Error getProgress(const std::string& documentHash, KOReaderProgress& outProgress); + + /** + * Update reading progress for a document. + * @param progress The progress data to upload + * @return OK on success, error code on failure + */ + static Error updateProgress(const KOReaderProgress& progress); + + /** + * Get human-readable error message. + */ + static const char* errorString(Error error); +}; diff --git a/lib/KOReaderSync/ProgressMapper.cpp b/lib/KOReaderSync/ProgressMapper.cpp new file mode 100644 index 00000000..f4c4f8d9 --- /dev/null +++ b/lib/KOReaderSync/ProgressMapper.cpp @@ -0,0 +1,125 @@ +#include "ProgressMapper.h" + +#include + +#include +#include + +KOReaderPosition ProgressMapper::toKOReader(const std::shared_ptr& epub, const CrossPointPosition& pos) { + KOReaderPosition result; + + // Calculate page progress within current spine item + float intraSpineProgress = 0.0f; + if (pos.totalPages > 0) { + intraSpineProgress = static_cast(pos.pageNumber) / static_cast(pos.totalPages); + } + + // Calculate overall book progress (0-100 from Epub, convert to 0.0-1.0) + const uint8_t progressPercent = epub->calculateProgress(pos.spineIndex, intraSpineProgress); + result.percentage = static_cast(progressPercent) / 100.0f; + + // Generate XPath with estimated paragraph position based on page + result.xpath = generateXPath(pos.spineIndex, pos.pageNumber, pos.totalPages); + + // Get chapter info for logging + const int tocIndex = epub->getTocIndexForSpineIndex(pos.spineIndex); + const std::string chapterName = (tocIndex >= 0) ? epub->getTocItem(tocIndex).title : "unknown"; + + Serial.printf("[%lu] [ProgressMapper] CrossPoint -> KOReader: chapter='%s', page=%d/%d -> %.2f%% at %s\n", millis(), + chapterName.c_str(), pos.pageNumber, pos.totalPages, result.percentage * 100, result.xpath.c_str()); + + return result; +} + +CrossPointPosition ProgressMapper::toCrossPoint(const std::shared_ptr& epub, const KOReaderPosition& koPos, + int totalPagesInSpine) { + CrossPointPosition result; + result.spineIndex = 0; + result.pageNumber = 0; + result.totalPages = totalPagesInSpine; + + const size_t bookSize = epub->getBookSize(); + if (bookSize == 0) { + Serial.printf("[%lu] [ProgressMapper] Book size is 0\n", millis()); + return result; + } + + // First, try to get spine index from XPath (DocFragment) + int xpathSpineIndex = parseDocFragmentIndex(koPos.xpath); + if (xpathSpineIndex >= 0 && xpathSpineIndex < epub->getSpineItemsCount()) { + result.spineIndex = xpathSpineIndex; + Serial.printf("[%lu] [ProgressMapper] Got spine index from XPath: %d\n", millis(), result.spineIndex); + } else { + // Fall back to percentage-based lookup + const size_t targetBytes = static_cast(bookSize * koPos.percentage); + + // Find the spine item that contains this byte position + for (int i = 0; i < epub->getSpineItemsCount(); i++) { + const size_t cumulativeSize = epub->getCumulativeSpineItemSize(i); + if (cumulativeSize >= targetBytes) { + result.spineIndex = i; + break; + } + } + Serial.printf("[%lu] [ProgressMapper] Got spine index from percentage (%.2f%%): %d\n", millis(), + koPos.percentage * 100, result.spineIndex); + } + + // Estimate page number within the spine item using percentage + if (totalPagesInSpine > 0 && result.spineIndex < epub->getSpineItemsCount()) { + // Calculate what percentage through the spine item we should be + const size_t prevCumSize = (result.spineIndex > 0) ? epub->getCumulativeSpineItemSize(result.spineIndex - 1) : 0; + const size_t currentCumSize = epub->getCumulativeSpineItemSize(result.spineIndex); + const size_t spineSize = currentCumSize - prevCumSize; + + if (spineSize > 0) { + const size_t targetBytes = static_cast(bookSize * koPos.percentage); + const size_t bytesIntoSpine = (targetBytes > prevCumSize) ? (targetBytes - prevCumSize) : 0; + const float intraSpineProgress = static_cast(bytesIntoSpine) / static_cast(spineSize); + + // Clamp to valid range + const float clampedProgress = std::max(0.0f, std::min(1.0f, intraSpineProgress)); + result.pageNumber = static_cast(clampedProgress * totalPagesInSpine); + + // Ensure page number is valid + result.pageNumber = std::max(0, std::min(result.pageNumber, totalPagesInSpine - 1)); + } + } + + Serial.printf("[%lu] [ProgressMapper] KOReader -> CrossPoint: %.2f%% at %s -> spine=%d, page=%d\n", millis(), + koPos.percentage * 100, koPos.xpath.c_str(), result.spineIndex, result.pageNumber); + + return result; +} + +std::string ProgressMapper::generateXPath(int spineIndex, int pageNumber, int totalPages) { + // KOReader uses 1-based DocFragment indices + // Estimate paragraph number based on page position + // Assume ~3 paragraphs per page on average for e-reader screens + constexpr int paragraphsPerPage = 3; + const int estimatedParagraph = (pageNumber * paragraphsPerPage) + 1; // 1-based + + return "/body/DocFragment[" + std::to_string(spineIndex + 1) + "]/body/p[" + std::to_string(estimatedParagraph) + "]"; +} + +int ProgressMapper::parseDocFragmentIndex(const std::string& xpath) { + // Look for DocFragment[N] pattern + const size_t start = xpath.find("DocFragment["); + if (start == std::string::npos) { + return -1; + } + + const size_t numStart = start + 12; // Length of "DocFragment[" + const size_t numEnd = xpath.find(']', numStart); + if (numEnd == std::string::npos) { + return -1; + } + + try { + const int docFragmentIndex = std::stoi(xpath.substr(numStart, numEnd - numStart)); + // KOReader uses 1-based indices, we use 0-based + return docFragmentIndex - 1; + } catch (...) { + return -1; + } +} diff --git a/lib/KOReaderSync/ProgressMapper.h b/lib/KOReaderSync/ProgressMapper.h new file mode 100644 index 00000000..d508ff01 --- /dev/null +++ b/lib/KOReaderSync/ProgressMapper.h @@ -0,0 +1,72 @@ +#pragma once +#include + +#include +#include + +/** + * CrossPoint position representation. + */ +struct CrossPointPosition { + int spineIndex; // Current spine item (chapter) index + int pageNumber; // Current page within the spine item + int totalPages; // Total pages in the current spine item +}; + +/** + * KOReader position representation. + */ +struct KOReaderPosition { + std::string xpath; // XPath-like progress string + float percentage; // Progress percentage (0.0 to 1.0) +}; + +/** + * Maps between CrossPoint and KOReader position formats. + * + * CrossPoint tracks position as (spineIndex, pageNumber). + * KOReader uses XPath-like strings + percentage. + * + * Since CrossPoint discards HTML structure during parsing, we generate + * synthetic XPath strings based on spine index, using percentage as the + * primary sync mechanism. + */ +class ProgressMapper { + public: + /** + * Convert CrossPoint position to KOReader format. + * + * @param epub The EPUB book + * @param pos CrossPoint position + * @return KOReader position + */ + static KOReaderPosition toKOReader(const std::shared_ptr& epub, const CrossPointPosition& pos); + + /** + * Convert KOReader position to CrossPoint format. + * + * Note: The returned pageNumber may be approximate since different + * rendering settings produce different page counts. + * + * @param epub The EPUB book + * @param koPos KOReader position + * @param totalPagesInSpine Total pages in the target spine item (for page estimation) + * @return CrossPoint position + */ + static CrossPointPosition toCrossPoint(const std::shared_ptr& epub, const KOReaderPosition& koPos, + int totalPagesInSpine = 0); + + private: + /** + * Generate XPath for KOReader compatibility. + * Format: /body/DocFragment[spineIndex+1]/body/p[estimatedParagraph] + * Paragraph is estimated based on page position within the chapter. + */ + static std::string generateXPath(int spineIndex, int pageNumber, int totalPages); + + /** + * Parse DocFragment index from XPath string. + * Returns -1 if not found. + */ + static int parseDocFragmentIndex(const std::string& xpath); +}; diff --git a/src/activities/reader/EpubReaderActivity.cpp b/src/activities/reader/EpubReaderActivity.cpp index d3cd5016..fb4cf6c2 100644 --- a/src/activities/reader/EpubReaderActivity.cpp +++ b/src/activities/reader/EpubReaderActivity.cpp @@ -120,9 +120,11 @@ void EpubReaderActivity::loop() { if (mappedInput.wasReleased(MappedInputManager::Button::Confirm)) { // Don't start activity transition while rendering xSemaphoreTake(renderingMutex, portMAX_DELAY); + const int currentPage = section ? section->currentPage : 0; + const int totalPages = section ? section->pageCount : 0; exitActivity(); enterNewActivity(new EpubReaderChapterSelectionActivity( - this->renderer, this->mappedInput, epub, currentSpineIndex, + this->renderer, this->mappedInput, epub, epub->getPath(), currentSpineIndex, currentPage, totalPages, [this] { exitActivity(); updateRequired = true; @@ -135,6 +137,16 @@ void EpubReaderActivity::loop() { } exitActivity(); updateRequired = true; + }, + [this](const int newSpineIndex, const int newPage) { + // Handle sync position + if (currentSpineIndex != newSpineIndex || (section && section->currentPage != newPage)) { + currentSpineIndex = newSpineIndex; + nextPageNumber = newPage; + section.reset(); + } + exitActivity(); + updateRequired = true; })); xSemaphoreGive(renderingMutex); } diff --git a/src/activities/reader/EpubReaderChapterSelectionActivity.cpp b/src/activities/reader/EpubReaderChapterSelectionActivity.cpp index 63f1e5a7..49c3051f 100644 --- a/src/activities/reader/EpubReaderChapterSelectionActivity.cpp +++ b/src/activities/reader/EpubReaderChapterSelectionActivity.cpp @@ -2,14 +2,25 @@ #include +#include "KOReaderCredentialStore.h" +#include "KOReaderSyncActivity.h" #include "MappedInputManager.h" #include "fontIds.h" namespace { // Time threshold for treating a long press as a page-up/page-down constexpr int SKIP_PAGE_MS = 700; + +// Sync option is shown at index 0 if credentials are configured +constexpr int SYNC_ITEM_INDEX = 0; } // namespace +int EpubReaderChapterSelectionActivity::getTotalItems() const { + // Add 1 for sync option if credentials are configured + const int syncOffset = KOREADER_STORE.hasCredentials() ? 1 : 0; + return epub->getTocItemsCount() + syncOffset; +} + int EpubReaderChapterSelectionActivity::getPageItems() const { // Layout constants used in renderScreen constexpr int startY = 60; @@ -32,17 +43,21 @@ void EpubReaderChapterSelectionActivity::taskTrampoline(void* param) { } void EpubReaderChapterSelectionActivity::onEnter() { - Activity::onEnter(); + ActivityWithSubactivity::onEnter(); if (!epub) { return; } renderingMutex = xSemaphoreCreateMutex(); + + // Account for sync option offset when finding current TOC index + const int syncOffset = KOREADER_STORE.hasCredentials() ? 1 : 0; selectorIndex = epub->getTocIndexForSpineIndex(currentSpineIndex); if (selectorIndex == -1) { selectorIndex = 0; } + selectorIndex += syncOffset; // Offset for sync option // Trigger first update updateRequired = true; @@ -55,7 +70,7 @@ void EpubReaderChapterSelectionActivity::onEnter() { } void EpubReaderChapterSelectionActivity::onExit() { - Activity::onExit(); + ActivityWithSubactivity::onExit(); // Wait until not rendering to delete task to avoid killing mid-instruction to EPD xSemaphoreTake(renderingMutex, portMAX_DELAY); @@ -67,7 +82,30 @@ void EpubReaderChapterSelectionActivity::onExit() { renderingMutex = nullptr; } +void EpubReaderChapterSelectionActivity::launchSyncActivity() { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + exitActivity(); + enterNewActivity(new KOReaderSyncActivity( + renderer, mappedInput, epub, epubPath, currentSpineIndex, currentPage, totalPagesInSpine, + [this]() { + // On cancel + exitActivity(); + updateRequired = true; + }, + [this](int newSpineIndex, int newPage) { + // On sync complete + exitActivity(); + onSyncPosition(newSpineIndex, newPage); + })); + xSemaphoreGive(renderingMutex); +} + void EpubReaderChapterSelectionActivity::loop() { + if (subActivity) { + subActivity->loop(); + return; + } + const bool prevReleased = mappedInput.wasReleased(MappedInputManager::Button::Up) || mappedInput.wasReleased(MappedInputManager::Button::Left); const bool nextReleased = mappedInput.wasReleased(MappedInputManager::Button::Down) || @@ -75,9 +113,19 @@ void EpubReaderChapterSelectionActivity::loop() { const bool skipPage = mappedInput.getHeldTime() > SKIP_PAGE_MS; const int pageItems = getPageItems(); + const int totalItems = getTotalItems(); + const int syncOffset = KOREADER_STORE.hasCredentials() ? 1 : 0; if (mappedInput.wasReleased(MappedInputManager::Button::Confirm)) { - const auto newSpineIndex = epub->getSpineIndexForTocIndex(selectorIndex); + // Check if sync option is selected + if (syncOffset > 0 && selectorIndex == SYNC_ITEM_INDEX) { + launchSyncActivity(); + return; + } + + // Get TOC index (account for sync offset) + const int tocIndex = selectorIndex - syncOffset; + const auto newSpineIndex = epub->getSpineIndexForTocIndex(tocIndex); if (newSpineIndex == -1) { onGoBack(); } else { @@ -87,17 +135,16 @@ void EpubReaderChapterSelectionActivity::loop() { onGoBack(); } else if (prevReleased) { if (skipPage) { - selectorIndex = - ((selectorIndex / pageItems - 1) * pageItems + epub->getTocItemsCount()) % epub->getTocItemsCount(); + selectorIndex = ((selectorIndex / pageItems - 1) * pageItems + totalItems) % totalItems; } else { - selectorIndex = (selectorIndex + epub->getTocItemsCount() - 1) % epub->getTocItemsCount(); + selectorIndex = (selectorIndex + totalItems - 1) % totalItems; } updateRequired = true; } else if (nextReleased) { if (skipPage) { - selectorIndex = ((selectorIndex / pageItems + 1) * pageItems) % epub->getTocItemsCount(); + selectorIndex = ((selectorIndex / pageItems + 1) * pageItems) % totalItems; } else { - selectorIndex = (selectorIndex + 1) % epub->getTocItemsCount(); + selectorIndex = (selectorIndex + 1) % totalItems; } updateRequired = true; } @@ -105,7 +152,7 @@ void EpubReaderChapterSelectionActivity::loop() { void EpubReaderChapterSelectionActivity::displayTaskLoop() { while (true) { - if (updateRequired) { + if (updateRequired && !subActivity) { updateRequired = false; xSemaphoreTake(renderingMutex, portMAX_DELAY); renderScreen(); @@ -120,6 +167,8 @@ void EpubReaderChapterSelectionActivity::renderScreen() { const auto pageWidth = renderer.getScreenWidth(); const int pageItems = getPageItems(); + const int totalItems = getTotalItems(); + const int syncOffset = KOREADER_STORE.hasCredentials() ? 1 : 0; const std::string title = renderer.truncatedText(UI_12_FONT_ID, epub->getTitle().c_str(), pageWidth - 40, EpdFontFamily::BOLD); @@ -127,11 +176,20 @@ void EpubReaderChapterSelectionActivity::renderScreen() { const auto pageStartIndex = selectorIndex / pageItems * pageItems; renderer.fillRect(0, 60 + (selectorIndex % pageItems) * 30 - 2, pageWidth - 1, 30); - for (int tocIndex = pageStartIndex; tocIndex < epub->getTocItemsCount() && tocIndex < pageStartIndex + pageItems; - tocIndex++) { - auto item = epub->getTocItem(tocIndex); - renderer.drawText(UI_10_FONT_ID, 20 + (item.level - 1) * 15, 60 + (tocIndex % pageItems) * 30, item.title.c_str(), - tocIndex != selectorIndex); + + for (int itemIndex = pageStartIndex; itemIndex < totalItems && itemIndex < pageStartIndex + pageItems; itemIndex++) { + const int displayY = 60 + (itemIndex % pageItems) * 30; + const bool isSelected = (itemIndex == selectorIndex); + + if (syncOffset > 0 && itemIndex == SYNC_ITEM_INDEX) { + // Draw sync option + renderer.drawText(UI_10_FONT_ID, 20, displayY, ">> Sync Progress", !isSelected); + } else { + // Draw TOC item (account for sync offset) + const int tocIndex = itemIndex - syncOffset; + auto item = epub->getTocItem(tocIndex); + renderer.drawText(UI_10_FONT_ID, 20 + (item.level - 1) * 15, displayY, item.title.c_str(), !isSelected); + } } renderer.displayBuffer(); diff --git a/src/activities/reader/EpubReaderChapterSelectionActivity.h b/src/activities/reader/EpubReaderChapterSelectionActivity.h index cf3f1905..4bfef6da 100644 --- a/src/activities/reader/EpubReaderChapterSelectionActivity.h +++ b/src/activities/reader/EpubReaderChapterSelectionActivity.h @@ -6,36 +6,50 @@ #include -#include "../Activity.h" +#include "../ActivityWithSubactivity.h" -class EpubReaderChapterSelectionActivity final : public Activity { +class EpubReaderChapterSelectionActivity final : public ActivityWithSubactivity { std::shared_ptr epub; + std::string epubPath; TaskHandle_t displayTaskHandle = nullptr; SemaphoreHandle_t renderingMutex = nullptr; int currentSpineIndex = 0; + int currentPage = 0; + int totalPagesInSpine = 0; int selectorIndex = 0; bool updateRequired = false; const std::function onGoBack; const std::function onSelectSpineIndex; + const std::function onSyncPosition; // Number of items that fit on a page, derived from logical screen height. // This adapts automatically when switching between portrait and landscape. int getPageItems() const; + // Total items including the sync option + int getTotalItems() const; + static void taskTrampoline(void* param); [[noreturn]] void displayTaskLoop(); void renderScreen(); + void launchSyncActivity(); public: explicit EpubReaderChapterSelectionActivity(GfxRenderer& renderer, MappedInputManager& mappedInput, - const std::shared_ptr& epub, const int currentSpineIndex, - const std::function& onGoBack, - const std::function& onSelectSpineIndex) - : Activity("EpubReaderChapterSelection", renderer, mappedInput), + const std::shared_ptr& epub, const std::string& epubPath, + const int currentSpineIndex, const int currentPage, + const int totalPagesInSpine, const std::function& onGoBack, + const std::function& onSelectSpineIndex, + const std::function& onSyncPosition) + : ActivityWithSubactivity("EpubReaderChapterSelection", renderer, mappedInput), epub(epub), + epubPath(epubPath), currentSpineIndex(currentSpineIndex), + currentPage(currentPage), + totalPagesInSpine(totalPagesInSpine), onGoBack(onGoBack), - onSelectSpineIndex(onSelectSpineIndex) {} + onSelectSpineIndex(onSelectSpineIndex), + onSyncPosition(onSyncPosition) {} void onEnter() override; void onExit() override; void loop() override; diff --git a/src/activities/reader/KOReaderSyncActivity.cpp b/src/activities/reader/KOReaderSyncActivity.cpp new file mode 100644 index 00000000..f267e124 --- /dev/null +++ b/src/activities/reader/KOReaderSyncActivity.cpp @@ -0,0 +1,423 @@ +#include "KOReaderSyncActivity.h" + +#include +#include +#include + +#include "KOReaderCredentialStore.h" +#include "KOReaderDocumentId.h" +#include "MappedInputManager.h" +#include "activities/network/WifiSelectionActivity.h" +#include "fontIds.h" + +namespace { +void syncTimeWithNTP() { + // Configure SNTP + esp_sntp_setoperatingmode(ESP_SNTP_OPMODE_POLL); + esp_sntp_setservername(0, "pool.ntp.org"); + esp_sntp_init(); + + // Wait for time to sync (with timeout) + int retry = 0; + const int maxRetries = 50; // 5 seconds max + while (sntp_get_sync_status() != SNTP_SYNC_STATUS_COMPLETED && retry < maxRetries) { + vTaskDelay(100 / portTICK_PERIOD_MS); + retry++; + } + + if (retry < maxRetries) { + Serial.printf("[%lu] [KOSync] NTP time synced\n", millis()); + } else { + Serial.printf("[%lu] [KOSync] NTP sync timeout, using fallback\n", millis()); + } +} +} // namespace + +void KOReaderSyncActivity::taskTrampoline(void* param) { + auto* self = static_cast(param); + self->displayTaskLoop(); +} + +void KOReaderSyncActivity::onWifiSelectionComplete(const bool success) { + exitActivity(); + + if (!success) { + Serial.printf("[%lu] [KOSync] WiFi connection failed, exiting\n", millis()); + onCancel(); + return; + } + + Serial.printf("[%lu] [KOSync] WiFi connected, starting sync\n", millis()); + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = SYNCING; + statusMessage = "Syncing time..."; + xSemaphoreGive(renderingMutex); + updateRequired = true; + + // Sync time with NTP before making API requests + syncTimeWithNTP(); + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + statusMessage = "Calculating document hash..."; + xSemaphoreGive(renderingMutex); + updateRequired = true; + + performSync(); +} + +void KOReaderSyncActivity::performSync() { + // Calculate document hash + documentHash = KOReaderDocumentId::calculate(epubPath); + if (documentHash.empty()) { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = SYNC_FAILED; + statusMessage = "Failed to calculate document hash"; + xSemaphoreGive(renderingMutex); + updateRequired = true; + return; + } + + Serial.printf("[%lu] [KOSync] Document hash: %s\n", millis(), documentHash.c_str()); + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + statusMessage = "Fetching remote progress..."; + xSemaphoreGive(renderingMutex); + updateRequired = true; + vTaskDelay(10 / portTICK_PERIOD_MS); + + // Fetch remote progress + const auto result = KOReaderSyncClient::getProgress(documentHash, remoteProgress); + + if (result == KOReaderSyncClient::NOT_FOUND) { + // No remote progress - offer to upload + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = NO_REMOTE_PROGRESS; + hasRemoteProgress = false; + xSemaphoreGive(renderingMutex); + updateRequired = true; + return; + } + + if (result != KOReaderSyncClient::OK) { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = SYNC_FAILED; + statusMessage = KOReaderSyncClient::errorString(result); + xSemaphoreGive(renderingMutex); + updateRequired = true; + return; + } + + // Convert remote progress to CrossPoint position + hasRemoteProgress = true; + KOReaderPosition koPos = {remoteProgress.progress, remoteProgress.percentage}; + remotePosition = ProgressMapper::toCrossPoint(epub, koPos, totalPagesInSpine); + + // Calculate local progress in KOReader format (for display) + CrossPointPosition localPos = {currentSpineIndex, currentPage, totalPagesInSpine}; + localProgress = ProgressMapper::toKOReader(epub, localPos); + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = SHOWING_RESULT; + selectedOption = 0; // Default to "Apply" + xSemaphoreGive(renderingMutex); + updateRequired = true; +} + +void KOReaderSyncActivity::performUpload() { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = UPLOADING; + statusMessage = "Uploading progress..."; + xSemaphoreGive(renderingMutex); + updateRequired = true; + vTaskDelay(10 / portTICK_PERIOD_MS); + + // Convert current position to KOReader format + CrossPointPosition localPos = {currentSpineIndex, currentPage, totalPagesInSpine}; + KOReaderPosition koPos = ProgressMapper::toKOReader(epub, localPos); + + KOReaderProgress progress; + progress.document = documentHash; + progress.progress = koPos.xpath; + progress.percentage = koPos.percentage; + + const auto result = KOReaderSyncClient::updateProgress(progress); + + if (result != KOReaderSyncClient::OK) { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = SYNC_FAILED; + statusMessage = KOReaderSyncClient::errorString(result); + xSemaphoreGive(renderingMutex); + updateRequired = true; + return; + } + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = UPLOAD_COMPLETE; + xSemaphoreGive(renderingMutex); + updateRequired = true; +} + +void KOReaderSyncActivity::onEnter() { + ActivityWithSubactivity::onEnter(); + + renderingMutex = xSemaphoreCreateMutex(); + + xTaskCreate(&KOReaderSyncActivity::taskTrampoline, "KOSyncTask", + 4096, // Stack size (larger for network operations) + this, // Parameters + 1, // Priority + &displayTaskHandle // Task handle + ); + + // Check for credentials first + if (!KOREADER_STORE.hasCredentials()) { + state = NO_CREDENTIALS; + updateRequired = true; + return; + } + + // Turn on WiFi + Serial.printf("[%lu] [KOSync] Turning on WiFi...\n", millis()); + WiFi.mode(WIFI_STA); + + // Check if already connected + if (WiFi.status() == WL_CONNECTED) { + Serial.printf("[%lu] [KOSync] Already connected to WiFi\n", millis()); + state = SYNCING; + statusMessage = "Syncing time..."; + updateRequired = true; + + // Perform sync directly (will be handled in loop) + xTaskCreate( + [](void* param) { + auto* self = static_cast(param); + // Sync time first + syncTimeWithNTP(); + xSemaphoreTake(self->renderingMutex, portMAX_DELAY); + self->statusMessage = "Calculating document hash..."; + xSemaphoreGive(self->renderingMutex); + self->updateRequired = true; + self->performSync(); + vTaskDelete(nullptr); + }, + "SyncTask", 4096, this, 1, nullptr); + return; + } + + // Launch WiFi selection subactivity + Serial.printf("[%lu] [KOSync] Launching WifiSelectionActivity...\n", millis()); + enterNewActivity( + new WifiSelectionActivity(renderer, mappedInput, [this](const bool connected) { onWifiSelectionComplete(connected); })); +} + +void KOReaderSyncActivity::onExit() { + ActivityWithSubactivity::onExit(); + + // Turn off wifi + WiFi.disconnect(false); + delay(100); + WiFi.mode(WIFI_OFF); + delay(100); + + // Wait until not rendering to delete task + xSemaphoreTake(renderingMutex, portMAX_DELAY); + if (displayTaskHandle) { + vTaskDelete(displayTaskHandle); + displayTaskHandle = nullptr; + } + vSemaphoreDelete(renderingMutex); + renderingMutex = nullptr; +} + +void KOReaderSyncActivity::displayTaskLoop() { + while (true) { + if (updateRequired) { + updateRequired = false; + xSemaphoreTake(renderingMutex, portMAX_DELAY); + render(); + xSemaphoreGive(renderingMutex); + } + vTaskDelay(10 / portTICK_PERIOD_MS); + } +} + +void KOReaderSyncActivity::render() { + if (subActivity) { + return; + } + + const auto pageWidth = renderer.getScreenWidth(); + + renderer.clearScreen(); + renderer.drawCenteredText(UI_12_FONT_ID, 15, "KOReader Sync", true, EpdFontFamily::BOLD); + + if (state == NO_CREDENTIALS) { + renderer.drawCenteredText(UI_10_FONT_ID, 280, "No credentials configured", true, EpdFontFamily::BOLD); + renderer.drawCenteredText(UI_10_FONT_ID, 320, "Set up KOReader account in Settings"); + + const auto labels = mappedInput.mapLabels("Back", "", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } + + if (state == SYNCING || state == UPLOADING) { + renderer.drawCenteredText(UI_10_FONT_ID, 300, statusMessage.c_str(), true, EpdFontFamily::BOLD); + renderer.displayBuffer(); + return; + } + + if (state == SHOWING_RESULT) { + // Show comparison + renderer.drawCenteredText(UI_10_FONT_ID, 120, "Progress found!", true, EpdFontFamily::BOLD); + + // Get chapter names from TOC + const int remoteTocIndex = epub->getTocIndexForSpineIndex(remotePosition.spineIndex); + const int localTocIndex = epub->getTocIndexForSpineIndex(currentSpineIndex); + const std::string remoteChapter = + (remoteTocIndex >= 0) ? epub->getTocItem(remoteTocIndex).title : ("Section " + std::to_string(remotePosition.spineIndex + 1)); + const std::string localChapter = + (localTocIndex >= 0) ? epub->getTocItem(localTocIndex).title : ("Section " + std::to_string(currentSpineIndex + 1)); + + // Remote progress - chapter and page + renderer.drawText(UI_10_FONT_ID, 20, 160, "Remote:", true); + char remoteChapterStr[128]; + snprintf(remoteChapterStr, sizeof(remoteChapterStr), " %s", remoteChapter.c_str()); + renderer.drawText(UI_10_FONT_ID, 20, 185, remoteChapterStr); + char remotePageStr[64]; + snprintf(remotePageStr, sizeof(remotePageStr), " Page %d, %.0f%% overall", remotePosition.pageNumber + 1, remoteProgress.percentage * 100); + renderer.drawText(UI_10_FONT_ID, 20, 210, remotePageStr); + + if (!remoteProgress.device.empty()) { + char deviceStr[64]; + snprintf(deviceStr, sizeof(deviceStr), " From: %s", remoteProgress.device.c_str()); + renderer.drawText(UI_10_FONT_ID, 20, 235, deviceStr); + } + + // Local progress - chapter and page + renderer.drawText(UI_10_FONT_ID, 20, 270, "Local:", true); + char localChapterStr[128]; + snprintf(localChapterStr, sizeof(localChapterStr), " %s", localChapter.c_str()); + renderer.drawText(UI_10_FONT_ID, 20, 295, localChapterStr); + char localPageStr[64]; + snprintf(localPageStr, sizeof(localPageStr), " Page %d/%d, %.0f%% overall", currentPage + 1, totalPagesInSpine, localProgress.percentage * 100); + renderer.drawText(UI_10_FONT_ID, 20, 320, localPageStr); + + // Options + const int optionY = 350; + const int optionHeight = 30; + + // Apply option + if (selectedOption == 0) { + renderer.fillRect(0, optionY - 2, pageWidth - 1, optionHeight); + } + renderer.drawText(UI_10_FONT_ID, 20, optionY, "Apply remote progress", selectedOption != 0); + + // Upload option + if (selectedOption == 1) { + renderer.fillRect(0, optionY + optionHeight - 2, pageWidth - 1, optionHeight); + } + renderer.drawText(UI_10_FONT_ID, 20, optionY + optionHeight, "Upload local progress", selectedOption != 1); + + // Cancel option + if (selectedOption == 2) { + renderer.fillRect(0, optionY + optionHeight * 2 - 2, pageWidth - 1, optionHeight); + } + renderer.drawText(UI_10_FONT_ID, 20, optionY + optionHeight * 2, "Cancel", selectedOption != 2); + + const auto labels = mappedInput.mapLabels("", "Select", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } + + if (state == NO_REMOTE_PROGRESS) { + renderer.drawCenteredText(UI_10_FONT_ID, 280, "No remote progress found", true, EpdFontFamily::BOLD); + renderer.drawCenteredText(UI_10_FONT_ID, 320, "Upload current position?"); + + const auto labels = mappedInput.mapLabels("Cancel", "Upload", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } + + if (state == UPLOAD_COMPLETE) { + renderer.drawCenteredText(UI_10_FONT_ID, 300, "Progress uploaded!", true, EpdFontFamily::BOLD); + + const auto labels = mappedInput.mapLabels("Back", "", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } + + if (state == SYNC_FAILED) { + renderer.drawCenteredText(UI_10_FONT_ID, 280, "Sync failed", true, EpdFontFamily::BOLD); + renderer.drawCenteredText(UI_10_FONT_ID, 320, statusMessage.c_str()); + + const auto labels = mappedInput.mapLabels("Back", "", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } +} + +void KOReaderSyncActivity::loop() { + if (subActivity) { + subActivity->loop(); + return; + } + + if (state == NO_CREDENTIALS || state == SYNC_FAILED || state == UPLOAD_COMPLETE) { + if (mappedInput.wasPressed(MappedInputManager::Button::Back)) { + onCancel(); + } + return; + } + + if (state == SHOWING_RESULT) { + // Navigate options + if (mappedInput.wasPressed(MappedInputManager::Button::Up) || + mappedInput.wasPressed(MappedInputManager::Button::Left)) { + selectedOption = (selectedOption + 2) % 3; // Wrap around + updateRequired = true; + } else if (mappedInput.wasPressed(MappedInputManager::Button::Down) || + mappedInput.wasPressed(MappedInputManager::Button::Right)) { + selectedOption = (selectedOption + 1) % 3; + updateRequired = true; + } + + if (mappedInput.wasPressed(MappedInputManager::Button::Confirm)) { + if (selectedOption == 0) { + // Apply remote progress + onSyncComplete(remotePosition.spineIndex, remotePosition.pageNumber); + } else if (selectedOption == 1) { + // Upload local progress + performUpload(); + } else { + // Cancel + onCancel(); + } + } + + if (mappedInput.wasPressed(MappedInputManager::Button::Back)) { + onCancel(); + } + return; + } + + if (state == NO_REMOTE_PROGRESS) { + if (mappedInput.wasPressed(MappedInputManager::Button::Confirm)) { + // Calculate hash if not done yet + if (documentHash.empty()) { + documentHash = KOReaderDocumentId::calculate(epubPath); + } + performUpload(); + } + + if (mappedInput.wasPressed(MappedInputManager::Button::Back)) { + onCancel(); + } + return; + } +} diff --git a/src/activities/reader/KOReaderSyncActivity.h b/src/activities/reader/KOReaderSyncActivity.h new file mode 100644 index 00000000..7f1948d1 --- /dev/null +++ b/src/activities/reader/KOReaderSyncActivity.h @@ -0,0 +1,95 @@ +#pragma once +#include +#include +#include +#include + +#include +#include + +#include "KOReaderSyncClient.h" +#include "ProgressMapper.h" +#include "activities/ActivityWithSubactivity.h" + +/** + * Activity for syncing reading progress with KOReader sync server. + * + * Flow: + * 1. Connect to WiFi (if not connected) + * 2. Calculate document hash + * 3. Fetch remote progress + * 4. Show comparison and options (Apply/Upload/Cancel) + * 5. Apply or upload progress + */ +class KOReaderSyncActivity final : public ActivityWithSubactivity { + public: + using OnCancelCallback = std::function; + using OnSyncCompleteCallback = std::function; + + explicit KOReaderSyncActivity(GfxRenderer& renderer, MappedInputManager& mappedInput, + const std::shared_ptr& epub, const std::string& epubPath, int currentSpineIndex, + int currentPage, int totalPagesInSpine, OnCancelCallback onCancel, + OnSyncCompleteCallback onSyncComplete) + : ActivityWithSubactivity("KOReaderSync", renderer, mappedInput), + epub(epub), + epubPath(epubPath), + currentSpineIndex(currentSpineIndex), + currentPage(currentPage), + totalPagesInSpine(totalPagesInSpine), + onCancel(std::move(onCancel)), + onSyncComplete(std::move(onSyncComplete)) {} + + void onEnter() override; + void onExit() override; + void loop() override; + bool preventAutoSleep() override { return state == CONNECTING || state == SYNCING; } + + private: + enum State { + WIFI_SELECTION, + CONNECTING, + SYNCING, + SHOWING_RESULT, + UPLOADING, + UPLOAD_COMPLETE, + NO_REMOTE_PROGRESS, + SYNC_FAILED, + NO_CREDENTIALS + }; + + std::shared_ptr epub; + std::string epubPath; + int currentSpineIndex; + int currentPage; + int totalPagesInSpine; + + TaskHandle_t displayTaskHandle = nullptr; + SemaphoreHandle_t renderingMutex = nullptr; + bool updateRequired = false; + + State state = WIFI_SELECTION; + std::string statusMessage; + std::string documentHash; + + // Remote progress data + bool hasRemoteProgress = false; + KOReaderProgress remoteProgress; + CrossPointPosition remotePosition; + + // Local progress as KOReader format (for display) + KOReaderPosition localProgress; + + // Selection in result screen (0=Apply, 1=Upload, 2=Cancel) + int selectedOption = 0; + + OnCancelCallback onCancel; + OnSyncCompleteCallback onSyncComplete; + + void onWifiSelectionComplete(bool success); + void performSync(); + void performUpload(); + + static void taskTrampoline(void* param); + [[noreturn]] void displayTaskLoop(); + void render(); +}; diff --git a/src/activities/settings/KOReaderAuthActivity.cpp b/src/activities/settings/KOReaderAuthActivity.cpp new file mode 100644 index 00000000..8681812f --- /dev/null +++ b/src/activities/settings/KOReaderAuthActivity.cpp @@ -0,0 +1,167 @@ +#include "KOReaderAuthActivity.h" + +#include +#include + +#include "KOReaderCredentialStore.h" +#include "KOReaderSyncClient.h" +#include "MappedInputManager.h" +#include "activities/network/WifiSelectionActivity.h" +#include "fontIds.h" + +void KOReaderAuthActivity::taskTrampoline(void* param) { + auto* self = static_cast(param); + self->displayTaskLoop(); +} + +void KOReaderAuthActivity::onWifiSelectionComplete(const bool success) { + exitActivity(); + + if (!success) { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = FAILED; + errorMessage = "WiFi connection failed"; + xSemaphoreGive(renderingMutex); + updateRequired = true; + return; + } + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + state = AUTHENTICATING; + statusMessage = "Authenticating..."; + xSemaphoreGive(renderingMutex); + updateRequired = true; + + performAuthentication(); +} + +void KOReaderAuthActivity::performAuthentication() { + const auto result = KOReaderSyncClient::authenticate(); + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + if (result == KOReaderSyncClient::OK) { + state = SUCCESS; + statusMessage = "Successfully authenticated!"; + } else { + state = FAILED; + errorMessage = KOReaderSyncClient::errorString(result); + } + xSemaphoreGive(renderingMutex); + updateRequired = true; +} + +void KOReaderAuthActivity::onEnter() { + ActivityWithSubactivity::onEnter(); + + renderingMutex = xSemaphoreCreateMutex(); + + xTaskCreate(&KOReaderAuthActivity::taskTrampoline, "KOAuthTask", + 4096, // Stack size + this, // Parameters + 1, // Priority + &displayTaskHandle // Task handle + ); + + // Turn on WiFi + WiFi.mode(WIFI_STA); + + // Check if already connected + if (WiFi.status() == WL_CONNECTED) { + state = AUTHENTICATING; + statusMessage = "Authenticating..."; + updateRequired = true; + + // Perform authentication in a separate task + xTaskCreate( + [](void* param) { + auto* self = static_cast(param); + self->performAuthentication(); + vTaskDelete(nullptr); + }, + "AuthTask", 4096, this, 1, nullptr); + return; + } + + // Launch WiFi selection + enterNewActivity(new WifiSelectionActivity(renderer, mappedInput, + [this](const bool connected) { onWifiSelectionComplete(connected); })); +} + +void KOReaderAuthActivity::onExit() { + ActivityWithSubactivity::onExit(); + + // Turn off wifi + WiFi.disconnect(false); + delay(100); + WiFi.mode(WIFI_OFF); + delay(100); + + xSemaphoreTake(renderingMutex, portMAX_DELAY); + if (displayTaskHandle) { + vTaskDelete(displayTaskHandle); + displayTaskHandle = nullptr; + } + vSemaphoreDelete(renderingMutex); + renderingMutex = nullptr; +} + +void KOReaderAuthActivity::displayTaskLoop() { + while (true) { + if (updateRequired && !subActivity) { + updateRequired = false; + xSemaphoreTake(renderingMutex, portMAX_DELAY); + render(); + xSemaphoreGive(renderingMutex); + } + vTaskDelay(10 / portTICK_PERIOD_MS); + } +} + +void KOReaderAuthActivity::render() { + if (subActivity) { + return; + } + + renderer.clearScreen(); + renderer.drawCenteredText(UI_12_FONT_ID, 15, "KOReader Auth", true, EpdFontFamily::BOLD); + + if (state == AUTHENTICATING) { + renderer.drawCenteredText(UI_10_FONT_ID, 300, statusMessage.c_str(), true, EpdFontFamily::BOLD); + renderer.displayBuffer(); + return; + } + + if (state == SUCCESS) { + renderer.drawCenteredText(UI_10_FONT_ID, 280, "Success!", true, EpdFontFamily::BOLD); + renderer.drawCenteredText(UI_10_FONT_ID, 320, "KOReader sync is ready to use"); + + const auto labels = mappedInput.mapLabels("Done", "", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } + + if (state == FAILED) { + renderer.drawCenteredText(UI_10_FONT_ID, 280, "Authentication Failed", true, EpdFontFamily::BOLD); + renderer.drawCenteredText(UI_10_FONT_ID, 320, errorMessage.c_str()); + + const auto labels = mappedInput.mapLabels("Back", "", "", ""); + renderer.drawButtonHints(UI_10_FONT_ID, labels.btn1, labels.btn2, labels.btn3, labels.btn4); + renderer.displayBuffer(); + return; + } +} + +void KOReaderAuthActivity::loop() { + if (subActivity) { + subActivity->loop(); + return; + } + + if (state == SUCCESS || state == FAILED) { + if (mappedInput.wasPressed(MappedInputManager::Button::Back) || + mappedInput.wasPressed(MappedInputManager::Button::Confirm)) { + onComplete(); + } + } +} diff --git a/src/activities/settings/KOReaderAuthActivity.h b/src/activities/settings/KOReaderAuthActivity.h new file mode 100644 index 00000000..a6ed0d3e --- /dev/null +++ b/src/activities/settings/KOReaderAuthActivity.h @@ -0,0 +1,44 @@ +#pragma once +#include +#include +#include + +#include + +#include "activities/ActivityWithSubactivity.h" + +/** + * Activity for testing KOReader credentials. + * Connects to WiFi and authenticates with the KOReader sync server. + */ +class KOReaderAuthActivity final : public ActivityWithSubactivity { + public: + explicit KOReaderAuthActivity(GfxRenderer& renderer, MappedInputManager& mappedInput, + const std::function& onComplete) + : ActivityWithSubactivity("KOReaderAuth", renderer, mappedInput), onComplete(onComplete) {} + + void onEnter() override; + void onExit() override; + void loop() override; + bool preventAutoSleep() override { return state == CONNECTING || state == AUTHENTICATING; } + + private: + enum State { WIFI_SELECTION, CONNECTING, AUTHENTICATING, SUCCESS, FAILED }; + + TaskHandle_t displayTaskHandle = nullptr; + SemaphoreHandle_t renderingMutex = nullptr; + bool updateRequired = false; + + State state = WIFI_SELECTION; + std::string statusMessage; + std::string errorMessage; + + const std::function onComplete; + + void onWifiSelectionComplete(bool success); + void performAuthentication(); + + static void taskTrampoline(void* param); + [[noreturn]] void displayTaskLoop(); + void render(); +}; diff --git a/src/activities/settings/SettingsActivity.cpp b/src/activities/settings/SettingsActivity.cpp index a242389d..979648a8 100644 --- a/src/activities/settings/SettingsActivity.cpp +++ b/src/activities/settings/SettingsActivity.cpp @@ -3,13 +3,16 @@ #include #include "CrossPointSettings.h" +#include "KOReaderAuthActivity.h" +#include "KOReaderCredentialStore.h" #include "MappedInputManager.h" #include "OtaUpdateActivity.h" +#include "activities/util/KeyboardEntryActivity.h" #include "fontIds.h" // Define the static settings list namespace { -constexpr int settingsCount = 14; +constexpr int settingsCount = 18; const SettingInfo settingsList[settingsCount] = { // Should match with SLEEP_SCREEN_MODE {"Sleep Screen", SettingType::ENUM, &CrossPointSettings::sleepScreen, {"Dark", "Light", "Custom", "Cover"}}, @@ -46,8 +49,56 @@ const SettingInfo settingsList[settingsCount] = { SettingType::ENUM, &CrossPointSettings::refreshFrequency, {"1 page", "5 pages", "10 pages", "15 pages", "30 pages"}}, + {"KOReader Username", SettingType::ACTION, nullptr, {}}, + {"KOReader Password", SettingType::ACTION, nullptr, {}}, + {"Authenticate KOReader", SettingType::ACTION, nullptr, {}}, {"Check for updates", SettingType::ACTION, nullptr, {}}, }; + +// Check if a setting should be visible +bool isSettingVisible(int index) { + // Hide "Authenticate KOReader" if credentials are not set + if (std::string(settingsList[index].name) == "Authenticate KOReader") { + return KOREADER_STORE.hasCredentials(); + } + return true; +} + +// Get visible settings count +int getVisibleSettingsCount() { + int count = 0; + for (int i = 0; i < settingsCount; i++) { + if (isSettingVisible(i)) { + count++; + } + } + return count; +} + +// Convert visible index to actual settings index +int visibleToActualIndex(int visibleIndex) { + int count = 0; + for (int i = 0; i < settingsCount; i++) { + if (isSettingVisible(i)) { + if (count == visibleIndex) { + return i; + } + count++; + } + } + return -1; +} + +// Convert actual index to visible index +int actualToVisibleIndex(int actualIndex) { + int count = 0; + for (int i = 0; i < actualIndex; i++) { + if (isSettingVisible(i)) { + count++; + } + } + return count; +} } // namespace void SettingsActivity::taskTrampoline(void* param) { @@ -106,16 +157,17 @@ void SettingsActivity::loop() { return; } - // Handle navigation + // Handle navigation (using visible settings count) + const int visibleCount = getVisibleSettingsCount(); if (mappedInput.wasPressed(MappedInputManager::Button::Up) || mappedInput.wasPressed(MappedInputManager::Button::Left)) { // Move selection up (with wrap-around) - selectedSettingIndex = (selectedSettingIndex > 0) ? (selectedSettingIndex - 1) : (settingsCount - 1); + selectedSettingIndex = (selectedSettingIndex > 0) ? (selectedSettingIndex - 1) : (visibleCount - 1); updateRequired = true; } else if (mappedInput.wasPressed(MappedInputManager::Button::Down) || mappedInput.wasPressed(MappedInputManager::Button::Right)) { // Move selection down - if (selectedSettingIndex < settingsCount - 1) { + if (selectedSettingIndex < visibleCount - 1) { selectedSettingIndex++; updateRequired = true; } @@ -123,12 +175,13 @@ void SettingsActivity::loop() { } void SettingsActivity::toggleCurrentSetting() { - // Validate index - if (selectedSettingIndex < 0 || selectedSettingIndex >= settingsCount) { + // Convert visible index to actual index + const int actualIndex = visibleToActualIndex(selectedSettingIndex); + if (actualIndex < 0 || actualIndex >= settingsCount) { return; } - const auto& setting = settingsList[selectedSettingIndex]; + const auto& setting = settingsList[actualIndex]; if (setting.type == SettingType::TOGGLE && setting.valuePtr != nullptr) { // Toggle the boolean value using the member pointer @@ -146,6 +199,54 @@ void SettingsActivity::toggleCurrentSetting() { updateRequired = true; })); xSemaphoreGive(renderingMutex); + } else if (std::string(setting.name) == "KOReader Username") { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + exitActivity(); + enterNewActivity(new KeyboardEntryActivity( + renderer, mappedInput, "KOReader Username", KOREADER_STORE.getUsername(), 10, + 64, // maxLength + false, // not password + [this](const std::string& username) { + KOREADER_STORE.setCredentials(username, KOREADER_STORE.getPassword()); + KOREADER_STORE.saveToFile(); + exitActivity(); + updateRequired = true; + }, + [this]() { + exitActivity(); + updateRequired = true; + })); + xSemaphoreGive(renderingMutex); + } else if (std::string(setting.name) == "KOReader Password") { + xSemaphoreTake(renderingMutex, portMAX_DELAY); + exitActivity(); + enterNewActivity(new KeyboardEntryActivity( + renderer, mappedInput, "KOReader Password", KOREADER_STORE.getPassword(), 10, + 64, // maxLength + false, // not password mode - show characters + [this](const std::string& password) { + KOREADER_STORE.setCredentials(KOREADER_STORE.getUsername(), password); + KOREADER_STORE.saveToFile(); + exitActivity(); + updateRequired = true; + }, + [this]() { + exitActivity(); + updateRequired = true; + })); + xSemaphoreGive(renderingMutex); + } else if (std::string(setting.name) == "Authenticate KOReader") { + // Only allow if credentials are set + if (!KOREADER_STORE.hasCredentials()) { + return; + } + xSemaphoreTake(renderingMutex, portMAX_DELAY); + exitActivity(); + enterNewActivity(new KOReaderAuthActivity(renderer, mappedInput, [this] { + exitActivity(); + updateRequired = true; + })); + xSemaphoreGive(renderingMutex); } } else { // Only toggle if it's a toggle type and has a value pointer @@ -177,15 +278,21 @@ void SettingsActivity::render() const { // Draw header renderer.drawCenteredText(UI_12_FONT_ID, 15, "Settings", true, EpdFontFamily::BOLD); - // Draw selection + // Draw selection highlight renderer.fillRect(0, 60 + selectedSettingIndex * 30 - 2, pageWidth - 1, 30); - // Draw all settings + // Draw only visible settings + int visibleIndex = 0; for (int i = 0; i < settingsCount; i++) { - const int settingY = 60 + i * 30; // 30 pixels between settings + if (!isSettingVisible(i)) { + continue; + } + + const int settingY = 60 + visibleIndex * 30; // 30 pixels between settings + const bool isSelected = (visibleIndex == selectedSettingIndex); // Draw setting name - renderer.drawText(UI_10_FONT_ID, 20, settingY, settingsList[i].name, i != selectedSettingIndex); + renderer.drawText(UI_10_FONT_ID, 20, settingY, settingsList[i].name, !isSelected); // Draw value based on setting type std::string valueText = ""; @@ -195,9 +302,18 @@ void SettingsActivity::render() const { } else if (settingsList[i].type == SettingType::ENUM && settingsList[i].valuePtr != nullptr) { const uint8_t value = SETTINGS.*(settingsList[i].valuePtr); valueText = settingsList[i].enumValues[value]; + } else if (settingsList[i].type == SettingType::ACTION) { + // Show status for KOReader settings + if (std::string(settingsList[i].name) == "KOReader Username") { + valueText = KOREADER_STORE.getUsername().empty() ? "[Not Set]" : "[Set]"; + } else if (std::string(settingsList[i].name) == "KOReader Password") { + valueText = KOREADER_STORE.getPassword().empty() ? "[Not Set]" : "[Set]"; + } } const auto width = renderer.getTextWidth(UI_10_FONT_ID, valueText.c_str()); - renderer.drawText(UI_10_FONT_ID, pageWidth - 20 - width, settingY, valueText.c_str(), i != selectedSettingIndex); + renderer.drawText(UI_10_FONT_ID, pageWidth - 20 - width, settingY, valueText.c_str(), !isSelected); + + visibleIndex++; } // Draw version text above button hints diff --git a/src/main.cpp b/src/main.cpp index e81448bd..73de236c 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -10,6 +10,7 @@ #include "Battery.h" #include "CrossPointSettings.h" #include "CrossPointState.h" +#include "KOReaderCredentialStore.h" #include "MappedInputManager.h" #include "activities/boot_sleep/BootActivity.h" #include "activities/boot_sleep/SleepActivity.h" @@ -276,6 +277,7 @@ void setup() { } SETTINGS.loadFromFile(); + KOREADER_STORE.loadFromFile(); // verify power button press duration after we've read settings. verifyWakeupLongPress();