From 9303df6afe297cefc35b9e71873e0ded4c3df237 Mon Sep 17 00:00:00 2001
From: Marwan Tammam <marwan.01010100@gmail.com>
Date: Mon, 18 Mar 2019 11:42:46 -0700
Subject: [PATCH] add -c flag for Starlark-in-java

**Related issues:**
- add -c flag [#35](https://github.com/bazelbuild/starlark/issues/35)
- improve starlark-in-java installation documentation [#24](https://github.com/bazelbuild/starlark/issues/24)

**Examples:**
$ bazel run :Starlark -- -c "print(1)"
1
$ bazel run :Starlark -- -c "print(1)" arg3
USAGE: Starlark [-c "\<cmdLineProgram\>" | \<fileName\>]

Closes #7683.

PiperOrigin-RevId: 239028987
---
 .../google/devtools/starlark/Starlark.java    | 20 +++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/google/devtools/starlark/Starlark.java b/src/main/java/com/google/devtools/starlark/Starlark.java
index 4ff32d99b0d7b4..31799b53cdb0b1 100644
--- a/src/main/java/com/google/devtools/starlark/Starlark.java
+++ b/src/main/java/com/google/devtools/starlark/Starlark.java
@@ -100,10 +100,20 @@ public void readEvalPrintLoop() {
   }
 
   /** Execute a Starlark file. */
-  public int execute(String path) {
+  public int executeFile(String path) {
     String content;
     try {
       content = new String(Files.readAllBytes(Paths.get(path)), CHARSET);
+      return execute(content);
+    } catch (Exception e) {
+      e.printStackTrace();
+      return 1;
+    }
+  }
+
+  /** Execute a Starlark command. */
+  public int execute(String content) {
+    try {
       BuildFileAST.eval(env, content);
       return 0;
     } catch (EvalException e) {
@@ -119,10 +129,12 @@ public static void main(String[] args) {
     int ret = 0;
     if (args.length == 0) {
       new Starlark().readEvalPrintLoop();
-    } else if (args.length == 1) {
-      ret = new Starlark().execute(args[0]);
+    } 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(args[1]);
     } else {
-      System.err.println("too many arguments");
+      System.err.println("USAGE: Starlark [-c \"<cmdLineProgram>\" | <fileName>]");
       ret = 1;
     }
     System.exit(ret);