//
//  WRPreloadBookManager.h
//  WeRead (微信读书)
//  Reverse-engineered header
//
//  Preload manager. Predicts which chapters to preload based on book ranking
//  and user behavior. Manages encryption keys for preloaded content.
//  Supports whole-book preloading and incremental chapter preloading.
//

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

@class WRBook;

// ---------------------------------------------------------------------------
// Preload scene types
// ---------------------------------------------------------------------------
typedef NS_ENUM(NSInteger, WRPreloadScene) {
    WRPreloadSceneNone        = 0,
    WRPreloadSceneShelf       = 1,  // Preload from bookshelf
    WRPreloadSceneReading     = 2,  // Preload next chapters while reading
    WRPreloadSceneWiFi        = 3,  // Aggressive preload on WiFi
    WRPreloadSceneManual      = 4,  // User-initiated preload
};

// ---------------------------------------------------------------------------
// WRPreloadBookManager
// ---------------------------------------------------------------------------
@interface WRPreloadBookManager : NSObject

// --- Ivars (from binary analysis) ---
// {
//     NSMutableDictionary *_preloadState;   // bookId -> preload state dict
// }

@property (nonatomic, strong, readonly) NSMutableDictionary *preloadState;

#pragma mark - Class Methods: Encryption Key Storage

/// Save an encryption key for a preloaded book.
/// @param key    The encryption key data (hex or raw).
/// @param path   The file path where the key is associated.
/// @param bookId The book identifier.
+ (void)saveEncryptKey:(NSString *)key
               forPath:(NSString *)path
                bookId:(NSString *)bookId;

/// Retrieve the stored encryption key for a book.
/// @param path   The associated file path.
/// @param bookId The book identifier.
/// @return The encryption key string, or nil.
+ (nullable NSString *)encryptKeyForPath:(NSString *)path
                                  bookId:(NSString *)bookId;

/// Remove a stored encryption key.
+ (void)removeEncryptKeyForPath:(NSString *)path
                         bookId:(NSString *)bookId;

#pragma mark - Class Methods: Filename Dictionary

/// Save a filename mapping dictionary for a book.
/// Maps content keys to actual filenames in the tar archive.
+ (void)saveFileNameDict:(NSDictionary<NSString *, NSString *> *)dict
                  bookId:(NSString *)bookId;

/// Look up a filename by key for a given book.
+ (nullable NSString *)fileNameForKey:(NSString *)key
                               bookId:(NSString *)bookId;

/// Remove a filename mapping entry.
+ (void)removeFileNameForKey:(NSString *)key
                      bookId:(NSString *)bookId;

#pragma mark - Class Methods: Cache Management

/// Clear all preload key-value caches.
+ (void)clearKV;

/// Remove all preload data for a specific book.
+ (void)removeKVWithBookId:(NSString *)bookId;

#pragma mark - Instance Methods: Preloading

/// Preload an entire book (all chapters).
/// @param book  The book model to preload.
/// @param scene The preload context/scene.
- (void)_preloadWholeBook:(WRBook *)book scene:(WRPreloadScene)scene;

/// Preload chapters starting from a given index.
/// @param book         The book model.
/// @param fromChapterIdx Starting chapter index.
- (void)preloadWithBook:(WRBook *)book fromChapterIdx:(NSInteger)fromChapterIdx;

/// Download specific chapters for a book.
/// @param book    The book model.
/// @param uids    Array of chapter UIDs to download.
- (void)downloadBook:(WRBook *)book uids:(NSArray<NSString *> *)uids;

/// Clean up all preloaded book data.
- (void)cleanUpPreloadBook;

/// Calculate preload cache size and optionally clear it.
/// @param completion Called with the total size in bytes.
/// @param onlyCalc   YES to only calculate, NO to also clear.
- (void)calcAndClearPreloadBookWithCompletion:(void (^)(NSUInteger totalSize))completion
                                      onlyCalc:(BOOL)onlyCalc;

#pragma mark - Additional Methods (23 total, reconstructed)

/// Check if a book is currently being preloaded.
- (BOOL)isPreloadingBookId:(NSString *)bookId;

/// Get the preload progress for a book (0.0 - 1.0).
- (float)preloadProgressForBookId:(NSString *)bookId;

/// Cancel an ongoing preload for a book.
- (void)cancelPreloadForBookId:(NSString *)bookId;

/// Determine which chapters should be preloaded next.
- (NSArray<NSString *> *)predictedChapterUidsForBook:(WRBook *)book;

@end

NS_ASSUME_NONNULL_END