|
| 1 | +local path = require("plenary.path") |
| 2 | +local companies_api = require("leetcode.api.companies") |
| 3 | + |
| 4 | +local log = require("leetcode.logger") |
| 5 | +local config = require("leetcode.config") |
| 6 | +local interval = config.user.cache.update_interval |
| 7 | + |
| 8 | +---@type Path |
| 9 | +local file = config.storage.cache:joinpath(("companylist%s"):format(config.is_cn and "_cn" or "")) |
| 10 | + |
| 11 | +---@type { at: integer, payload: lc.cache.payload } |
| 12 | +local hist = nil |
| 13 | + |
| 14 | +---@class lc.cache.Company |
| 15 | +---@field name string |
| 16 | +---@field slug string |
| 17 | +---@field questionCount number |
| 18 | + |
| 19 | + |
| 20 | +---@class lc.cache.Copmanylist |
| 21 | +local Companylist = {} |
| 22 | + |
| 23 | +---@return lc.cache.Company[] |
| 24 | +function Companylist.get() |
| 25 | + return Companylist.read().data |
| 26 | +end |
| 27 | + |
| 28 | +---@return lc.cache.payload |
| 29 | +function Companylist.read() |
| 30 | + if not file:exists() then |
| 31 | + return Companylist.populate() |
| 32 | + end |
| 33 | + |
| 34 | + local time = os.time() |
| 35 | + if hist and (time - hist.at) <= math.min(60, interval) then |
| 36 | + return hist.payload |
| 37 | + end |
| 38 | + |
| 39 | + local contents = file:read() |
| 40 | + if not contents or type(contents) ~= "string" then |
| 41 | + return Companylist.populate() |
| 42 | + end |
| 43 | + |
| 44 | + local cached = Companylist.parse(contents) |
| 45 | + |
| 46 | + if not cached or (cached.version ~= config.version or cached.username ~= config.auth.name) then |
| 47 | + return Companylist.populate() |
| 48 | + end |
| 49 | + |
| 50 | + hist = { at = time, payload = cached } |
| 51 | + if (time - cached.updated_at) > interval then |
| 52 | + Companylist.update() |
| 53 | + end |
| 54 | + |
| 55 | + return cached |
| 56 | +end |
| 57 | + |
| 58 | +---@return lc.cache.payload |
| 59 | +function Companylist.populate() |
| 60 | + local res, err = companies_api.all(nil, true) |
| 61 | + |
| 62 | + if not res or err then |
| 63 | + local msg = (err or {}).msg or "failed to fetch company list" |
| 64 | + error(msg) |
| 65 | + end |
| 66 | + |
| 67 | + Companylist.write({ data = res }) |
| 68 | + return hist.payload |
| 69 | +end |
| 70 | + |
| 71 | +function Companylist.update() |
| 72 | + companies_api.all(function(res, err) |
| 73 | + if not err then |
| 74 | + Companylist.write({ data = res }) |
| 75 | + end |
| 76 | + end, true) |
| 77 | +end |
| 78 | + |
| 79 | +---@return lc.cache.Company |
| 80 | +function Companylist.get_by_title_slug(title_slug) |
| 81 | + local companies = Companylist.get() |
| 82 | + |
| 83 | + local company = vim.tbl_filter(function(e) |
| 84 | + return e.title_slug == slug |
| 85 | + end, companies)[1] |
| 86 | + |
| 87 | + assert(company("Company `%s` not found. Try updating cache?"):format(title_slug)) |
| 88 | + return company |
| 89 | +end |
| 90 | + |
| 91 | +---@param payload? lc.cache.payload |
| 92 | +function Companylist.write(payload) |
| 93 | + payload = vim.tbl_deep_extend("force", { |
| 94 | + version = config.version, |
| 95 | + updated_at = os.time(), |
| 96 | + username = config.auth.name, |
| 97 | + }, payload) |
| 98 | + |
| 99 | + if not payload.data then |
| 100 | + payload.data = Companylist.get() |
| 101 | + end |
| 102 | + |
| 103 | + file:write(vim.json.encode(payload), "w") |
| 104 | + hist = { at = os.time(), payload = payload } |
| 105 | +end |
| 106 | + |
| 107 | +---@alias lc.cache.payload { version: string, data: lc.cache.Company[], updated_at: integer, username: string } |
| 108 | + |
| 109 | +---@param str string |
| 110 | +--- |
| 111 | +---@return lc.cache.payload |
| 112 | +function Companylist.parse(str) |
| 113 | + return vim.json.decode(str) |
| 114 | +end |
| 115 | + |
| 116 | +function Companylist.delete() |
| 117 | + if not file:exists() then |
| 118 | + return false |
| 119 | + end |
| 120 | + return pcall(path.rm, file) |
| 121 | +end |
| 122 | + |
| 123 | +return Companylist |
0 commit comments