Basic integration testing; uses struct instead of asssociativeArray; ensure easyinstall is usable by pip/setup.py

1 files changed, 96 insertions, 0 deletions

diff --git a/source/util.d b/source/util.d
new file mode 100644
index 0000000..d49f9d7
--- /dev/null
+++ b/source/util.d
@@ -0,0 +1,96 @@
+module util;
+import std.stdio;
+import std.string;
+import std.process;
+
+
+enum byte MAXCOLS = 80;
+
+
+void banner(const char ch, string s) {
+    string ruler;
+    byte i = 0;
+    while (i < MAXCOLS) {
+        ruler ~= ch;
+        i++;
+    }
+
+    string result;
+    string[] tmpstr = splitLines(wrap(s, MAXCOLS - 2, ch ~ " ", ch ~ "    "));
+    foreach (idx, line; tmpstr) {
+        if (idx < tmpstr.length - 1) {
+            line ~= " \\";
+        }
+        result ~= line ~ "\n";
+    }
+
+    writeln(ruler);
+    write(result);
+    writeln(ruler);
+}
+
+
+static auto getenv(string[string] base=null, string preface=null) {
+    const char delim = '=';
+    char delim_line = '\n';
+    string[string] env;
+    string cmd = "env";
+
+    version (linux) {
+        cmd ~= " -0";
+        delim_line = '\0';
+    }
+
+    version (Windows) {
+        cmd = "set";
+        delim_line = "\r\n";
+    }
+
+    // Execute a command before dumping the environment
+    if (preface !is null) {
+        cmd = preface ~ " && " ~ cmd;
+    }
+
+    auto env_sh = executeShell(cmd, env=base);
+    if (env_sh.status) {
+        throw new Exception("Unable to read shell environment:" ~ env_sh.output);
+    }
+
+    foreach (string line; split(env_sh.output, delim_line)) {
+        if (line.empty) {
+            continue;
+        }
+        auto data = split(line, delim);
+
+        // Recombine extra '=' chars
+        if (data.length > 2) {
+           data[1] = join(data[1 .. $], delim);
+        }
+        env[data[0]] = data[1];
+    }
+    return env;
+}
+
+
+string safe_spec(string s) {
+    return "'" ~ s ~ "'";
+}
+
+
+string safe_install(string[] specs) {
+    string[] result;
+    foreach (record; specs) {
+        result ~= safe_spec(record);
+    }
+    return result.join(" ");
+}
+
+
+string safe_install(string specs) {
+    string[] result;
+    foreach (record; specs.split(" ")) {
+        result ~= safe_spec(record);
+    }
+    return result.join(" ");
+}
+