138 lines
3.4 KiB
Lua
138 lines
3.4 KiB
Lua
local Job = require("plenary.job")
|
||
local config = require("memos").config
|
||
|
||
local M = {}
|
||
|
||
-- 【新增】URL 编码辅助函数
|
||
local function url_encode(str)
|
||
if str then
|
||
str = string.gsub(str, "\n", "\r\n")
|
||
str = string.gsub(str, "([^%w %-%_%.%~])", function(c)
|
||
return string.format("%%%02X", string.byte(c))
|
||
end)
|
||
str = string.gsub(str, " ", "%%20")
|
||
end
|
||
return str
|
||
end
|
||
|
||
local function run_curl(args, on_exit)
|
||
table.insert(args, "-H")
|
||
table.insert(args, "Authorization: Bearer " .. config.token)
|
||
Job:new({ command = "curl", args = args, on_exit = on_exit }):start()
|
||
end
|
||
|
||
function M.get_current_user(callback)
|
||
run_curl({
|
||
"-s",
|
||
"--fail",
|
||
"-X",
|
||
"GET",
|
||
config.host .. "/api/v1/auth/me",
|
||
}, function(job, return_val)
|
||
if return_val == 0 then
|
||
local result_string = table.concat(job:result(), "")
|
||
local data = vim.json.decode(result_string)
|
||
callback(data and data.user or nil)
|
||
else
|
||
vim.schedule(function()
|
||
vim.notify("Failed to get user info.", vim.log.levels.ERROR)
|
||
end)
|
||
callback(nil)
|
||
end
|
||
end)
|
||
end
|
||
|
||
function M.list_memos(parent, filter, page_size, pageToken, callback)
|
||
local list_url = config.host .. "/api/v1/memos"
|
||
local params = {}
|
||
table.insert(params, "parent=" .. parent)
|
||
table.insert(params, "page_size=" .. tostring(page_size))
|
||
if pageToken and pageToken ~= "" then
|
||
table.insert(params, "pageToken=" .. pageToken)
|
||
end
|
||
if filter and filter ~= "" then
|
||
-- 1. 构造原始的过滤字符串，例如：content.contains("hello world")
|
||
local raw_filter = 'content.contains("' .. vim.fn.escape(filter, '"') .. '")'
|
||
-- 2. 对整个值进行 URL 编码，变成：content.contains%28%22hello%20world%22%29
|
||
table.insert(params, "filter=" .. url_encode(raw_filter))
|
||
end
|
||
list_url = list_url .. "?" .. table.concat(params, "&")
|
||
|
||
run_curl({
|
||
"-s",
|
||
"--fail",
|
||
"-X",
|
||
"GET",
|
||
list_url,
|
||
}, function(job, return_val)
|
||
if return_val == 0 then
|
||
local result_string = table.concat(job:result(), "")
|
||
local data = vim.json.decode(result_string)
|
||
callback({ memos = data.memos or {}, nextPageToken = data.nextPageToken or "" })
|
||
else
|
||
vim.schedule(function()
|
||
vim.notify("Failed to fetch memos.", vim.log.levels.ERROR)
|
||
end)
|
||
callback(nil)
|
||
end
|
||
end)
|
||
end
|
||
|
||
-- 【修改】让 create_memo 的回调函数返回新创建的 memo 对象
|
||
function M.create_memo(content, callback)
|
||
local create_url = config.host .. "/api/v1/memos"
|
||
local json_data = vim.json.encode({ content = content })
|
||
run_curl({
|
||
"-s",
|
||
"--fail",
|
||
"-X",
|
||
"POST",
|
||
create_url,
|
||
"-H",
|
||
"Content-Type: application/json",
|
||
"--data",
|
||
json_data,
|
||
}, function(job, return_val)
|
||
if return_val == 0 then
|
||
local result_string = table.concat(job:result(), "")
|
||
local new_memo = vim.json.decode(result_string)
|
||
callback(new_memo)
|
||
else
|
||
callback(nil)
|
||
end
|
||
end)
|
||
end
|
||
|
||
function M.update_memo(memo_name, content, callback)
|
||
local update_url = config.host .. "/api/v1/" .. memo_name
|
||
local json_data = vim.json.encode({ content = content })
|
||
run_curl({
|
||
"-s",
|
||
"--fail",
|
||
"-X",
|
||
"PATCH",
|
||
update_url,
|
||
"-H",
|
||
"Content-Type: application/json",
|
||
"--data",
|
||
json_data,
|
||
}, function(job, return_val)
|
||
callback(return_val == 0)
|
||
end)
|
||
end
|
||
|
||
function M.delete_memo(memo_name, callback)
|
||
local delete_url = config.host .. "/api/v1/" .. memo_name
|
||
run_curl({
|
||
"-s",
|
||
"--fail",
|
||
"-X",
|
||
"DELETE",
|
||
delete_url,
|
||
}, function(job, return_val)
|
||
callback(return_val == 0)
|
||
end)
|
||
end
|
||
|
||
return M
|