1 files changed, 168 insertions, 0 deletions
diff --git a/src/tools/wast-desugar.cc b/src/tools/wast-desugar.cc
new file mode 100644
index 00000000..d3e9d931
--- /dev/null
+++ b/src/tools/wast-desugar.cc
@@ -0,0 +1,168 @@
+/*
+ * Copyright 2016 WebAssembly Community Group participants
+ *
+ * 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.
+ */
+
+#include <assert.h>
+#include <stdarg.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "apply-names.h"
+#include "ast.h"
+#include "ast-parser.h"
+#include "ast-writer.h"
+#include "common.h"
+#include "config.h"
+#include "generate-names.h"
+#include "option-parser.h"
+#include "stream.h"
+#include "writer.h"
+
+#define PROGRAM_NAME "wast-desugar"
+
+static const char* s_infile;
+static const char* s_outfile;
+static bool s_generate_names;
+
+static WabtSourceErrorHandler s_error_handler =
+    WABT_SOURCE_ERROR_HANDLER_DEFAULT;
+
+enum {
+  FLAG_HELP,
+  FLAG_OUTPUT,
+  FLAG_GENERATE_NAMES,
+  NUM_FLAGS
+};
+
+static const char s_description[] =
+    "  read a file in the wasm s-expression format and format it.\n"
+    "\n"
+    "examples:\n"
+    "  # write output to stdout\n"
+    "  $ wast-desugar test.wast\n"
+    "\n"
+    "  # write output to test2.wast\n"
+    "  $ wast-desugar test.wast -o test2.wast\n"
+    "\n"
+    "  # generate names for indexed variables\n"
+    "  $ wast-desugar --generate-names test.wast\n";
+
+static WabtOption s_options[] = {
+    {FLAG_HELP, 'h', "help", NULL, WABT_OPTION_NO_ARGUMENT,
+     "print this help message"},
+    {FLAG_OUTPUT, 'o', "output", "FILE", WABT_OPTION_HAS_ARGUMENT,
+     "output file for the formatted file"},
+    {FLAG_GENERATE_NAMES, 0, "generate-names", NULL, WABT_OPTION_NO_ARGUMENT,
+     "Give auto-generated names to non-named functions, types, etc."},
+};
+WABT_STATIC_ASSERT(NUM_FLAGS == WABT_ARRAY_SIZE(s_options));
+
+static void on_option(struct WabtOptionParser* parser,
+                      struct WabtOption* option,
+                      const char* argument) {
+  switch (option->id) {
+    case FLAG_HELP:
+      wabt_print_help(parser, PROGRAM_NAME);
+      exit(0);
+      break;
+
+    case FLAG_OUTPUT:
+      s_outfile = argument;
+      break;
+
+    case FLAG_GENERATE_NAMES:
+      s_generate_names = true;
+      break;
+  }
+}
+
+static void on_argument(struct WabtOptionParser* parser, const char* argument) {
+  s_infile = argument;
+}
+
+static void on_option_error(struct WabtOptionParser* parser,
+                            const char* message) {
+  WABT_FATAL("%s\n", message);
+}
+
+static void parse_options(int argc, char** argv) {
+  WabtOptionParser parser;
+  WABT_ZERO_MEMORY(parser);
+  parser.description = s_description;
+  parser.options = s_options;
+  parser.num_options = WABT_ARRAY_SIZE(s_options);
+  parser.on_option = on_option;
+  parser.on_argument = on_argument;
+  parser.on_error = on_option_error;
+  wabt_parse_options(&parser, argc, argv);
+
+  if (!s_infile) {
+    wabt_print_help(&parser, PROGRAM_NAME);
+    WABT_FATAL("No filename given.\n");
+  }
+}
+
+typedef struct Context {
+  WabtMemoryWriter json_writer;
+  WabtMemoryWriter module_writer;
+  WabtStream json_stream;
+  WabtStringSlice output_filename_noext;
+  char* module_filename;
+  WabtResult result;
+} Context;
+
+int main(int argc, char** argv) {
+  wabt_init_stdio();
+  parse_options(argc, argv);
+
+  WabtAstLexer* lexer = wabt_new_ast_file_lexer(s_infile);
+  if (!lexer)
+    WABT_FATAL("unable to read %s\n", s_infile);
+
+  WabtScript script;
+  WabtResult result = wabt_parse_ast(lexer, &script, &s_error_handler);
+
+  if (WABT_SUCCEEDED(result)) {
+    WabtModule* module = wabt_get_first_module(&script);
+    if (!module)
+      WABT_FATAL("no module in file.\n");
+
+    if (s_generate_names)
+      result = wabt_generate_names(module);
+
+    if (WABT_SUCCEEDED(result))
+      result = wabt_apply_names(module);
+
+    if (WABT_SUCCEEDED(result)) {
+      WabtFileWriter file_writer;
+      if (s_outfile) {
+        result = wabt_init_file_writer(&file_writer, s_outfile);
+      } else {
+        wabt_init_file_writer_existing(&file_writer, stdout);
+      }
+
+      if (WABT_SUCCEEDED(result)) {
+        result = wabt_write_ast(&file_writer.base, module);
+        wabt_close_file_writer(&file_writer);
+      }
+    }
+  }
+
+  wabt_destroy_ast_lexer(lexer);
+  wabt_destroy_script(&script);
+  return result;
+}
+