AWS Lambda that formats code in a request (#198)

Summary: Pull Request resolved: https://github.com/facebookincubator/ktfmt/pull/198

Reviewed By: hick209

Differential Revision: D27163581

fbshipit-source-id: e717e537ecf988c00edc96317dc2cde7374d77ce
diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
index fa9448a..f75004e 100644
--- a/.github/workflows/build_and_test.yml
+++ b/.github/workflows/build_and_test.yml
@@ -31,3 +31,8 @@
         pushd ktfmt_idea_plugin
         ./gradlew build
         popd
+    - name: Build the Online Formatter
+      run: |
+        pushd online_formatter
+        ./gradlew build
+        popd
diff --git a/.gitignore b/.gitignore
index 2b8f1e6..a747046 100644
--- a/.gitignore
+++ b/.gitignore
@@ -17,3 +17,6 @@
 
 .DS_Store
 classes/
+
+online_formatter/build/
+online_formatter/out.yml
diff --git a/online_formatter/README.md b/online_formatter/README.md
new file mode 100644
index 0000000..139e582
--- /dev/null
+++ b/online_formatter/README.md
@@ -0,0 +1,15 @@
+# AWS Lambda to format Kotlin code using ktfmt
+
+## Build
+
+```
+./gradlew build
+```
+
+## Deploy
+
+```
+./build_and_deploy.sh
+```
+
+The script creates two jars, one with the `com.facebook.ktfmt.onlineformatter` package, and the other with all of its dependencies (including ktfmt itself). This makes deploying just the Lambda faster.
diff --git a/online_formatter/build.gradle.kts b/online_formatter/build.gradle.kts
new file mode 100644
index 0000000..1c322e4
--- /dev/null
+++ b/online_formatter/build.gradle.kts
@@ -0,0 +1,63 @@
+/*
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * 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.
+ */
+
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
+
+plugins { kotlin("jvm") version "1.4.21" }
+
+repositories {
+  mavenLocal()
+  mavenCentral()
+}
+
+val ktfmtVersion = rootProject.file("../version.txt").readText().trim()
+
+dependencies {
+  implementation("com.facebook:ktfmt:$ktfmtVersion")
+  implementation(platform("software.amazon.awssdk:bom:2.10.73"))
+  implementation("software.amazon.awssdk:lambda")
+  implementation("com.amazonaws:aws-lambda-java-core:1.2.1")
+  implementation("com.amazonaws:aws-lambda-java-events:2.2.9")
+  implementation("com.google.code.gson:gson:2.8.6")
+  testImplementation(kotlin("test-junit"))
+}
+
+tasks {
+  test { useJUnit() }
+
+  withType<KotlinCompile>() { kotlinOptions.jvmTarget = "11" }
+
+  val packageFat by creating(Zip::class) {
+    from(compileKotlin)
+    from(processResources)
+    into("lib") { from(configurations.runtimeClasspath) }
+    dirMode = 0b111101101 // 0755
+    fileMode = 0b111101101 // 0755
+  }
+
+  val packageLibs by creating(Zip::class) {
+    into("java/lib") { from(configurations.runtimeClasspath) }
+    dirMode = 0b111101101 // 0755
+    fileMode = 0b111101101 // 0755
+  }
+
+  val packageSkinny by creating(Zip::class) {
+    from(compileKotlin)
+    from(processResources)
+  }
+
+  build { dependsOn(packageSkinny) }
+}
diff --git a/online_formatter/build_and_deploy.sh b/online_formatter/build_and_deploy.sh
new file mode 100755
index 0000000..b5c84d9
--- /dev/null
+++ b/online_formatter/build_and_deploy.sh
@@ -0,0 +1,25 @@
+#!/bin/bash
+# Copyright (c) Facebook, Inc. and its affiliates.
+#
+# 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.
+
+set -eo pipefail
+./gradlew -q packageLibs
+mv build/distributions/lambda.zip build/lambda-lib.zip
+
+./gradlew build
+
+ARTIFACT_BUCKET=ktfmt-online-formatter-lambda
+TEMPLATE=format.yaml
+aws cloudformation package --template-file $TEMPLATE --s3-bucket $ARTIFACT_BUCKET --output-template-file out.yml
+aws cloudformation deploy --template-file out.yml --stack-name ktfmt --capabilities CAPABILITY_NAMED_IAM --region=us-east-2
diff --git a/online_formatter/format.yaml b/online_formatter/format.yaml
new file mode 100644
index 0000000..e937b1d
--- /dev/null
+++ b/online_formatter/format.yaml
@@ -0,0 +1,29 @@
+AWSTemplateFormatVersion: '2010-09-09'
+Transform: 'AWS::Serverless-2016-10-31'
+Description: An AWS Serverless Specification template describing your function.
+Resources:
+  format:
+    Type: 'AWS::Serverless::Function'
+    Properties:
+      CodeUri: build/distributions/lambda.zip
+      Handler: 'com.facebook.ktfmt.onlineformatter.Handler::handleRequest'
+      Runtime: java11
+      Description: ''
+      MemorySize: 1024
+      Timeout: 15
+      Policies:
+        - AWSLambdaBasicExecutionRole
+        - AWSLambdaReadOnlyAccess
+        - AWSXrayWriteOnlyAccess
+        - AWSLambdaVPCAccessExecutionRole
+      Tracing: Active
+      Layers:
+        - !Ref libs
+  libs:
+    Type: AWS::Serverless::LayerVersion
+    Properties:
+      LayerName: ktfmt-lib
+      Description: Dependencies for ktfmt
+      ContentUri: build/lambda-lib.zip
+      CompatibleRuntimes:
+        - java11
diff --git a/online_formatter/gradle.properties b/online_formatter/gradle.properties
new file mode 100644
index 0000000..7fc6f1f
--- /dev/null
+++ b/online_formatter/gradle.properties
@@ -0,0 +1 @@
+kotlin.code.style=official
diff --git a/online_formatter/gradle/wrapper/gradle-wrapper.jar b/online_formatter/gradle/wrapper/gradle-wrapper.jar
new file mode 100644
index 0000000..e708b1c
--- /dev/null
+++ b/online_formatter/gradle/wrapper/gradle-wrapper.jar
Binary files differ
diff --git a/online_formatter/gradle/wrapper/gradle-wrapper.properties b/online_formatter/gradle/wrapper/gradle-wrapper.properties
new file mode 100644
index 0000000..be52383
--- /dev/null
+++ b/online_formatter/gradle/wrapper/gradle-wrapper.properties
@@ -0,0 +1,5 @@
+distributionBase=GRADLE_USER_HOME
+distributionPath=wrapper/dists
+distributionUrl=https\://services.gradle.org/distributions/gradle-6.7-bin.zip
+zipStoreBase=GRADLE_USER_HOME
+zipStorePath=wrapper/dists
diff --git a/online_formatter/gradlew b/online_formatter/gradlew
new file mode 100755
index 0000000..4f906e0
--- /dev/null
+++ b/online_formatter/gradlew
@@ -0,0 +1,185 @@
+#!/usr/bin/env sh
+
+#
+# Copyright 2015 the original author or authors.
+#
+# 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
+#
+#      https://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.
+#
+
+##############################################################################
+##
+##  Gradle start up script for UN*X
+##
+##############################################################################
+
+# Attempt to set APP_HOME
+# Resolve links: $0 may be a link
+PRG="$0"
+# Need this for relative symlinks.
+while [ -h "$PRG" ] ; do
+    ls=`ls -ld "$PRG"`
+    link=`expr "$ls" : '.*-> \(.*\)$'`
+    if expr "$link" : '/.*' > /dev/null; then
+        PRG="$link"
+    else
+        PRG=`dirname "$PRG"`"/$link"
+    fi
+done
+SAVED="`pwd`"
+cd "`dirname \"$PRG\"`/" >/dev/null
+APP_HOME="`pwd -P`"
+cd "$SAVED" >/dev/null
+
+APP_NAME="Gradle"
+APP_BASE_NAME=`basename "$0"`
+
+# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
+
+# Use the maximum available, or set MAX_FD != -1 to use that value.
+MAX_FD="maximum"
+
+warn () {
+    echo "$*"
+}
+
+die () {
+    echo
+    echo "$*"
+    echo
+    exit 1
+}
+
+# OS specific support (must be 'true' or 'false').
+cygwin=false
+msys=false
+darwin=false
+nonstop=false
+case "`uname`" in
+  CYGWIN* )
+    cygwin=true
+    ;;
+  Darwin* )
+    darwin=true
+    ;;
+  MINGW* )
+    msys=true
+    ;;
+  NONSTOP* )
+    nonstop=true
+    ;;
+esac
+
+CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
+
+
+# Determine the Java command to use to start the JVM.
+if [ -n "$JAVA_HOME" ] ; then
+    if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
+        # IBM's JDK on AIX uses strange locations for the executables
+        JAVACMD="$JAVA_HOME/jre/sh/java"
+    else
+        JAVACMD="$JAVA_HOME/bin/java"
+    fi
+    if [ ! -x "$JAVACMD" ] ; then
+        die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+    fi
+else
+    JAVACMD="java"
+    which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+fi
+
+# Increase the maximum file descriptors if we can.
+if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
+    MAX_FD_LIMIT=`ulimit -H -n`
+    if [ $? -eq 0 ] ; then
+        if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
+            MAX_FD="$MAX_FD_LIMIT"
+        fi
+        ulimit -n $MAX_FD
+        if [ $? -ne 0 ] ; then
+            warn "Could not set maximum file descriptor limit: $MAX_FD"
+        fi
+    else
+        warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
+    fi
+fi
+
+# For Darwin, add options to specify how the application appears in the dock
+if $darwin; then
+    GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
+fi
+
+# For Cygwin or MSYS, switch paths to Windows format before running java
+if [ "$cygwin" = "true" -o "$msys" = "true" ] ; then
+    APP_HOME=`cygpath --path --mixed "$APP_HOME"`
+    CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
+
+    JAVACMD=`cygpath --unix "$JAVACMD"`
+
+    # We build the pattern for arguments to be converted via cygpath
+    ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
+    SEP=""
+    for dir in $ROOTDIRSRAW ; do
+        ROOTDIRS="$ROOTDIRS$SEP$dir"
+        SEP="|"
+    done
+    OURCYGPATTERN="(^($ROOTDIRS))"
+    # Add a user-defined pattern to the cygpath arguments
+    if [ "$GRADLE_CYGPATTERN" != "" ] ; then
+        OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
+    fi
+    # Now convert the arguments - kludge to limit ourselves to /bin/sh
+    i=0
+    for arg in "$@" ; do
+        CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
+        CHECK2=`echo "$arg"|egrep -c "^-"`                                 ### Determine if an option
+
+        if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then                    ### Added a condition
+            eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
+        else
+            eval `echo args$i`="\"$arg\""
+        fi
+        i=`expr $i + 1`
+    done
+    case $i in
+        0) set -- ;;
+        1) set -- "$args0" ;;
+        2) set -- "$args0" "$args1" ;;
+        3) set -- "$args0" "$args1" "$args2" ;;
+        4) set -- "$args0" "$args1" "$args2" "$args3" ;;
+        5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
+        6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
+        7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
+        8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
+        9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
+    esac
+fi
+
+# Escape application args
+save () {
+    for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
+    echo " "
+}
+APP_ARGS=`save "$@"`
+
+# Collect all arguments for the java command, following the shell quoting and substitution rules
+eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
+
+exec "$JAVACMD" "$@"
diff --git a/online_formatter/gradlew.bat b/online_formatter/gradlew.bat
new file mode 100644
index 0000000..ac1b06f
--- /dev/null
+++ b/online_formatter/gradlew.bat
@@ -0,0 +1,89 @@
+@rem

