From 8d3f75242915821611ea9ef0b9f71173d5f8d12b Mon Sep 17 00:00:00 2001
From: ValKmjolnir <lhk101lhk101@qq.com>
Date: Fri, 2 Aug 2024 22:34:10 +0800
Subject: [PATCH] :sparkles: add native terminal_size

---
 src/nasal_codegen.h     |  2 +-
 src/natives/builtin.cpp | 23 +++++++++++++++++++++++
 src/natives/builtin.h   |  1 +
 std/utils.nas           |  4 ++++
 4 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/src/nasal_codegen.h b/src/nasal_codegen.h
index 938cdb8..9a9d749 100644
--- a/src/nasal_codegen.h
+++ b/src/nasal_codegen.h
@@ -46,7 +46,7 @@ private:
     // under limited mode, unsafe system api will be banned
     const std::unordered_set<std::string> unsafe_system_api = {
         // builtin
-        "__system", "__input",
+        "__system", "__input", "__terminal_size",
         // io
         "__fout", "__open", "__write", "__stat"
         // bits
diff --git a/src/natives/builtin.cpp b/src/natives/builtin.cpp
index 4291442..8a4e151 100644
--- a/src/natives/builtin.cpp
+++ b/src/natives/builtin.cpp
@@ -5,6 +5,9 @@
 
 #ifdef _WIN32
 #include <windows.h>
+#else
+#include <sys/ioctl.h>
+#include <unistd.h>
 #endif
 
 namespace nasal {
@@ -786,6 +789,25 @@ var builtin_set_utf8_output(context* ctx, gc* ngc) {
     return nil;
 }
 
+var builtin_terminal_size(context* ctx, gc* ngc) {
+    var res = ngc->alloc(vm_type::vm_hash);
+#ifdef _WIN32
+    CONSOLE_SCREEN_BUFFER_INFO csbi;
+    if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &csbi)) {
+        auto rows = csbi.srWindow.Bottom - csbi.srWindow.Top + 1;
+        auto cols = csbi.srWindow.Right - csbi.srWindow.Left + 1;
+        res.hash().elems["rows"] = var::num(rows);
+        res.hash().elems["cols"] = var::num(cols);
+    }
+#else
+    struct winsize w;
+    ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
+    res.hash().elems["rows"] = var::num(w.ws_row);
+    res.hash().elems["cols"] = var::num(w.ws_col);
+#endif
+    return res;
+}
+
 nasal_builtin_table builtin[] = {
     {"__print", builtin_print},
     {"__println", builtin_println},
@@ -835,6 +857,7 @@ nasal_builtin_table builtin[] = {
     {"__logtime", builtin_logtime},
     {"__ghosttype", builtin_ghosttype},
     {"__set_utf8_output", builtin_set_utf8_output},
+    {"__terminal_size", builtin_terminal_size},
     {nullptr, nullptr}
 };
 
diff --git a/src/natives/builtin.h b/src/natives/builtin.h
index b5a16f9..ce63e08 100644
--- a/src/natives/builtin.h
+++ b/src/natives/builtin.h
@@ -85,6 +85,7 @@ var builtin_ghosttype(context*, gc*);
 
 // only useful on windows platform
 var builtin_set_utf8_output(context*, gc*);
+var builtin_terminal_size(context*, gc*);
 
 // register builtin function's name and it's address here in this table below
 // this table must end with {nullptr, nullptr}
diff --git a/std/utils.nas b/std/utils.nas
index e309483..182b937 100644
--- a/std/utils.nas
+++ b/std/utils.nas
@@ -5,4 +5,8 @@ use std.math;
 # when count can be divided exactly by times, return true
 var times_trigger = func(times, count) {
     return math.mod(times, count)==0;
+}
+
+var terminal_size = func {
+    return __terminal_size;
 }
\ No newline at end of file