Add Shell dsl type

3 files changed, 90 insertions, 0 deletions

diff --git a/builtins/shell.c b/builtins/shell.c
new file mode 100644
index 00000000..da6cbe5b
--- /dev/null
+++ b/builtins/shell.c
@@ -0,0 +1,67 @@
+// Boolean methods/type info
+#include <stdbool.h>
+#include <stdint.h>
+
+#include "array.h"
+#include "functions.h"
+#include "integers.h"
+#include "shell.h"
+#include "text.h"
+#include "types.h"
+#include "util.h"
+
+public Pattern_t Shell$escape_text(Text_t text)
+{
+    // TODO: optimize for ASCII and short strings
+    Array_t shell_graphemes = {.atomic=1};
+#define add_char(c) Array$insert(&shell_graphemes, (uint32_t[1]){c}, I_small(0), sizeof(uint32_t))
+    add_char('\'');
+    const char *text_utf8 = Text$as_c_string(text);
+    for (const char *p = text_utf8; *p; p++) {
+        if (*p == '\'') {
+            add_char('\'');
+            add_char('"');
+            add_char('\'');
+            add_char('"');
+            add_char('\'');
+        } else
+            add_char(*p);
+    }
+    add_char('\'');
+#undef add_char
+    return (Text_t){.length=shell_graphemes.length, .tag=TEXT_GRAPHEMES, .graphemes=shell_graphemes.data};
+}
+
+public Text_t Shell$run(Shell_t command, int32_t *status)
+{
+    const char *cmd_str = Text$as_c_string(command);
+    FILE *prog = popen(cmd_str, "r");
+
+    const int chunk_size = 256;
+    char *buf = GC_MALLOC_ATOMIC(chunk_size);
+    Text_t output = Text("");
+    size_t just_read;
+    do {
+        just_read = fread(buf, sizeof(char), chunk_size, prog);
+        if (just_read > 0) {
+            output = Texts(output, Text$from_strn(buf, just_read));
+            buf = GC_MALLOC_ATOMIC(chunk_size);
+        }
+    } while (just_read > 0);
+
+    if (status)
+        *status = WEXITSTATUS(pclose(prog));
+    else
+        pclose(prog);
+
+    return Text$trim(output, Pattern("{1 nl}"), false, true);
+}
+
+public const TypeInfo Shell$info = {
+    .size=sizeof(Shell_t),
+    .align=__alignof__(Shell_t),
+    .tag=TextInfo,
+    .TextInfo={.lang="Shell"},
+};
+
+// vim: ts=4 sw=0 et cino=L2,l1,(0,W4,m1,\:0
diff --git a/builtins/shell.h b/builtins/shell.h
new file mode 100644
index 00000000..0429d496
--- /dev/null
+++ b/builtins/shell.h
@@ -0,0 +1,22 @@
+#pragma once
+
+// A lang for Shell Command Language
+
+#include <gc/cord.h>
+#include <stdbool.h>
+#include <stdint.h>
+
+#include "types.h"
+#include "datatypes.h"
+
+#define Shell_t Text_t
+#define Shell(text) ((Shell_t)Text(text))
+#define Shells(...) ((Shell_t)Texts(__VA_ARGS__))
+
+Text_t Shell$run(Shell_t command, int32_t *status);
+Shell_t Shell$escape_text(Text_t text);
+
+extern const TypeInfo Shell$info;
+
+// vim: ts=4 sw=0 et cino=L2,l1,(0,W4,m1,\:0
+
diff --git a/builtins/tomo.h b/builtins/tomo.h
index 8d6661ca..68e115bb 100644
--- a/builtins/tomo.h
+++ b/builtins/tomo.h
@@ -22,6 +22,7 @@
 #include "nums.h"
 #include "pointer.h"
 #include "range.h"
+#include "shell.h"
 #include "siphash.h"
 #include "table.h"
 #include "text.h"