From 509be95a292710231a56ff790b76dc285dfd547e Mon Sep 17 00:00:00 2001
From: blueloveTH <blueloveth@foxmail.com>
Date: Wed, 25 Sep 2024 15:29:14 +0800
Subject: [PATCH] add `input()`

---
 src/public/modules.c | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/src/public/modules.c b/src/public/modules.c
index aa57ba42..a4b1548c 100644
--- a/src/public/modules.c
+++ b/src/public/modules.c
@@ -160,11 +160,6 @@ __SUCCESS:
 
 //////////////////////////
 
-static bool builtins_repr(int argc, py_Ref argv) {
-    PY_CHECK_ARGC(1);
-    return py_repr(argv);
-}
-
 static bool builtins_exit(int argc, py_Ref argv) {
     int code = 0;
     if(argc > 1) return TypeError("exit() takes at most 1 argument");
@@ -181,6 +176,26 @@ static bool builtins_exit(int argc, py_Ref argv) {
     // return py_raise(py_retval());
 }
 
+static bool builtins_input(int argc, py_Ref argv) {
+    if(argc > 1) return TypeError("input() takes at most 1 argument");
+    const char* prompt = "";
+    if(argc == 1) {
+        if(!py_checkstr(argv)) return false;
+        prompt = py_tostr(argv);
+    }
+    printf("%s", prompt);
+    char buf[2048];
+    scanf("%s", buf);
+    getchar();
+    py_newstr(py_retval(), buf);
+    return true;
+}
+
+static bool builtins_repr(int argc, py_Ref argv) {
+    PY_CHECK_ARGC(1);
+    return py_repr(argv);
+}
+
 static bool builtins_len(int argc, py_Ref argv) {
     PY_CHECK_ARGC(1);
     return py_len(argv);
@@ -640,8 +655,9 @@ static bool NotImplementedType__repr__(int argc, py_Ref argv) {
 
 py_TValue pk_builtins__register() {
     py_Ref builtins = py_newmodule("builtins");
-    py_bindfunc(builtins, "repr", builtins_repr);
     py_bindfunc(builtins, "exit", builtins_exit);
+    py_bindfunc(builtins, "input", builtins_input);
+    py_bindfunc(builtins, "repr", builtins_repr);
     py_bindfunc(builtins, "len", builtins_len);
     py_bindfunc(builtins, "hex", builtins_hex);
     py_bindfunc(builtins, "iter", builtins_iter);