+@rem Copyright 2015 the original author or authors.

+@rem

+@rem Licensed under the Apache License, Version 2.0 (the "License");

+@rem you may not use this file except in compliance with the License.

+@rem You may obtain a copy of the License at

+@rem

+@rem      https://www.apache.org/licenses/LICENSE-2.0

+@rem

+@rem Unless required by applicable law or agreed to in writing, software

+@rem distributed under the License is distributed on an "AS IS" BASIS,

+@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

+@rem See the License for the specific language governing permissions and

+@rem limitations under the License.

+@rem

+

+@if "%DEBUG%" == "" @echo off

+@rem ##########################################################################

+@rem

+@rem  Gradle startup script for Windows

+@rem

+@rem ##########################################################################

+

+@rem Set local scope for the variables with windows NT shell

+if "%OS%"=="Windows_NT" setlocal

+

+set DIRNAME=%~dp0

+if "%DIRNAME%" == "" set DIRNAME=.

+set APP_BASE_NAME=%~n0

+set APP_HOME=%DIRNAME%

+

+@rem Resolve any "." and ".." in APP_HOME to make it shorter.

+for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi

+

+@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.

+set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"

+

+@rem Find java.exe

+if defined JAVA_HOME goto findJavaFromJavaHome

+

+set JAVA_EXE=java.exe

