koreader/frontend/readcollection.lua

local DataStorage = require("datastorage")
local FFIUtil = require("ffi/util")
local LuaSettings = require("luasettings")
local lfs = require("libs/libkoreader-lfs")
local util = require("util")

local DEFAULT_COLLECTION_NAME = "favorites"
local collection_file = DataStorage:getSettingsDir() .. "/collection.lua"

local ReadCollection = {}

function ReadCollection:read(collection_name)
    if not collection_name then collection_name = DEFAULT_COLLECTION_NAME end
    local collections = LuaSettings:open(collection_file)
    local coll = collections:readSetting(collection_name) or {}
    local coll_max_item = 0
    for _, v in pairs(coll) do
        if v.order > coll_max_item then
            coll_max_item = v.order
        end
    end
    return coll, coll_max_item
end

function ReadCollection:readAllCollection()
    local collection = LuaSettings:open(collection_file)
    if collection and collection.data then
        return collection.data
    else
        return {}
    end
end

function ReadCollection:prepareList(collection_name)
    local data = self:read(collection_name)
    local list = {}
    for _, v in pairs(data) do
        local file_path = FFIUtil.realpath(v.file) or v.file -- keep orig file path of deleted files
        local file_exists = lfs.attributes(file_path, "mode") == "file"
        table.insert(list, {
            order = v.order,
            file = file_path,
            text = v.file:gsub(".*/", ""),
            dim = not file_exists,
            mandatory = file_exists and util.getFriendlySize(lfs.attributes(file_path, "size") or 0) or "",
            select_enabled = file_exists,
        })
    end
    table.sort(list, function(v1,v2)
        return v1.order < v2.order
    end)
    return list
end

function ReadCollection:removeItemByPath(path, is_dir)
    local dir
    local should_write = false
    if is_dir then
        path = path .. "/"
    end
    local coll = self:readAllCollection()
    for i in pairs(coll) do
        local single_collection = coll[i]
        for item = #single_collection, 1, -1 do
            if not is_dir and single_collection[item].file == path then
                should_write = true
                table.remove(single_collection, item)
            elseif is_dir then
                dir = util.splitFilePathName(single_collection[item].file)
                if dir == path then
                    should_write = true
                    table.remove(single_collection, item)
                end
            end
        end
    end
    if should_write then
        local collection = LuaSettings:open(collection_file)
        collection.data = coll
        collection:flush()
    end
end

function ReadCollection:updateItemByPath(old_path, new_path)
    local is_dir = false
    local dir, file
    if lfs.attributes(new_path, "mode") == "directory" then
        is_dir = true
        old_path = old_path .. "/"
    end
    local should_write = false
    local coll = self:readAllCollection()
    for i, j in pairs(coll) do
        for k, v in pairs(j) do
            if not is_dir and v.file == old_path then
                should_write = true
                coll[i][k].file = new_path
            elseif is_dir then
                dir, file = util.splitFilePathName(v.file)
                if dir == old_path then
                    should_write = true
                    coll[i][k].file = string.format("%s/%s", new_path, file)
                end
            end
        end
    end
    if should_write then
        local collection = LuaSettings:open(collection_file)
        collection.data = coll
        collection:flush()
    end
end

function ReadCollection:removeItem(item, collection_name)
    local coll = self:read(collection_name)
    for k, v in pairs(coll) do
        if v.file == item then
            table.remove(coll, k)
            break
        end
    end
    self:writeCollection(coll, collection_name)
end

function ReadCollection:writeCollection(coll_items, collection_name)
    local collection = LuaSettings:open(collection_file)
    collection:saveSetting(collection_name or DEFAULT_COLLECTION_NAME, coll_items)
    collection:flush()
end

function ReadCollection:addItem(file, collection_name)
    local coll, coll_max_item = self:read(collection_name)
    local collection_item = {
        file = file,
        order = coll_max_item + 1,
    }
    table.insert(coll, collection_item)
    self:writeCollection(coll, collection_name)
end

function ReadCollection:checkItemExist(item, collection_name)
    local coll = self:read(collection_name)
    for _, v in pairs(coll) do
        if v.file == item then
            return true
        end
    end
end

return ReadCollection