barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 13

feat: format

Browse source
This commit is contained in:
Barrett Ruth 2025-09-15 22:40:16 -04:00
parent d5b3c9a881
commit 0951944e8c
2 changed files with 18 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

21
lua/cp/scrape.lua
View file

@ -163,7 +163,7 @@ function M.scrape_problem(ctx)
table.insert(test_cases, {
index = i,
input = input_content,
output = expected_content
output = expected_content,
})
i = i + 1
else
@ -228,7 +228,6 @@ function M.scrape_problem(ctx)
timeout = 30000,
}):wait()
if result.code ~= 0 then
return {
success = false,
@ -246,7 +245,6 @@ function M.scrape_problem(ctx)
}
end
if not data.success then
return data
end
@ -269,8 +267,20 @@ function M.scrape_problem(ctx)
vim.fn.writefile(vim.split(expected_content, "\n", true), expected_file)
end
local combined_input = data.combined and data.combined.input:gsub("\r", "") or table.concat(vim.tbl_map(function(tc) return tc.input end, data.test_cases), "\n")
local combined_output = data.combined and data.combined.output:gsub("\r", "") or table.concat(vim.tbl_map(function(tc) return tc.output end, data.test_cases), "\n")
local combined_input = data.combined and data.combined.input:gsub("\r", "")
or table.concat(
vim.tbl_map(function(tc)
return tc.input
end, data.test_cases),
"\n"
)
local combined_output = data.combined and data.combined.output:gsub("\r", "")
or table.concat(
vim.tbl_map(function(tc)
return tc.output
end, data.test_cases),
"\n"
)
if ctx.contest == "atcoder" then
combined_input = tostring(#data.test_cases) .. "\n" .. combined_input
@ -280,7 +290,6 @@ function M.scrape_problem(ctx)
vim.fn.writefile(vim.split(combined_output, "\n", true), ctx.expected_file)
end
return {
success = true,
problem_id = ctx.problem_name,