+%JAVA_EXE% -version >NUL 2>&1

+if "%ERRORLEVEL%" == "0" goto execute

+

+echo.

+echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.

+echo.

+echo Please set the JAVA_HOME variable in your environment to match the

+echo location of your Java installation.

+

+goto fail

+

+:findJavaFromJavaHome

+set JAVA_HOME=%JAVA_HOME:"=%

+set JAVA_EXE=%JAVA_HOME%/bin/java.exe

+

+if exist "%JAVA_EXE%" goto execute

+

+echo.

+echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%

+echo.

+echo Please set the JAVA_HOME variable in your environment to match the

+echo location of your Java installation.

+

+goto fail

+

+:execute

+@rem Setup the command line

+

+set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar

+

+

+@rem Execute Gradle

+"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %*

+

+:end

+@rem End local scope for the variables with windows NT shell

+if "%ERRORLEVEL%"=="0" goto mainEnd

+

+:fail

+rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of

+rem the _cmd.exe /c_ return code!

+if  not "" == "%GRADLE_EXIT_CONSOLE%" exit 1

+exit /b 1

+

+:mainEnd

+if "%OS%"=="Windows_NT" endlocal

+

+:omega

diff --git a/online_formatter/settings.gradle.kts b/online_formatter/settings.gradle.kts
new file mode 100644
index 0000000..c9ee2d5
--- /dev/null
+++ b/online_formatter/settings.gradle.kts
@@ -0,0 +1,18 @@
+/*
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * 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.
+ */
+
+rootProject.name = "lambda"
+
diff --git a/online_formatter/src/main/kotlin/main.kt b/online_formatter/src/main/kotlin/main.kt
new file mode 100644
index 0000000..4260919
--- /dev/null
+++ b/online_formatter/src/main/kotlin/main.kt
@@ -0,0 +1,65 @@
+/*
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * 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.facebook.ktfmt.onlineformatter
+
+import com.amazonaws.services.lambda.runtime.Context
+import com.amazonaws.services.lambda.runtime.RequestHandler
+import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent
+import com.facebook.ktfmt.KOTLINLANG_FORMAT
+import com.facebook.ktfmt.format
+import com.facebook.ktfmt.parseOptions
+import com.google.gson.Gson
+import java.io.ByteArrayOutputStream
+import java.io.PrintStream
+
+class Handler : RequestHandler<APIGatewayProxyRequestEvent, String> {
+  init {
+    // Warm up.
+    gson.toJson(format(KOTLINLANG_FORMAT, "/* foo */ fun foo() {}"))
+  }
+
+  override fun handleRequest(event: APIGatewayProxyRequestEvent, context: Context?): String {
+    return gson.toJson(
+        try {
+          val request = gson.fromJson(event.body, Request::class.java)
+          val parsingErrors = ByteArrayOutputStream()
+          val style = request.style
+          val parsedArgs =
+              parseOptions(
+                  PrintStream(parsingErrors), if (style == null) arrayOf() else arrayOf(style))
+          Response(
+              format(
+                  parsedArgs.formattingOptions.copy(maxWidth = request.maxWidth ?: 100),
+                  request.source ?: ""),
+              parsingErrors.toString().ifEmpty { null })
+        } catch (e: Exception) {
+          Response(null, e.message)
+        })
+  }
+
+  companion object {
+    val gson = Gson()
+  }
+}
+
+class Request {
+  var source: String? = ""
+  var style: String? = ""
+  var maxWidth: Int? = 100
+}
+
+data class Response(val source: String?, val err: String?)
diff --git a/online_formatter/src/test/kotlin/HandlerTest.kt b/online_formatter/src/test/kotlin/HandlerTest.kt
new file mode 100644
index 0000000..2a41ab9
--- /dev/null
+++ b/online_formatter/src/test/kotlin/HandlerTest.kt
@@ -0,0 +1,120 @@
+/*
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * 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.facebook.ktfmt.onlineformatter
+
+import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent
+import kotlin.test.assertEquals
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.junit.runners.JUnit4
+
+@RunWith(JUnit4::class)
+class HandlerTest {
+
+  @Test
+  fun dontCrash_success() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }","style": "--kotlinlang-style"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"fun foo() {}\n"}""", response)
+  }
+
+  @Test
+  fun dontCrash_malformedRequestJson() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }"style": "--kotlinlang-style"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals(
+        """{"err":"com.google.gson.stream.MalformedJsonException: """ +
+            """Unterminated object at line 1 column 30 path ${'$'}.source"}""",
+        response)
+  }
+
+  @Test
+  fun dontCrash_emptyRequestJson() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = ""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("{}", response)
+  }
+
+  @Test
+  fun dontCrash_malformedSourceCode() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( { }","style": "--kotlinlang-style"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"err":"1:10: error: Expecting \u0027)\u0027"}""", response)
+  }
+
+  @Test
+  fun dontCrash_allNullRequest() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"\n"}""", response)
+  }
+
+  @Test
+  fun dontCrash_malformedStyle() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }","style": "blabla"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"fun foo() {}\n"}""", response)
+  }
+
+  @Test
+  fun dontCrash_noStyle() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"fun foo() {}\n"}""", response)
+  }
+
+  @Test
+  fun dontCrash_negativeMaxWidth() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }", "maxWidth": -100}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"fun foo() {}\n"}""", response)
+  }
+
+  /** maxWidth == null is treated as if it were 100 */
+  @Test
+  fun dontCrash_nullMaxWidth() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }", "maxWidth": null}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"fun foo() {}\n"}""", response)
+  }
+
+  @Test
+  fun dontCrash_nonIntegerMaxWidth() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }", "maxWidth": "bla"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals(
+        """{"err":"java.lang.NumberFormatException: For input string: \"bla\""}""", response)
+  }
+
+  @Test
+  fun dontCrash_coerceMaxWidthToInt() {
+    val event = APIGatewayProxyRequestEvent()
+    event.body = """{"source": "fun foo ( ) { }", "maxWidth": "100"}"""
+    val response = Handler().handleRequest(event, null)
+    assertEquals("""{"source":"fun foo() {}\n"}""", response)
+  }
+}