From 3bf1846063ad30ecc0fc90d34490bf70776faef0 Mon Sep 17 00:00:00 2001 From: Karen Arutyunov Date: Sat, 15 Dec 2018 17:23:37 +0300 Subject: Add manifest_rewriter class --- tests/manifest-rewriter/buildfile | 8 ++ tests/manifest-rewriter/driver.cxx | 185 +++++++++++++++++++++++++++++++++++++ 2 files changed, 193 insertions(+) create mode 100644 tests/manifest-rewriter/buildfile create mode 100644 tests/manifest-rewriter/driver.cxx (limited to 'tests/manifest-rewriter') diff --git a/tests/manifest-rewriter/buildfile b/tests/manifest-rewriter/buildfile new file mode 100644 index 0000000..4c6e90f --- /dev/null +++ b/tests/manifest-rewriter/buildfile @@ -0,0 +1,8 @@ +# file : tests/manifest-rewriter/buildfile +# copyright : Copyright (c) 2014-2018 Code Synthesis Ltd +# license : MIT; see accompanying LICENSE file + +import libs = libbutl%lib{butl} +libs += $stdmod_lib + +exe{driver}: {hxx cxx}{*} $libs diff --git a/tests/manifest-rewriter/driver.cxx b/tests/manifest-rewriter/driver.cxx new file mode 100644 index 0000000..f206c22 --- /dev/null +++ b/tests/manifest-rewriter/driver.cxx @@ -0,0 +1,185 @@ +// file : tests/manifest-rewriter/driver.cxx -*- C++ -*- +// copyright : Copyright (c) 2014-2018 Code Synthesis Ltd +// license : MIT; see accompanying LICENSE file + +#include + +#ifndef __cpp_lib_modules +#include +#include +#include // uint64_t +#include // move() +#include +#include +#endif + +// Other includes. + +#ifdef __cpp_modules +#ifdef __cpp_lib_modules +import std.core; +import std.io; +#endif +import butl.path; +import butl.optional; +import butl.fdstream; +import butl.manifest_parser; +import butl.manifest_rewriter; +#else +#include +#include +#include +#include +#include +#endif + +using namespace std; + +namespace butl +{ + using butl::optional; + using butl::nullopt; + + // Value rewriting or insertion command. + // + struct edit_cmd + { + string name; + string value; + optional after; // Rewrite an existing value if nullopt. + + edit_cmd (string n, string v) + : name (move (n)), value (move (v)) {} + + edit_cmd (string n, string v, string a) + : name (move (n)), value (move (v)), after (move (a)) {} + }; + + using edit_cmds = vector; + + // Dump the manifest into the file, edit and return the resulting manifest. + // + // The file will stay in the filesystem for troubleshooting in case of an + // assertion failure and will be deleted otherwise. + // + static path temp_file (path::temp_path ("butl-manifest-rewriter")); + + static string + edit (const char* manifest, const edit_cmds&); + + int + main () + { + auto_rmfile rm (temp_file); + + assert (edit (":1\n# Comment\n# Comment\n a : b \n# Comment\n\nc:d\n", + {{"a", "xyz"}}) == + ":1\n# Comment\n# Comment\n a : xyz\n# Comment\n\nc:d\n"); + + assert (edit (":1\n\n a: b\n", {{"a", "xyz"}}) == ":1\n\n a: xyz\n"); + assert (edit (":1\na: b", {{"a", "xyz"}}) == ":1\na: xyz"); + + assert (edit (":1\na:b\nc:d\ne:f", + {{"a", "xyz"}, edit_cmd {"x", "y", "c"}, {"e", "123"}}) == + ":1\na: xyz\nc:d\nx: y\ne: 123"); + + assert (edit (":1\na: b", {{"a", "xy\nz"}}) == ":1\na: \\\nxy\nz\n\\"); + + assert (edit (":1\n", {{"a", "b", ""}}) == ":1\na: b\n"); + + assert (edit (":1\n abc: b", + {{"abc", "xyz"}}) == + ":1\n abc: \\\nxyz\n\\"); + + // Test editing of manifests that contains CR characters. + // + assert (edit (":1\r\na: b\r\r\n", {{"a", "xyz"}}) == ":1\r\na: xyz\r\r\n"); + + assert (edit (":1\ra: b\r", {{"a", "xyz"}}) == ":1\ra: xyz\r"); + + assert (edit (":1\na: \\s", {{"a", "xyz"}}) == ":1\na: xyz"); + + assert (edit (":1\na: \\\nx\ny\nz\n\\\r", {{"a", "b"}}) == ":1\na: b\r"); + + return 0; + } + + static string + edit (const char* manifest, const edit_cmds& cmds) + { + { + ofdstream os (temp_file); + os << manifest; + os.close (); + } + + struct insertion + { + manifest_name_value value; + optional pos; // Rewrite existing value if nullopt. + }; + + vector insertions; + { + ifdstream is (temp_file); + manifest_parser p (is, temp_file.string ()); + + for (manifest_name_value nv; !(nv = p.next ()).empty (); ) + { + for (const edit_cmd& c: cmds) + { + if (c.after) + { + if (nv.name == *c.after) + { + // Note: new value lines, columns and positions are all zero as + // are not used for an insertion. + // + insertions.push_back ( + insertion { + manifest_name_value {c.name, c.value, 0, 0, 0, 0, 0, 0, 0}, + move (nv)}); + + break; + } + } + else if (nv.name == c.name) + { + nv.value = c.value; + insertions.push_back (insertion {move (nv), nullopt /* pos */}); + break; + } + } + } + } + + { + manifest_rewriter rw (temp_file); + + for (const auto& ins: reverse_iterate (insertions)) + { + if (ins.pos) + rw.insert (*ins.pos, ins.value); + else + rw.replace (ins.value); + } + } + + ifdstream is (temp_file); + return is.read_text (); + } +} + +int +main () +{ + try + { + return butl::main (); + } + catch (const exception& e) + { + cerr << e << endl; + return 1; + } +} -- cgit v1.1