[truffle] Implement nqp::getstdin

Author: MasterDuke17

src/vm/jvm/Truffle.nqp

@@ -172,6 +172,7 @@ class QAST::OperationsTruffle {
 
     add_simple_op('getstdout', $OBJ, [], :side_effects);
     add_simple_op('getstderr', $OBJ, [], :side_effects);
+    add_simple_op('getstdin', $OBJ, [], :side_effects);
 
     add_simple_op('say', $STR, [$STR], :side_effects);
     add_simple_op('print', $STR, [$STR], :side_effects);

src/vm/jvm/runtime/org/perl6/nqp/truffle/io/IIOInteractive.java

@@ -0,0 +1,7 @@
+package org.perl6.nqp.truffle.io;
+
+import org.perl6.nqp.runtime.ThreadContext;
+
+public interface IIOInteractive {
+    public String readlineInteractive(ThreadContext tc, String prompt);
+}

src/vm/jvm/runtime/org/perl6/nqp/truffle/io/IIOSyncReadable.java

@@ -0,0 +1,11 @@
+package org.perl6.nqp.truffle.io;
+
+import org.perl6.nqp.runtime.ThreadContext;
+
+public interface IIOSyncReadable {
+    public String slurp();
+    public String readline();
+    public String readchars(int chars);
+    public byte[] read(int bytes);
+    public boolean eof();
+}

src/vm/jvm/runtime/org/perl6/nqp/truffle/io/StandardReadHandle.java

@@ -0,0 +1,132 @@
+package org.perl6.nqp.truffle.io;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.OutputStreamWriter;
+import java.nio.charset.Charset;
+
+import jline.ConsoleReader;
+
+import org.perl6.nqp.runtime.ExceptionHandling;
+import org.perl6.nqp.runtime.ThreadContext;
+
+public class StandardReadHandle implements IIOClosable, IIOEncodable, IIOSyncReadable, IIOInteractive, IIOPossiblyTTY {
+    private InputStream is;
+    private BufferedReader br;
+    private ConsoleReader cr;
+    private boolean eof = false;
+    private Charset cs;
+
+    public StandardReadHandle(InputStream is) {
+        this.is = is;
+        setEncoding(Charset.forName("UTF-8"));
+    }
+
+    public void close() {
+        try {
+            is.close();
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public void setEncoding(Charset cs) {
+        this.cs = cs;
+    }
+
+    public byte[] read(int bytes) {
+        try {
+            byte[] array = new byte[bytes];
+            int read = 0;
+            int offset = 0;
+            while (offset < bytes) {
+                if ((read = is.read(array, offset, bytes - offset)) == -1) {
+                    eof = true;
+                    break;
+                }
+                offset += read;
+            }
+            byte[] compact = new byte[offset];
+            System.arraycopy(array, 0, compact, 0, offset);
+            return compact;
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public synchronized String slurp() {
+        try {
+            if (br == null)
+                br = new BufferedReader(new InputStreamReader(is, cs));
+            StringBuffer data = new StringBuffer();
+            char[] buf = new char[4096];
+            int read = 0;
+            while((read = br.read(buf)) != -1)
+                data.append(String.valueOf(buf, 0, read));
+            eof = true;
+            return data.toString();
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public synchronized String readline() {
+        try {
+            if (br == null)
+                br = new BufferedReader(new InputStreamReader(is, cs));
+            String line = br.readLine();
+            if (line == null) {
+                eof = true;
+                line = "";
+            }
+            return line;
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    /* TODO - think about unicode in readchars and getc */
+    public synchronized String readchars(int count) {
+        try {
+            if (br == null)
+                br = new BufferedReader(new InputStreamReader(is, cs));
+            char[] chars = new char[count];
+
+            int actuallyRead = br.read(chars, 0, count);
+
+            if (actuallyRead == -1) {
+                return "";
+            }
+            else {
+                return new String(chars, 0, actuallyRead);
+            }
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public synchronized String readlineInteractive(ThreadContext tc, String prompt) {
+        try {
+            if (cr == null)
+                cr = new ConsoleReader(is, new OutputStreamWriter(tc.gc.out));
+            String line = cr.readLine(prompt);
+            if (line == null) {
+                eof = true;
+                line = "";
+            }
+            return line;
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public boolean eof() {
+        return eof;
+    }
+
+    public boolean isTTY() {
+        return System.console() != null;
+    }
+}

src/vm/jvm/runtime/org/perl6/nqp/truffle/nodes/io/NQPGetstdinNode.java

@@ -0,0 +1,29 @@
+package org.perl6.nqp.truffle.nodes.io;
+import com.oracle.truffle.api.frame.VirtualFrame;
+import com.oracle.truffle.api.nodes.NodeInfo;
+import org.perl6.nqp.truffle.nodes.NQPNode;
+import org.perl6.nqp.truffle.nodes.NQPObjNode;
+import org.perl6.nqp.dsl.Deserializer;
+import org.perl6.nqp.truffle.runtime.HLL;
+import org.perl6.nqp.truffle.GlobalContext;
+import org.perl6.nqp.truffle.io.StandardReadHandle;
+import org.perl6.nqp.truffle.sixmodel.reprs.IOHandleInstance;
+
+@NodeInfo(shortName = "getstdin")
+public final class NQPGetstdinNode extends NQPObjNode {
+    HLL hll;
+    GlobalContext globalContext;
+
+    @Deserializer
+    public NQPGetstdinNode(HLL hll, GlobalContext globalContext) {
+        this.hll = hll;
+        this.globalContext = globalContext;
+    }
+
+    @Override
+    public Object execute(VirtualFrame frame) {
+        IOHandleInstance handle = (IOHandleInstance) hll.ioType.stable.repr.allocate();
+        handle.handle = new StandardReadHandle(globalContext.in);
+        return handle;
+    }
+}