From 03807e46e054337bcb54726600598f46264efbb7 Mon Sep 17 00:00:00 2001 From: Barrett Ruth Date: Fri, 12 Sep 2025 17:20:38 -0500 Subject: [PATCH] fix formatting; --- lua/cp/init.lua | 3 +-- lua/cp/scrape.lua | 8 ++++---- lua/cp/window.lua | 12 ++++++------ scrapers/atcoder.py | 6 ++++-- 4 files changed, 15 insertions(+), 14 deletions(-) diff --git a/lua/cp/init.lua b/lua/cp/init.lua index 55680b3..e8ac6a9 100644 --- a/lua/cp/init.lua +++ b/lua/cp/init.lua @@ -21,7 +21,6 @@ local function get_plugin_path() return vim.fn.fnamemodify(plugin_path, ":h:h:h") end - local competition_types = { "atcoder", "codeforces", "cses" } local function setup_contest(contest_type) @@ -58,7 +57,7 @@ local function setup_problem(problem_id, problem_letter) vim.g.cp_diff_mode = false end - vim.cmd('silent only') + vim.cmd("silent only") local scrape_result = scrape.scrape_problem(vim.g.cp_contest, problem_id, problem_letter) diff --git a/lua/cp/scrape.lua b/lua/cp/scrape.lua index 656ebd8..be62e2c 100644 --- a/lua/cp/scrape.lua +++ b/lua/cp/scrape.lua @@ -90,20 +90,20 @@ function M.scrape_problem(contest, problem_id, problem_letter) if #data.test_cases > 0 then local all_inputs = {} local all_outputs = {} - + for i, test_case in ipairs(data.test_cases) do local input_lines = vim.split(test_case.input:gsub("\r", ""):gsub("\n+$", ""), "\n") local output_lines = vim.split(test_case.output:gsub("\r", ""):gsub("\n+$", ""), "\n") - + for _, line in ipairs(input_lines) do table.insert(all_inputs, line) end - + for _, line in ipairs(output_lines) do table.insert(all_outputs, line) end end - + vim.fn.writefile(all_inputs, input_file) vim.fn.writefile(all_outputs, expected_file) end diff --git a/lua/cp/window.lua b/lua/cp/window.lua index df3e4f2..e4d7b6d 100644 --- a/lua/cp/window.lua +++ b/lua/cp/window.lua @@ -35,7 +35,7 @@ function M.restore_layout(state) end vim.cmd.diffoff() - + local problem_id = vim.fn.expand("%:t:r") if problem_id == "" then for win, win_state in pairs(state.windows) do @@ -50,13 +50,13 @@ function M.restore_layout(state) end if problem_id ~= "" then - vim.cmd('silent only') - + vim.cmd("silent only") + local base_fp = vim.fn.getcwd() local input = ("%s/io/%s.in"):format(base_fp, problem_id) local output = ("%s/io/%s.out"):format(base_fp, problem_id) local source = problem_id .. ".cc" - + vim.cmd.edit(source) vim.cmd.vsplit(output) vim.bo.filetype = "cp" @@ -68,7 +68,7 @@ function M.restore_layout(state) vim.cmd.wincmd("h") else vim.cmd(state.layout) - + for win, win_state in pairs(state.windows) do if vim.api.nvim_win_is_valid(win) then vim.api.nvim_set_current_win(win) @@ -86,7 +86,7 @@ end function M.setup_diff_layout(actual_output, expected_output, input_file) vim.cmd.diffoff() - vim.cmd('silent only') + vim.cmd("silent only") local output_lines = vim.split(actual_output, "\n") local output_buf = vim.api.nvim_create_buf(false, true) diff --git a/scrapers/atcoder.py b/scrapers/atcoder.py index 9eff826..1f84f2e 100644 --- a/scrapers/atcoder.py +++ b/scrapers/atcoder.py @@ -88,9 +88,11 @@ def main(): test_cases = [] for input_data, output_data in tests: test_cases.append({"input": input_data, "output": output_data}) - + if test_cases: - combined_input = str(len(test_cases)) + "\n" + "\n".join(tc["input"] for tc in test_cases) + combined_input = ( + str(len(test_cases)) + "\n" + "\n".join(tc["input"] for tc in test_cases) + ) combined_output = "\n".join(tc["output"] for tc in test_cases) test_cases = [{"input": combined_input, "output": combined_output}]