From f6e14f4b2b15b0ace8ed312252ae107f139bd33d Mon Sep 17 00:00:00 2001
From: nic-gaffney <gaffney_nic@protonmail.com>
Date: Fri, 14 Apr 2023 03:20:23 -0500
Subject: Terminal and file stuff

---
 crates/sloth_vm/src/sloth_std/stdio.rs | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

(limited to 'crates/sloth_vm/src/sloth_std/stdio.rs')

diff --git a/crates/sloth_vm/src/sloth_std/stdio.rs b/crates/sloth_vm/src/sloth_std/stdio.rs
index a743ad1..75029bd 100644
--- a/crates/sloth_vm/src/sloth_std/stdio.rs
+++ b/crates/sloth_vm/src/sloth_std/stdio.rs
@@ -18,7 +18,7 @@ fn write(vm: &mut VM, args: &[Primitive]) -> NativeFunctionResult {
         return Err(native::Error::InvalidArgument);
     };
 
-    println!("{str}");
+    print!("{str}");
 
     Ok(Primitive::Empty)
 }
@@ -30,6 +30,32 @@ pub const WRITE_FUNCTION: NativeFunction = NativeFunction {
     returns_value: false,
 };
 
+fn writeln(vm: &mut VM, args: &[Primitive]) -> NativeFunctionResult {
+    let Some(Primitive::Object(ptr)) = args.get(0).cloned() else {
+        return Err(native::Error::InvalidArgument);
+    };
+
+    let object = vm
+        .objects()
+        .get(ptr as usize)
+        .ok_or(native::Error::InvalidArgument)?;
+
+    let ObjectType::String(str) = &object.typ else {
+        return Err(native::Error::InvalidArgument);
+    };
+
+    println!("{str}");
+
+    Ok(Primitive::Empty)
+}
+
+pub const WRITELN_FUNCTION: NativeFunction = NativeFunction {
+    name: "writeln",
+    function: writeln,
+    arity: 1,
+    returns_value: false,
+};
+
 fn read(vm: &mut VM, _args: &[Primitive]) -> NativeFunctionResult {
     let mut line = String::new();
     stdin()
-- 
cgit v1.2.3