blob: c1877795907153d01a62b09f207c97df9cf674d2 [file] [log] [blame]
// Copyright 2016 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package com.google.devtools.starlark;
import com.google.devtools.build.lib.events.Event;
import com.google.devtools.build.lib.syntax.EvalException;
import com.google.devtools.build.lib.syntax.EvalUtils;
import com.google.devtools.build.lib.syntax.Module;
import com.google.devtools.build.lib.syntax.Mutability;
import com.google.devtools.build.lib.syntax.ParserInput;
import com.google.devtools.build.lib.syntax.StarlarkThread;
import com.google.devtools.build.lib.syntax.SyntaxError;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
/**
* Starlark is a standalone starlark intepreter. The environment doesn't
* contain Bazel-specific functions and variables. Load statements are
* forbidden for the moment.
*/
class Starlark {
private static final String START_PROMPT = ">> ";
private static final String CONTINUATION_PROMPT = ".. ";
private static final Charset CHARSET = StandardCharsets.ISO_8859_1;
private final BufferedReader reader =
new BufferedReader(new InputStreamReader(System.in, CHARSET));
private final Mutability mutability = Mutability.create("interpreter");
private final StarlarkThread thread;
{
thread =
StarlarkThread.builder(mutability)
.useDefaultSemantics()
.setGlobals(
Module.createForBuiltins(com.google.devtools.build.lib.syntax.Starlark.UNIVERSE))
.build();
thread.setPrintHandler((th, msg) -> System.out.println(msg));
}
private String prompt() {
StringBuilder input = new StringBuilder();
System.out.print(START_PROMPT);
try {
String lineSeparator = "";
while (true) {
String line = reader.readLine();
if (line == null) {
return null;
}
if (line.isEmpty()) {
return input.toString();
}
input.append(lineSeparator).append(line);
lineSeparator = "\n";
System.out.print(CONTINUATION_PROMPT);
}
} catch (IOException io) {
io.printStackTrace();
return null;
}
}
/** Provide a REPL evaluating Starlark code. */
@SuppressWarnings("CatchAndPrintStackTrace")
private void readEvalPrintLoop() {
String line;
// TODO(adonovan): parse a compound statement, like the Python and
// go.starlark.net REPLs. This requires a new grammar production, and
// integration with the lexer so that it consumes new
// lines only until the parse is complete.
while ((line = prompt()) != null) {
ParserInput input = ParserInput.create(line, "<stdin>");
try {
Object result = EvalUtils.execAndEvalOptionalFinalExpression(input, thread);
if (result != null) {
System.out.println(com.google.devtools.build.lib.syntax.Starlark.repr(result));
}
} catch (SyntaxError ex) {
for (Event ev : ex.errors()) {
System.err.println(ev);
}
} catch (EvalException ex) {
System.err.println(ex.print());
} catch (InterruptedException ex) {
System.err.println("Interrupted");
}
}
}
/** Execute a Starlark file. */
private int executeFile(String filename) {
String content;
try {
content = new String(Files.readAllBytes(Paths.get(filename)), CHARSET);
return execute(filename, content);
} catch (Exception e) {
e.printStackTrace();
return 1;
}
}
/** Execute a Starlark file. */
private int execute(String filename, String content) {
try {
EvalUtils.exec(ParserInput.create(content, filename), thread);
return 0;
} catch (SyntaxError ex) {
for (Event ev : ex.errors()) {
System.err.println(ev);
}
return 1;
} catch (EvalException ex) {
System.err.println(ex.print());
return 1;
} catch (Exception e) {
e.printStackTrace(System.err);
return 1;
}
}
public static void main(String[] args) {
int ret = 0;
if (args.length == 0) {
new Starlark().readEvalPrintLoop();
} else if (args.length == 1 && !args[0].equals("-c")) {
ret = new Starlark().executeFile(args[0]);
} else if (args.length == 2 && args[0].equals("-c")) {
ret = new Starlark().execute("<command-line>", args[1]);
} else {
System.err.println("USAGE: Starlark [-c \"<cmdLineProgram>\" | <fileName>]");
ret = 1;
}
System.exit(ret);
}
}