Merge branch 'master' of github.com:PolymerLabs/arcs into kt-gen-utils

Author: alxmrs

java/arcs/sdk/BUILD

@@ -18,6 +18,7 @@ arcs_kt_jvm_library(
     ]),
     visibility = ["//visibility:public"],
     exports = [
+        "//java/arcs/core/data",
         "//java/arcs/core/data:rawentity",
         "//java/arcs/core/data/util:data-util",
         "//java/arcs/core/host",

src/tools/schema2kotlin.ts

@@ -198,6 +198,41 @@ export class KotlinGenerator implements ClassGenerator {
     }
   }
 
+  mapOf(items: string[]): string {
+    switch (items.length) {
+      case 0:
+        return `emptyMap()`;
+      case 1:
+        return `mapOf(${items[0]})`;
+      default:
+        return `\
+mapOf(
+${this.leftPad(items.join(',\n'), 4)}
+)`;
+    }
+
+  }
+
+  createSchema(schemaHash: string): string {
+    const schemaNames = this.node.schema.names.map(n => `SchemaName("${n}")`);
+    return `\
+Schema(
+    listOf(${schemaNames.join(',\n' + ' '.repeat(8))}),
+    SchemaFields(
+        singletons = ${this.leftPad(this.mapOf(this.singletonSchemaFields), 8, true)},
+        collections = ${this.leftPad(this.mapOf(this.collectionSchemaFields), 8, true)}
+    ),
+    "${schemaHash}"
+)`;
+  }
+
+  leftPad(input: string, indent: number, skipFirst: boolean = false) {
+    return input
+      .split('\n')
+      .map((line: string, idx: number) => (idx === 0 && skipFirst) ? line : ' '.repeat(indent) + line)
+      .join('\n');
+  }
+
 
   createSchema(schemaHash: string): string {
     const schemaNames = this.node.schema.names.map(n => `SchemaName("${n}")`);
@@ -289,7 +324,7 @@ class ${name}_Spec() : ${this.getType('EntitySpec')}<${name}> {
 ${this.opts.wasm ? '' : `\
 
     companion object {
-        val schema = ${leftPad(this.createSchema(schemaHash), 8, true)}
+        val schema = ${this.leftPad(this.createSchema(schemaHash), 8, true)}
         
         init {
             SchemaRegistry.register(schema)

src/tools/tests/schema2kotlin-test.ts

@@ -0,0 +1,41 @@
+/**
+ * @license
+ * Copyright (c) 2020 Google Inc. All rights reserved.
+ * This code may only be used under the BSD style license found at
+ * http://polymer.github.io/LICENSE.txt
+ * Code distributed by Google as part of this project is also
+ * subject to an additional IP rights grant found at
+ * http://polymer.github.io/PATENTS.txt
+ */
+
+
+import {assert} from '../../platform/chai-node.js';
+import {KotlinGenerator} from '../schema2kotlin.js';
+import {SchemaNode} from '../schema2graph.js';
+import {Schema} from '../../runtime/schema.js';
+
+
+describe('schema2wasm', () => {
+  describe('kotlin-generator', () => {
+    const ktGen = new KotlinGenerator(new SchemaNode(new Schema([], {}), 'dummyNode'), {arg: '', _: []});
+    it('when no items are present, it creates an empty map', () => {
+      const actual = ktGen.mapOf([]);
+
+      assert.strictEqual('emptyMap()', actual);
+    });
+    it('when one item is present, it creates a single-line map', () => {
+      const actual = ktGen.mapOf([`"a" to "b"`]);
+
+      assert.strictEqual('mapOf("a" to "b")', actual);
+    });
+    it('when multiple items are present, it creates a multi-line map', () => {
+      const actual = ktGen.mapOf([`"a" to "b"`, `"b" to "c"`]);
+
+      assert.strictEqual(`\
+mapOf(
+    "a" to "b",
+    "b" to "c"
+)`, actual);
+    });
+  });
+});