Skip to content

Commit

Permalink
added io/reader and io/writer function
Browse files Browse the repository at this point in the history
  • Loading branch information
@jlangch
jlangch committed Oct 16, 2023
1 parent df1fbf2 commit f49b7c1
Show file tree
Hide file tree
Showing 2 changed files with 227 additions and 2 deletions.
2 changes: 2 additions & 0 deletions src/main/java/com/github/jlangch/venice/impl/docgen/cheatsheet/section/IoSection.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,8 @@ public DocSection section() {
all.addSection(rd_wr);
rd_wr.addItem(diBuilder.getDocItem("io/buffered-reader"));
rd_wr.addItem(diBuilder.getDocItem("io/buffered-writer"));
rd_wr.addItem(diBuilder.getDocItem("io/reader"));
rd_wr.addItem(diBuilder.getDocItem("io/writer"));
rd_wr.addItem(diBuilder.getDocItem("io/string-reader"));
rd_wr.addItem(diBuilder.getDocItem("io/string-writer"));
rd_wr.addItem(diBuilder.getDocItem("io/flush"));
Expand Down
227 changes: 225 additions & 2 deletions src/main/java/com/github/jlangch/venice/impl/functions/IOFunctionsStreams.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,7 @@
import com.github.jlangch.venice.impl.Printer;
import com.github.jlangch.venice.impl.thread.ThreadContext;
import com.github.jlangch.venice.impl.types.VncBoolean;
import com.github.jlangch.venice.impl.types.VncByteBuffer;
import com.github.jlangch.venice.impl.types.VncChar;
import com.github.jlangch.venice.impl.types.VncFunction;
import com.github.jlangch.venice.impl.types.VncJavaObject;
Expand Down Expand Up @@ -824,6 +825,103 @@ else if (delegate instanceof Writer) {
private static final long serialVersionUID = -1848883965231344442L;
};


public static VncFunction io_writer =
new VncFunction(
"io/writer",
VncFunction
.meta()
.arglists(
"(io/writer f & options)" )
.doc(
"Creates a `java.io.Writer` for f.\n\n" +
"f may be a file or a string (file path). " +
"Options: \n\n" +
"| :append true/false | e.g.: `:append true`, defaults to false |\n" +
"| :encoding enc | e.g.: `:encoding :utf-8`, defaults to :utf-8 |\n\n" +
"`io/writer` supports load paths. See the `loadpath/paths` " +
"doc for a description of the *load path* feature.")
.seeAlso(
"str", "io/string-reader")
.build()
) {
@Override
public VncVal apply(final VncList args) {
ArityExceptions.assertMinArity(this, args, 2);

sandboxFunctionCallValidation();

final ILoadPaths loadpaths = ThreadContext.getInterceptor().getLoadPaths();

final VncHashMap options = VncHashMap.ofAll(args.slice(1));
final VncVal append = options.get(new VncKeyword("append"));
final VncVal encVal = options.get(new VncKeyword("encoding"));
final Charset charset = CharsetUtil.charset(encVal);

final File file = convertToFile(args.first());
if (file != null) {
try {
final OutputStream os = loadpaths.getOutputStream(
file,
StandardOpenOption.CREATE,
StandardOpenOption.WRITE,
VncBoolean.isTrue(append)
? StandardOpenOption.APPEND
: StandardOpenOption.TRUNCATE_EXISTING);
if (os != null) {
return new VncJavaObject(
new BufferedWriter(
new OutputStreamWriter(os, charset)));
}
else {
throw new SecurityException(
String.format(
"Failed to create writer to the file %s. " +
"The load paths configuration prevented this action!\n" +
"Load-Paths: unlimited-access=%b, paths=%s",
file.getPath(),
loadpaths.isUnlimitedAccess(),
loadpaths.getPaths().toString()));
}
}
catch (VncException ex) {
throw ex;
}
catch (Exception ex) {
throw new VncException(
"Failed to create writer to the file " + file.getPath(),
ex);
}
}
else if (Types.isVncJavaObject(args.first(), OutputStream.class)) {
try {
final OutputStream os = Coerce.toVncJavaObject(args.first(), OutputStream.class);
return new VncJavaObject(
new BufferedWriter(
new OutputStreamWriter(os, charset)));
}
catch (VncException ex) {
throw ex;
}
catch (Exception ex) {
throw new VncException(
"Failed to create writer to the OutputStream!",
ex);
}
}
else if (Types.isVncJavaObject(args.first(), Writer.class)) {
return new VncJavaObject(args.first());
}
else {
throw new VncException(String.format(
"Function 'io/writer' does not allow %s as f",
Types.getType(args.first())));
}
}

private static final long serialVersionUID = -1848883965231344442L;
};

public static VncFunction io_string_writer =
new VncFunction(
"io/string-writer",
Expand Down Expand Up @@ -857,6 +955,129 @@ public VncVal apply(final VncList args) {
private static final long serialVersionUID = -1848883965231344442L;
};

public static VncFunction io_reader =
new VncFunction(
"io/reader",
VncFunction
.meta()
.arglists(
"(io/reader f & options)" )
.doc(
"Create a `java.io.Reader` from f. \n\n" +
"f may be a: \n\n" +
" * string file path, e.g: \"/temp/foo.json\" \n" +
" * bytebuffer \n" +
" * `java.io.File`, e.g: `(io/file \"/temp/foo.json\")` \n" +
" * `java.io.InputStream` \n" +
" * `java.io.Reader` \n" +
" * `java.nio.file.Path` \n" +
" * `java.net.URL` \n" +
" * `java.net.URI` \n\n" +
"Options: \n\n" +
"| :encoding enc | e.g.: `:encoding :utf-8`, defaults to :utf-8 | \n\n" +
"`io/reader` supports load paths. See the `loadpath/paths` " +
"doc for a description of the *load path* feature. \n\n" +
"Note: The caller is responsible for closing the reader!")
.seeAlso(
"io/string-reader")
.build()
) {
@Override
public VncVal apply(final VncList args) {
ArityExceptions.assertMinArity(this, args, 1);

sandboxFunctionCallValidation();

final VncVal arg = args.first();

final VncHashMap options = VncHashMap.ofAll(args.rest());

final VncVal encVal = options.get(new VncKeyword("encoding"));
final Charset charset = CharsetUtil.charset(encVal);

final ILoadPaths loadpaths = ThreadContext.getInterceptor().getLoadPaths();

final File file = convertToFile(arg);
if (file != null) {
try {
final Reader rd = loadpaths.getBufferedReader(file, charset);
if (rd == null) {
if (file.exists()) {
throw new VncException(
"Failed to create reader from the file " + file.getPath() +
". The file does not exists!");
}
else {
throw new SecurityException(
"Failed to create reader from the file " + file.getPath() +
". The load paths configuration prevented this action!");
}
}
return new VncJavaObject(rd);
}
catch (VncException ex) {
throw ex;
}
catch (Exception ex) {
throw new VncException("Failed to create reader from the file " + file.getPath(), ex);
}
}
else if (Types.isVncString(arg)) {
return new VncJavaObject(
new StringReader(
Coerce.toVncString(arg).getValue()));
}
else if (Types.isVncByteBuffer(arg)) {
try {
final VncByteBuffer buf = (VncByteBuffer)arg;
final InputStream is = new ByteArrayInputStream(buf.getBytes());
return new VncJavaObject(
new BufferedReader(
new InputStreamReader(is, charset)));
}
catch (VncException ex) {
throw ex;
}
catch (Exception ex) {
throw new VncException("Failed to create reader from a bytebuffer", ex);
}
}
else if (Types.isVncJavaObject(arg, InputStream.class)) {
try {
final InputStream is = Coerce.toVncJavaObject(args.first(), InputStream.class);
return new VncJavaObject(
new BufferedReader(
new InputStreamReader(is, charset)));
}
catch (VncException ex) {
throw ex;
}
catch (Exception ex) {
throw new VncException("Failed to create reader from a :java.io.InputStream", ex);
}
}
else if (Types.isVncJavaObject(arg, Reader.class)) {
try {
final Reader rd = Coerce.toVncJavaObject(args.first(), Reader.class);
return new VncJavaObject(rd);
}
catch (VncException ex) {
throw ex;
}
catch (Exception ex) {
throw new VncException("Failed to create reader from a :java.io.Reader", ex);
}
}
else {
throw new VncException(String.format(
"Function 'io/reader' does not allow %s as f",
Types.getType(args.first())));
}
}

private static final long serialVersionUID = -1848883965231344442L;
};

public static VncFunction io_string_reader =
new VncFunction(
"io/string-reader",
Expand All @@ -878,7 +1099,7 @@ public VncVal apply(final VncList args) {
" (println (read-line br)) \n" +
" (println (read-line br)))) ")
.seeAlso(
"io/string-writer", "io/buffered-reader")
"io/reader", "io/string-writer", "io/buffered-reader")
.build()
) {
@Override
Expand Down Expand Up @@ -1111,9 +1332,11 @@ public static void validateReadableFile(final File file) {
.add(io_wrap_os_with_buffered_writer)
.add(io_wrap_os_with_print_writer)
.add(io_wrap_is_with_buffered_reader)
.add(io_buffered_reader)
.add(io_buffered_writer)
.add(io_writer)
.add(io_string_writer)
.add(io_buffered_reader)
.add(io_reader)
.add(io_string_reader)
.add(io_capturing_print_stream)
.add(io_print)
Expand Down

0 comments on commit f49b7c1

Please sign in to comment.